@mingxy/cerebro 2.3.4 → 2.3.5

package/src/config.ts

@@ -1,275 +1,277 @@
-import { readFileSync, appendFileSync, mkdirSync } from "node:fs";
-import { homedir } from "node:os";
-import { join } from "node:path";
-
-// ── Nested config interface ──────────────────────────────────────────
-
-export interface CerebroPluginConfig {
-  connection: {
-    apiUrl: string;
-    apiKey: string;
-    requestTimeoutMs: number;
-  };
-  content: {
-    maxQueryLength: number;
-    maxContentChars: number;
-    maxContentLength: number;
-  };
-  injection: {
-    recentCount: number;
-    searchCount: number;
-    recentTruncateChars: number;
-    searchTruncateChars: number;
-    recentTimeoutMs: number;
-    searchTimeoutMs: number;
-    profileTimeoutMs: number;
-  };
-  ingest: {
-    autoCaptureThreshold: number;
-    ingestMode: "smart" | "raw";
-  };
-  logging: {
-    logEnabled: boolean;
-    logLevel: "DEBUG" | "INFO" | "WARN" | "ERROR";
-    logDir: string;
-  };
-  ui: {
-    toastDelayMs: number;
-  };
-  web?: {
-    enabled?: boolean;
-    port?: number;
-  };
-  agentMemoryPolicy?: Record<string, "none" | "readonly" | "readwrite">;
-  defaultPolicy?: "none" | "readonly" | "readwrite";
-  autoUpdate?: boolean;  // default false
-}
-
-// ── Defaults ─────────────────────────────────────────────────────────
-
-const DEFAULTS: CerebroPluginConfig = {
-  connection: {
-    apiUrl: "https://www.mengxy.cc",
-    apiKey: "",
-    requestTimeoutMs: 15000,
-  },
-  content: {
-    maxQueryLength: 200,
-    maxContentChars: 30000,
-    maxContentLength: 3000,
-  },
-  injection: {
-    recentCount: 5,
-    searchCount: 10,
-    recentTruncateChars: 0,   // 0 = 不截断
-    searchTruncateChars: 0,   // 0 = 不截断
-    recentTimeoutMs: 3000,
-    searchTimeoutMs: 5000,
-    profileTimeoutMs: 2000,
-  },
-  ingest: {
-    autoCaptureThreshold: 5,
-    ingestMode: "smart",
-  },
-  logging: {
-    logEnabled: true,
-    logLevel: "INFO",
-    logDir: join(homedir(), ".config", "cerebro"),
-  },
-  ui: {
-    toastDelayMs: 7000,
-  },
-  web: {
-    enabled: true,
-  },
-  autoUpdate: false,
-};
-
-// ── Flat-to-nested migration ─────────────────────────────────────────
-
-/** Shape of legacy flat config (pre-nesting). */
-interface FlatConfig {
-  apiUrl?: string;
-  apiKey?: string;
-  requestTimeoutMs?: number;
-  maxQueryLength?: number;
-  maxContentChars?: number;
-  maxContentLength?: number;
-  autoCaptureThreshold?: number;
-  ingestMode?: "smart" | "raw";
-  toastDelayMs?: number;
-  logEnabled?: boolean;
-  logLevel?: "DEBUG" | "INFO" | "WARN" | "ERROR";
-  logDir?: string;
-  // Nested fields that would indicate new format
-  connection?: unknown;
-}
-
-function isFlatConfig(cfg: Record<string, unknown>): boolean {
-  return "apiUrl" in cfg && !("connection" in cfg);
-}
-
-function migrateFlatToNested(flat: FlatConfig): CerebroPluginConfig {
-  return {
-    connection: {
-      apiUrl: flat.apiUrl ?? DEFAULTS.connection.apiUrl,
-      apiKey: flat.apiKey ?? DEFAULTS.connection.apiKey,
-      requestTimeoutMs: flat.requestTimeoutMs ?? DEFAULTS.connection.requestTimeoutMs,
-    },
-    content: {
-      maxQueryLength: flat.maxQueryLength ?? DEFAULTS.content.maxQueryLength,
-      maxContentChars: flat.maxContentChars ?? DEFAULTS.content.maxContentChars,
-      maxContentLength: flat.maxContentLength ?? DEFAULTS.content.maxContentLength,
-    },
-    injection: { ...DEFAULTS.injection },
-    ingest: {
-      autoCaptureThreshold: flat.autoCaptureThreshold ?? DEFAULTS.ingest.autoCaptureThreshold,
-      ingestMode: flat.ingestMode ?? DEFAULTS.ingest.ingestMode,
-    },
-    logging: {
-      logEnabled: flat.logEnabled ?? DEFAULTS.logging.logEnabled,
-      logLevel: flat.logLevel ?? DEFAULTS.logging.logLevel,
-      logDir: flat.logDir ?? DEFAULTS.logging.logDir,
-    },
-    ui: {
-      toastDelayMs: flat.toastDelayMs ?? DEFAULTS.ui.toastDelayMs,
-    },
-  };
-}
-
-// ── Helpers ──────────────────────────────────────────────────────────
-
-type IngestMode = "smart" | "raw";
-const INGEST_MODES: ReadonlySet<string> = new Set<IngestMode>(["smart", "raw"]);
-
-function deepMerge(base: CerebroPluginConfig, overrides: Partial<CerebroPluginConfig>): CerebroPluginConfig {
-  const result: CerebroPluginConfig = {
-    connection: { ...base.connection, ...overrides.connection },
-    content: { ...base.content, ...overrides.content },
-    injection: { ...base.injection, ...overrides.injection },
-    ingest: { ...base.ingest, ...overrides.ingest },
-    logging: { ...base.logging, ...overrides.logging },
-    ui: { ...base.ui, ...overrides.ui },
-  };
-  result.web = { ...base.web!, ...overrides.web };
-  if (overrides.agentMemoryPolicy) result.agentMemoryPolicy = overrides.agentMemoryPolicy;
-  if (overrides.defaultPolicy) result.defaultPolicy = overrides.defaultPolicy;
-  if (overrides.autoUpdate !== undefined) result.autoUpdate = overrides.autoUpdate;
-  return result;
-}
-
-// ── Load config ──────────────────────────────────────────────────────
-
-const LEVEL_MAP: Record<string, number> = { DEBUG: 0, INFO: 1, WARN: 2, ERROR: 3 };
-
-function readConfiguredLogLevel(): number {
-  try {
-    const cfgPath = join(homedir(), ".config", "cerebro", "config.json");
-    const raw = JSON.parse(readFileSync(cfgPath, "utf-8")) as Record<string, unknown>;
-    const nested = (raw?.logging as Record<string, unknown>)?.logLevel as string | undefined;
-    const flat = raw?.logLevel as string | undefined;
-    const level = nested ?? flat ?? "INFO";
-    return LEVEL_MAP[level] ?? LEVEL_MAP.INFO;
-  } catch {
-    return LEVEL_MAP.INFO;
-  }
-}
-
-const CONFIGURED_MIN_LEVEL = readConfiguredLogLevel();
-
-/** File-only logger for config.ts (cannot import logger.ts due to circular dependency). */
-function configLog(message: string, fields?: Record<string, unknown>, level: string = "WARN"): void {
-  const lvl = LEVEL_MAP[level] ?? 0;
-  if (lvl < CONFIGURED_MIN_LEVEL) return;
-  try {
-    const logDir = join(homedir(), ".config", "cerebro", "logs");
-    const logPath = join(logDir, "cerebro.log");
-    const ts = new Date().toISOString().replace("T", " ").replace(/\.\d+Z$/, "");
-    const parts = [`${level.padEnd(5)} ${ts} service=cerebro ${message}`];
-    if (fields) {
-      for (const [k, v] of Object.entries(fields)) {
-        parts.push(`${k}=${typeof v === "string" ? v : JSON.stringify(v)}`);
-      }
-    }
-    mkdirSync(logDir, { recursive: true });
-    appendFileSync(logPath, parts.join(" ") + "\n");
-  } catch (writeErr) {
-    process.stderr.write(`[cerebro] configLog write failed: ${writeErr instanceof Error ? writeErr.message : String(writeErr)}\n`);
-  }
-}
-
-export function loadPluginConfig(overrides?: Partial<CerebroPluginConfig>): CerebroPluginConfig {
-  let config: CerebroPluginConfig = structuredClone(DEFAULTS);
-
-  // Try loading from config file
-  try {
-    const cfgPath = join(homedir(), ".config", "cerebro", "config.json");
-    const raw = JSON.parse(readFileSync(cfgPath, "utf-8")) as Record<string, unknown>;
-
-    // Auto-migrate flat format
-    const parsed: CerebroPluginConfig = isFlatConfig(raw) ? migrateFlatToNested(raw as FlatConfig) : raw as unknown as CerebroPluginConfig;
-
-    // Merge nested groups with defaults for safety
-    config = deepMerge(config, parsed);
-  } catch (e) {
-    configLog("config.json load failed, using defaults", { error: String(e) });
-  }
-
-  // Apply explicit overrides (from opencode.json)
-  if (overrides) {
-    config = deepMerge(config, overrides);
-  }
-
-  // Apply environment variable overrides last — env vars have highest priority
-  if (process.env.OMEM_API_URL) config.connection.apiUrl = process.env.OMEM_API_URL;
-  if (process.env.OMEM_API_KEY) config.connection.apiKey = process.env.OMEM_API_KEY;
-  if (process.env.OMEM_REQUEST_TIMEOUT_MS) {
-    config.connection.requestTimeoutMs = parseInt(process.env.OMEM_REQUEST_TIMEOUT_MS, 10) || DEFAULTS.connection.requestTimeoutMs;
-  }
-  if (process.env.OMEM_AUTO_CAPTURE_THRESHOLD) {
-    config.ingest.autoCaptureThreshold = parseInt(process.env.OMEM_AUTO_CAPTURE_THRESHOLD, 10) || DEFAULTS.ingest.autoCaptureThreshold;
-  }
-  if (INGEST_MODES.has(process.env.OMEM_INGEST_MODE ?? "")) {
-    config.ingest.ingestMode = process.env.OMEM_INGEST_MODE as IngestMode;
-  }
-
-  if (process.env.OMEM_WEB_ENABLED === "false" || process.env.OMEM_WEB_ENABLED === "0") {
-    config.web = { ...config.web!, enabled: false };
-  }
-  if (process.env.OMEM_LOCAL_PORT) {
-    config.web = { ...config.web!, port: parseInt(process.env.OMEM_LOCAL_PORT, 10) || DEFAULTS.web!.port };
-  }
-
-  // Expand ~ to home directory in logDir
-  if (config.logging.logDir?.startsWith("~")) {
-    config.logging.logDir = config.logging.logDir.replace(/^~/, homedir());
-  }
-
-  return config;
-}
-
-// ── Agent policy resolver ────────────────────────────────────────────
-
-export type AgentPolicy = "none" | "readonly" | "readwrite";
-
-export function resolveAgentPolicy(
-  agentName: string,
-  config: Partial<CerebroPluginConfig>,
-): AgentPolicy {
-  const policies = config.agentMemoryPolicy;
-  if (policies) {
-    const exact = policies[agentName];
-    if (exact) return exact;
-    const lower = agentName.toLowerCase();
-    for (const [key, policy] of Object.entries(policies)) {
-      if (lower.startsWith(key.toLowerCase()) || key.toLowerCase().startsWith(lower)) {
-        return policy;
-      }
-    }
-  }
-  if (config.defaultPolicy) return config.defaultPolicy;
-  configLog("resolveAgentPolicy: defaulting to readwrite", { agentName }, "DEBUG");
-  return "readwrite";
-}
-
-export { DEFAULTS };
+import { readFileSync, appendFileSync, mkdirSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+
+// ── Nested config interface ──────────────────────────────────────────
+
+export interface CerebroPluginConfig {
+  connection: {
+    apiUrl: string;
+    apiKey: string;
+    requestTimeoutMs: number;
+  };
+  content: {
+    maxQueryLength: number;
+    maxContentChars: number;
+    maxContentLength: number;
+  };
+  injection: {
+    recentCount: number;
+    searchCount: number;
+    globalCount: number;
+    recentTruncateChars: number;
+    searchTruncateChars: number;
+    recentTimeoutMs: number;
+    searchTimeoutMs: number;
+    profileTimeoutMs: number;
+  };
+  ingest: {
+    autoCaptureThreshold: number;
+    ingestMode: "smart" | "raw";
+  };
+  logging: {
+    logEnabled: boolean;
+    logLevel: "DEBUG" | "INFO" | "WARN" | "ERROR";
+    logDir: string;
+  };
+  ui: {
+    toastDelayMs: number;
+  };
+  web?: {
+    enabled?: boolean;
+    port?: number;
+  };
+  agentMemoryPolicy?: Record<string, "none" | "readonly" | "readwrite">;
+  defaultPolicy?: "none" | "readonly" | "readwrite";
+  autoUpdate?: boolean;  // default false
+}
+
+// ── Defaults ─────────────────────────────────────────────────────────
+
+const DEFAULTS: CerebroPluginConfig = {
+  connection: {
+    apiUrl: "https://www.mengxy.cc",
+    apiKey: "",
+    requestTimeoutMs: 15000,
+  },
+  content: {
+    maxQueryLength: 200,
+    maxContentChars: 30000,
+    maxContentLength: 3000,
+  },
+  injection: {
+    recentCount: 5,
+    searchCount: 10,
+    globalCount: 3,
+    recentTruncateChars: 0,   // 0 = 不截断
+    searchTruncateChars: 0,   // 0 = 不截断
+    recentTimeoutMs: 3000,
+    searchTimeoutMs: 5000,
+    profileTimeoutMs: 2000,
+  },
+  ingest: {
+    autoCaptureThreshold: 5,
+    ingestMode: "smart",
+  },
+  logging: {
+    logEnabled: true,
+    logLevel: "INFO",
+    logDir: join(homedir(), ".config", "cerebro"),
+  },
+  ui: {
+    toastDelayMs: 7000,
+  },
+  web: {
+    enabled: true,
+  },
+  autoUpdate: false,
+};
+
+// ── Flat-to-nested migration ─────────────────────────────────────────
+
+/** Shape of legacy flat config (pre-nesting). */
+interface FlatConfig {
+  apiUrl?: string;
+  apiKey?: string;
+  requestTimeoutMs?: number;
+  maxQueryLength?: number;
+  maxContentChars?: number;
+  maxContentLength?: number;
+  autoCaptureThreshold?: number;
+  ingestMode?: "smart" | "raw";
+  toastDelayMs?: number;
+  logEnabled?: boolean;
+  logLevel?: "DEBUG" | "INFO" | "WARN" | "ERROR";
+  logDir?: string;
+  // Nested fields that would indicate new format
+  connection?: unknown;
+}
+
+function isFlatConfig(cfg: Record<string, unknown>): boolean {
+  return "apiUrl" in cfg && !("connection" in cfg);
+}
+
+function migrateFlatToNested(flat: FlatConfig): CerebroPluginConfig {
+  return {
+    connection: {
+      apiUrl: flat.apiUrl ?? DEFAULTS.connection.apiUrl,
+      apiKey: flat.apiKey ?? DEFAULTS.connection.apiKey,
+      requestTimeoutMs: flat.requestTimeoutMs ?? DEFAULTS.connection.requestTimeoutMs,
+    },
+    content: {
+      maxQueryLength: flat.maxQueryLength ?? DEFAULTS.content.maxQueryLength,
+      maxContentChars: flat.maxContentChars ?? DEFAULTS.content.maxContentChars,
+      maxContentLength: flat.maxContentLength ?? DEFAULTS.content.maxContentLength,
+    },
+    injection: { ...DEFAULTS.injection },
+    ingest: {
+      autoCaptureThreshold: flat.autoCaptureThreshold ?? DEFAULTS.ingest.autoCaptureThreshold,
+      ingestMode: flat.ingestMode ?? DEFAULTS.ingest.ingestMode,
+    },
+    logging: {
+      logEnabled: flat.logEnabled ?? DEFAULTS.logging.logEnabled,
+      logLevel: flat.logLevel ?? DEFAULTS.logging.logLevel,
+      logDir: flat.logDir ?? DEFAULTS.logging.logDir,
+    },
+    ui: {
+      toastDelayMs: flat.toastDelayMs ?? DEFAULTS.ui.toastDelayMs,
+    },
+  };
+}
+
+// ── Helpers ──────────────────────────────────────────────────────────
+
+type IngestMode = "smart" | "raw";
+const INGEST_MODES: ReadonlySet<string> = new Set<IngestMode>(["smart", "raw"]);
+
+function deepMerge(base: CerebroPluginConfig, overrides: Partial<CerebroPluginConfig>): CerebroPluginConfig {
+  const result: CerebroPluginConfig = {
+    connection: { ...base.connection, ...overrides.connection },
+    content: { ...base.content, ...overrides.content },
+    injection: { ...base.injection, ...overrides.injection },
+    ingest: { ...base.ingest, ...overrides.ingest },
+    logging: { ...base.logging, ...overrides.logging },
+    ui: { ...base.ui, ...overrides.ui },
+  };
+  result.web = { ...base.web!, ...overrides.web };
+  if (overrides.agentMemoryPolicy) result.agentMemoryPolicy = overrides.agentMemoryPolicy;
+  if (overrides.defaultPolicy) result.defaultPolicy = overrides.defaultPolicy;
+  if (overrides.autoUpdate !== undefined) result.autoUpdate = overrides.autoUpdate;
+  return result;
+}
+
+// ── Load config ──────────────────────────────────────────────────────
+
+const LEVEL_MAP: Record<string, number> = { DEBUG: 0, INFO: 1, WARN: 2, ERROR: 3 };
+
+function readConfiguredLogLevel(): number {
+  try {
+    const cfgPath = join(homedir(), ".config", "cerebro", "config.json");
+    const raw = JSON.parse(readFileSync(cfgPath, "utf-8")) as Record<string, unknown>;
+    const nested = (raw?.logging as Record<string, unknown>)?.logLevel as string | undefined;
+    const flat = raw?.logLevel as string | undefined;
+    const level = nested ?? flat ?? "INFO";
+    return LEVEL_MAP[level] ?? LEVEL_MAP.INFO;
+  } catch {
+    return LEVEL_MAP.INFO;
+  }
+}
+
+const CONFIGURED_MIN_LEVEL = readConfiguredLogLevel();
+
+/** File-only logger for config.ts (cannot import logger.ts due to circular dependency). */
+function configLog(message: string, fields?: Record<string, unknown>, level: string = "WARN"): void {
+  const lvl = LEVEL_MAP[level] ?? 0;
+  if (lvl < CONFIGURED_MIN_LEVEL) return;
+  try {
+    const logDir = join(homedir(), ".config", "cerebro", "logs");
+    const logPath = join(logDir, "cerebro.log");
+    const ts = new Date().toISOString().replace("T", " ").replace(/\.\d+Z$/, "");
+    const parts = [`${level.padEnd(5)} ${ts} service=cerebro ${message}`];
+    if (fields) {
+      for (const [k, v] of Object.entries(fields)) {
+        parts.push(`${k}=${typeof v === "string" ? v : JSON.stringify(v)}`);
+      }
+    }
+    mkdirSync(logDir, { recursive: true });
+    appendFileSync(logPath, parts.join(" ") + "\n");
+  } catch (writeErr) {
+    process.stderr.write(`[cerebro] configLog write failed: ${writeErr instanceof Error ? writeErr.message : String(writeErr)}\n`);
+  }
+}
+
+export function loadPluginConfig(overrides?: Partial<CerebroPluginConfig>): CerebroPluginConfig {
+  let config: CerebroPluginConfig = structuredClone(DEFAULTS);
+
+  // Try loading from config file
+  try {
+    const cfgPath = join(homedir(), ".config", "cerebro", "config.json");
+    const raw = JSON.parse(readFileSync(cfgPath, "utf-8")) as Record<string, unknown>;
+
+    // Auto-migrate flat format
+    const parsed: CerebroPluginConfig = isFlatConfig(raw) ? migrateFlatToNested(raw as FlatConfig) : raw as unknown as CerebroPluginConfig;
+
+    // Merge nested groups with defaults for safety
+    config = deepMerge(config, parsed);
+  } catch (e) {
+    configLog("config.json load failed, using defaults", { error: String(e) });
+  }
+
+  // Apply explicit overrides (from opencode.json)
+  if (overrides) {
+    config = deepMerge(config, overrides);
+  }
+
+  // Apply environment variable overrides last — env vars have highest priority
+  if (process.env.OMEM_API_URL) config.connection.apiUrl = process.env.OMEM_API_URL;
+  if (process.env.OMEM_API_KEY) config.connection.apiKey = process.env.OMEM_API_KEY;
+  if (process.env.OMEM_REQUEST_TIMEOUT_MS) {
+    config.connection.requestTimeoutMs = parseInt(process.env.OMEM_REQUEST_TIMEOUT_MS, 10) || DEFAULTS.connection.requestTimeoutMs;
+  }
+  if (process.env.OMEM_AUTO_CAPTURE_THRESHOLD) {
+    config.ingest.autoCaptureThreshold = parseInt(process.env.OMEM_AUTO_CAPTURE_THRESHOLD, 10) || DEFAULTS.ingest.autoCaptureThreshold;
+  }
+  if (INGEST_MODES.has(process.env.OMEM_INGEST_MODE ?? "")) {
+    config.ingest.ingestMode = process.env.OMEM_INGEST_MODE as IngestMode;
+  }
+
+  if (process.env.OMEM_WEB_ENABLED === "false" || process.env.OMEM_WEB_ENABLED === "0") {
+    config.web = { ...config.web!, enabled: false };
+  }
+  if (process.env.OMEM_LOCAL_PORT) {
+    config.web = { ...config.web!, port: parseInt(process.env.OMEM_LOCAL_PORT, 10) || DEFAULTS.web!.port };
+  }
+
+  // Expand ~ to home directory in logDir
+  if (config.logging.logDir?.startsWith("~")) {
+    config.logging.logDir = config.logging.logDir.replace(/^~/, homedir());
+  }
+
+  return config;
+}
+
+// ── Agent policy resolver ────────────────────────────────────────────
+
+export type AgentPolicy = "none" | "readonly" | "readwrite";
+
+export function resolveAgentPolicy(
+  agentName: string,
+  config: Partial<CerebroPluginConfig>,
+): AgentPolicy {
+  const policies = config.agentMemoryPolicy;
+  if (policies) {
+    const exact = policies[agentName];
+    if (exact) return exact;
+    const lower = agentName.toLowerCase();
+    for (const [key, policy] of Object.entries(policies)) {
+      if (lower.startsWith(key.toLowerCase()) || key.toLowerCase().startsWith(lower)) {
+        return policy;
+      }
+    }
+  }
+  if (config.defaultPolicy) return config.defaultPolicy;
+  configLog("resolveAgentPolicy: defaulting to readwrite", { agentName }, "DEBUG");
+  return "readwrite";
+}
+
+export { DEFAULTS };