stagent 0.1.0 → 0.1.2

package/src/lib/db/__tests__/bootstrap.test.ts

@@ -0,0 +1,56 @@
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import Database from "better-sqlite3";
+import { join } from "path";
+import { mkdtempSync, rmSync } from "fs";
+import { tmpdir } from "os";
+import { drizzle } from "drizzle-orm/better-sqlite3";
+import { migrate } from "drizzle-orm/better-sqlite3/migrator";
+import {
+  bootstrapStagentDatabase,
+  hasLegacyStagentTables,
+  hasMigrationHistory,
+  markAllMigrationsApplied,
+} from "../bootstrap";
+
+const migrationsFolder = join(process.cwd(), "src", "lib", "db", "migrations");
+
+describe("database bootstrap recovery", () => {
+  let tempDir: string;
+  let dbPath: string;
+
+  beforeEach(() => {
+    tempDir = mkdtempSync(join(tmpdir(), "stagent-db-bootstrap-"));
+    dbPath = join(tempDir, "stagent.db");
+  });
+
+  afterEach(() => {
+    rmSync(tempDir, { recursive: true, force: true });
+  });
+
+  it("recovers a bootstrapped database that has no drizzle migration history", () => {
+    const bootstrapDb = new Database(dbPath);
+    bootstrapStagentDatabase(bootstrapDb);
+
+    expect(hasLegacyStagentTables(bootstrapDb)).toBe(true);
+    expect(hasMigrationHistory(bootstrapDb)).toBe(false);
+
+    markAllMigrationsApplied(bootstrapDb, migrationsFolder);
+    expect(hasMigrationHistory(bootstrapDb)).toBe(true);
+    bootstrapDb.close();
+
+    const migratedDb = new Database(dbPath);
+    const drizzleDb = drizzle(migratedDb);
+
+    expect(() =>
+      migrate(drizzleDb, {
+        migrationsFolder,
+      })
+    ).not.toThrow();
+
+    const migrationCount = migratedDb
+      .prepare("SELECT COUNT(*) AS count FROM __drizzle_migrations")
+      .get() as { count: number };
+    expect(migrationCount.count).toBe(9);
+    migratedDb.close();
+  });
+});

package/src/lib/db/bootstrap.ts

