bopodev-db 0.1.30 → 0.1.32

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bopodev-db",
-  "version": "0.1.30",
+  "version": "0.1.32",
   "license": "MIT",
   "type": "module",
   "main": "src/index.ts",

package/src/migrations/0005_work_loops.sql

@@ -0,0 +1,67 @@
+CREATE TABLE "work_loops" (
+	"id" text PRIMARY KEY NOT NULL,
+	"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 REFERENCES "issues"("id") ON DELETE SET NULL,
+	"goal_ids_json" text DEFAULT '[]' NOT NULL,
+	"title" text NOT NULL,
+	"description" text,
+	"assignee_agent_id" text NOT NULL REFERENCES "agents"("id") ON DELETE RESTRICT,
+	"priority" text DEFAULT 'medium' NOT NULL,
+	"status" text DEFAULT 'active' NOT NULL,
+	"concurrency_policy" text DEFAULT 'coalesce_if_active' NOT NULL,
+	"catch_up_policy" text DEFAULT 'skip_missed' NOT NULL,
+	"last_triggered_at" timestamp,
+	"created_at" timestamp DEFAULT CURRENT_TIMESTAMP NOT NULL,
+	"updated_at" timestamp DEFAULT CURRENT_TIMESTAMP NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "work_loop_triggers" (
+	"id" text PRIMARY KEY NOT NULL,
+	"company_id" text NOT NULL REFERENCES "companies"("id") ON DELETE CASCADE,
+	"work_loop_id" text NOT NULL REFERENCES "work_loops"("id") ON DELETE CASCADE,
+	"kind" text DEFAULT 'schedule' NOT NULL,
+	"label" text,
+	"enabled" boolean DEFAULT true NOT NULL,
+	"cron_expression" text NOT NULL,
+	"timezone" text DEFAULT 'UTC' NOT NULL,
+	"next_run_at" timestamp,
+	"last_fired_at" timestamp,
+	"last_result" text,
+	"created_at" timestamp DEFAULT CURRENT_TIMESTAMP NOT NULL,
+	"updated_at" timestamp DEFAULT CURRENT_TIMESTAMP NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "work_loop_runs" (
+	"id" text PRIMARY KEY NOT NULL,
+	"company_id" text NOT NULL REFERENCES "companies"("id") ON DELETE CASCADE,
+	"work_loop_id" text NOT NULL REFERENCES "work_loops"("id") ON DELETE CASCADE,
+	"trigger_id" text REFERENCES "work_loop_triggers"("id") ON DELETE SET NULL,
+	"source" text NOT NULL,
+	"status" text DEFAULT 'received' NOT NULL,
+	"triggered_at" timestamp DEFAULT CURRENT_TIMESTAMP NOT NULL,
+	"idempotency_key" text,
+	"payload_json" text DEFAULT '{}' NOT NULL,
+	"linked_issue_id" text REFERENCES "issues"("id") ON DELETE SET NULL,
+	"coalesced_into_run_id" text REFERENCES "work_loop_runs"("id") ON DELETE SET NULL,
+	"failure_reason" text,
+	"completed_at" timestamp,
+	"created_at" timestamp DEFAULT CURRENT_TIMESTAMP NOT NULL,
+	"updated_at" timestamp DEFAULT CURRENT_TIMESTAMP NOT NULL
+);
+--> statement-breakpoint
+ALTER TABLE "issues" ADD COLUMN "loop_id" text REFERENCES "work_loops"("id") ON DELETE SET NULL;
+--> statement-breakpoint
+ALTER TABLE "issues" ADD COLUMN "loop_run_id" text REFERENCES "work_loop_runs"("id") ON DELETE SET NULL;
+--> statement-breakpoint
+CREATE INDEX "work_loops_company_status_idx" ON "work_loops" ("company_id","status");
+--> statement-breakpoint
+CREATE INDEX "work_loop_triggers_company_next_run_idx" ON "work_loop_triggers" ("company_id","next_run_at");
+--> statement-breakpoint
+CREATE INDEX "work_loop_triggers_work_loop_idx" ON "work_loop_triggers" ("work_loop_id");
+--> statement-breakpoint
+CREATE INDEX "work_loop_runs_work_loop_created_idx" ON "work_loop_runs" ("work_loop_id","created_at");
+--> statement-breakpoint
+CREATE INDEX "work_loop_runs_trigger_idempotency_idx" ON "work_loop_runs" ("trigger_id","idempotency_key");
+--> statement-breakpoint
+CREATE INDEX "issues_loop_id_idx" ON "issues" ("loop_id");

package/src/migrations/0006_company_assistant_chat.sql

@@ -0,0 +1,20 @@
+CREATE TABLE "company_assistant_threads" (
+	"id" text PRIMARY KEY NOT NULL,
+	"company_id" text NOT NULL REFERENCES "companies"("id") ON DELETE CASCADE,
+	"created_at" timestamp DEFAULT now() NOT NULL,
+	"updated_at" timestamp DEFAULT now() NOT NULL
+);
+
+CREATE INDEX "company_assistant_threads_company_idx" ON "company_assistant_threads" ("company_id");
+
+CREATE TABLE "company_assistant_messages" (
+	"id" text PRIMARY KEY NOT NULL,
+	"thread_id" text NOT NULL REFERENCES "company_assistant_threads"("id") ON DELETE CASCADE,
+	"company_id" text NOT NULL REFERENCES "companies"("id") ON DELETE CASCADE,
+	"role" text NOT NULL,
+	"body" text NOT NULL,
+	"metadata_json" text,
+	"created_at" timestamp DEFAULT now() NOT NULL
+);
+
+CREATE INDEX "company_assistant_messages_thread_created_idx" ON "company_assistant_messages" ("thread_id","created_at");

package/src/migrations/0007_cost_ledger_company_assistant.sql

@@ -0,0 +1,7 @@
+ALTER TABLE "cost_ledger" ADD COLUMN "cost_category" text;
+--> statement-breakpoint
+ALTER TABLE "cost_ledger" ADD COLUMN "assistant_thread_id" text REFERENCES "company_assistant_threads"("id") ON DELETE SET NULL;
+--> statement-breakpoint
+ALTER TABLE "cost_ledger" ADD COLUMN "assistant_message_id" text REFERENCES "company_assistant_messages"("id") ON DELETE SET NULL;
+--> statement-breakpoint
+CREATE INDEX "idx_cost_ledger_company_category_created" ON "cost_ledger" ("company_id", "cost_category", "created_at");

package/src/migrations/meta/_journal.json

@@ -36,6 +36,27 @@
       "when": 1742900000000,
       "tag": "0004_agents_capabilities",
       "breakpoints": true
+    },
+    {
+      "idx": 5,
+      "version": "7",
+      "when": 1743000000000,
+      "tag": "0005_work_loops",
+      "breakpoints": true
+    },
+    {
+      "idx": 6,
+      "version": "7",
+      "when": 1743100000000,
+      "tag": "0006_company_assistant_chat",
+      "breakpoints": true
+    },
+    {
+      "idx": 7,
+      "version": "7",
+      "when": 1743200000000,
+      "tag": "0007_cost_ledger_company_assistant",
+      "breakpoints": true
     }
   ]
 }

