@vibeorm/migrate 1.0.0

package/src/migration-runner.ts

@@ -0,0 +1,103 @@
+/**
+ * Migration Runner — Apply SQL files with tracking
+ *
+ * Handles creating the migration tracking table, querying applied migrations,
+ * executing migration SQL files, and recording results.
+ */
+
+import type { SqlExecutor } from "./types.ts";
+import { splitSqlStatements } from "./sql-utils.ts";
+
+// ─── Types ────────────────────────────────────────────────────────
+
+export type AppliedMigration = {
+  id: number;
+  migrationName: string;
+  checksum: string;
+  appliedAt: Date;
+  executionTime: number | null;
+};
+
+// ─── Migration Table Management ───────────────────────────────────
+
+const CREATE_TABLE_SQL = `
+CREATE TABLE IF NOT EXISTS "_vibeorm_migrations" (
+  "id"             SERIAL PRIMARY KEY,
+  "migration_name" TEXT NOT NULL UNIQUE,
+  "checksum"       TEXT NOT NULL,
+  "applied_at"     TIMESTAMPTZ NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  "execution_time" INTEGER
+);
+`;
+
+export async function ensureMigrationTable(params: { executor: SqlExecutor }): Promise<void> {
+  await params.executor({ text: CREATE_TABLE_SQL });
+}
+
+export async function getAppliedMigrations(params: { executor: SqlExecutor }): Promise<AppliedMigration[]> {
+  const rows = await params.executor({
+    text: `SELECT id, migration_name, checksum, applied_at, execution_time FROM "_vibeorm_migrations" ORDER BY id;`,
+  });
+
+  return rows.map((row) => ({
+    id: row.id as number,
+    migrationName: row.migration_name as string,
+    checksum: row.checksum as string,
+    appliedAt: new Date(row.applied_at as string),
+    executionTime: row.execution_time as number | null,
+  }));
+}
+
+// ─── Apply Migration ──────────────────────────────────────────────
+
+export async function applyMigration(params: {
+  executor: SqlExecutor;
+  migrationName: string;
+  sql: string;
+  checksum: string;
+}): Promise<{ executionTime: number }> {
+  const { executor, migrationName, sql, checksum } = params;
+  const start = Date.now();
+
+  // Split migration SQL into individual statements to handle
+  // DO $$ blocks and multi-statement migrations correctly
+  const statements = splitSqlStatements({ sql });
+  for (const stmt of statements) {
+    await executor({ text: stmt });
+  }
+
+  const executionTime = Date.now() - start;
+
+  // Record the applied migration
+  await executor({
+    text: `INSERT INTO "_vibeorm_migrations" ("migration_name", "checksum", "execution_time") VALUES ($1, $2, $3);`,
+    values: [migrationName, checksum, executionTime],
+  });
+
+  return { executionTime };
+}
+
+// ─── Repair Tools ─────────────────────────────────────────────────
+
+export async function markMigrationApplied(params: {
+  executor: SqlExecutor;
+  migrationName: string;
+  checksum: string;
+}): Promise<void> {
+  const { executor, migrationName, checksum } = params;
+  await executor({
+    text: `INSERT INTO "_vibeorm_migrations" ("migration_name", "checksum", "execution_time") VALUES ($1, $2, $3) ON CONFLICT ("migration_name") DO NOTHING;`,
+    values: [migrationName, checksum, 0],
+  });
+}
+
+export async function removeMigrationRecord(params: {
+  executor: SqlExecutor;
+  migrationName: string;
+}): Promise<void> {
+  const { executor, migrationName } = params;
+  await executor({
+    text: `DELETE FROM "_vibeorm_migrations" WHERE "migration_name" = $1;`,
+    values: [migrationName],
+  });
+}