@@ -0,0 +1,301 @@
+import type Database from "better-sqlite3";
+import { readMigrationFiles } from "drizzle-orm/migrator";
+
+const STAGENT_TABLES = [
+  "projects",
+  "tasks",
+  "workflows",
+  "agent_logs",
+  "notifications",
+  "settings",
+  "documents",
+  "schedules",
+  "usage_ledger",
+  "learned_context",
+] as const;
+
+export function bootstrapStagentDatabase(sqlite: Database.Database): void {
+  sqlite.exec(`
+    CREATE TABLE IF NOT EXISTS projects (
+      id TEXT PRIMARY KEY NOT NULL,
+      name TEXT NOT NULL,
+      description TEXT,
+      status TEXT DEFAULT 'active' NOT NULL,
+      created_at INTEGER NOT NULL,
+      updated_at INTEGER NOT NULL
+    );
+
+    CREATE TABLE IF NOT EXISTS tasks (
+      id TEXT PRIMARY KEY NOT NULL,
+      project_id TEXT,
+      workflow_id TEXT,
+      schedule_id TEXT,
+      title TEXT NOT NULL,
+      description TEXT,
+      status TEXT DEFAULT 'planned' NOT NULL,
+      assigned_agent TEXT,
+      agent_profile TEXT,
+      priority INTEGER DEFAULT 2 NOT NULL,
+      result TEXT,
+      session_id TEXT,
+      resume_count INTEGER DEFAULT 0 NOT NULL,
+      created_at INTEGER NOT NULL,
+      updated_at INTEGER NOT NULL,
+      FOREIGN KEY (project_id) REFERENCES projects(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
+      FOREIGN KEY (workflow_id) REFERENCES workflows(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
+      FOREIGN KEY (schedule_id) REFERENCES schedules(id) ON UPDATE NO ACTION ON DELETE NO ACTION
+    );
+
+    CREATE TABLE IF NOT EXISTS workflows (
+      id TEXT PRIMARY KEY NOT NULL,
+      project_id TEXT,
+      name TEXT NOT NULL,
+      definition TEXT NOT NULL,
+      status TEXT DEFAULT 'draft' NOT NULL,
+      created_at INTEGER NOT NULL,
+      updated_at INTEGER NOT NULL,
+      FOREIGN KEY (project_id) REFERENCES projects(id) ON UPDATE NO ACTION ON DELETE NO ACTION
+    );
+
+    CREATE TABLE IF NOT EXISTS agent_logs (
+      id TEXT PRIMARY KEY NOT NULL,
+      task_id TEXT,
+      agent_type TEXT NOT NULL,
+      event TEXT NOT NULL,
+      payload TEXT,
+      timestamp INTEGER NOT NULL,
+      FOREIGN KEY (task_id) REFERENCES tasks(id) ON UPDATE NO ACTION ON DELETE NO ACTION
+    );
+
+    CREATE TABLE IF NOT EXISTS notifications (
+      id TEXT PRIMARY KEY NOT NULL,
+      task_id TEXT,
+      type TEXT NOT NULL,
+      title TEXT NOT NULL,
+      body TEXT,
+      read INTEGER DEFAULT 0 NOT NULL,
+      tool_name TEXT,
+      tool_input TEXT,
+      response TEXT,
+      responded_at INTEGER,
+      created_at INTEGER NOT NULL,
+      FOREIGN KEY (task_id) REFERENCES tasks(id) ON UPDATE NO ACTION ON DELETE NO ACTION
+    );
+
+    CREATE TABLE IF NOT EXISTS settings (
+      key TEXT PRIMARY KEY,
+      value TEXT NOT NULL,
+      updated_at INTEGER NOT NULL
+    );
+
+    CREATE INDEX IF NOT EXISTS idx_tasks_status ON tasks(status);
+    CREATE INDEX IF NOT EXISTS idx_tasks_project_id ON tasks(project_id);
+    CREATE INDEX IF NOT EXISTS idx_agent_logs_task_id ON agent_logs(task_id);
+    CREATE INDEX IF NOT EXISTS idx_agent_logs_timestamp ON agent_logs(timestamp);
+
+    CREATE TABLE IF NOT EXISTS documents (
+      id TEXT PRIMARY KEY NOT NULL,
+      task_id TEXT,
+      project_id TEXT,
+      filename TEXT NOT NULL,
+      original_name TEXT NOT NULL,
+      mime_type TEXT NOT NULL,
+      size INTEGER NOT NULL,
+      storage_path TEXT NOT NULL,
+      version INTEGER DEFAULT 1 NOT NULL,
+      direction TEXT DEFAULT 'input' NOT NULL,
+      category TEXT,
+      status TEXT DEFAULT 'uploaded' NOT NULL,
+      extracted_text TEXT,
+      processed_path TEXT,
+      processing_error TEXT,
+      created_at INTEGER NOT NULL,
+      updated_at INTEGER NOT NULL,
+      FOREIGN KEY (task_id) REFERENCES tasks(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
+      FOREIGN KEY (project_id) REFERENCES projects(id) ON UPDATE NO ACTION ON DELETE NO ACTION
+    );
+
+    CREATE TABLE IF NOT EXISTS schedules (
+      id TEXT PRIMARY KEY NOT NULL,
+      project_id TEXT,
+      name TEXT NOT NULL,
+      prompt TEXT NOT NULL,
+      cron_expression TEXT NOT NULL,
+      assigned_agent TEXT,
+      agent_profile TEXT,
+      recurs INTEGER DEFAULT 1 NOT NULL,
+      status TEXT DEFAULT 'active' NOT NULL,
+      max_firings INTEGER,
+      firing_count INTEGER DEFAULT 0 NOT NULL,
+      expires_at INTEGER,
+      last_fired_at INTEGER,
+      next_fire_at INTEGER,
+      created_at INTEGER NOT NULL,
+      updated_at INTEGER NOT NULL,
+      FOREIGN KEY (project_id) REFERENCES projects(id) ON UPDATE NO ACTION ON DELETE NO ACTION
+    );
+
+    CREATE INDEX IF NOT EXISTS idx_schedules_status ON schedules(status);
+    CREATE INDEX IF NOT EXISTS idx_schedules_next_fire_at ON schedules(next_fire_at);
+    CREATE INDEX IF NOT EXISTS idx_schedules_project_id ON schedules(project_id);
+
+    CREATE INDEX IF NOT EXISTS idx_notifications_task_id ON notifications(task_id);
+    CREATE INDEX IF NOT EXISTS idx_notifications_read ON notifications(read);
+    CREATE INDEX IF NOT EXISTS idx_documents_task_id ON documents(task_id);
+    CREATE INDEX IF NOT EXISTS idx_documents_project_id ON documents(project_id);
+
+    CREATE TABLE IF NOT EXISTS usage_ledger (
+      id TEXT PRIMARY KEY NOT NULL,
+      task_id TEXT,
+      workflow_id TEXT,
+      schedule_id TEXT,
+      project_id TEXT,
+      activity_type TEXT NOT NULL,
+      runtime_id TEXT NOT NULL,
+      provider_id TEXT NOT NULL,
+      model_id TEXT,
+      status TEXT NOT NULL,
+      input_tokens INTEGER,
+      output_tokens INTEGER,
+      total_tokens INTEGER,
+      cost_micros INTEGER,
+      pricing_version TEXT,
+      started_at INTEGER NOT NULL,
+      finished_at INTEGER NOT NULL,
+      FOREIGN KEY (task_id) REFERENCES tasks(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
+      FOREIGN KEY (workflow_id) REFERENCES workflows(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
+      FOREIGN KEY (schedule_id) REFERENCES schedules(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
+      FOREIGN KEY (project_id) REFERENCES projects(id) ON UPDATE NO ACTION ON DELETE NO ACTION
+    );
+
+    CREATE INDEX IF NOT EXISTS idx_usage_ledger_task_id ON usage_ledger(task_id);
+    CREATE INDEX IF NOT EXISTS idx_usage_ledger_activity_type ON usage_ledger(activity_type);
+    CREATE INDEX IF NOT EXISTS idx_usage_ledger_runtime_id ON usage_ledger(runtime_id);
+    CREATE INDEX IF NOT EXISTS idx_usage_ledger_provider_model ON usage_ledger(provider_id, model_id);
+    CREATE INDEX IF NOT EXISTS idx_usage_ledger_finished_at ON usage_ledger(finished_at);
+
+    CREATE TABLE IF NOT EXISTS learned_context (
+      id TEXT PRIMARY KEY NOT NULL,
+      profile_id TEXT NOT NULL,
+      version INTEGER NOT NULL,
+      content TEXT,
+      diff TEXT,
+      change_type TEXT NOT NULL,
+      source_task_id TEXT,
+      proposal_notification_id TEXT,
+      proposed_additions TEXT,
+      approved_by TEXT,
+      created_at INTEGER NOT NULL,
+      FOREIGN KEY (source_task_id) REFERENCES tasks(id) ON UPDATE NO ACTION ON DELETE NO ACTION
+    );
+
+    CREATE INDEX IF NOT EXISTS idx_learned_context_profile_version ON learned_context(profile_id, version);
+    CREATE INDEX IF NOT EXISTS idx_learned_context_change_type ON learned_context(change_type);
+  `);
+
+  try {
+    sqlite.exec(`ALTER TABLE tasks ADD COLUMN agent_profile TEXT;`);
+  } catch {
+    // Column already exists.
+  }
+  sqlite.exec(`CREATE INDEX IF NOT EXISTS idx_tasks_agent_profile ON tasks(agent_profile);`);
+
+  try {
+    sqlite.exec(`ALTER TABLE tasks ADD COLUMN workflow_id TEXT REFERENCES workflows(id);`);
+  } catch {
+    // Column already exists.
+  }
+  sqlite.exec(`CREATE INDEX IF NOT EXISTS idx_tasks_workflow_id ON tasks(workflow_id);`);
+
+  try {
+    sqlite.exec(`ALTER TABLE tasks ADD COLUMN schedule_id TEXT REFERENCES schedules(id);`);
+  } catch {
+    // Column already exists.
+  }
+  sqlite.exec(`CREATE INDEX IF NOT EXISTS idx_tasks_schedule_id ON tasks(schedule_id);`);
+
+  try {
+    sqlite.exec(`ALTER TABLE projects ADD COLUMN working_directory TEXT;`);
+  } catch {
+    // Column already exists.
+  }
+
+  try {
+    sqlite.exec(`ALTER TABLE schedules ADD COLUMN assigned_agent TEXT;`);
+  } catch {
+    // Column already exists.
+  }
+
+  try {
+    sqlite.exec(`ALTER TABLE documents ADD COLUMN version INTEGER NOT NULL DEFAULT 1;`);
+  } catch {
+    // Column already exists.
+  }
+}
+
+export function hasLegacyStagentTables(sqlite: Database.Database): boolean {
+  const placeholders = STAGENT_TABLES.map(() => "?").join(", ");
+  const row = sqlite
+    .prepare(
+      `SELECT COUNT(*) AS count
+       FROM sqlite_master
+       WHERE type = 'table' AND name IN (${placeholders})`
+    )
+    .get(...STAGENT_TABLES) as { count: number };
+
+  return row.count > 0;
+}
+
+export function hasMigrationHistory(
+  sqlite: Database.Database,
+  migrationsTable = "__drizzle_migrations"
+): boolean {
+  const tableRow = sqlite
+    .prepare(
+      `SELECT COUNT(*) AS count
+       FROM sqlite_master
+       WHERE type = 'table' AND name = ?`
+    )
+    .get(migrationsTable) as { count: number };
+
+  if (tableRow.count === 0) {
+    return false;
+  }
+
+  const row = sqlite
+    .prepare(`SELECT COUNT(*) AS count FROM ${migrationsTable}`)
+    .get() as { count: number };
+
+  return row.count > 0;
+}
+
+export function markAllMigrationsApplied(
+  sqlite: Database.Database,
+  migrationsFolder: string,
+  migrationsTable = "__drizzle_migrations"
+): void {
+  const migrations = readMigrationFiles({ migrationsFolder, migrationsTable });
+
+  sqlite.exec(`
+    CREATE TABLE IF NOT EXISTS ${migrationsTable} (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      hash TEXT NOT NULL,
+      created_at NUMERIC
+    )
+  `);
+
+  const existing = sqlite
+    .prepare(`SELECT hash, created_at FROM ${migrationsTable}`)
+    .all() as Array<{ hash: string; created_at: number }>;
+  const seen = new Set(existing.map((row) => `${row.hash}:${row.created_at}`));
+  const insert = sqlite.prepare(
+    `INSERT INTO ${migrationsTable} (hash, created_at) VALUES (?, ?)`
+  );
+
+  for (const migration of migrations) {
+    const key = `${migration.hash}:${migration.folderMillis}`;
+    if (!seen.has(key)) {
+      insert.run(migration.hash, migration.folderMillis);
+    }
+  }
+}

