Run DB migrations automatically on server startup

Extract migration logic into reusable runMigrations() in src/db.ts,
called from instrumentation register() before any DB queries.
Simplify scripts/migrate.ts to reuse the same function.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: hubyrod

instrumentation.ts

@@ -1,5 +1,6 @@
+import { join } from "node:path";
 import { loadAppConfig } from "@/src/config";
-import { getAllRoutes } from "@/src/db";
+import { runMigrations, getAllRoutes } from "@/src/db";
 import { validateConnection, type SlashworkConnection } from "@/src/slashwork";
 import { validateCalendarAuth } from "@/src/calendar/auth";
 import { startCalendarPoller } from "@/src/calendar/poller";
@@ -11,6 +12,8 @@ function log(level: string, message: string) {
 export async function register() {
   const config = loadAppConfig();
 
+  await runMigrations(join(process.cwd(), "migrations"), log);
+
   log("info", `Slashwork URL: ${config.slashwork.graphqlUrl}`);
 
   try {

scripts/migrate.ts

@@ -1,87 +1,15 @@
-import pg from "pg";
-import { readdir, readFile } from "node:fs/promises";
 import { join } from "node:path";
+import { runMigrations } from "../src/db";
 
-const connectionString = process.env.POSTGRESQL_ADDON_URI;
-if (!connectionString) {
-  console.error("POSTGRESQL_ADDON_URI environment variable is required");
-  process.exit(1);
-}
-
-const pool = new pg.Pool({ connectionString });
 const migrationsDir = join(import.meta.dirname, "..", "migrations");
 
-async function ensureMigrationsTable() {
-  await pool.query(`
-    CREATE TABLE IF NOT EXISTS schema_migrations (
-      version TEXT PRIMARY KEY,
-      applied_at TIMESTAMPTZ NOT NULL DEFAULT now()
-    )
-  `);
-}
-
-async function getAppliedMigrations(): Promise<Set<string>> {
-  const result = await pool.query<{ version: string }>(
-    "SELECT version FROM schema_migrations ORDER BY version",
-  );
-  return new Set(result.rows.map((r) => r.version));
-}
-
-async function getPendingMigrations(applied: Set<string>): Promise<string[]> {
-  const files = await readdir(migrationsDir);
-  return files
-    .filter((f) => f.endsWith(".sql"))
-    .sort()
-    .filter((f) => !applied.has(f));
-}
-
-async function runMigration(filename: string) {
-  const sql = await readFile(join(migrationsDir, filename), "utf-8");
-  const client = await pool.connect();
-  try {
-    await client.query("BEGIN");
-    await client.query(sql);
-    await client.query(
-      "INSERT INTO schema_migrations (version) VALUES ($1)",
-      [filename],
-    );
-    await client.query("COMMIT");
-  } catch (err) {
-    await client.query("ROLLBACK");
-    throw err;
-  } finally {
-    client.release();
-  }
-}
-
-async function migrate() {
-  await ensureMigrationsTable();
-  const applied = await getAppliedMigrations();
-
-  if (applied.size > 0) {
-    console.log(`Already applied: ${[...applied].join(", ")}`);
-  }
-
-  const pending = await getPendingMigrations(applied);
-  if (pending.length === 0) {
-    console.log("No pending migrations.");
-    await pool.end();
-    return;
-  }
-
-  console.log(`Pending: ${pending.join(", ")}`);
-
-  for (const filename of pending) {
-    console.log(`Applying ${filename}...`);
-    await runMigration(filename);
-    console.log(`  Done.`);
-  }
-
-  console.log(`Applied ${pending.length} migration(s).`);
-  await pool.end();
+function log(level: string, message: string) {
+  console.log(`[${level}] ${message}`);
 }
 
-migrate().catch((err) => {
-  console.error("Migration failed:", err);
-  process.exit(1);
-});
+runMigrations(migrationsDir, log)
+  .then(() => process.exit(0))
+  .catch((err) => {
+    console.error("Migration failed:", err);
+    process.exit(1);
+  });

src/db.ts

@@ -1,4 +1,6 @@
 import pg from "pg";
+import { readdir, readFile } from "node:fs/promises";
+import { join } from "node:path";
 
 let pool: pg.Pool | null = null;
 
@@ -13,6 +15,55 @@ function getPool(): pg.Pool {
   return pool;
 }
 
+export async function runMigrations(
+  migrationsDir: string,
+  log: (level: string, message: string) => void = () => {},
+): Promise<void> {
+  const db = getPool();
+
+  await db.query(`
+    CREATE TABLE IF NOT EXISTS schema_migrations (
+      version TEXT PRIMARY KEY,
+      applied_at TIMESTAMPTZ NOT NULL DEFAULT now()
+    )
+  `);
+
+  const applied = new Set(
+    (await db.query<{ version: string }>("SELECT version FROM schema_migrations ORDER BY version"))
+      .rows.map((r) => r.version),
+  );
+
+  const files = await readdir(migrationsDir);
+  const pending = files
+    .filter((f) => f.endsWith(".sql"))
+    .sort()
+    .filter((f) => !applied.has(f));
+
+  if (pending.length === 0) {
+    log("info", "Migrations: all up to date");
+    return;
+  }
+
+  for (const filename of pending) {
+    const sql = await readFile(join(migrationsDir, filename), "utf-8");
+    const client = await db.connect();
+    try {
+      await client.query("BEGIN");
+      await client.query(sql);
+      await client.query("INSERT INTO schema_migrations (version) VALUES ($1)", [filename]);
+      await client.query("COMMIT");
+      log("info", `Migrations: applied ${filename}`);
+    } catch (err) {
+      await client.query("ROLLBACK");
+      throw err;
+    } finally {
+      client.release();
+    }
+  }
+
+  log("info", `Migrations: applied ${pending.length} migration(s)`);
+}
+
 export interface ResolvedRoute {
   targetId: string;
   authToken: string;