@hasna/conversations 0.2.25 → 0.2.26

package/bin/hook.js

@@ -9467,6 +9467,7 @@ function getDb() {
   db.exec(`
     CREATE TABLE IF NOT EXISTS messages (
       id INTEGER PRIMARY KEY AUTOINCREMENT,
+      uuid TEXT NOT NULL DEFAULT (lower(hex(randomblob(16)))),
       session_id TEXT NOT NULL,
       from_agent TEXT NOT NULL,
       to_agent TEXT NOT NULL,
@@ -9486,6 +9487,7 @@ function getDb() {
   db.exec("CREATE INDEX IF NOT EXISTS idx_messages_to ON messages(to_agent)");
   db.exec("CREATE INDEX IF NOT EXISTS idx_messages_created ON messages(created_at)");
   db.exec("CREATE INDEX IF NOT EXISTS idx_messages_space ON messages(space)");
+  db.exec("CREATE UNIQUE INDEX IF NOT EXISTS idx_messages_uuid ON messages(uuid)");
   db.exec(`
     CREATE TABLE IF NOT EXISTS projects (
       id TEXT PRIMARY KEY,
@@ -9633,6 +9635,11 @@ function getDb() {
     db.exec("ALTER TABLE messages ADD COLUMN project_id TEXT");
     db.exec("CREATE INDEX IF NOT EXISTS idx_messages_project ON messages(project_id)");
   }
+  if (!colNames2.includes("uuid")) {
+    db.exec("ALTER TABLE messages ADD COLUMN uuid TEXT");
+    db.exec("UPDATE messages SET uuid = lower(hex(randomblob(16))) WHERE uuid IS NULL");
+    db.exec("CREATE UNIQUE INDEX IF NOT EXISTS idx_messages_uuid ON messages(uuid)");
+  }
   const presenceCols = db.prepare("PRAGMA table_info(agent_presence)").all();
   const presenceColNames = presenceCols.map((c) => c.name);
   if (!presenceColNames.includes("id")) {

package/bin/index.js

@@ -13382,6 +13382,7 @@ function getDb() {
   db.exec(`
     CREATE TABLE IF NOT EXISTS messages (
       id INTEGER PRIMARY KEY AUTOINCREMENT,
+      uuid TEXT NOT NULL DEFAULT (lower(hex(randomblob(16)))),
       session_id TEXT NOT NULL,
       from_agent TEXT NOT NULL,
       to_agent TEXT NOT NULL,
@@ -13401,6 +13402,7 @@ function getDb() {
   db.exec("CREATE INDEX IF NOT EXISTS idx_messages_to ON messages(to_agent)");
   db.exec("CREATE INDEX IF NOT EXISTS idx_messages_created ON messages(created_at)");
   db.exec("CREATE INDEX IF NOT EXISTS idx_messages_space ON messages(space)");
+  db.exec("CREATE UNIQUE INDEX IF NOT EXISTS idx_messages_uuid ON messages(uuid)");
   db.exec(`
     CREATE TABLE IF NOT EXISTS projects (
       id TEXT PRIMARY KEY,
@@ -13548,6 +13550,11 @@ function getDb() {
     db.exec("ALTER TABLE messages ADD COLUMN project_id TEXT");
     db.exec("CREATE INDEX IF NOT EXISTS idx_messages_project ON messages(project_id)");
   }
+  if (!colNames2.includes("uuid")) {
+    db.exec("ALTER TABLE messages ADD COLUMN uuid TEXT");
+    db.exec("UPDATE messages SET uuid = lower(hex(randomblob(16))) WHERE uuid IS NULL");
+    db.exec("CREATE UNIQUE INDEX IF NOT EXISTS idx_messages_uuid ON messages(uuid)");
+  }
   const presenceCols = db.prepare("PRAGMA table_info(agent_presence)").all();
   const presenceColNames = presenceCols.map((c) => c.name);
   if (!presenceColNames.includes("id")) {
@@ -14089,7 +14096,26 @@ function guessMimeType(name) {
   };
   return mimeMap[ext || ""] || "application/octet-stream";
 }
+function checkRateLimit(agentId) {
+  const dbPath = process.env.CONVERSATIONS_DB_PATH ?? process.env.HASNA_CONVERSATIONS_DB_PATH ?? "";
+  if (dbPath === ":memory:" || dbPath.includes("test") || dbPath.includes("tmp"))
+    return;
+  const now = Date.now();
+  const entry = _rateLimitCounters.get(agentId);
+  if (!entry || now - entry.windowStart > RATE_LIMIT_WINDOW_MS) {
+    _rateLimitCounters.set(agentId, { count: 1, windowStart: now });
+    return;
+  }
+  entry.count++;
+  if (entry.count > RATE_LIMIT_MAX) {
+    throw new Error(`Rate limit exceeded: ${agentId} may send at most ${RATE_LIMIT_MAX} messages per minute.`);
+  }
+}
 function sendMessage(opts) {
+  if (Buffer.byteLength(opts.content, "utf8") > MAX_MESSAGE_BYTES) {
+    throw new Error(`Message content exceeds maximum size of ${MAX_MESSAGE_BYTES} bytes (64 KB).`);
+  }
+  checkRateLimit(opts.from);
   const db2 = getDb();
   const explicitSession = opts.session_id && opts.session_id.trim().length > 0 ? opts.session_id : undefined;
   const sessionId = explicitSession ?? (opts.space ? `space:${opts.space}` : `${[opts.from, opts.to].sort().join("-")}-${randomUUID().slice(0, 8)}`);
@@ -14617,9 +14643,11 @@ function getMessageReadStatus(messageId, space) {
   const unread_by = members.map((m) => m.agent).filter((a) => !readers.has(a));
   return { receipts, unread_by };
 }
+var MAX_MESSAGE_BYTES = 65536, RATE_LIMIT_MAX = 60, RATE_LIMIT_WINDOW_MS = 60000, _rateLimitCounters;
 var init_messages = __esm(() => {
   init_db();
   init_webhooks();
+  _rateLimitCounters = new Map;
 });
 
 // src/lib/poll.ts
@@ -14900,7 +14928,7 @@ var init_presence = __esm(() => {
 var require_package = __commonJS((exports, module) => {
   module.exports = {
     name: "@hasna/conversations",
-    version: "0.2.25",
+    version: "0.2.26",
     description: "Real-time CLI messaging for AI agents",
     type: "module",
     bin: {
@@ -14929,7 +14957,7 @@ var require_package = __commonJS((exports, module) => {
       test: "bun test",
       dev: "bun run ./src/cli/index.tsx",
       typecheck: "tsc --noEmit",
-      prepublishOnly: "bun run build",
+      prepublishOnly: "bun run build:dashboard && bun run build",
       postinstall: "mkdir -p $HOME/.hasna/conversations $HOME/.hasna/conversations/training 2>/dev/null || true"
     },
     keywords: [
@@ -46272,6 +46300,187 @@ var init_advanced = __esm(() => {
   init_graph();
 });
 
+// src/lib/pg-migrations.ts
+var exports_pg_migrations = {};
+__export(exports_pg_migrations, {
+  PG_MIGRATIONS: () => PG_MIGRATIONS
+});
+var PG_MIGRATIONS;
+var init_pg_migrations = __esm(() => {
+  PG_MIGRATIONS = [
+    `
+  CREATE TABLE IF NOT EXISTS projects (
+    id TEXT PRIMARY KEY,
+    name TEXT NOT NULL UNIQUE,
+    description TEXT,
+    path TEXT,
+    created_by TEXT NOT NULL,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+    metadata TEXT,
+    tags TEXT,
+    status TEXT NOT NULL DEFAULT 'active',
+    repository TEXT,
+    settings TEXT
+  );
+  CREATE INDEX IF NOT EXISTS idx_projects_name ON projects(name);
+  CREATE INDEX IF NOT EXISTS idx_projects_status ON projects(status);
+
+  CREATE TABLE IF NOT EXISTS spaces (
+    name TEXT PRIMARY KEY,
+    description TEXT,
+    parent_id TEXT REFERENCES spaces(name),
+    project_id TEXT REFERENCES projects(id),
+    created_by TEXT NOT NULL,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+    archived_at TEXT,
+    topic TEXT
+  );
+  CREATE INDEX IF NOT EXISTS idx_spaces_parent ON spaces(parent_id);
+  CREATE INDEX IF NOT EXISTS idx_spaces_project ON spaces(project_id);
+
+  CREATE TABLE IF NOT EXISTS space_members (
+    space TEXT NOT NULL REFERENCES spaces(name),
+    agent TEXT NOT NULL,
+    joined_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+    PRIMARY KEY (space, agent)
+  );
+
+  CREATE TABLE IF NOT EXISTS messages (
+    id BIGINT GENERATED ALWAYS AS IDENTITY PRIMARY KEY,
+    uuid TEXT NOT NULL DEFAULT gen_random_uuid()::text UNIQUE,
+    session_id TEXT NOT NULL,
+    from_agent TEXT NOT NULL,
+    to_agent TEXT NOT NULL,
+    space TEXT,
+    project_id TEXT,
+    content TEXT NOT NULL,
+    priority TEXT NOT NULL DEFAULT 'normal',
+    working_dir TEXT,
+    repository TEXT,
+    branch TEXT,
+    metadata TEXT,
+    edited_at TEXT,
+    pinned_at TEXT,
+    blocking BOOLEAN NOT NULL DEFAULT FALSE,
+    attachments TEXT,
+    reply_to BIGINT,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+    read_at TEXT
+  );
+  CREATE INDEX IF NOT EXISTS idx_messages_session ON messages(session_id);
+  CREATE INDEX IF NOT EXISTS idx_messages_to ON messages(to_agent);
+  CREATE INDEX IF NOT EXISTS idx_messages_created ON messages(created_at);
+  CREATE INDEX IF NOT EXISTS idx_messages_space ON messages(space);
+  CREATE INDEX IF NOT EXISTS idx_messages_pinned ON messages(pinned_at);
+  CREATE INDEX IF NOT EXISTS idx_messages_blocking ON messages(blocking);
+  CREATE INDEX IF NOT EXISTS idx_messages_reply_to ON messages(reply_to);
+  CREATE INDEX IF NOT EXISTS idx_messages_project ON messages(project_id);
+
+  CREATE TABLE IF NOT EXISTS agent_presence (
+    id TEXT NOT NULL DEFAULT '',
+    agent TEXT PRIMARY KEY,
+    session_id TEXT,
+    role TEXT NOT NULL DEFAULT 'agent',
+    project_id TEXT,
+    status TEXT NOT NULL DEFAULT 'online',
+    last_seen_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+    created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+    metadata TEXT
+  );
+
+  CREATE TABLE IF NOT EXISTS resource_locks (
+    resource_type TEXT NOT NULL,
+    resource_id TEXT NOT NULL,
+    agent_id TEXT NOT NULL,
+    lock_type TEXT NOT NULL DEFAULT 'advisory',
+    locked_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+    expires_at TIMESTAMPTZ NOT NULL,
+    UNIQUE(resource_type, resource_id, lock_type)
+  );
+  CREATE INDEX IF NOT EXISTS idx_locks_resource ON resource_locks(resource_type, resource_id);
+  CREATE INDEX IF NOT EXISTS idx_locks_agent ON resource_locks(agent_id);
+  CREATE INDEX IF NOT EXISTS idx_locks_expires ON resource_locks(expires_at);
+
+  CREATE TABLE IF NOT EXISTS reactions (
+    id BIGINT GENERATED ALWAYS AS IDENTITY PRIMARY KEY,
+    message_id BIGINT NOT NULL REFERENCES messages(id) ON DELETE CASCADE,
+    agent TEXT NOT NULL,
+    emoji TEXT NOT NULL,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+    UNIQUE(message_id, agent, emoji)
+  );
+  CREATE INDEX IF NOT EXISTS idx_reactions_message ON reactions(message_id);
+
+  CREATE TABLE IF NOT EXISTS message_read_receipts (
+    message_id BIGINT NOT NULL REFERENCES messages(id) ON DELETE CASCADE,
+    agent TEXT NOT NULL,
+    read_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+    PRIMARY KEY (message_id, agent)
+  );
+  CREATE INDEX IF NOT EXISTS idx_read_receipts_message ON message_read_receipts(message_id);
+  CREATE INDEX IF NOT EXISTS idx_read_receipts_agent ON message_read_receipts(agent);
+
+  CREATE TABLE IF NOT EXISTS message_mentions (
+    id BIGINT GENERATED ALWAYS AS IDENTITY PRIMARY KEY,
+    message_id BIGINT NOT NULL REFERENCES messages(id) ON DELETE CASCADE,
+    mentioned_agent TEXT NOT NULL,
+    from_agent TEXT NOT NULL,
+    space TEXT,
+    notified_at TEXT,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
+  );
+  CREATE INDEX IF NOT EXISTS idx_mentions_agent ON message_mentions(mentioned_agent);
+  CREATE INDEX IF NOT EXISTS idx_mentions_message ON message_mentions(message_id);
+  CREATE INDEX IF NOT EXISTS idx_mentions_notified ON message_mentions(notified_at);
+
+  -- Full-text search using PostgreSQL tsvector
+  ALTER TABLE messages ADD COLUMN IF NOT EXISTS search_vector tsvector;
+  CREATE INDEX IF NOT EXISTS idx_messages_search ON messages USING GIN(search_vector);
+
+  CREATE OR REPLACE FUNCTION messages_search_vector_update() RETURNS trigger AS $$
+  BEGIN
+    NEW.search_vector :=
+      setweight(to_tsvector('english', COALESCE(NEW.content, '')), 'A') ||
+      setweight(to_tsvector('english', COALESCE(NEW.from_agent, '')), 'B') ||
+      setweight(to_tsvector('english', COALESCE(NEW.to_agent, '')), 'B') ||
+      setweight(to_tsvector('english', COALESCE(NEW.space, '')), 'C');
+    RETURN NEW;
+  END;
+  $$ LANGUAGE plpgsql;
+
+  DROP TRIGGER IF EXISTS messages_search_vector_trigger ON messages;
+  CREATE TRIGGER messages_search_vector_trigger
+    BEFORE INSERT OR UPDATE OF content, from_agent, to_agent, space ON messages
+    FOR EACH ROW EXECUTE FUNCTION messages_search_vector_update();
+
+  -- Backfill existing rows
+  UPDATE messages SET search_vector =
+    setweight(to_tsvector('english', COALESCE(content, '')), 'A') ||
+    setweight(to_tsvector('english', COALESCE(from_agent, '')), 'B') ||
+    setweight(to_tsvector('english', COALESCE(to_agent, '')), 'B') ||
+    setweight(to_tsvector('english', COALESCE(space, '')), 'C')
+  WHERE search_vector IS NULL;
+
+  -- Feedback table
+  CREATE TABLE IF NOT EXISTS feedback (
+    id TEXT PRIMARY KEY DEFAULT gen_random_uuid()::text,
+    message TEXT NOT NULL,
+    email TEXT,
+    category TEXT DEFAULT 'general',
+    version TEXT,
+    machine_id TEXT,
+    created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
+  );
+
+  CREATE TABLE IF NOT EXISTS _migrations (
+    id INTEGER PRIMARY KEY,
+    applied_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
+  );
+  INSERT INTO _migrations (id) VALUES (1) ON CONFLICT DO NOTHING;
+  `
+  ];
+});
+
 // src/mcp/tools/cloud.ts
 async function detectAndLogConflicts(local, cloud, table) {
   if (!CONFLICT_TABLES.has(table))
@@ -46355,7 +46564,7 @@ function registerCloudSyncTools(server) {
       const localPath = cloudGetDbPath("conversations");
       const local = new SqliteAdapter2(localPath);
       const cloud = new PgAdapterAsync2(getConnectionString2("conversations"));
-      const tableList = tablesStr ? tablesStr.split(",").map((t) => t.trim()) : listSqliteTables2(local).filter((t) => !t.startsWith("_") && !t.endsWith("_fts") && !t.startsWith("messages_fts"));
+      const tableList = tablesStr ? tablesStr.split(",").map((t) => t.trim()) : listSqliteTables2(local).filter((t) => !SYNC_EXCLUDED.has(t));
       let totalConflicts = 0;
       for (const table of tableList) {
         totalConflicts += await detectAndLogConflicts(local, cloud, table);
@@ -46400,7 +46609,7 @@ function registerCloudSyncTools(server) {
         tableList = tablesStr.split(",").map((t) => t.trim());
       } else {
         try {
-          tableList = (await listPgTables2(cloud)).filter((t) => !t.startsWith("_"));
+          tableList = (await listPgTables2(cloud)).filter((t) => !SYNC_EXCLUDED.has(t));
         } catch {
           local.close();
           await cloud.close();
@@ -46422,6 +46631,100 @@ function registerCloudSyncTools(server) {
       if (errors4.length > 0)
         lines.push(`Errors: ${errors4.join("; ")}`);
       return { content: [{ type: "text", text: lines.join(`
+`) }] };
+    } catch (e) {
+      return { content: [{ type: "text", text: formatError2(e) }], isError: true };
+    }
+  });
+  server.tool("conversations_cloud_sync", "Bidirectional cloud sync \u2014 pull remote changes then push local changes. Detects and logs conflicts.", {
+    tables: exports_external2.string().optional().describe("Comma-separated table names (default: all syncable tables)")
+  }, async ({ tables: tablesStr }) => {
+    try {
+      const {
+        getCloudConfig: getCloudConfig2,
+        getConnectionString: getConnectionString2,
+        syncPush: syncPush2,
+        syncPull: syncPull2,
+        listSqliteTables: listSqliteTables2,
+        listPgTables: listPgTables2,
+        SqliteAdapter: SqliteAdapter2,
+        PgAdapterAsync: PgAdapterAsync2,
+        getDbPath: cloudGetDbPath
+      } = await Promise.resolve().then(() => (init_dist(), exports_dist));
+      const config2 = getCloudConfig2();
+      if (config2.mode === "local") {
+        return { content: [{ type: "text", text: "Error: cloud mode not configured." }], isError: true };
+      }
+      const local = new SqliteAdapter2(cloudGetDbPath("conversations"));
+      const cloud = new PgAdapterAsync2(getConnectionString2("conversations"));
+      let tableList;
+      if (tablesStr) {
+        tableList = tablesStr.split(",").map((t) => t.trim());
+      } else {
+        const localTables = new Set(listSqliteTables2(local).filter((t) => !SYNC_EXCLUDED.has(t)));
+        let remoteTables;
+        try {
+          remoteTables = new Set((await listPgTables2(cloud)).filter((t) => !SYNC_EXCLUDED.has(t)));
+        } catch {
+          local.close();
+          await cloud.close();
+          return { content: [{ type: "text", text: "Error: failed to list cloud tables." }], isError: true };
+        }
+        tableList = [...new Set([...localTables, ...remoteTables])];
+      }
+      let totalConflicts = 0;
+      for (const table of tableList) {
+        totalConflicts += await detectAndLogConflicts(local, cloud, table);
+      }
+      const pullResults = await syncPull2(cloud, local, { tables: tableList });
+      const pullTotal = pullResults.reduce((s, r) => s + r.rowsWritten, 0);
+      const pushResults = await syncPush2(local, cloud, { tables: tableList });
+      const pushTotal = pushResults.reduce((s, r) => s + r.rowsWritten, 0);
+      local.close();
+      await cloud.close();
+      const allErrors = [
+        ...pullResults.flatMap((r) => r.errors.map((e) => `pull: ${e}`)),
+        ...pushResults.flatMap((r) => r.errors.map((e) => `push: ${e}`))
+      ];
+      const lines = [
+        `Sync complete: pulled ${pullTotal} rows, pushed ${pushTotal} rows across ${tableList.length} table(s).`
+      ];
+      if (totalConflicts > 0)
+        lines.push(`Conflicts detected: ${totalConflicts} (logged to _sync_conflicts)`);
+      if (allErrors.length > 0)
+        lines.push(`Errors: ${allErrors.join("; ")}`);
+      return { content: [{ type: "text", text: lines.join(`
+`) }] };
+    } catch (e) {
+      return { content: [{ type: "text", text: formatError2(e) }], isError: true };
+    }
+  });
+  server.tool("conversations_cloud_migrate", "Run PostgreSQL migrations against the configured RDS instance to initialize the cloud schema", {
+    dry_run: exports_external2.boolean().optional().describe("Print SQL without executing")
+  }, async ({ dry_run }) => {
+    try {
+      const { getCloudConfig: getCloudConfig2, getConnectionString: getConnectionString2, PgAdapterAsync: PgAdapterAsync2 } = await Promise.resolve().then(() => (init_dist(), exports_dist));
+      const { PG_MIGRATIONS: PG_MIGRATIONS2 } = await Promise.resolve().then(() => (init_pg_migrations(), exports_pg_migrations));
+      const config2 = getCloudConfig2();
+      if (config2.mode === "local") {
+        return { content: [{ type: "text", text: "Error: cloud mode not configured." }], isError: true };
+      }
+      if (dry_run) {
+        return { content: [{ type: "text", text: PG_MIGRATIONS2.join(`
+
+---
+
+`) }] };
+      }
+      const pg = new PgAdapterAsync2(getConnectionString2("conversations"));
+      const lines = [];
+      for (let i = 0;i < PG_MIGRATIONS2.length; i++) {
+        await pg.run(PG_MIGRATIONS2[i]);
+        lines.push(`Migration ${i + 1}/${PG_MIGRATIONS2.length}: applied`);
+      }
+      await pg.close();
+      lines.push("All migrations applied successfully.");
+      return { content: [{ type: "text", text: lines.join(`
 `) }] };
     } catch (e) {
       return { content: [{ type: "text", text: formatError2(e) }], isError: true };
@@ -46452,10 +46755,19 @@ function formatError2(e) {
     return e.message;
   return String(e);
 }
-var CONFLICT_TABLES;
+var SYNC_EXCLUDED, CONFLICT_TABLES;
 var init_cloud = __esm(() => {
   init_zod2();
-  CONFLICT_TABLES = new Set(["messages", "spaces", "projects", "agent_presence"]);
+  SYNC_EXCLUDED = new Set([
+    "messages",
+    "reactions",
+    "message_read_receipts",
+    "message_mentions",
+    "messages_fts",
+    "_sync_conflicts",
+    "_migrations"
+  ]);
+  CONFLICT_TABLES = new Set(["spaces", "projects", "agent_presence"]);
 });
 
 // src/mcp/index.ts
@@ -49719,6 +50031,40 @@ program2.command("dashboard").description("Start web dashboard").option("--port
 });
 registerBrainsCommand(program2);
 registerCloudCommands(program2, "conversations");
+{
+  const cloudCmd = program2.commands.find((c) => c.name() === "cloud");
+  if (cloudCmd) {
+    cloudCmd.command("migrate").description("Run PostgreSQL migrations against the configured RDS instance").option("--dry-run", "Print SQL without executing").action(async (opts) => {
+      try {
+        const { getCloudConfig: getCloudConfig2, getConnectionString: getConnectionString2, PgAdapterAsync: PgAdapterAsync2 } = await Promise.resolve().then(() => (init_dist(), exports_dist));
+        const { PG_MIGRATIONS: PG_MIGRATIONS2 } = await Promise.resolve().then(() => (init_pg_migrations(), exports_pg_migrations));
+        const config2 = getCloudConfig2();
+        if (config2.mode === "local") {
+          console.error(chalk9.red("Error: cloud mode not configured. Set RDS credentials first."));
+          process.exit(1);
+        }
+        if (opts.dryRun) {
+          console.log(chalk9.dim(`-- Dry run: SQL that would be executed --
+`));
+          for (const sql of PG_MIGRATIONS2)
+            console.log(sql);
+          return;
+        }
+        const pg = new PgAdapterAsync2(getConnectionString2("conversations"));
+        for (let i = 0;i < PG_MIGRATIONS2.length; i++) {
+          process.stdout.write(chalk9.dim(`Running migration ${i + 1}/${PG_MIGRATIONS2.length}...`));
+          await pg.run(PG_MIGRATIONS2[i]);
+          console.log(chalk9.green(" done"));
+        }
+        await pg.close();
+        console.log(chalk9.green("\u2713 All migrations applied."));
+      } catch (e) {
+        console.error(chalk9.red(`Migration failed: ${e?.message ?? e}`));
+        process.exit(1);
+      }
+    });
+  }
+}
 program2.action(() => {
   if (!process.stdin.isTTY) {
     console.error(chalk9.red("Interactive mode requires a TTY terminal."));