@brianli/kimaki 0.4.72-brianli.1

package/src/schema.sql

@@ -0,0 +1,158 @@
+-- This file is generated by pnpm generate:sql. Do not edit manually.
+CREATE TABLE IF NOT EXISTS "thread_sessions" (
+    "thread_id" TEXT NOT NULL PRIMARY KEY,
+    "session_id" TEXT NOT NULL,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+CREATE TABLE IF NOT EXISTS "part_messages" (
+    "part_id" TEXT NOT NULL PRIMARY KEY,
+    "message_id" TEXT NOT NULL,
+    "thread_id" TEXT NOT NULL,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "part_messages_thread_id_fkey" FOREIGN KEY ("thread_id") REFERENCES "thread_sessions" ("thread_id") ON DELETE RESTRICT ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "bot_tokens" (
+    "app_id" TEXT NOT NULL PRIMARY KEY,
+    "token" TEXT NOT NULL,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+CREATE TABLE IF NOT EXISTS "channel_directories" (
+    "channel_id" TEXT NOT NULL PRIMARY KEY,
+    "directory" TEXT NOT NULL,
+    "channel_type" TEXT NOT NULL,
+    "app_id" TEXT,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "channel_directories_app_id_fkey" FOREIGN KEY ("app_id") REFERENCES "bot_tokens" ("app_id") ON DELETE SET NULL ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "bot_api_keys" (
+    "app_id" TEXT NOT NULL PRIMARY KEY,
+    "gemini_api_key" TEXT,
+    "openai_api_key" TEXT,
+    "xai_api_key" TEXT,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "bot_api_keys_app_id_fkey" FOREIGN KEY ("app_id") REFERENCES "bot_tokens" ("app_id") ON DELETE RESTRICT ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "thread_worktrees" (
+    "thread_id" TEXT NOT NULL PRIMARY KEY,
+    "worktree_name" TEXT NOT NULL,
+    "worktree_directory" TEXT,
+    "project_directory" TEXT NOT NULL,
+    "status" TEXT DEFAULT 'pending',
+    "error_message" TEXT,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "thread_worktrees_thread_id_fkey" FOREIGN KEY ("thread_id") REFERENCES "thread_sessions" ("thread_id") ON DELETE RESTRICT ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "channel_models" (
+    "channel_id" TEXT NOT NULL PRIMARY KEY,
+    "model_id" TEXT NOT NULL,
+    "variant" TEXT,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "channel_models_channel_id_fkey" FOREIGN KEY ("channel_id") REFERENCES "channel_directories" ("channel_id") ON DELETE RESTRICT ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "session_models" (
+    "session_id" TEXT NOT NULL PRIMARY KEY,
+    "model_id" TEXT NOT NULL,
+    "variant" TEXT,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+CREATE TABLE IF NOT EXISTS "channel_agents" (
+    "channel_id" TEXT NOT NULL PRIMARY KEY,
+    "agent_name" TEXT NOT NULL,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "channel_agents_channel_id_fkey" FOREIGN KEY ("channel_id") REFERENCES "channel_directories" ("channel_id") ON DELETE RESTRICT ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "session_agents" (
+    "session_id" TEXT NOT NULL PRIMARY KEY,
+    "agent_name" TEXT NOT NULL,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP
+);
+CREATE TABLE IF NOT EXISTS "channel_worktrees" (
+    "channel_id" TEXT NOT NULL PRIMARY KEY,
+    "enabled" INTEGER NOT NULL DEFAULT 0,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "channel_worktrees_channel_id_fkey" FOREIGN KEY ("channel_id") REFERENCES "channel_directories" ("channel_id") ON DELETE RESTRICT ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "channel_verbosity" (
+    "channel_id" TEXT NOT NULL PRIMARY KEY,
+    "verbosity" TEXT NOT NULL DEFAULT 'tools-and-text',
+    "updated_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "channel_verbosity_channel_id_fkey" FOREIGN KEY ("channel_id") REFERENCES "channel_directories" ("channel_id") ON DELETE RESTRICT ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "channel_mention_mode" (
+    "channel_id" TEXT NOT NULL PRIMARY KEY,
+    "enabled" INTEGER NOT NULL DEFAULT 0,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "channel_mention_mode_channel_id_fkey" FOREIGN KEY ("channel_id") REFERENCES "channel_directories" ("channel_id") ON DELETE RESTRICT ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "global_models" (
+    "app_id" TEXT NOT NULL PRIMARY KEY,
+    "model_id" TEXT NOT NULL,
+    "variant" TEXT,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "global_models_app_id_fkey" FOREIGN KEY ("app_id") REFERENCES "bot_tokens" ("app_id") ON DELETE RESTRICT ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "scheduled_tasks" (
+    "id" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
+    "status" TEXT NOT NULL DEFAULT 'planned',
+    "schedule_kind" TEXT NOT NULL,
+    "run_at" DATETIME,
+    "cron_expr" TEXT,
+    "timezone" TEXT,
+    "next_run_at" DATETIME NOT NULL,
+    "running_started_at" DATETIME,
+    "last_run_at" DATETIME,
+    "last_error" TEXT,
+    "attempts" INTEGER NOT NULL DEFAULT 0,
+    "payload_json" TEXT NOT NULL,
+    "prompt_preview" TEXT NOT NULL,
+    "channel_id" TEXT,
+    "thread_id" TEXT,
+    "session_id" TEXT,
+    "project_directory" TEXT,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "scheduled_tasks_channel_id_fkey" FOREIGN KEY ("channel_id") REFERENCES "channel_directories" ("channel_id") ON DELETE SET NULL ON UPDATE CASCADE,
+    CONSTRAINT "scheduled_tasks_thread_id_fkey" FOREIGN KEY ("thread_id") REFERENCES "thread_sessions" ("thread_id") ON DELETE SET NULL ON UPDATE CASCADE
+);
+CREATE TABLE sqlite_sequence(name,seq);
+CREATE TABLE IF NOT EXISTS "session_start_sources" (
+    "session_id" TEXT NOT NULL PRIMARY KEY,
+    "schedule_kind" TEXT NOT NULL,
+    "scheduled_task_id" INTEGER,
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "session_start_sources_scheduled_task_id_fkey" FOREIGN KEY ("scheduled_task_id") REFERENCES "scheduled_tasks" ("id") ON DELETE SET NULL ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "forum_sync_configs" (
+    "id" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
+    "app_id" TEXT NOT NULL,
+    "forum_channel_id" TEXT NOT NULL,
+    "output_dir" TEXT NOT NULL,
+    "direction" TEXT NOT NULL DEFAULT 'bidirectional',
+    "created_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" DATETIME DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "forum_sync_configs_app_id_fkey" FOREIGN KEY ("app_id") REFERENCES "bot_tokens" ("app_id") ON DELETE CASCADE ON UPDATE CASCADE
+);
+CREATE TABLE IF NOT EXISTS "ipc_requests" (
+    "id" TEXT NOT NULL PRIMARY KEY,
+    "type" TEXT NOT NULL,
+    "session_id" TEXT NOT NULL,
+    "thread_id" TEXT NOT NULL,
+    "payload" TEXT NOT NULL,
+    "response" TEXT,
+    "status" TEXT NOT NULL DEFAULT 'pending',
+    "created_at" DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    "updated_at" DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
+    CONSTRAINT "ipc_requests_thread_id_fkey" FOREIGN KEY ("thread_id") REFERENCES "thread_sessions" ("thread_id") ON DELETE RESTRICT ON UPDATE CASCADE
+);
+CREATE INDEX "scheduled_tasks_status_next_run_at_idx" ON "scheduled_tasks"("status", "next_run_at");
+CREATE INDEX "scheduled_tasks_channel_id_status_idx" ON "scheduled_tasks"("channel_id", "status");
+CREATE INDEX "scheduled_tasks_thread_id_status_idx" ON "scheduled_tasks"("thread_id", "status");
+CREATE INDEX "session_start_sources_scheduled_task_id_idx" ON "session_start_sources"("scheduled_task_id");
+CREATE UNIQUE INDEX "forum_sync_configs_app_id_forum_channel_id_key" ON "forum_sync_configs"("app_id", "forum_channel_id");
+CREATE INDEX "ipc_requests_status_created_at_idx" ON "ipc_requests"("status", "created_at");

package/src/sentry.ts

@@ -0,0 +1,137 @@
+// Sentry error tracking initialization and notifyError helper.
+// Uses @sentry/node for the Node.js runtime (bot process, plugin process, worker threads).
+// Must be initialized early in both the bot process (cli.ts) and plugin process
+// (opencode-plugin.ts). The plugin process receives the DSN via KIMAKI_SENTRY_DSN env var.
+
+import * as Sentry from '@sentry/node'
+import * as errore from 'errore'
+import { createRequire } from 'node:module'
+import { sanitizeSensitiveText, sanitizeUnknownValue } from './privacy-sanitizer.js'
+
+// DSN placeholder — replace with your Sentry project DSN.
+// Users can also set KIMAKI_SENTRY_DSN env var.
+const HARDCODED_DSN = 'https://3b87e21ac01cb9c66225719ea65111d2@o4510952031715328.ingest.us.sentry.io/4510952088469504'
+
+function readKimakiVersion(): string {
+  try {
+    const require = createRequire(import.meta.url)
+    const pkg = require('../package.json') as { version?: string }
+    const version = pkg.version
+    if (!version) {
+      return 'unknown'
+    }
+    return version
+  } catch {
+    return 'unknown'
+  }
+}
+
+const kimakiVersion = readKimakiVersion()
+const kimakiRelease = `kimaki@${kimakiVersion}`
+
+let initialized = false
+
+/**
+ * Initialize Sentry. Call once at process startup.
+ * No-op if DSN is empty or --no-sentry was passed.
+ */
+export function initSentry({ dsn }: { dsn?: string } = {}): void {
+  if (process.env.KIMAKI_SENTRY_DISABLED === '1') {
+    return
+  }
+
+  const resolvedDsn = dsn || process.env.KIMAKI_SENTRY_DSN || HARDCODED_DSN
+  if (!resolvedDsn || initialized) {
+    return
+  }
+
+  Sentry.init({
+    dsn: resolvedDsn,
+    release: kimakiRelease,
+    integrations: [],
+    tracesSampleRate: 0,
+    sendDefaultPii: false,
+    profilesSampleRate: 0,
+    beforeSend(event, hint) {
+      // Skip in development — too noisy, errors appear in terminal
+      if (process.env.NODE_ENV === 'development') {
+        return null
+      }
+      // Skip abort errors — walks the cause chain so wrapped aborts are caught
+      if (errore.isAbortError(hint.originalException)) {
+        return null
+      }
+
+      try {
+        const sanitizedEvent = sanitizeUnknownValue(event, {
+          redactPaths: false,
+        })
+        if (sanitizedEvent && typeof sanitizedEvent === 'object') {
+          return sanitizedEvent as typeof event
+        }
+      } catch {
+        return event
+      }
+      return event
+    },
+  })
+
+  Sentry.setTag('kimaki_version', kimakiVersion)
+
+  initialized = true
+}
+
+/**
+ * Report an unexpected error to Sentry.
+ * Safe to call even if Sentry is not initialized.
+ * Fire-and-forget only: use `void notifyError(error, msg)` and never await it.
+ * This helper must never throw.
+ * Use this at terminal error handlers — the "last catch" in a chain
+ * where the error would otherwise be invisible.
+ */
+export function notifyError(error: unknown, msg?: string): void {
+  try {
+    if (!initialized) {
+      return
+    }
+
+    const safeMsg = (() => {
+      if (!msg) {
+        return undefined
+      }
+      try {
+        return sanitizeSensitiveText(msg, { redactPaths: false })
+      } catch {
+        return msg
+      }
+    })()
+    const safeError = (() => {
+      try {
+        return sanitizeUnknownValue(error, { redactPaths: false })
+      } catch {
+        return error
+      }
+    })()
+
+    Sentry.captureException(error, {
+      tags: { kimaki_version: kimakiVersion },
+      extra: { msg: safeMsg, kimakiVersion, error: safeError },
+    })
+    void Sentry.flush(1000).catch(() => {
+      return
+    })
+  } catch {
+    return
+  }
+}
+
+/**
+ * User-readable error class. Messages from AppError instances
+ * are forwarded to the user as-is; regular Error messages may be obfuscated.
+ */
+export class AppError extends Error {
+  constructor(message: string) {
+    super(message)
+    this.name = 'AppError'
+  }
+}

package/src/session-handler/state.ts

@@ -0,0 +1,232 @@
+// Session handler run-state store and transitions.
+// Centralizes prompt/idle race state so interrupt handling stays consistent.
+
+import { createStore, type StoreApi } from 'zustand/vanilla'
+
+export type MainRunPhase =
+  | 'waiting-dispatch'
+  | 'collecting-baseline'
+  | 'dispatching'
+  | 'prompt-resolved'
+  | 'finished'
+  | 'aborted'
+
+export type MainIdleState = 'none' | 'deferred'
+
+export type MainRunState = {
+  phase: MainRunPhase
+  idleState: MainIdleState
+  baselineAssistantIds: Set<string>
+  currentAssistantMessageId: string | undefined
+  eventSeq: number
+  evidenceSeq: number | undefined
+  deferredIdleSeq: number | undefined
+}
+
+export type MainRunStore = StoreApi<MainRunState>
+
+export type MainSessionIdleDecision =
+  | 'deferred'
+  | 'ignore-no-evidence'
+  | 'process'
+
+export type DeferredIdleDecision =
+  | 'none'
+  | 'ignore-no-evidence'
+  | 'ignore-before-evidence'
+  | 'process'
+
+function initialMainRunState(): MainRunState {
+  return {
+    phase: 'waiting-dispatch',
+    idleState: 'none',
+    baselineAssistantIds: new Set<string>(),
+    currentAssistantMessageId: undefined,
+    eventSeq: 0,
+    evidenceSeq: undefined,
+    deferredIdleSeq: undefined,
+  }
+}
+
+export function createMainRunStore(): MainRunStore {
+  return createStore<MainRunState>(() => {
+    return initialMainRunState()
+  })
+}
+
+export function beginPromptCycle({ store }: { store: MainRunStore }): void {
+  store.setState((state) => {
+    return {
+      ...state,
+      phase: 'collecting-baseline',
+      idleState: 'none',
+      baselineAssistantIds: new Set<string>(),
+      currentAssistantMessageId: undefined,
+      eventSeq: 0,
+      evidenceSeq: undefined,
+      deferredIdleSeq: undefined,
+    }
+  })
+}
+
+export function setBaselineAssistantIds({
+  store,
+  messageIds,
+}: {
+  store: MainRunStore
+  messageIds: Set<string>
+}): void {
+  store.setState((state) => {
+    return {
+      ...state,
+      baselineAssistantIds: new Set<string>(messageIds),
+    }
+  })
+}
+
+export function markDispatching({ store }: { store: MainRunStore }): void {
+  store.setState((state) => {
+    return {
+      ...state,
+      phase: 'dispatching',
+    }
+  })
+}
+
+export function markCurrentPromptEvidence({
+  store,
+  messageId,
+}: {
+  store: MainRunStore
+  messageId: string
+}): void {
+  store.setState((state) => {
+    const eventSeq = state.eventSeq + 1
+    const canTrackCurrentPrompt =
+      state.phase === 'dispatching' || state.phase === 'prompt-resolved'
+    if (!canTrackCurrentPrompt) {
+      return {
+        ...state,
+        eventSeq,
+      }
+    }
+    if (state.baselineAssistantIds.has(messageId)) {
+      return {
+        ...state,
+        eventSeq,
+      }
+    }
+    return {
+      ...state,
+      eventSeq,
+      currentAssistantMessageId: messageId,
+      evidenceSeq: state.evidenceSeq ?? eventSeq,
+    }
+  })
+}
+
+export function handleMainSessionIdle({
+  store,
+}: {
+  store: MainRunStore
+}): MainSessionIdleDecision {
+  const state = store.getState()
+  const idleSeq = state.eventSeq + 1
+
+  if (state.phase !== 'prompt-resolved') {
+    store.setState((current) => {
+      return {
+        ...current,
+        eventSeq: idleSeq,
+        idleState: 'deferred',
+        deferredIdleSeq: idleSeq,
+      }
+    })
+    return 'deferred'
+  }
+
+  if (!state.currentAssistantMessageId) {
+    store.setState((current) => {
+      return {
+        ...current,
+        eventSeq: idleSeq,
+      }
+    })
+    return 'ignore-no-evidence'
+  }
+
+  store.setState((current) => {
+    return {
+      ...current,
+      eventSeq: idleSeq,
+    }
+  })
+  return 'process'
+}
+
+export function markPromptResolvedAndConsumeDeferredIdle({
+  store,
+}: {
+  store: MainRunStore
+}): DeferredIdleDecision {
+  const state = store.getState()
+
+  const nextState: MainRunState = {
+    ...state,
+    phase: 'prompt-resolved',
+  }
+
+  if (state.idleState !== 'deferred') {
+    store.setState(nextState)
+    return 'none'
+  }
+
+  nextState.idleState = 'none'
+  nextState.deferredIdleSeq = undefined
+
+  if (!state.currentAssistantMessageId) {
+    store.setState(nextState)
+    return 'ignore-no-evidence'
+  }
+
+  if (
+    typeof state.deferredIdleSeq === 'number' &&
+    typeof state.evidenceSeq === 'number' &&
+    state.deferredIdleSeq <= state.evidenceSeq
+  ) {
+    store.setState(nextState)
+    return 'ignore-before-evidence'
+  }
+
+  store.setState(nextState)
+  return 'process'
+}
+
+export function markFinished({ store }: { store: MainRunStore }): void {
+  store.setState((state) => {
+    return {
+      ...state,
+      phase: 'finished',
+    }
+  })
+}
+
+export function markAborted({ store }: { store: MainRunStore }): void {
+  store.setState((state) => {
+    if (state.phase === 'finished') {
+      return state
+    }
+    return {
+      ...state,
+      phase: 'aborted',
+    }
+  })
+}
+
+export function hasCurrentPromptEvidence({
+  store,
+}: {
+  store: MainRunStore
+}): boolean {
+  return Boolean(store.getState().currentAssistantMessageId)
+}