package/src/lib/db/index.ts

@@ -4,6 +4,7 @@ import * as schema from "./schema";
 import { join } from "path";
 import { mkdirSync } from "fs";
 import { getStagentDataDir } from "@/lib/utils/stagent-paths";
+import { bootstrapStagentDatabase } from "./bootstrap";
 
 const dataDir = getStagentDataDir();
 mkdirSync(dataDir, { recursive: true });
@@ -12,210 +13,6 @@ const dbPath = join(dataDir, "stagent.db");
 const sqlite = new Database(dbPath);
 sqlite.pragma("journal_mode = WAL");
 sqlite.pragma("foreign_keys = ON");
-
-// Bootstrap all tables (migrations may not have been applied on fresh install)
-// Note: sqlite.exec() here is better-sqlite3's synchronous DDL method, not child_process
-sqlite.exec(`
-  CREATE TABLE IF NOT EXISTS projects (
-    id TEXT PRIMARY KEY NOT NULL,
-    name TEXT NOT NULL,
-    description TEXT,
-    status TEXT DEFAULT 'active' NOT NULL,
-    created_at INTEGER NOT NULL,
-    updated_at INTEGER NOT NULL
-  );
-
-  CREATE TABLE IF NOT EXISTS tasks (
-    id TEXT PRIMARY KEY NOT NULL,
-    project_id TEXT,
-    workflow_id TEXT,
-    schedule_id TEXT,
-    title TEXT NOT NULL,
-    description TEXT,
-    status TEXT DEFAULT 'planned' NOT NULL,
-    assigned_agent TEXT,
-    agent_profile TEXT,
-    priority INTEGER DEFAULT 2 NOT NULL,
-    result TEXT,
-    session_id TEXT,
-    resume_count INTEGER DEFAULT 0 NOT NULL,
-    created_at INTEGER NOT NULL,
-    updated_at INTEGER NOT NULL,
-    FOREIGN KEY (project_id) REFERENCES projects(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
-    FOREIGN KEY (workflow_id) REFERENCES workflows(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
-    FOREIGN KEY (schedule_id) REFERENCES schedules(id) ON UPDATE NO ACTION ON DELETE NO ACTION
-  );
-
-  CREATE TABLE IF NOT EXISTS workflows (
-    id TEXT PRIMARY KEY NOT NULL,
-    project_id TEXT,
-    name TEXT NOT NULL,
-    definition TEXT NOT NULL,
-    status TEXT DEFAULT 'draft' NOT NULL,
-    created_at INTEGER NOT NULL,
-    updated_at INTEGER NOT NULL,
-    FOREIGN KEY (project_id) REFERENCES projects(id) ON UPDATE NO ACTION ON DELETE NO ACTION
-  );
-
-  CREATE TABLE IF NOT EXISTS agent_logs (
-    id TEXT PRIMARY KEY NOT NULL,
-    task_id TEXT,
-    agent_type TEXT NOT NULL,
-    event TEXT NOT NULL,
-    payload TEXT,
-    timestamp INTEGER NOT NULL,
-    FOREIGN KEY (task_id) REFERENCES tasks(id) ON UPDATE NO ACTION ON DELETE NO ACTION
-  );
-
-  CREATE TABLE IF NOT EXISTS notifications (
-    id TEXT PRIMARY KEY NOT NULL,
-    task_id TEXT,
-    type TEXT NOT NULL,
-    title TEXT NOT NULL,
-    body TEXT,
-    read INTEGER DEFAULT 0 NOT NULL,
-    tool_name TEXT,
-    tool_input TEXT,
-    response TEXT,
-    responded_at INTEGER,
-    created_at INTEGER NOT NULL,
-    FOREIGN KEY (task_id) REFERENCES tasks(id) ON UPDATE NO ACTION ON DELETE NO ACTION
-  );
-
-  CREATE TABLE IF NOT EXISTS settings (
-    key TEXT PRIMARY KEY,
-    value TEXT NOT NULL,
-    updated_at INTEGER NOT NULL
-  );
-
-  CREATE INDEX IF NOT EXISTS idx_tasks_status ON tasks(status);
-  CREATE INDEX IF NOT EXISTS idx_tasks_project_id ON tasks(project_id);
-  CREATE INDEX IF NOT EXISTS idx_agent_logs_task_id ON agent_logs(task_id);
-  CREATE INDEX IF NOT EXISTS idx_agent_logs_timestamp ON agent_logs(timestamp);
-  CREATE TABLE IF NOT EXISTS documents (
-    id TEXT PRIMARY KEY NOT NULL,
-    task_id TEXT,
-    project_id TEXT,
-    filename TEXT NOT NULL,
-    original_name TEXT NOT NULL,
-    mime_type TEXT NOT NULL,
-    size INTEGER NOT NULL,
-    storage_path TEXT NOT NULL,
-    version INTEGER DEFAULT 1 NOT NULL,
-    direction TEXT DEFAULT 'input' NOT NULL,
-    category TEXT,
-    status TEXT DEFAULT 'uploaded' NOT NULL,
-    extracted_text TEXT,
-    processed_path TEXT,
-    processing_error TEXT,
-    created_at INTEGER NOT NULL,
-    updated_at INTEGER NOT NULL,
-    FOREIGN KEY (task_id) REFERENCES tasks(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
-    FOREIGN KEY (project_id) REFERENCES projects(id) ON UPDATE NO ACTION ON DELETE NO ACTION
-  );
-
-  CREATE TABLE IF NOT EXISTS schedules (
-    id TEXT PRIMARY KEY NOT NULL,
-    project_id TEXT,
-    name TEXT NOT NULL,
-    prompt TEXT NOT NULL,
-    cron_expression TEXT NOT NULL,
-    assigned_agent TEXT,
-    agent_profile TEXT,
-    recurs INTEGER DEFAULT 1 NOT NULL,
-    status TEXT DEFAULT 'active' NOT NULL,
-    max_firings INTEGER,
-    firing_count INTEGER DEFAULT 0 NOT NULL,
-    expires_at INTEGER,
-    last_fired_at INTEGER,
-    next_fire_at INTEGER,
-    created_at INTEGER NOT NULL,
-    updated_at INTEGER NOT NULL,
-    FOREIGN KEY (project_id) REFERENCES projects(id) ON UPDATE NO ACTION ON DELETE NO ACTION
-  );
-
-  CREATE INDEX IF NOT EXISTS idx_schedules_status ON schedules(status);
-  CREATE INDEX IF NOT EXISTS idx_schedules_next_fire_at ON schedules(next_fire_at);
-  CREATE INDEX IF NOT EXISTS idx_schedules_project_id ON schedules(project_id);
-
-  CREATE INDEX IF NOT EXISTS idx_notifications_task_id ON notifications(task_id);
-  CREATE INDEX IF NOT EXISTS idx_notifications_read ON notifications(read);
-  CREATE INDEX IF NOT EXISTS idx_documents_task_id ON documents(task_id);
-  CREATE INDEX IF NOT EXISTS idx_documents_project_id ON documents(project_id);
-
-  CREATE TABLE IF NOT EXISTS usage_ledger (
-    id TEXT PRIMARY KEY NOT NULL,
-    task_id TEXT,
-    workflow_id TEXT,
-    schedule_id TEXT,
-    project_id TEXT,
-    activity_type TEXT NOT NULL,
-    runtime_id TEXT NOT NULL,
-    provider_id TEXT NOT NULL,
-    model_id TEXT,
-    status TEXT NOT NULL,
-    input_tokens INTEGER,
-    output_tokens INTEGER,
-    total_tokens INTEGER,
-    cost_micros INTEGER,
-    pricing_version TEXT,
-    started_at INTEGER NOT NULL,
-    finished_at INTEGER NOT NULL,
-    FOREIGN KEY (task_id) REFERENCES tasks(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
-    FOREIGN KEY (workflow_id) REFERENCES workflows(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
-    FOREIGN KEY (schedule_id) REFERENCES schedules(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
-    FOREIGN KEY (project_id) REFERENCES projects(id) ON UPDATE NO ACTION ON DELETE NO ACTION
-  );
-
-  CREATE INDEX IF NOT EXISTS idx_usage_ledger_task_id ON usage_ledger(task_id);
-  CREATE INDEX IF NOT EXISTS idx_usage_ledger_activity_type ON usage_ledger(activity_type);
-  CREATE INDEX IF NOT EXISTS idx_usage_ledger_runtime_id ON usage_ledger(runtime_id);
-  CREATE INDEX IF NOT EXISTS idx_usage_ledger_provider_model ON usage_ledger(provider_id, model_id);
-  CREATE INDEX IF NOT EXISTS idx_usage_ledger_finished_at ON usage_ledger(finished_at);
-`);
-
-// Migration: add agent_profile column to existing tasks table (safe to re-run)
-// Note: sqlite.exec() here is better-sqlite3's synchronous DDL method, not child_process
-try {
-  sqlite.exec(`ALTER TABLE tasks ADD COLUMN agent_profile TEXT;`);
-} catch {
-  // Column already exists — ignore
-}
-sqlite.exec(`CREATE INDEX IF NOT EXISTS idx_tasks_agent_profile ON tasks(agent_profile);`);
-
-try {
-  sqlite.exec(`ALTER TABLE tasks ADD COLUMN workflow_id TEXT REFERENCES workflows(id);`);
-} catch {
-  // Column already exists — ignore
-}
-sqlite.exec(`CREATE INDEX IF NOT EXISTS idx_tasks_workflow_id ON tasks(workflow_id);`);
-
-try {
-  sqlite.exec(`ALTER TABLE tasks ADD COLUMN schedule_id TEXT REFERENCES schedules(id);`);
-} catch {
-  // Column already exists — ignore
-}
-sqlite.exec(`CREATE INDEX IF NOT EXISTS idx_tasks_schedule_id ON tasks(schedule_id);`);
-
-// Migration: add working_directory column to existing projects table (safe to re-run)
-// Note: sqlite.exec() here is better-sqlite3's synchronous DDL method, not child_process
-try {
-  sqlite.exec(`ALTER TABLE projects ADD COLUMN working_directory TEXT;`);
-} catch {
-  // Column already exists — ignore
-}
-
-// Migration: add assigned_agent column to existing schedules table (safe to re-run)
-try {
-  sqlite.exec(`ALTER TABLE schedules ADD COLUMN assigned_agent TEXT;`);
-} catch {
-  // Column already exists — ignore
-}
-
-try {
-  sqlite.exec(`ALTER TABLE documents ADD COLUMN version INTEGER NOT NULL DEFAULT 1;`);
-} catch {
-  // Column already exists — ignore
-}
+bootstrapStagentDatabase(sqlite);
 
 export const db = drizzle(sqlite, { schema });

