bopodev-db 0.1.1

package/package.json

@@ -0,0 +1,19 @@
+{
+  "name": "bopodev-db",
+  "version": "0.1.1",
+  "license": "MIT",
+  "type": "module",
+  "main": "src/index.ts",
+  "types": "src/index.ts",
+  "dependencies": {
+    "@electric-sql/pglite": "^0.3.7",
+    "drizzle-orm": "^0.44.5",
+    "drizzle-zod": "^0.8.2",
+    "nanoid": "^5.1.5"
+  },
+  "scripts": {
+    "build": "tsc -p tsconfig.json --emitDeclarationOnly",
+    "lint": "tsc -p tsconfig.json --noEmit",
+    "typecheck": "tsc -p tsconfig.json --noEmit"
+  }
+}

package/src/bootstrap.ts

@@ -0,0 +1,213 @@
+import { sql } from "drizzle-orm";
+import { createDb } from "./client";
+
+export async function bootstrapDatabase(dbPath?: string) {
+  const { db, client } = await createDb(dbPath);
+
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS companies (
+      id TEXT PRIMARY KEY,
+      name TEXT NOT NULL,
+      mission TEXT,
+      created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+    );
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS projects (
+      id TEXT PRIMARY KEY,
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      name TEXT NOT NULL,
+      description TEXT,
+      status TEXT NOT NULL DEFAULT 'planned',
+      planned_start_at TIMESTAMP,
+      workspace_local_path TEXT,
+      workspace_github_repo TEXT,
+      created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+    );
+  `);
+  await db.execute(sql`
+    ALTER TABLE projects
+    ADD COLUMN IF NOT EXISTS status TEXT NOT NULL DEFAULT 'planned';
+  `);
+  await db.execute(sql`
+    ALTER TABLE projects
+    ADD COLUMN IF NOT EXISTS planned_start_at TIMESTAMP;
+  `);
+  await db.execute(sql`
+    ALTER TABLE projects
+    ADD COLUMN IF NOT EXISTS workspace_local_path TEXT;
+  `);
+  await db.execute(sql`
+    ALTER TABLE projects
+    ADD COLUMN IF NOT EXISTS workspace_github_repo TEXT;
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS goals (
+      id TEXT PRIMARY KEY,
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      project_id TEXT REFERENCES projects(id) ON DELETE SET NULL,
+      parent_goal_id TEXT,
+      level TEXT NOT NULL,
+      title TEXT NOT NULL,
+      description TEXT,
+      status TEXT NOT NULL DEFAULT 'draft',
+      created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+      updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+    );
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS agents (
+      id TEXT PRIMARY KEY,
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      manager_agent_id TEXT,
+      role TEXT NOT NULL,
+      name TEXT NOT NULL,
+      provider_type TEXT NOT NULL,
+      status TEXT NOT NULL DEFAULT 'idle',
+      heartbeat_cron TEXT NOT NULL,
+      monthly_budget_usd NUMERIC(12, 4) NOT NULL DEFAULT 0,
+      used_budget_usd NUMERIC(12, 4) NOT NULL DEFAULT 0,
+      token_usage INTEGER NOT NULL DEFAULT 0,
+      can_hire_agents BOOLEAN NOT NULL DEFAULT false,
+      state_blob TEXT NOT NULL DEFAULT '{}',
+      created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+      updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+    );
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS issues (
+      id TEXT PRIMARY KEY,
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
+      parent_issue_id TEXT,
+      title TEXT NOT NULL,
+      body TEXT,
+      status TEXT NOT NULL DEFAULT 'todo',
+      priority TEXT NOT NULL DEFAULT 'none',
+      assignee_agent_id TEXT,
+      labels_json TEXT NOT NULL DEFAULT '[]',
+      tags_json TEXT NOT NULL DEFAULT '[]',
+      is_claimed BOOLEAN NOT NULL DEFAULT false,
+      claimed_by_heartbeat_run_id TEXT,
+      created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+      updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+    );
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS issue_comments (
+      id TEXT PRIMARY KEY,
+      issue_id TEXT NOT NULL REFERENCES issues(id) ON DELETE CASCADE,
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      author_type TEXT NOT NULL,
+      author_id TEXT,
+      body TEXT NOT NULL,
+      created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+    );
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS activity_logs (
+      id TEXT PRIMARY KEY,
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      issue_id TEXT REFERENCES issues(id) ON DELETE SET NULL,
+      actor_type TEXT NOT NULL,
+      actor_id TEXT,
+      event_type TEXT NOT NULL,
+      payload_json TEXT NOT NULL DEFAULT '{}',
+      created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+    );
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS heartbeat_runs (
+      id TEXT PRIMARY KEY,
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      agent_id TEXT NOT NULL REFERENCES agents(id) ON DELETE CASCADE,
+      status TEXT NOT NULL DEFAULT 'started',
+      started_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+      finished_at TIMESTAMP,
+      message TEXT
+    );
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS approval_requests (
+      id TEXT PRIMARY KEY,
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      requested_by_agent_id TEXT,
+      action TEXT NOT NULL,
+      payload_json TEXT NOT NULL DEFAULT '{}',
+      status TEXT NOT NULL DEFAULT 'pending',
+      created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+      resolved_at TIMESTAMP
+    );
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS approval_inbox_states (
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      actor_id TEXT NOT NULL,
+      approval_id TEXT NOT NULL REFERENCES approval_requests(id) ON DELETE CASCADE,
+      seen_at TIMESTAMP,
+      dismissed_at TIMESTAMP,
+      created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+      updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+      PRIMARY KEY (company_id, actor_id, approval_id)
+    );
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS cost_ledger (
+      id TEXT PRIMARY KEY,
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      project_id TEXT REFERENCES projects(id) ON DELETE SET NULL,
+      issue_id TEXT REFERENCES issues(id) ON DELETE SET NULL,
+      agent_id TEXT REFERENCES agents(id) ON DELETE SET NULL,
+      provider_type TEXT NOT NULL,
+      token_input INTEGER NOT NULL DEFAULT 0,
+      token_output INTEGER NOT NULL DEFAULT 0,
+      usd_cost NUMERIC(12, 6) NOT NULL DEFAULT 0,
+      created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+    );
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS audit_events (
+      id TEXT PRIMARY KEY,
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      actor_type TEXT NOT NULL,
+      actor_id TEXT,
+      event_type TEXT NOT NULL,
+      entity_type TEXT NOT NULL,
+      entity_id TEXT NOT NULL,
+      correlation_id TEXT,
+      payload_json TEXT NOT NULL DEFAULT '{}',
+      created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
+    );
+  `);
+  await db.execute(sql`
+    CREATE TABLE IF NOT EXISTS agent_issue_labels (
+      company_id TEXT NOT NULL REFERENCES companies(id) ON DELETE CASCADE,
+      issue_id TEXT NOT NULL REFERENCES issues(id) ON DELETE CASCADE,
+      label TEXT NOT NULL,
+      PRIMARY KEY (company_id, issue_id, label)
+    );
+  `);
+  await db.execute(sql`
+    CREATE INDEX IF NOT EXISTS idx_issues_company_status
+      ON issues (company_id, status, updated_at);
+  `);
+  await db.execute(sql`
+    CREATE INDEX IF NOT EXISTS idx_audit_events_company_created
+      ON audit_events (company_id, created_at DESC);
+  `);
+  await db.execute(sql`
+    CREATE INDEX IF NOT EXISTS idx_cost_ledger_company_created
+      ON cost_ledger (company_id, created_at DESC);
+  `);
+  await db.execute(sql`
+    CREATE UNIQUE INDEX IF NOT EXISTS idx_heartbeat_runs_single_started
+      ON heartbeat_runs (company_id, agent_id)
+      WHERE status = 'started';
+  `);
+  await db.execute(sql`
+    CREATE INDEX IF NOT EXISTS idx_approval_inbox_states_company_actor_updated
+      ON approval_inbox_states (company_id, actor_id, updated_at DESC);
+  `);
+
+  return { db, client };
+}

package/src/client.ts

@@ -0,0 +1,16 @@
+import { mkdir } from "node:fs/promises";
+import { dirname, resolve } from "node:path";
+import { PGlite } from "@electric-sql/pglite";
+import { drizzle } from "drizzle-orm/pglite";
+import * as dbSchema from "./schema";
+
+export type BopoDb = ReturnType<typeof drizzle<typeof dbSchema>>;
+
+const defaultDbPath = resolve(process.cwd(), "data", "bopohq.db");
+
+export async function createDb(dbPath = defaultDbPath) {
+  await mkdir(dirname(dbPath), { recursive: true });
+  const client = new PGlite(dbPath);
+  const db = drizzle({ client, schema: dbSchema });
+  return { db, client };
+}

package/src/index.ts

@@ -0,0 +1,4 @@
+export * from "./bootstrap";
+export * from "./client";
+export * from "./repositories";
+export * from "./schema";