package/src/repositories/company-assistant-chat.ts

@@ -0,0 +1,126 @@
+import { and, asc, count, desc, eq, gte, lt } from "drizzle-orm";
+import { nanoid } from "nanoid";
+import type { BopoDb } from "../client";
+import { companyAssistantMessages, companyAssistantThreads } from "../schema";
+
+export type AssistantMessageRole = "user" | "assistant" | "system";
+
+export async function getOrCreateAssistantThread(db: BopoDb, companyId: string) {
+  const [existing] = await db
+    .select()
+    .from(companyAssistantThreads)
+    .where(eq(companyAssistantThreads.companyId, companyId))
+    .orderBy(desc(companyAssistantThreads.updatedAt))
+    .limit(1);
+  if (existing) {
+    return existing;
+  }
+  return createAssistantThreadRow(db, companyId);
+}
+
+async function createAssistantThreadRow(db: BopoDb, companyId: string) {
+  const id = nanoid(16);
+  const now = new Date();
+  await db.insert(companyAssistantThreads).values({
+    id,
+    companyId,
+    createdAt: now,
+    updatedAt: now
+  });
+  const [row] = await db.select().from(companyAssistantThreads).where(eq(companyAssistantThreads.id, id)).limit(1);
+  return row!;
+}
+
+/** New empty thread; previous threads and messages remain in the database. */
+export async function createAssistantThread(db: BopoDb, companyId: string) {
+  return createAssistantThreadRow(db, companyId);
+}
+
+export async function getAssistantThreadById(db: BopoDb, companyId: string, threadId: string) {
+  const [row] = await db
+    .select()
+    .from(companyAssistantThreads)
+    .where(and(eq(companyAssistantThreads.id, threadId), eq(companyAssistantThreads.companyId, companyId)))
+    .limit(1);
+  return row ?? null;
+}
+
+export async function touchAssistantThread(db: BopoDb, threadId: string) {
+  await db
+    .update(companyAssistantThreads)
+    .set({ updatedAt: new Date() })
+    .where(eq(companyAssistantThreads.id, threadId));
+}
+
+export async function insertAssistantMessage(
+  db: BopoDb,
+  input: {
+    threadId: string;
+    companyId: string;
+    role: AssistantMessageRole;
+    body: string;
+    metadataJson?: string | null;
+  }
+) {
+  const id = nanoid(16);
+  await db.insert(companyAssistantMessages).values({
+    id,
+    threadId: input.threadId,
+    companyId: input.companyId,
+    role: input.role,
+    body: input.body,
+    metadataJson: input.metadataJson ?? null
+  });
+  await touchAssistantThread(db, input.threadId);
+  const [row] = await db.select().from(companyAssistantMessages).where(eq(companyAssistantMessages.id, id)).limit(1);
+  return row!;
+}
+
+export async function listAssistantMessages(db: BopoDb, threadId: string, limit = 100) {
+  const capped = Math.min(Math.max(1, limit), 200);
+  return db
+    .select()
+    .from(companyAssistantMessages)
+    .where(eq(companyAssistantMessages.threadId, threadId))
+    .orderBy(asc(companyAssistantMessages.createdAt))
+    .limit(capped);
+}
+
+/** Threads with at least one message in `[startInclusive, endExclusive)` on `created_at`. */
+export async function listAssistantChatThreadStatsInCreatedAtRange(
+  db: BopoDb,
+  companyId: string,
+  startInclusive: Date,
+  endExclusive: Date
+): Promise<Array<{ threadId: string; messageCount: number }>> {
+  const rows = await db
+    .select({
+      threadId: companyAssistantMessages.threadId,
+      messageCount: count()
+    })
+    .from(companyAssistantMessages)
+    .where(
+      and(
+        eq(companyAssistantMessages.companyId, companyId),
+        gte(companyAssistantMessages.createdAt, startInclusive),
+        lt(companyAssistantMessages.createdAt, endExclusive)
+      )
+    )
+    .groupBy(companyAssistantMessages.threadId);
+  return rows.map((r) => ({
+    threadId: r.threadId,
+    messageCount: Number(r.messageCount) || 0
+  }));
+}
+
+/** Threads with at least one message in the UTC calendar month (for callers without local bounds). */
+export async function listAssistantChatThreadStatsInUtcMonth(
+  db: BopoDb,
+  companyId: string,
+  year: number,
+  month1Based: number
+): Promise<Array<{ threadId: string; messageCount: number }>> {
+  const startUtc = new Date(Date.UTC(year, month1Based - 1, 1, 0, 0, 0, 0));
+  const endExclusiveUtc = new Date(Date.UTC(year, month1Based, 1, 0, 0, 0, 0));
+  return listAssistantChatThreadStatsInCreatedAtRange(db, companyId, startUtc, endExclusiveUtc);
+}

