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

package/src/renderer/index.ts

@@ -0,0 +1,507 @@
+/**
+ * Electron 渲染进程 Adapter
+ * 
+ * 在渲染进程中使用，创建 ChatAdapter
+ */
+
+import type { AiChatBridge } from '../preload';
+
+// 事件类型
+export type ChatEventType = 
+  | 'thinking_start'
+  | 'thinking_delta'
+  | 'thinking_end'
+  | 'search_start'
+  | 'search_result'
+  | 'search_end'
+  | 'tool_approval_request'
+  | 'tool_call_start'
+  | 'tool_call_result'
+  | 'text_delta'
+  | 'image'
+  | 'video'
+  | 'stream_start'
+  | 'done'
+  | 'error'
+  | 'abort'
+  | 'step_start'
+  | 'step_end';
+
+// ChatEvent 接口
+export interface ChatEvent {
+  type: ChatEventType;
+  data: unknown;
+}
+
+// 对话模式
+export type ChatMode = 'agent' | 'plan' | 'ask';
+
+// 供应商类型
+export type ProviderType = 'ark' | 'qwen' | 'gemini' | 'openrouter';
+
+// 深度思考模式
+export type ThinkingMode = 'enabled' | 'disabled';
+
+// 自动运行模式
+export type AutoRunMode = 'run-everything' | 'manual';
+
+// 自动运行配置
+export interface AutoRunConfig {
+  /** 
+   * 自动运行模式
+   * - 'run-everything': 运行所有内容（自动执行）
+   * - 'manual': 手动批准（每次执行前询问）
+   */
+  mode?: AutoRunMode;
+}
+
+/** 聊天历史消息（无状态架构） */
+export interface ChatHistoryMessage {
+  role: 'user' | 'assistant' | 'system' | 'tool';
+  content: string;
+}
+
+// 聊天选项
+export interface ChatOptions {
+  mode?: ChatMode;
+  model?: string;
+  provider?: ProviderType;
+  enableWebSearch?: boolean;
+  /** 深度思考开关（每个 provider 内部使用最优参数） */
+  thinkingMode?: ThinkingMode;
+  /** 自动运行配置 */
+  autoRunConfig?: AutoRunConfig;
+  /** 对话历史（无状态架构，历史从前端传入） */
+  history?: ChatHistoryMessage[];
+}
+
+/**
+ * 简化的模型配置
+ */
+export interface ModelOption {
+  /** 模型 ID（发送给 API） */
+  modelId: string;
+  /** 显示名称 */
+  displayName: string;
+  /** 分组名称（由后端决定，前端只负责渲染，必填） */
+  group: string;
+  /** 是否来自 OpenRouter（保留用于兼容，后续可能移除） */
+  isOpenRouter?: boolean;
+  /** 提供商名称（保留用于兼容，后续可能移除） */
+  provider?: string;
+}
+
+// 会话记录
+export interface SessionRecord {
+  id: string;
+  appId: string | null;
+  userId: string | null;
+  title: string;
+  model: string;
+  mode: ChatMode;
+  webSearchEnabled: boolean;
+  thinkingEnabled: boolean;
+  /** 是否在 tab 栏隐藏（关闭但不删除） */
+  hidden: boolean;
+  createdAt: Date;
+  updatedAt: Date;
+}
+
+// 消息记录
+export 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;
+}
+
+// 操作记录
+export interface OperationRecord {
+  id: string;
+  sessionId: string;
+  messageId?: string | null;
+  command: string;
+  operationType: string;
+  affectedFiles: string;
+  status: string;
+  timestamp: Date;
+}
+
+// 回收站记录
+export interface TrashRecord {
+  id: string;
+  sessionId: string;
+  originalPath: string;
+  trashPath: string;
+  deletedAt: Date;
+  autoDeleteAt: Date;
+}
+
+// 文件信息
+export interface FileInfo {
+  name: string;
+  path: string;
+  isDirectory: boolean;
+  size: number;
+  modifiedAt: Date;
+  extension: string;
+}
+
+// ChatAdapter 接口
+export interface ChatAdapter {
+  // ============ Chat API ============
+  /** 获取可用模型 */
+  getModels(): Promise<ModelOption[]>;
+  /** 发送消息，返回异步迭代器 */
+  sendMessage(message: string, options?: ChatOptions, images?: string[], sessionId?: string): AsyncIterable<ChatEvent>;
+  /** 取消当前请求 */
+  cancel(): void;
+  // 无状态架构：历史通过 ChatOptions.history 传入
+  /** 设置当前工作目录 */
+  setCwd?(dir: string): 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<void>;
+
+  // ============ 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<void>;
+  /** 删除指定时间之后的消息（用于分叉） */
+  deleteMessagesAfter(sessionId: string, timestamp: number): Promise<void>;
+
+  // ============ Operations API ============
+  /** 获取操作日志 */
+  getOperations(sessionId: string): Promise<OperationRecord[]>;
+
+  // ============ Trash API ============
+  /** 获取回收站 */
+  getTrashItems(): Promise<TrashRecord[]>;
+  /** 恢复回收站项目 */
+  restoreFromTrash(id: string): Promise<TrashRecord | undefined>;
+
+  // ============ 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<void>;
+  /** 获取所有用户设置 */
+  getAllSettings?(): Promise<Record<string, string>>;
+  /** 删除用户设置 */
+  deleteSetting?(key: string): Promise<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>;
+}
+
+// 声明 window 上的 bridge
+declare global {
+  interface Window {
+    aiChatBridge?: AiChatBridge;
+  }
+}
+
+export interface CreateAdapterOptions {
+  /** 全局变量名 */
+  bridgeName?: string;
+}
+
+/**
+ * 创建 Electron Adapter
+ */
+export function createElectronAdapter(options: CreateAdapterOptions = {}): ChatAdapter {
+  const { bridgeName = 'aiChatBridge' } = options;
+  
+  const getBridge = (): AiChatBridge => {
+    const bridge = (window as unknown as Record<string, unknown>)[bridgeName] as AiChatBridge | undefined;
+    if (!bridge) {
+      throw new Error(`AI Chat Bridge not found. Make sure to call exposeElectronBridge() in preload.`);
+    }
+    return bridge;
+  };
+
+  return {
+    // ============ Chat API ============
+    async getModels(): Promise<ModelOption[]> {
+      const bridge = getBridge();
+      return bridge.getModels();
+    },
+
+    async *sendMessage(
+      message: string, 
+      options?: ChatOptions,
+      images?: string[],
+      sessionId?: string
+    ): AsyncGenerator<ChatEvent> {
+      const bridge = getBridge();
+      
+      const queue: ChatEvent[] = [];
+      let resolveNext: (() => void) | null = null;
+      let done = false;
+
+      const cleanup = bridge.onProgress((progress) => {
+        const eventWithSession = progress as ChatEvent & { sessionId?: string };
+        
+        // 过滤：只处理属于当前会话的事件（或没有 sessionId 的旧事件）
+        if (sessionId && eventWithSession.sessionId && eventWithSession.sessionId !== sessionId) {
+          return; // 忽略不属于当前会话的事件
+        }
+        
+        queue.push(progress as ChatEvent);
+        resolveNext?.();
+        
+        if ((progress as ChatEvent).type === 'done' || (progress as ChatEvent).type === 'error') {
+          done = true;
+        }
+      });
+
+      bridge.send({ message, images, options, sessionId });
+
+      try {
+        while (!done || queue.length > 0) {
+          while (queue.length === 0 && !done) {
+            await new Promise<void>(r => resolveNext = r);
+          }
+
+          if (queue.length > 0) {
+            const progress = queue.shift()!;
+            yield progress;
+            
+            if (progress.type === 'done' || progress.type === 'error') {
+              break;
+            }
+          }
+        }
+      } finally {
+        cleanup();
+      }
+    },
+
+    cancel() {
+      getBridge().cancel();
+    },
+
+    // 无状态架构：历史通过 ChatOptions.history 传入
+
+    setCwd(dir: string) {
+      getBridge().setCwd(dir);
+    },
+
+    // ============ Sessions API ============
+    async getSessions(): Promise<SessionRecord[]> {
+      return getBridge().getSessions();
+    },
+
+    async getSession(id: string): Promise<SessionRecord | null> {
+      return getBridge().getSession(id);
+    },
+
+    async createSession(params): Promise<SessionRecord> {
+      return getBridge().createSession(params);
+    },
+
+    async updateSession(id: string, data): Promise<SessionRecord | null> {
+      return getBridge().updateSession(id, data);
+    },
+
+    async deleteSession(id: string): Promise<void> {
+      await getBridge().deleteSession(id);
+    },
+
+    // ============ Messages API ============
+    async getMessages(sessionId: string): Promise<MessageRecord[]> {
+      return getBridge().getMessages(sessionId);
+    },
+
+    async saveMessage(params): Promise<MessageRecord> {
+      return getBridge().saveMessage(params);
+    },
+
+    async updateMessage(params): Promise<void> {
+      await getBridge().updateMessage(params);
+    },
+
+    async deleteMessagesAfter(sessionId: string, timestamp: number): Promise<void> {
+      await getBridge().deleteMessagesAfter(sessionId, timestamp);
+    },
+
+    // ============ Operations API ============
+    async getOperations(sessionId: string): Promise<OperationRecord[]> {
+      return getBridge().getOperations(sessionId);
+    },
+
+    // ============ Trash API ============
+    async getTrashItems(): Promise<TrashRecord[]> {
+      return getBridge().getTrashItems();
+    },
+
+    async restoreFromTrash(id: string): Promise<TrashRecord | undefined> {
+      return getBridge().restoreFromTrash(id);
+    },
+
+    // ============ File System API ============
+    async listDir(dirPath: string): Promise<FileInfo[]> {
+      return getBridge().listDir(dirPath);
+    },
+
+    async exists(filePath: string): Promise<boolean> {
+      return getBridge().exists(filePath);
+    },
+
+    async stat(filePath: string): Promise<FileInfo | null> {
+      return getBridge().stat(filePath);
+    },
+
+    async readFile(filePath: string): Promise<string | null> {
+      return getBridge().readFile(filePath);
+    },
+
+    async readFileBase64(filePath: string): Promise<string | null> {
+      return getBridge().readFileBase64(filePath);
+    },
+
+    async homeDir(): Promise<string> {
+      return getBridge().homeDir();
+    },
+
+    async resolvePath(inputPath: string): Promise<string> {
+      return getBridge().resolvePath(inputPath);
+    },
+
+    async parentDir(dirPath: string): Promise<string> {
+      return getBridge().parentDir(dirPath);
+    },
+
+    async watchDir(dirPath: string): Promise<boolean> {
+      return getBridge().watchDir(dirPath);
+    },
+
+    async unwatchDir(dirPath: string): Promise<void> {
+      return getBridge().unwatchDir(dirPath);
+    },
+
+    onDirChange(callback: (data: { dirPath: string; eventType: string; filename: string | null }) => void): () => void {
+      return getBridge().onDirChange(callback);
+    },
+
+    // ============ Settings API ============
+    async getSetting(key: string): Promise<string | null> {
+      const bridge = getBridge();
+      if (bridge.getSetting) {
+        return bridge.getSetting(key);
+      }
+      return null;
+    },
+
+    async setSetting(key: string, value: string): Promise<void> {
+      const bridge = getBridge();
+      if (bridge.setSetting) {
+        await bridge.setSetting(key, value);
+      }
+    },
+
+    async getAllSettings(): Promise<Record<string, string>> {
+      const bridge = getBridge();
+      if (bridge.getAllSettings) {
+        return bridge.getAllSettings();
+      }
+      return {};
+    },
+
+    async deleteSetting(key: string): Promise<void> {
+      const bridge = getBridge();
+      if (bridge.deleteSetting) {
+        await bridge.deleteSetting(key);
+      }
+    },
+
+    // ============ Tool Approval API ============
+    onToolApprovalRequest(callback: (request: { id: string; name: string; args: Record<string, unknown> }) => void): () => void {
+      const bridge = getBridge();
+      if (bridge.onToolApprovalRequest) {
+        return bridge.onToolApprovalRequest(callback);
+      }
+      // 如果没有实现，返回空清理函数
+      return () => {};
+    },
+
+    async respondToolApproval(id: string, approved: boolean): Promise<void> {
+      const bridge = getBridge();
+      if (bridge.respondToolApproval) {
+        return bridge.respondToolApproval(id, approved);
+      }
+    },
+  };
+}