Migrator init prototype

Author: Sukairo-02

drizzle-orm/src/neon-http/migrator.ts

@@ -2,6 +2,7 @@ import type { MigrationConfig } from '~/migrator.ts';
 import { readMigrationFiles } from '~/migrator.ts';
 import type { AnyRelations } from '~/relations.ts';
 import { type SQL, sql } from '~/sql/sql.ts';
+import type { MigratorInitFailResponse } from '~/utils.ts';
 import type { NeonHttpDatabase } from './driver.ts';
 
 /**
@@ -15,7 +16,8 @@ import type { NeonHttpDatabase } from './driver.ts';
 export async function migrate<TSchema extends Record<string, unknown>, TRelations extends AnyRelations>(
 	db: NeonHttpDatabase<TSchema, TRelations>,
 	config: MigrationConfig,
-) {
+	init?: boolean,
+): Promise<void | MigratorInitFailResponse> {
 	const migrations = readMigrationFiles(config);
 	const migrationsTable = config.migrationsTable ?? '__drizzle_migrations';
 	const migrationsSchema = config.migrationsSchema ?? 'drizzle';
@@ -35,6 +37,16 @@ export async function migrate<TSchema extends Record<string, unknown>, TRelation
 		} order by created_at desc limit 1`,
 	);
 
+	if (init) {
+		if (dbMigrations.length) {
+			return { exitCode: 'manyMigrationsExist' };
+		}
+
+		if (migrations.length > 1) {
+			return { exitCode: 'manyMigrationsExist' };
+		}
+	}
+
 	const lastDbMigration = dbMigrations[0];
 	const rowsToInsert: SQL[] = [];
 	for await (const migration of migrations) {

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

@@ -57,7 +57,7 @@ import {
 } from '~/sql/sql.ts';
 import { Subquery } from '~/subquery.ts';
 import { getTableName, getTableUniqueName, Table, TableColumns } from '~/table.ts';
-import { type Casing, orderSelectedFields, type UpdateSet } from '~/utils.ts';
+import { type Casing, type MigratorInitFailResponse, orderSelectedFields, type UpdateSet } from '~/utils.ts';
 import { ViewBaseConfig } from '~/view-common.ts';
 import type { PgSession } from './session.ts';
 import { PgViewBase } from './view-base.ts';
@@ -77,7 +77,12 @@ export class PgDialect {
 		this.casing = new CasingCache(config?.casing);
 	}
 
-	async migrate(migrations: MigrationMeta[], session: PgSession, config: string | MigrationConfig): Promise<void> {
+	async migrate(
+		migrations: MigrationMeta[],
+		session: PgSession,
+		config: string | MigrationConfig,
+		init?: boolean,
+	): Promise<void | MigratorInitFailResponse> {
 		const migrationsTable = typeof config === 'string'
 			? '__drizzle_migrations'
 			: config.migrationsTable ?? '__drizzle_migrations';
@@ -98,6 +103,16 @@ export class PgDialect {
 			} order by created_at desc limit 1`,
 		);
 
+		if (init) {
+			if (dbMigrations.length) {
+				return { exitCode: 'manyMigrationsExist' };
+			}
+
+			if (migrations.length > 1) {
+				return { exitCode: 'manyMigrationsExist' };
+			}
+		}
+
 		const lastDbMigration = dbMigrations[0];
 		await session.transaction(async (tx) => {
 			for await (const migration of migrations) {

drizzle-orm/src/postgres-js/migrator.ts

@@ -6,7 +6,8 @@ import type { PostgresJsDatabase } from './driver.ts';
 export async function migrate<TSchema extends Record<string, unknown>, TRelations extends AnyRelations>(
 	db: PostgresJsDatabase<TSchema, TRelations>,
 	config: MigrationConfig,
+	init?: boolean,
 ) {
 	const migrations = readMigrationFiles(config);
-	await db.dialect.migrate(migrations, db.session, config);
+	return await db.dialect.migrate(migrations, db.session, config, init);
 }

drizzle-orm/src/utils.ts

@@ -336,3 +336,5 @@ export function isConfig(data: any): boolean {
 export type NeonAuthToken = string | (() => string | Promise<string>);
 
 export const textDecoder = typeof TextDecoder === 'undefined' ? null : new TextDecoder();
+
+export type MigratorInitFailResponse = { exitCode: 'migrationsTableExists' | 'manyMigrationsExist' };