metona-ai-desktop/electron/services/config.service.ts

/**
 * Config Service — 配置读写
 *
 * 管理 app_config 表的读写操作。
 * 所有配置值以 JSON 格式存储。
 *
 * v0.3.17: 引入全局配置层（GlobalConfigService）。
 *   - 全局 key（llm.* / agent.* / onboarding.completed 等）读取时优先工作空间 DB，
 *     miss 时回退到全局 JSON，确保切换工作空间后 LLM 凭据和引导状态不丢失。
 *   - 写入时同时写工作空间 DB（兼容）和全局 JSON（跨工作空间共享）。
 *
 * @see docs/MetonaAI-Desktop 架构与交互设计.html — 数据库配置
 */

import type Database from 'better-sqlite3';
import log from 'electron-log';
import { GlobalConfigService, isGlobalKey, isUnconfiguredGlobalKey } from './global-config.service';
import {
  decryptConfigValue,
  encryptConfigValue,
  isSensitiveConfigKey,
} from '../utils/secure-config';

export interface ConfigRow {
  key: string;
  value: string;
  category: string;
  updated_at: number;
}

export class ConfigService {
  private globalConfig: GlobalConfigService | null = null;

  constructor(private getDB: () => Database.Database) {}

  /**
   * 注入全局配置服务（可选，由 main.ts 在启动时调用）
   * 注入后，全局 key 的读取会回退到全局层，写入会双写。
   */
  setGlobalConfig(globalConfig: GlobalConfigService): void {
    this.globalConfig = globalConfig;
  }

  /**
   * 获取配置值
   * 读取顺序：工作空间 DB → 全局 JSON（仅全局 key）
   *
   * 特殊处理：对于"空值视为未配置"的全局 key（如 llm.apiKey=''、onboarding.completed=false），
   * 当工作空间 DB 的值为空/默认值时，回退到全局层读取真实值。
   * 这解决了新工作空间 DB 被 seedDefaults 灌入空值导致配置丢失的问题。
   */
  get<T = unknown>(key: string): T | null {
    const db = this.getDB();
    const row = db.prepare('SELECT value FROM app_config WHERE key = ?').get(key) as
      | ConfigRow
      | undefined;

    if (row) {
      let parsed: unknown;
      try {
        parsed = JSON.parse(row.value);
      } catch {
        parsed = row.value;
      }

      // P0-1: 敏感 key 落盘为密文，读取时解密（含历史明文的平滑兼容——明文原样返回）
      if (isSensitiveConfigKey(key)) {
        parsed = decryptConfigValue(parsed);
      }

      // 全局 key 的空值回退：若工作空间 DB 存在该 key 但值为"未配置"（如 llm.apiKey=''），
      // 且全局层有真实值，则返回全局层的值。
      // 解决场景：新工作空间 DB 被 seedDefaults 灌入 llm.apiKey='' / onboarding.completed=false，
      // 但用户在全局层已配置过，应回退到全局层。
      if (this.globalConfig && isGlobalKey(key) && isUnconfiguredGlobalKey(key, parsed)) {
        const globalValue = this.globalConfig.get<T>(key);
        if (globalValue !== null && !isUnconfiguredGlobalKey(key, globalValue)) {
          return globalValue;
        }
      }

      return parsed as T;
    }

    // 工作空间 DB miss，若为全局 key 则回退到全局层
    if (this.globalConfig && isGlobalKey(key)) {
      return this.globalConfig.get<T>(key);
    }

    return null;
  }

  /**
   * 设置配置值（保留已有 category）
   * 全局 key 同时写入工作空间 DB 和全局 JSON
   * P0-1: 敏感 key（apiKey 等）加密后再落盘（DB 与全局 JSON 均为密文）
   */
  set(key: string, value: unknown): void {
    const db = this.getDB();
    const storedValue = isSensitiveConfigKey(key) ? encryptConfigValue(value) : value;
    const jsonValue = JSON.stringify(storedValue);

    // 查询已有 category，避免 INSERT OR REPLACE 覆盖为默认值
    const existing = db.prepare('SELECT category FROM app_config WHERE key = ?').get(key) as
      | ConfigRow
      | undefined;
    const category = existing?.category ?? 'general';

    db.prepare(
      `
      INSERT OR REPLACE INTO app_config (key, value, category, updated_at)
      VALUES (?, ?, ?, ?)
    `,
    ).run(key, jsonValue, category, Date.now());

    // 全局 key 同步写入全局层（跨工作空间共享；全局层内部同样加密）
    if (this.globalConfig && isGlobalKey(key)) {
      this.globalConfig.set(key, storedValue);
    }

    log.debug(`Config set: ${key}`);
  }