package/src/lib/db/migrations/0004_add_documents.sql

@@ -15,6 +15,7 @@ CREATE TABLE IF NOT EXISTS documents (
   FOREIGN KEY (task_id) REFERENCES tasks(id) ON UPDATE NO ACTION ON DELETE NO ACTION,
   FOREIGN KEY (project_id) REFERENCES projects(id) ON UPDATE NO ACTION ON DELETE NO ACTION
 );
-
+--> statement-breakpoint
 CREATE INDEX IF NOT EXISTS idx_documents_task_id ON documents(task_id);
+--> statement-breakpoint
 CREATE INDEX IF NOT EXISTS idx_documents_project_id ON documents(project_id);

package/src/lib/db/migrations/0005_add_document_preprocessing.sql

@@ -1,4 +1,6 @@
 -- Add preprocessing columns to documents table
 ALTER TABLE documents ADD COLUMN extracted_text TEXT;
+--> statement-breakpoint
 ALTER TABLE documents ADD COLUMN processed_path TEXT;
+--> statement-breakpoint
 ALTER TABLE documents ADD COLUMN processing_error TEXT;

package/src/lib/db/migrations/0006_add_agent_profile.sql

@@ -1,2 +1,3 @@
 ALTER TABLE tasks ADD COLUMN agent_profile TEXT;
