agent-relay-server 0.4.22 → 0.4.24

package/src/db.ts

@@ -3,6 +3,8 @@ import { randomUUID } from "node:crypto";
 import { VERSION } from "./config.ts";
 import type {
   AgentCard,
+  ActivityEvent,
+  ActivityEventInput,
   AgentSessionGuard,
   CreatePairInput,
   HealthCheck,
@@ -21,6 +23,9 @@ import type {
   TaskSeverity,
   TaskStatus,
   IntegrationEventInput,
+  InboxDraft,
+  InboxState,
+  InboxThreadState,
   TaskStatusInput,
 } from "./types";
 import { STALE_TTL_MS, DAY_MS, CLAIM_LEASE_MS } from "./config";
@@ -144,6 +149,48 @@ export function initDb(path: string = "agent-relay.db"): Database {
     CREATE INDEX IF NOT EXISTS idx_pairs_requester ON pairs(requester_id);
     CREATE INDEX IF NOT EXISTS idx_pairs_target ON pairs(target_id);
     CREATE INDEX IF NOT EXISTS idx_pairs_status ON pairs(status);
+
+    CREATE TABLE IF NOT EXISTS inbox_thread_state (
+      operator_id TEXT NOT NULL,
+      peer_id TEXT NOT NULL,
+      read_cursor_message_id INTEGER,
+      archived_at_message_id INTEGER,
+      updated_at INTEGER NOT NULL,
+      PRIMARY KEY (operator_id, peer_id)
+    );
+    CREATE INDEX IF NOT EXISTS idx_inbox_thread_operator ON inbox_thread_state(operator_id);
+
+    CREATE TABLE IF NOT EXISTS inbox_drafts (
+      operator_id TEXT NOT NULL,
+      peer_id TEXT NOT NULL,
+      body TEXT NOT NULL,
+      subject TEXT,
+      channel TEXT,
+      updated_at INTEGER NOT NULL,
+      PRIMARY KEY (operator_id, peer_id)
+    );
+    CREATE INDEX IF NOT EXISTS idx_inbox_drafts_operator ON inbox_drafts(operator_id);
+
+    CREATE TABLE IF NOT EXISTS activity_events (
+      id INTEGER PRIMARY KEY AUTOINCREMENT,
+      operator_id TEXT,
+      client_id TEXT UNIQUE,
+      kind TEXT NOT NULL,
+      title TEXT NOT NULL,
+      body TEXT,
+      meta_text TEXT,
+      icon TEXT,
+      view TEXT,
+      peer_id TEXT,
+      message_id INTEGER,
+      pair_id TEXT,
+      task_id INTEGER,
+      agent_id TEXT,
+      metadata TEXT NOT NULL DEFAULT '{}',
+      created_at INTEGER NOT NULL
+    );
+    CREATE INDEX IF NOT EXISTS idx_activity_operator ON activity_events(operator_id, created_at);
+    CREATE INDEX IF NOT EXISTS idx_activity_created ON activity_events(created_at);
   `);
 
   // Migrations
@@ -257,6 +304,44 @@ function parseStringArray(raw: string): string[] {
   return parsed.filter((value): value is string => typeof value === "string");
 }
 
+function normalizeTags(tags: string[] | undefined): string[] {
+  return [...new Set((tags ?? []).map((tag) => tag.trim()).filter(Boolean))];
+}
+
+function stringValue(value: unknown): string | undefined {
+  return typeof value === "string" && value.trim() ? value.trim() : undefined;
+}
+
+function inferProviderTag(input: RegisterAgentInput): "claude" | "codex" | undefined {
+  const meta = input.meta ?? {};
+  const values = [
+    input.id,
+    input.name,
+    input.rig,
+    ...(input.tags ?? []),
+    stringValue(meta.provider),
+    stringValue(meta.client),
+    stringValue(meta.runtime),
+    stringValue(meta.agentType),
+  ].filter((value): value is string => typeof value === "string");
+
+  if (values.some((value) => value.toLowerCase().includes("codex"))) return "codex";
+  if (values.some((value) => value.toLowerCase().includes("claude"))) return "claude";
+
+  // Older Claude hooks did not always send an explicit provider tag, but did
+  // send Claude-style approval metadata. Codex also sends approvalMode, so this
+  // fallback only runs after Codex signals have been ruled out.
+  if (Object.prototype.hasOwnProperty.call(meta, "approvalMode")) return "claude";
+  return undefined;
+}
+
+function tagsWithProvider(input: RegisterAgentInput): string[] {
+  const tags = normalizeTags(input.tags);
+  const provider = inferProviderTag(input);
+  if (!provider || tags.includes(provider)) return tags;
+  return [provider, ...tags];
+}
+
 function rowToAgent(row: any): AgentCard {
   return {
     id: row.id,
@@ -355,6 +440,48 @@ function rowToPair(row: any): PairSession {
   };
 }
 
+function rowToInboxThreadState(row: any): InboxThreadState {
+  return {
+    operatorId: row.operator_id,
+    peerId: row.peer_id,
+    readCursorMessageId: row.read_cursor_message_id ?? undefined,
+    archivedAtMessageId: row.archived_at_message_id ?? undefined,
+    updatedAt: row.updated_at,
+  };
+}
+
+function rowToInboxDraft(row: any): InboxDraft {
+  return {
+    operatorId: row.operator_id,
+    peerId: row.peer_id,
+    body: row.body,
+    subject: row.subject ?? undefined,
+    channel: row.channel ?? undefined,
+    updatedAt: row.updated_at,
+  };
+}
+
+function rowToActivityEvent(row: any): ActivityEvent {
+  return {
+    id: row.id,
+    operatorId: row.operator_id ?? undefined,
+    clientId: row.client_id ?? undefined,
+    kind: row.kind,
+    title: row.title,
+    body: row.body ?? undefined,
+    meta: row.meta_text ?? undefined,
+    icon: row.icon ?? undefined,
+    view: row.view ?? undefined,
+    peer: row.peer_id ?? undefined,
+    messageId: row.message_id ?? undefined,
+    pairId: row.pair_id ?? undefined,
+    taskId: row.task_id ?? undefined,
+    agentId: row.agent_id ?? undefined,
+    metadata: parseJson(row.metadata, {}),
+    createdAt: row.created_at,
+  };
+}
+
 const MSG_SELECT = `SELECT m.*, (
   SELECT json_group_array(agent_id) FROM message_reads WHERE message_id = m.id
 ) AS read_by_agents FROM messages m`;
@@ -363,6 +490,7 @@ const MSG_SELECT = `SELECT m.*, (
 
 export function upsertAgent(input: RegisterAgentInput): AgentCard {
   const now = Date.now();
+  const tags = tagsWithProvider(input);
   // Preserve the existing label across re-registrations unless the caller
   // explicitly sends one (including null to clear).
   const labelProvided = Object.prototype.hasOwnProperty.call(input, "label");
@@ -394,7 +522,7 @@ export function upsertAgent(input: RegisterAgentInput): AgentCard {
     $name: input.name,
     $label: input.label ?? null,
     $labelProvided: labelProvided ? 1 : 0,
-    $tags: JSON.stringify(input.tags ?? []),
+    $tags: JSON.stringify(tags),
     $machine: input.machine ?? null,
     $rig: input.rig ?? null,
     $capabilities: JSON.stringify(input.capabilities ?? []),
@@ -1423,6 +1551,129 @@ export function markRead(messageId: number, agentId: string): boolean {
   return true;
 }
 
+export function getInboxState(operatorId: string): InboxState {
+  const threads = (db.prepare(
+    "SELECT * FROM inbox_thread_state WHERE operator_id = ? ORDER BY updated_at DESC",
+  ).all(operatorId) as any[]).map(rowToInboxThreadState);
+  const drafts = (db.prepare(
+    "SELECT * FROM inbox_drafts WHERE operator_id = ? ORDER BY updated_at DESC",
+  ).all(operatorId) as any[]).map(rowToInboxDraft);
+  return { operatorId, threads, drafts };
+}
+
+export function setInboxThreadState(input: {
+  operatorId: string;
+  peerId: string;
+  readCursorMessageId?: number | null;
+  archivedAtMessageId?: number | null;
+}): InboxThreadState {
+  const now = Date.now();
+  const current = db.prepare(
+    "SELECT * FROM inbox_thread_state WHERE operator_id = ? AND peer_id = ?",
+  ).get(input.operatorId, input.peerId) as any | undefined;
+
+  const readCursorMessageId = Object.prototype.hasOwnProperty.call(input, "readCursorMessageId")
+    ? input.readCursorMessageId ?? null
+    : current?.read_cursor_message_id ?? null;
+  const archivedAtMessageId = Object.prototype.hasOwnProperty.call(input, "archivedAtMessageId")
+    ? input.archivedAtMessageId ?? null
+    : current?.archived_at_message_id ?? null;
+
+  db.prepare(`
+    INSERT INTO inbox_thread_state (operator_id, peer_id, read_cursor_message_id, archived_at_message_id, updated_at)
+    VALUES (?, ?, ?, ?, ?)
+    ON CONFLICT(operator_id, peer_id) DO UPDATE SET
+      read_cursor_message_id = excluded.read_cursor_message_id,
+      archived_at_message_id = excluded.archived_at_message_id,
+      updated_at = excluded.updated_at
+  `).run(input.operatorId, input.peerId, readCursorMessageId, archivedAtMessageId, now);
+
+  return rowToInboxThreadState(db.prepare(
+    "SELECT * FROM inbox_thread_state WHERE operator_id = ? AND peer_id = ?",
+  ).get(input.operatorId, input.peerId));
+}
+
+export function setInboxDraft(input: {
+  operatorId: string;
+  peerId: string;
+  body: string;
+  subject?: string | null;
+  channel?: string | null;
+}): InboxDraft {
+  const now = Date.now();
+  db.prepare(`
+    INSERT INTO inbox_drafts (operator_id, peer_id, body, subject, channel, updated_at)
+    VALUES (?, ?, ?, ?, ?, ?)
+    ON CONFLICT(operator_id, peer_id) DO UPDATE SET
+      body = excluded.body,
+      subject = excluded.subject,
+      channel = excluded.channel,
+      updated_at = excluded.updated_at
+  `).run(input.operatorId, input.peerId, input.body, input.subject ?? null, input.channel ?? null, now);
+
+  return rowToInboxDraft(db.prepare(
+    "SELECT * FROM inbox_drafts WHERE operator_id = ? AND peer_id = ?",
+  ).get(input.operatorId, input.peerId));
+}
+
+export function deleteInboxDraft(operatorId: string, peerId: string): boolean {
+  return db.prepare("DELETE FROM inbox_drafts WHERE operator_id = ? AND peer_id = ?").run(operatorId, peerId).changes > 0;
+}
+
+export function listActivityEvents(input: {
+  operatorId?: string;
+  limit?: number;
+  since?: number;
+} = {}): ActivityEvent[] {
+  const conditions: string[] = [];
+  const params: any[] = [];
+  if (input.operatorId) {
+    conditions.push("operator_id = ?");
+    params.push(input.operatorId);
+  }
+  if (input.since !== undefined) {
+    conditions.push("created_at >= ?");
+    params.push(input.since);
+  }
+  const where = conditions.length ? `WHERE ${conditions.join(" AND ")}` : "";
+  params.push(input.limit ?? 200);
+  return (db.prepare(
+    `SELECT * FROM activity_events ${where} ORDER BY created_at DESC, id DESC LIMIT ?`,
+  ).all(...params) as any[]).map(rowToActivityEvent);
+}
+
+export function createActivityEvent(input: ActivityEventInput): ActivityEvent {
+  if (input.clientId) {
+    const existing = db.prepare("SELECT * FROM activity_events WHERE client_id = ?").get(input.clientId) as any | undefined;
+    if (existing) return rowToActivityEvent(existing);
+  }
+  const now = Date.now();
+  const result = db.prepare(`
+    INSERT INTO activity_events (
+      operator_id, client_id, kind, title, body, meta_text, icon, view, peer_id,
+      message_id, pair_id, task_id, agent_id, metadata, created_at
+    )
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+  `).run(
+    input.operatorId ?? null,
+    input.clientId ?? null,
+    input.kind,
+    input.title,
+    input.body ?? null,
+    input.meta ?? null,
+    input.icon ?? null,
+    input.view ?? null,
+    input.peer ?? null,
+    input.messageId ?? null,
+    input.pairId ?? null,
+    input.taskId ?? null,
+    input.agentId ?? null,
+    JSON.stringify(input.metadata ?? {}),
+    now,
+  );
+  return rowToActivityEvent(db.prepare("SELECT * FROM activity_events WHERE id = ?").get(Number(result.lastInsertRowid)));
+}
+
 export function deleteMessage(id: number): boolean {
   return db.transaction(() => {
     // Break reply_to references from children so the FK doesn't block delete.