refactor: complete config unification and split db helpers

2026-04-08 19:05:19 +09:00
parent d996d5ff10
commit 8b9bebd29e
27 changed files with 4752 additions and 3652 deletions
--- a/src/db/router-state.ts
+++ b/src/db/router-state.ts
@@ -0,0 +1,87 @@
+import { Database } from 'bun:sqlite';
+
+import { normalizeServiceId } from '../config.js';
+import { AgentType } from '../types.js';
+
+export function getRouterStateFromDatabase(
+  database: Database,
+  key: string,
+  currentServiceId: string,
+): string | undefined {
+  const row = database
+    .prepare('SELECT value FROM router_state WHERE key = ?')
+    .get(key) as { value: string } | undefined;
+  if (row) return row.value;
+
+  const prefixedKey = `${normalizeServiceId(currentServiceId)}:${key}`;
+  const prefixedRow = database
+    .prepare('SELECT value FROM router_state WHERE key = ?')
+    .get(prefixedKey) as { value: string } | undefined;
+  if (!prefixedRow) return undefined;
+
+  database
+    .prepare('INSERT OR REPLACE INTO router_state (key, value) VALUES (?, ?)')
+    .run(key, prefixedRow.value);
+  database.prepare('DELETE FROM router_state WHERE key = ?').run(prefixedKey);
+  return prefixedRow.value;
+}
+
+export function getRouterStateForServiceFromDatabase(
+  database: Database,
+  key: string,
+  serviceId: string,
+): string | undefined {
+  const prefixedKey = `${normalizeServiceId(serviceId)}:${key}`;
+  const row = database
+    .prepare('SELECT value FROM router_state WHERE key = ?')
+    .get(prefixedKey) as { value: string } | undefined;
+  if (row) return row.value;
+
+  const canonical = database
+    .prepare('SELECT value FROM router_state WHERE key = ?')
+    .get(key) as { value: string } | undefined;
+  return canonical?.value;
+}
+
+export function setRouterStateInDatabase(
+  database: Database,
+  key: string,
+  value: string,
+): void {
+  database
+    .prepare('INSERT OR REPLACE INTO router_state (key, value) VALUES (?, ?)')
+    .run(key, value);
+}
+
+export function setRouterStateForServiceInDatabase(
+  database: Database,
+  key: string,
+  value: string,
+  serviceId: string,
+): void {
+  const prefixedKey = `${normalizeServiceId(serviceId)}:${key}`;
+  database
+    .prepare('INSERT OR REPLACE INTO router_state (key, value) VALUES (?, ?)')
+    .run(prefixedKey, value);
+}
+
+export function getLastRespondingAgentTypeFromDatabase(
+  database: Database,
+  chatJid: string,
+): AgentType | undefined {
+  const row = database
+    .prepare(
+      `SELECT sender FROM messages
+       WHERE chat_jid = ? AND is_bot_message = 1
+       ORDER BY timestamp DESC, seq DESC
+       LIMIT 1`,
+    )
+    .get(chatJid) as { sender: string } | undefined;
+
+  if (!row) return undefined;
+
+  const sender = row.sender.toLowerCase();
+  if (sender.includes('claude')) return 'claude-code';
+  if (sender.includes('codex')) return 'codex';
+  return undefined;
+}