Data migrator (#5265)

* Added migrator from migration journal to `bun-sqlite`

Author: Sukairo-02

drizzle-orm/src/bun-sqlite/migrator.ts

@@ -1,12 +1,45 @@
-import type { MigrationConfig } from '~/migrator.ts';
+import type {
+	MigrationConfig,
+	MigrationFromJournalConfig,
+	MigrationMeta,
+	MigrationsJournal,
+	MigratorInitFailResponse,
+} from '~/migrator.ts';
 import { readMigrationFiles } from '~/migrator.ts';
 import type { AnyRelations, EmptyRelations } from '~/relations.ts';
 import type { SQLiteBunDatabase } from './driver.ts';
 
 export function migrate<TSchema extends Record<string, unknown>, TRelations extends AnyRelations = EmptyRelations>(
 	db: SQLiteBunDatabase<TSchema, TRelations>,
 	config: MigrationConfig,
-) {
-	const migrations = readMigrationFiles(config);
-	return db.dialect.migrate(migrations, db.session, config);
+): void | MigratorInitFailResponse;
+export function migrate<
+	TSchema extends Record<string, unknown>,
+	TRelations extends AnyRelations = EmptyRelations,
+>(
+	db: SQLiteBunDatabase<TSchema, TRelations>,
+	config: MigrationFromJournalConfig | MigrationsJournal,
+): void;
+export function migrate<TSchema extends Record<string, unknown>, TRelations extends AnyRelations = EmptyRelations>(
+	db: SQLiteBunDatabase<TSchema, TRelations>,
+	config: MigrationConfig | MigrationFromJournalConfig | MigrationsJournal,
+): void | MigratorInitFailResponse {
+	if ('migrationsJournal' in config) {
+		const journal = config.migrationsJournal;
+		const migrationsTable = config.migrationsTable;
+
+		const migrations: MigrationMeta[] = journal.map((d) => ({
+			sql: d.sql.split('--> statement-breakpoint'),
+			folderMillis: d.timestamp,
+			hash: '',
+			bps: true,
+		}));
+
+		return db.dialect.migrate(migrations, db.session, {
+			migrationsTable,
+		});
+	}
+
+	const migrations = readMigrationFiles(config as MigrationConfig);
+	return db.dialect.migrate(migrations, db.session, config as MigrationConfig);
 }

drizzle-orm/src/migrator.ts

@@ -22,6 +22,16 @@ export interface MigrationMeta {
 	bps: boolean;
 }
 
+export interface MigrationFromJournalConfig {
+	migrationsJournal: MigrationsJournal;
+	migrationsTable?: string;
+}
+
+export type MigrationsJournal = {
+	sql: string;
+	timestamp: number;
+}[];
+
 /** Only gets returned if migrator failed with `init: true` used by `drizzle-kit pull --init`*/
 export interface MigratorInitFailResponse {
 	exitCode: 'databaseMigrations' | 'localMigrations';

drizzle-orm/src/sqlite-core/dialect.ts

@@ -1142,7 +1142,7 @@ export class SQLiteSyncDialect extends SQLiteDialect {
 			AnyRelations,
 			V1.TablesRelationalConfig
 		>,
-		config?: string | MigrationConfig,
+		config?: string | Omit<MigrationConfig, 'migrationsFolder'>,
 	): void | MigratorInitFailResponse {
 		const migrationsTable = config === undefined
 			? '__drizzle_migrations'
@@ -1222,7 +1222,7 @@ export class SQLiteAsyncDialect extends SQLiteDialect {
 			AnyRelations,
 			V1.TablesRelationalConfig
 		>,
-		config?: string | MigrationConfig,
+		config?: string | Omit<MigrationConfig, 'migrationsFolder'>,
 	): Promise<void | MigratorInitFailResponse> {
 		const migrationsTable = config === undefined
 			? '__drizzle_migrations'