  /**
   * v0.7.4 P3-6: 批量设置配置值（供 config:setBatch 使用）。
   *
   * 旧实现：setBatch 循环内逐条调用 configService.set —— 每条都对全局 key 触发
   * globalConfig.set → flush() 落盘一次文件，10 个 key 的批量保存 = 10 次同步磁盘 IO。
   * 本方法：DB 批量写入（单事务）+ 全局层累积后一次 setBatch（一次 flush 落盘）。
   *
   * 语义与 set() 完全一致：敏感 key 加密、保留已有 category、全局 key 双写。
   */
  setBatch(entries: Array<{ key: string; value: unknown }>): void {
    const db = this.getDB();
    const globalEntries: Array<{ key: string; value: unknown }> = [];

    db.transaction((items: Array<{ key: string; value: unknown }>) => {
      for (const { key, value } of items) {
        const storedValue = isSensitiveConfigKey(key) ? encryptConfigValue(value) : value;
        const jsonValue = JSON.stringify(storedValue);
        const existing = db.prepare('SELECT category FROM app_config WHERE key = ?').get(key) as
          | ConfigRow
          | undefined;
        const category = existing?.category ?? 'general';
        db.prepare(
          `
          INSERT OR REPLACE INTO app_config (key, value, category, updated_at)
          VALUES (?, ?, ?, ?)
        `,
        ).run(key, jsonValue, category, Date.now());
        if (this.globalConfig && isGlobalKey(key)) {
          globalEntries.push({ key, value: storedValue });
        }
      }
    })(entries);

    // 全局层一次批量落盘（替代逐条 set 的 N 次 flush）
    if (globalEntries.length > 0 && this.globalConfig) {
      this.globalConfig.setBatch(globalEntries);
    }

    log.debug(`Config setBatch: ${entries.length} keys`);
  }

  /**
   * 获取所有配置（工作空间 DB + 全局层合并）
   * 合并规则与 get() 一致：全局 key 在 DB 中为"未配置"时回退到全局层
   */
  getAll(): Record<string, unknown> {
    const db = this.getDB();
    const rows = db.prepare('SELECT key, value FROM app_config').all() as ConfigRow[];

    const config: Record<string, unknown> = {};
    for (const row of rows) {
      try {
        config[row.key] = JSON.parse(row.value);
      } catch {
        config[row.key] = row.value;
      }
      // P0-1: 敏感 key 解密后再返回（与 get() 行为一致）
      if (isSensitiveConfigKey(row.key)) {
        config[row.key] = decryptConfigValue(config[row.key]);
      }
    }

    // 全局层合并：与 get() 的回退逻辑保持一致
    if (this.globalConfig) {
      const globalAll = this.globalConfig.getAll();
      for (const [key, globalValue] of Object.entries(globalAll)) {
        if (!(key in config)) {
          // DB miss → 用全局层
          config[key] = globalValue;
        } else if (
          isGlobalKey(key) &&
          isUnconfiguredGlobalKey(key, config[key]) &&
          !isUnconfiguredGlobalKey(key, globalValue)
        ) {
          // DB 有值但为"未配置"（seedDefaults 默认值/空值），全局层有真实值 → 用全局层
          config[key] = globalValue;
        }
      }
    }

    return config;
  }

  /**
   * 获取指定分类的所有配置
   */
  getByCategory(category: string): Record<string, unknown> {
    const db = this.getDB();
    const rows = db
      .prepare('SELECT key, value FROM app_config WHERE category = ?')
      .all(category) as ConfigRow[];

    const config: Record<string, unknown> = {};
    for (const row of rows) {
      try {
        config[row.key] = JSON.parse(row.value);
      } catch {
        config[row.key] = row.value;
      }
    }

    return config;
  }

  /**
   * 删除配置
   */
  delete(key: string): boolean {
    const db = this.getDB();
    const result = db.prepare('DELETE FROM app_config WHERE key = ?').run(key);
    return result.changes > 0;
  }
}