+--> statement-breakpoint
 CREATE INDEX IF NOT EXISTS idx_tasks_agent_profile ON tasks(agent_profile);

package/src/lib/db/migrations/0007_add_usage_metering_ledger.sql

@@ -1,8 +1,11 @@
 ALTER TABLE tasks ADD COLUMN workflow_id TEXT REFERENCES workflows(id);
+--> statement-breakpoint
 ALTER TABLE tasks ADD COLUMN schedule_id TEXT REFERENCES schedules(id);
+--> statement-breakpoint
 CREATE INDEX IF NOT EXISTS idx_tasks_workflow_id ON tasks(workflow_id);
+--> statement-breakpoint
 CREATE INDEX IF NOT EXISTS idx_tasks_schedule_id ON tasks(schedule_id);
-
+--> statement-breakpoint
 CREATE TABLE IF NOT EXISTS usage_ledger (
   id TEXT PRIMARY KEY NOT NULL,
   task_id TEXT REFERENCES tasks(id),
@@ -22,9 +25,13 @@ CREATE TABLE IF NOT EXISTS usage_ledger (
   started_at INTEGER NOT NULL,
   finished_at INTEGER NOT NULL
 );
-
+--> statement-breakpoint
 CREATE INDEX IF NOT EXISTS idx_usage_ledger_task_id ON usage_ledger(task_id);
+--> statement-breakpoint
 CREATE INDEX IF NOT EXISTS idx_usage_ledger_activity_type ON usage_ledger(activity_type);
+--> statement-breakpoint
 CREATE INDEX IF NOT EXISTS idx_usage_ledger_runtime_id ON usage_ledger(runtime_id);
+--> statement-breakpoint
 CREATE INDEX IF NOT EXISTS idx_usage_ledger_provider_model ON usage_ledger(provider_id, model_id);
+--> statement-breakpoint
 CREATE INDEX IF NOT EXISTS idx_usage_ledger_finished_at ON usage_ledger(finished_at);

package/src/lib/db/migrations/meta/_journal.json

@@ -22,6 +22,48 @@
       "when": 1773187200000,
       "tag": "0002_add_resume_count",
       "breakpoints": true
+    },
+    {
+      "idx": 3,
+      "version": "6",
+      "when": 1773273600000,
+      "tag": "0003_add_settings",
+      "breakpoints": true
+    },
+    {
+      "idx": 4,
+      "version": "6",
+      "when": 1773360000000,
+      "tag": "0004_add_documents",
+      "breakpoints": true
+    },
+    {
+      "idx": 5,
+      "version": "6",
+      "when": 1773446400000,
+      "tag": "0005_add_document_preprocessing",
+      "breakpoints": true
+    },
+    {
+      "idx": 6,
+      "version": "6",
+      "when": 1773532800000,
+      "tag": "0006_add_agent_profile",
+      "breakpoints": true
+    },
+    {
+      "idx": 7,
+      "version": "6",
+      "when": 1773619200000,
+      "tag": "0007_add_usage_metering_ledger",
+      "breakpoints": true
+    },
+    {
+      "idx": 8,
+      "version": "6",
+      "when": 1773705600000,
+      "tag": "0008_add_document_version",
+      "breakpoints": true
     }
   ]
-}
+}