package/src/repositories/index.ts

@@ -1,3 +1,4 @@
 export * from "./helpers";
 export * from "./companies";
+export * from "./company-assistant-chat";
 export * from "./legacy";

package/src/repositories/legacy.ts

@@ -1,4 +1,4 @@
-import { and, asc, desc, eq, gt, inArray, notInArray, sql } from "drizzle-orm";
+import { and, asc, desc, eq, gt, gte, inArray, lt, notInArray, sql } from "drizzle-orm";
 import { nanoid } from "nanoid";
 import type { BopoDb } from "../client";
 import {
@@ -521,6 +521,8 @@ export async function createIssue(
     assigneeAgentId?: string | null;
     labels?: string[];
     tags?: string[];
+    loopId?: string | null;
+    loopRunId?: string | null;
   }
 ) {
   await assertProjectBelongsToCompany(db, input.companyId, input.projectId);
@@ -546,7 +548,9 @@ export async function createIssue(
         priority: input.priority ?? "none",
         assigneeAgentId: input.assigneeAgentId ?? null,
         labelsJson: JSON.stringify(input.labels ?? []),
-        tagsJson: JSON.stringify(input.tags ?? [])
+        tagsJson: JSON.stringify(input.tags ?? []),
+        loopId: input.loopId ?? null,
+        loopRunId: input.loopRunId ?? null
       })
       .returning();
     if (!row) {
@@ -1514,6 +1518,10 @@ export async function appendCost(
     projectId?: string | null;
     issueId?: string | null;
     agentId?: string | null;
+    /** Discriminator for reporting (e.g. `company_assistant`); null for heartbeat / legacy */
+    costCategory?: string | null;
+    assistantThreadId?: string | null;
+    assistantMessageId?: string | null;
   }
 ) {
   const id = nanoid(14);
@@ -1521,6 +1529,9 @@ export async function appendCost(
     id,
     companyId: input.companyId,
     runId: input.runId ?? null,
+    costCategory: input.costCategory ?? null,
+    assistantThreadId: input.assistantThreadId ?? null,
+    assistantMessageId: input.assistantMessageId ?? null,
     providerType: input.providerType,
     runtimeModelId: input.runtimeModelId ?? null,
     pricingProviderType: input.pricingProviderType ?? null,
@@ -1546,6 +1557,83 @@ export async function listCostEntries(db: BopoDb, companyId: string, limit = 200
     .limit(limit);
 }
 
+export type CostLedgerAggregate = {
+  rowCount: number;
+  tokenInput: number;
+  tokenOutput: number;
+  /** Sum of `usd_cost` as a decimal string (full precision from DB). */
+  usdTotal: string;
+};
+
+/** Sum every ledger row for the company in `[startInclusive, endExclusive)` (typically UTC month). */
+export async function aggregateCompanyCostLedgerInRange(
+  db: BopoDb,
+  companyId: string,
+  startInclusive: Date,
+  endExclusive: Date
+): Promise<CostLedgerAggregate> {
+  const [row] = await db
+    .select({
+      rowCount: sql<number>`count(*)::int`,
+      tokenInput: sql<string>`coalesce(sum(${costLedger.tokenInput})::text, '0')`,
+      tokenOutput: sql<string>`coalesce(sum(${costLedger.tokenOutput})::text, '0')`,
+      usdTotal: sql<string>`coalesce(sum(${costLedger.usdCost})::text, '0')`
+    })
+    .from(costLedger)
+    .where(
+      and(
+        eq(costLedger.companyId, companyId),
+        gte(costLedger.createdAt, startInclusive),
+        lt(costLedger.createdAt, endExclusive)
+      )
+    );
+
+  const parseBigIntish = (v: string) => {
+    try {
+      return Number(BigInt(v));
+    } catch {
+      return Number.parseInt(v, 10) || 0;
+    }
+  };
+
+  return {
+    rowCount: Number(row?.rowCount ?? 0),
+    tokenInput: parseBigIntish(String(row?.tokenInput ?? "0")),
+    tokenOutput: parseBigIntish(String(row?.tokenOutput ?? "0")),
+    usdTotal: String(row?.usdTotal ?? "0").trim() || "0"
+  };
+}
+
+export async function aggregateCompanyCostLedgerAllTime(
+  db: BopoDb,
+  companyId: string
+): Promise<CostLedgerAggregate> {
+  const [row] = await db
+    .select({
+      rowCount: sql<number>`count(*)::int`,
+      tokenInput: sql<string>`coalesce(sum(${costLedger.tokenInput})::text, '0')`,
+      tokenOutput: sql<string>`coalesce(sum(${costLedger.tokenOutput})::text, '0')`,
+      usdTotal: sql<string>`coalesce(sum(${costLedger.usdCost})::text, '0')`
+    })
+    .from(costLedger)
+    .where(eq(costLedger.companyId, companyId));
+
+  const parseBigIntish = (v: string) => {
+    try {
+      return Number(BigInt(v));
+    } catch {
+      return Number.parseInt(v, 10) || 0;
+    }
+  };
+
+  return {
+    rowCount: Number(row?.rowCount ?? 0),
+    tokenInput: parseBigIntish(String(row?.tokenInput ?? "0")),
+    tokenOutput: parseBigIntish(String(row?.tokenOutput ?? "0")),
+    usdTotal: String(row?.usdTotal ?? "0").trim() || "0"
+  };
+}
+
 export async function listHeartbeatRuns(db: BopoDb, companyId: string, limit = 100) {
   return db
     .select()

package/src/schema.ts

@@ -121,6 +121,75 @@ export const issues = pgTable("issues", {
   externalLink: text("external_link"),
   isClaimed: boolean("is_claimed").notNull().default(false),
   claimedByHeartbeatRunId: text("claimed_by_heartbeat_run_id"),
+  /** Set when issue was created by a scheduled/manual work loop run (FK enforced in SQL migration). */
+  loopId: text("loop_id"),
+  loopRunId: text("loop_run_id"),
+  createdAt: timestamp("created_at", { mode: "date" }).defaultNow().notNull(),
+  updatedAt: timestamp("updated_at", { mode: "date" }).defaultNow().notNull()
+});
+
+export const workLoops = pgTable("work_loops", {
+  id: text("id").primaryKey(),
+  companyId: text("company_id")
+    .notNull()
+    .references(() => companies.id, { onDelete: "cascade" }),
+  projectId: text("project_id")
+    .notNull()
+    .references(() => projects.id, { onDelete: "cascade" }),
+  parentIssueId: text("parent_issue_id").references(() => issues.id, { onDelete: "set null" }),
+  goalIdsJson: text("goal_ids_json").notNull().default("[]"),
+  title: text("title").notNull(),
+  description: text("description"),
+  assigneeAgentId: text("assignee_agent_id")
+    .notNull()
+    .references(() => agents.id, { onDelete: "restrict" }),
+  priority: text("priority").notNull().default("medium"),
+  status: text("status").notNull().default("active"),
+  concurrencyPolicy: text("concurrency_policy").notNull().default("coalesce_if_active"),
+  catchUpPolicy: text("catch_up_policy").notNull().default("skip_missed"),
+  lastTriggeredAt: timestamp("last_triggered_at", { mode: "date" }),
+  createdAt: timestamp("created_at", { mode: "date" }).defaultNow().notNull(),
+  updatedAt: timestamp("updated_at", { mode: "date" }).defaultNow().notNull()
+});
+
+export const workLoopTriggers = pgTable("work_loop_triggers", {
+  id: text("id").primaryKey(),
+  companyId: text("company_id")
+    .notNull()
+    .references(() => companies.id, { onDelete: "cascade" }),
+  workLoopId: text("work_loop_id")
+    .notNull()
+    .references(() => workLoops.id, { onDelete: "cascade" }),
+  kind: text("kind").notNull().default("schedule"),
+  label: text("label"),
+  enabled: boolean("enabled").notNull().default(true),
+  cronExpression: text("cron_expression").notNull(),
+  timezone: text("timezone").notNull().default("UTC"),
+  nextRunAt: timestamp("next_run_at", { mode: "date" }),
+  lastFiredAt: timestamp("last_fired_at", { mode: "date" }),
+  lastResult: text("last_result"),
+  createdAt: timestamp("created_at", { mode: "date" }).defaultNow().notNull(),
+  updatedAt: timestamp("updated_at", { mode: "date" }).defaultNow().notNull()
+});
+
+export const workLoopRuns = pgTable("work_loop_runs", {
+  id: text("id").primaryKey(),
+  companyId: text("company_id")
+    .notNull()
+    .references(() => companies.id, { onDelete: "cascade" }),
+  workLoopId: text("work_loop_id")
+    .notNull()
+    .references(() => workLoops.id, { onDelete: "cascade" }),
+  triggerId: text("trigger_id").references(() => workLoopTriggers.id, { onDelete: "set null" }),
+  source: text("source").notNull(),
+  status: text("status").notNull().default("received"),
+  triggeredAt: timestamp("triggered_at", { mode: "date" }).defaultNow().notNull(),
+  idempotencyKey: text("idempotency_key"),
+  payloadJson: text("payload_json").notNull().default("{}"),
+  linkedIssueId: text("linked_issue_id").references(() => issues.id, { onDelete: "set null" }),
+  coalescedIntoRunId: text("coalesced_into_run_id"),
+  failureReason: text("failure_reason"),
+  completedAt: timestamp("completed_at", { mode: "date" }),
   createdAt: timestamp("created_at", { mode: "date" }).defaultNow().notNull(),
   updatedAt: timestamp("updated_at", { mode: "date" }).defaultNow().notNull()
 });
@@ -297,6 +366,29 @@ export const attentionInboxStates = pgTable(
   (table) => [primaryKey({ columns: [table.companyId, table.actorId, table.itemKey] })]
 );
 
+export const companyAssistantThreads = pgTable("company_assistant_threads", {
+  id: text("id").primaryKey(),
+  companyId: text("company_id")
+    .notNull()
+    .references(() => companies.id, { onDelete: "cascade" }),
+  createdAt: timestamp("created_at", { mode: "date" }).defaultNow().notNull(),
+  updatedAt: timestamp("updated_at", { mode: "date" }).defaultNow().notNull()
+});
+
+export const companyAssistantMessages = pgTable("company_assistant_messages", {
+  id: text("id").primaryKey(),
+  threadId: text("thread_id")
+    .notNull()
+    .references(() => companyAssistantThreads.id, { onDelete: "cascade" }),
+  companyId: text("company_id")
+    .notNull()
+    .references(() => companies.id, { onDelete: "cascade" }),
+  role: text("role").notNull(),
+  body: text("body").notNull(),
+  metadataJson: text("metadata_json"),
+  createdAt: timestamp("created_at", { mode: "date" }).defaultNow().notNull()
+});
+
 export const costLedger = pgTable("cost_ledger", {
   id: text("id").primaryKey(),
   companyId: text("company_id")
@@ -306,6 +398,10 @@ export const costLedger = pgTable("cost_ledger", {
   projectId: text("project_id").references(() => projects.id, { onDelete: "set null" }),
   issueId: text("issue_id").references(() => issues.id, { onDelete: "set null" }),
   agentId: text("agent_id").references(() => agents.id, { onDelete: "set null" }),
+  /** e.g. `company_assistant` for owner-assistant chat; null for heartbeat and legacy rows */
+  costCategory: text("cost_category"),
+  assistantThreadId: text("assistant_thread_id").references(() => companyAssistantThreads.id, { onDelete: "set null" }),
+  assistantMessageId: text("assistant_message_id").references(() => companyAssistantMessages.id, { onDelete: "set null" }),
   providerType: text("provider_type").notNull(),
   runtimeModelId: text("runtime_model_id"),
   pricingProviderType: text("pricing_provider_type"),
@@ -445,6 +541,9 @@ export const schema = {
   goals,
   agents,
   issues,
+  workLoops,
+  workLoopTriggers,
+  workLoopRuns,
   issueGoals,
   issueComments,
   issueAttachments,
@@ -464,7 +563,9 @@ export const schema = {
   templateVersions,
   templateInstalls,
   agentIssueLabels,
-  projectWorkspaces
+  projectWorkspaces,
+  companyAssistantThreads,
+  companyAssistantMessages
 };
 
 export const touchUpdatedAtSql = sql`CURRENT_TIMESTAMP`;