@huyooo/ai-chat-bridge-electron 0.1.6 → 0.1.8

package/src/preload/index.ts

@@ -0,0 +1,457 @@
+/**
+ * Electron Preload 桥接
+ * 
+ * 在 preload 脚本中调用，暴露 API 给渲染进程
+ */
+
+import { contextBridge, ipcRenderer } from 'electron';
+
+/** 发送消息参数 */
+interface SendMessageParams {
+  message: string;
+  images?: string[];
+  sessionId?: string;
+  options?: {
+    mode?: string;
+    model?: string;
+    provider?: string;
+    enableWebSearch?: boolean;
+    thinkingMode?: string;
+    thinkingBudget?: number;
+  };
+}
+
+/** 模型配置 */
+interface ModelOption {
+  modelId: string;
+  displayName: string;
+  /** 分组名称（由后端决定，前端只负责渲染，必填） */
+  group: string;
+  /** 是否来自 OpenRouter（保留用于兼容，后续可能移除） */
+  isOpenRouter?: boolean;
+  /** 提供商名称（保留用于兼容，后续可能移除） */
+  provider?: string;
+}
+
+/** 会话记录 */
+interface SessionRecord {
+  id: string;
+  appId: string | null;
+  userId: string | null;
+  title: string;
+  model: string;
+  mode: 'agent' | 'plan' | 'ask';
+  webSearchEnabled: boolean;
+  thinkingEnabled: boolean;
+  /** 是否在 tab 栏隐藏（关闭但不删除） */
+  hidden: boolean;
+  createdAt: Date;
+  updatedAt: Date;
+}
+
+/** 消息记录 */
+interface MessageRecord {
+  id: string;
+  sessionId: string;
+  role: 'user' | 'assistant';
+  content: string;
+  /** 生成此消息时使用的模型 */
+  model?: string | null;
+  /** 生成此消息时使用的模式 (ask/agent) */
+  mode?: string | null;
+  /** 生成此消息时是否启用 web 搜索 */
+  webSearchEnabled?: boolean | null;
+  /** 生成此消息时是否启用深度思考 */
+  thinkingEnabled?: boolean | null;
+  /** 执行步骤列表 JSON */
+  steps?: string | null;
+  operationIds?: string | null;
+  timestamp: Date;
+}
+
+/** 操作记录 */
+interface OperationRecord {
+  id: string;
+  sessionId: string;
+  messageId?: string | null;
+  command: string;
+  operationType: string;
+  affectedFiles: string;
+  status: string;
+  timestamp: Date;
+}
+
+/** 回收站记录 */
+interface TrashRecord {
+  id: string;
+  sessionId: string;
+  originalPath: string;
+  trashPath: string;
+  deletedAt: Date;
+  autoDeleteAt: Date;
+}
+
+/** 文件信息 */
+interface FileInfo {
+  name: string;
+  path: string;
+  isDirectory: boolean;
+  size: number;
+  modifiedAt: Date;
+  extension: string;
+}
+
+export interface AiChatBridge {
+  // ============ Chat API ============
+  /** 获取可用模型 */
+  getModels(): Promise<ModelOption[]>;
+  /** 发送消息 */
+  send(params: SendMessageParams): Promise<void>;
+  /** 取消当前请求 */
+  cancel(): Promise<void>;
+  // 无状态架构：历史通过 ChatOptions.history 传入
+  /** 设置当前工作目录 */
+  setCwd(dir: string): Promise<void>;
+  /** 获取配置 */
+  getConfig(): Promise<unknown>;
+  /** 监听进度事件 */
+  onProgress(callback: (progress: unknown) => void): () => void;
+
+  // ============ Sessions API ============
+  /** 获取会话列表 */
+  getSessions(): Promise<SessionRecord[]>;
+  /** 获取单个会话 */
+  getSession(id: string): Promise<SessionRecord | null>;
+  /** 创建会话 */
+  createSession(params?: { title?: string; model?: string; mode?: string; webSearchEnabled?: boolean; thinkingEnabled?: boolean; hidden?: boolean }): Promise<SessionRecord>;
+  /** 更新会话 */
+  updateSession(id: string, data: { title?: string; model?: string; mode?: string; webSearchEnabled?: boolean; thinkingEnabled?: boolean; hidden?: boolean }): Promise<SessionRecord | null>;
+  /** 删除会话 */
+  deleteSession(id: string): Promise<{ success: boolean }>;
+
+  // ============ Messages API ============
+  /** 获取会话消息 */
+  getMessages(sessionId: string): Promise<MessageRecord[]>;
+  /** 保存消息 */
+  saveMessage(params: {
+    /** 消息 ID（可选，如果不传则自动生成） */
+    id?: string;
+    sessionId: string;
+    role: 'user' | 'assistant';
+    content: string;
+    /** 生成此消息时使用的模型 */
+    model?: string;
+    /** 生成此消息时使用的模式 (ask/agent) */
+    mode?: string;
+    /** 生成此消息时是否启用 web 搜索 */
+    webSearchEnabled?: boolean;
+    /** 生成此消息时是否启用深度思考 */
+    thinkingEnabled?: boolean;
+    /** 执行步骤列表 JSON */
+    steps?: string;
+    operationIds?: string;
+  }): Promise<MessageRecord>;
+  /** 更新消息（增量保存） */
+  updateMessage(params: {
+    id: string;
+    content?: string;
+    steps?: string;
+  }): Promise<{ success: boolean }>;
+  /** 删除指定时间之后的消息（用于分叉） */
+  deleteMessagesAfter(sessionId: string, timestamp: number): Promise<{ success: boolean }>;
+
+  // ============ Operations API ============
+  /** 获取操作日志 */
+  getOperations(sessionId: string): Promise<OperationRecord[]>;
+
+  // ============ Trash API ============
+  /** 获取回收站 */
+  getTrashItems(): Promise<TrashRecord[]>;
+  /** 恢复回收站项目 */
+  restoreFromTrash(id: string): Promise<TrashRecord | undefined>;
+
+  // ============ System API ============
+  /** 在系统默认浏览器中打开链接 */
+  openExternal(url: string): Promise<void>;
+
+  // ============ File System API ============
+  /** 列出目录内容 */
+  listDir(dirPath: string): Promise<FileInfo[]>;
+  /** 检查路径是否存在 */
+  exists(filePath: string): Promise<boolean>;
+  /** 获取文件信息 */
+  stat(filePath: string): Promise<FileInfo | null>;
+  /** 读取文件内容（文本） */
+  readFile(filePath: string): Promise<string | null>;
+  /** 读取文件为 base64 */
+  readFileBase64(filePath: string): Promise<string | null>;
+  /** 获取用户主目录 */
+  homeDir(): Promise<string>;
+  /** 解析路径（处理 ~） */
+  resolvePath(inputPath: string): Promise<string>;
+  /** 获取父目录 */
+  parentDir(dirPath: string): Promise<string>;
+  /** 监听目录变化 */
+  watchDir(dirPath: string): Promise<boolean>;
+  /** 停止监听目录 */
+  unwatchDir(dirPath: string): Promise<void>;
+  /** 监听目录变化事件 */
+  onDirChange(callback: (data: { dirPath: string; eventType: string; filename: string | null }) => void): () => void;
+
+  // ============ Settings API ============
+  /** 获取用户设置 */
+  getSetting(key: string): Promise<string | null>;
+  /** 设置用户设置 */
+  setSetting(key: string, value: string): Promise<{ success: boolean }>;
+  /** 获取所有用户设置 */
+  getAllSettings(): Promise<Record<string, string>>;
+  /** 删除用户设置 */
+  deleteSetting(key: string): Promise<{ success: boolean }>;
+
+  // ============ Index API ============
+  /** 获取索引统计信息 */
+  getIndexStats(): Promise<{
+    totalDocuments: number;
+    indexSize: number;
+    lastUpdated: string | null;
+  }>;
+  /** 检查索引状态 */
+  getIndexStatus(): Promise<{ 
+    isIndexing: boolean;
+    lastProgress: {
+      indexed: number;
+      total: number;
+      currentFile?: string;
+      stage: string;
+    } | null;
+  }>;
+  /** 同步索引（重新索引工作空间） */
+  syncIndex(): Promise<{ success: boolean }>;
+  /** 取消索引 */
+  cancelIndex(): Promise<{ success: boolean }>;
+  /** 删除索引 */
+  deleteIndex(): Promise<{ success: boolean }>;
+  /** 注册索引进度监听器 */
+  registerIndexListener(): Promise<{ success: boolean }>;
+  /** 注销索引进度监听器 */
+  unregisterIndexListener(): Promise<{ success: boolean }>;
+  /** 监听索引进度 */
+  onIndexProgress(callback: (progress: {
+    indexed: number;
+    total: number;
+    currentFile?: string;
+    stage: string;
+    error?: string;
+  }) => void): () => void;
+
+  // ============ Tool Approval API ============
+  /** 监听工具批准请求（manual 模式） */
+  onToolApprovalRequest(callback: (request: { id: string; name: string; args: Record<string, unknown> }) => void): () => void;
+  /** 响应工具批准请求 */
+  respondToolApproval(id: string, approved: boolean): Promise<void>;
+}
+
+export interface ExposeOptions {
+  /** IPC channel 前缀 */
+  channelPrefix?: string;
+  /** 暴露的全局变量名 */
+  exposeName?: string;
+}
+
+/**
+ * 暴露 AI Chat API 给渲染进程
+ */
+export function exposeElectronBridge(options: ExposeOptions = {}) {
+  const { channelPrefix = 'ai-chat', exposeName = 'aiChatBridge' } = options;
+
+  const bridge: AiChatBridge = {
+    // ============ Chat API ============
+    getModels: () => 
+      ipcRenderer.invoke(`${channelPrefix}:models`),
+
+    send: (params: SendMessageParams) => 
+      ipcRenderer.invoke(`${channelPrefix}:send`, params),
+
+    cancel: () => 
+      ipcRenderer.invoke(`${channelPrefix}:cancel`),
+
+    // 无状态架构：历史通过 ChatOptions.history 传入
+
+    setCwd: (dir: string) => 
+      ipcRenderer.invoke(`${channelPrefix}:setCwd`, dir),
+
+    getConfig: () => 
+      ipcRenderer.invoke(`${channelPrefix}:config`),
+
+    onProgress: (callback: (progress: unknown) => void) => {
+      const handler = (_event: Electron.IpcRendererEvent, progress: unknown) => {
+        callback(progress);
+      };
+      ipcRenderer.on(`${channelPrefix}:progress`, handler);
+      
+      return () => {
+        ipcRenderer.removeListener(`${channelPrefix}:progress`, handler);
+      };
+    },
+
+    onToolApprovalRequest: (callback: (request: { id: string; name: string; args: Record<string, unknown> }) => void) => {
+      const handler = (_event: Electron.IpcRendererEvent, request: { id: string; name: string; args: Record<string, unknown> }) => {
+        callback(request);
+      };
+      ipcRenderer.on(`${channelPrefix}:toolApprovalRequest`, handler);
+      
+      return () => {
+        ipcRenderer.removeListener(`${channelPrefix}:toolApprovalRequest`, handler);
+      };
+    },
+
+    respondToolApproval: (id: string, approved: boolean) =>
+      ipcRenderer.invoke(`${channelPrefix}:toolApprovalResponse`, { id, approved }),
+
+    // ============ Sessions API ============
+    getSessions: () =>
+      ipcRenderer.invoke(`${channelPrefix}:sessions:list`),
+
+    getSession: (id: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:sessions:get`, id),
+
+    createSession: (params) =>
+      ipcRenderer.invoke(`${channelPrefix}:sessions:create`, params || {}),
+
+    updateSession: (id: string, data) =>
+      ipcRenderer.invoke(`${channelPrefix}:sessions:update`, id, data),
+
+    deleteSession: (id: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:sessions:delete`, id),
+
+    // ============ Messages API ============
+    getMessages: (sessionId: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:messages:list`, sessionId),
+
+    saveMessage: (params) =>
+      ipcRenderer.invoke(`${channelPrefix}:messages:save`, params),
+
+    updateMessage: (params) =>
+      ipcRenderer.invoke(`${channelPrefix}:messages:update`, params),
+
+    deleteMessagesAfter: (sessionId: string, timestamp: number) =>
+      ipcRenderer.invoke(`${channelPrefix}:messages:deleteAfter`, sessionId, timestamp),
+
+    // ============ Operations API ============
+    getOperations: (sessionId: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:operations:list`, sessionId),
+
+    // ============ Trash API ============
+    getTrashItems: () =>
+      ipcRenderer.invoke(`${channelPrefix}:trash:list`),
+
+    restoreFromTrash: (id: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:trash:restore`, id),
+
+    // ============ System API ============
+    openExternal: (url: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:openExternal`, url),
+
+    // ============ File System API ============
+    listDir: (dirPath: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:fs:listDir`, dirPath),
+
+    exists: (filePath: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:fs:exists`, filePath),
+
+    stat: (filePath: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:fs:stat`, filePath),
+
+    readFile: (filePath: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:fs:readFile`, filePath),
+
+    readFileBase64: (filePath: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:fs:readFileBase64`, filePath),
+
+    homeDir: () =>
+      ipcRenderer.invoke(`${channelPrefix}:fs:homeDir`),
+
+    resolvePath: (inputPath: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:fs:resolvePath`, inputPath),
+
+    parentDir: (dirPath: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:fs:parentDir`, dirPath),
+
+    watchDir: (dirPath: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:fs:watchDir`, dirPath),
+
+    unwatchDir: (dirPath: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:fs:unwatchDir`, dirPath),
+
+    onDirChange: (callback: (data: { dirPath: string; eventType: string; filename: string | null }) => void) => {
+      const handler = (_event: Electron.IpcRendererEvent, data: { dirPath: string; eventType: string; filename: string | null }) => {
+        callback(data);
+      };
+      ipcRenderer.on(`${channelPrefix}:fs:dirChange`, handler);
+      
+      return () => {
+        ipcRenderer.removeListener(`${channelPrefix}:fs:dirChange`, handler);
+      };
+    },
+
+    // ============ Settings API ============
+    getSetting: (key: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:settings:get`, key),
+
+    setSetting: (key: string, value: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:settings:set`, key, value),
+
+    getAllSettings: () =>
+      ipcRenderer.invoke(`${channelPrefix}:settings:getAll`),
+
+    deleteSetting: (key: string) =>
+      ipcRenderer.invoke(`${channelPrefix}:settings:delete`, key),
+
+    // ============ Index API ============
+    getIndexStats: () =>
+      ipcRenderer.invoke(`${channelPrefix}:index:getStats`),
+    
+    getIndexStatus: () =>
+      ipcRenderer.invoke(`${channelPrefix}:index:status`),
+    
+    syncIndex: () =>
+      ipcRenderer.invoke(`${channelPrefix}:index:sync`),
+    
+    cancelIndex: () =>
+      ipcRenderer.invoke(`${channelPrefix}:index:cancel`),
+    
+    deleteIndex: () =>
+      ipcRenderer.invoke(`${channelPrefix}:index:delete`),
+    
+    registerIndexListener: () =>
+      ipcRenderer.invoke(`${channelPrefix}:index:registerListener`),
+    
+    unregisterIndexListener: () =>
+      ipcRenderer.invoke(`${channelPrefix}:index:unregisterListener`),
+    
+    onIndexProgress: (callback: (progress: {
+      indexed: number;
+      total: number;
+      currentFile?: string;
+      stage: string;
+      error?: string;
+    }) => void) => {
+      const handler = (_event: Electron.IpcRendererEvent, progress: {
+        indexed: number;
+        total: number;
+        currentFile?: string;
+        stage: string;
+        error?: string;
+      }) => {
+        callback(progress);
+      };
+      ipcRenderer.on(`${channelPrefix}:index:progress`, handler);
+      
+      return () => {
+        ipcRenderer.removeListener(`${channelPrefix}:index:progress`, handler);
+      };
+    },
+  };
+
+  contextBridge.exposeInMainWorld(exposeName, bridge);
+}