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

package/dist/main/index.cjs

@@ -1,7 +1,9 @@
 "use strict";
+var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __export = (target, all) => {
   for (var name in all)
@@ -15,71 +17,163 @@ var __copyProps = (to, from, except, desc) => {
   }
   return to;
 };
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/main/index.ts
 var main_exports = {};
 __export(main_exports, {
-  AVAILABLE_MODELS: () => import_ai_chat_core.AVAILABLE_MODELS,
+  MODELS: () => import_ai_chat_core.MODELS,
   createElectronBridge: () => createElectronBridge
 });
 module.exports = __toCommonJS(main_exports);
 var import_electron = require("electron");
+var fs = __toESM(require("fs"), 1);
+var path = __toESM(require("path"), 1);
 var import_ai_chat_core = require("@huyooo/ai-chat-core");
+var import_electron2 = require("@huyooo/ai-search/bridge/electron");
 var import_ai_chat_storage = require("@huyooo/ai-chat-storage");
 async function createElectronBridge(options) {
   const {
     channelPrefix = "ai-chat",
+    dataDir,
     storagePath,
     defaultContext = {},
     ...agentConfig
   } = options;
-  const agent = new import_ai_chat_core.HybridAgent(agentConfig);
+  const resolvedStoragePath = storagePath || (dataDir ? `${dataDir}/db.sqlite` : void 0);
+  (0, import_electron2.createSearchElectronBridge)({
+    ipcMain: import_electron.ipcMain,
+    channelPrefix
+  }).init();
+  const pendingApprovals = /* @__PURE__ */ new Map();
+  const toolApprovalCallback = async (toolCall) => {
+    console.log("[Main] \u5DE5\u5177\u6279\u51C6\u56DE\u8C03\u88AB\u8C03\u7528:", toolCall.name);
+    const currentWebContents = global.currentWebContents;
+    if (!currentWebContents) {
+      console.log("[Main] \u8B66\u544A: \u6CA1\u6709 webContents\uFF0C\u9ED8\u8BA4\u6279\u51C6");
+      return true;
+    }
+    return new Promise((resolve2, reject) => {
+      pendingApprovals.set(toolCall.id, {
+        resolve: resolve2,
+        reject,
+        webContents: currentWebContents
+      });
+      console.log("[Main] \u53D1\u9001\u5DE5\u5177\u6279\u51C6\u8BF7\u6C42\u5230\u524D\u7AEF:", toolCall.name);
+      currentWebContents.send(`${channelPrefix}:toolApprovalRequest`, {
+        id: toolCall.id,
+        name: toolCall.name,
+        args: toolCall.args
+      });
+    });
+  };
   const storage = await (0, import_ai_chat_storage.createStorage)({
     type: "sqlite",
-    sqlitePath: storagePath || (0, import_ai_chat_storage.getDefaultStoragePath)()
+    sqlitePath: resolvedStoragePath || (0, import_ai_chat_storage.getDefaultStoragePath)()
   });
   const getContext = () => defaultContext;
+  const getAutoRunConfig = async () => {
+    try {
+      const configJson = await storage.getUserSetting("autoRunConfig", getContext());
+      if (configJson) {
+        return JSON.parse(configJson);
+      }
+    } catch (error) {
+      console.error("[Main] \u83B7\u53D6 autoRunConfig \u5931\u8D25:", error);
+    }
+    return void 0;
+  };
+  const agent = new import_ai_chat_core.HybridAgent({
+    ...agentConfig,
+    // tools 直接传递 ToolConfigItem[]，HybridAgent 会在 asyncInit 中解析
+    tools: options.tools,
+    onToolApprovalRequest: toolApprovalCallback,
+    getAutoRunConfig
+  });
   import_electron.ipcMain.handle(`${channelPrefix}:models`, () => {
-    return import_ai_chat_core.AVAILABLE_MODELS;
+    return import_ai_chat_core.MODELS;
   });
   import_electron.ipcMain.handle(`${channelPrefix}:send`, async (event, params) => {
     const webContents = event.sender;
     const { message, images, options: options2 = {}, sessionId } = params;
+    global.currentWebContents = webContents;
     console.log("[AI-Chat] \u6536\u5230\u6D88\u606F:", { message, options: options2, images: images?.length || 0, sessionId });
+    console.log("[AI-Chat] autoRunConfig:", JSON.stringify(options2.autoRunConfig, null, 2));
     try {
-      for await (const progress of agent.chat(message, options2, images)) {
+      for await (const chatEvent of agent.chat(message, options2, images)) {
         console.log(
-          "[AI-Chat] \u8FDB\u5EA6:",
-          progress.type,
-          progress.type === "text_delta" ? progress.data.slice(0, 20) : progress.type === "thinking" ? JSON.stringify(progress.data) : progress.type === "search_result" ? `\u641C\u7D22\u5B8C\u6210 ${progress.data.results?.length || 0} \u6761` : JSON.stringify(progress.data).slice(0, 100)
+          "[AI-Chat] \u4E8B\u4EF6:",
+          chatEvent.type,
+          chatEvent.type === "text_delta" ? chatEvent.data.content.slice(0, 20) : chatEvent.type === "thinking_delta" ? chatEvent.data.content.slice(0, 20) : chatEvent.type === "search_result" ? `\u641C\u7D22\u5B8C\u6210 ${chatEvent.data.results?.length || 0} \u6761` : JSON.stringify(chatEvent.data).slice(0, 100)
         );
         if (!webContents.isDestroyed()) {
-          webContents.send(`${channelPrefix}:progress`, progress);
+          webContents.send(`${channelPrefix}:progress`, { ...chatEvent, sessionId });
         }
       }
       console.log("[AI-Chat] \u5B8C\u6210");
     } catch (error) {
       console.error("[AI-Chat] \u9519\u8BEF:", error);
+      if (error instanceof Error) {
+        console.error("[AI-Chat] \u9519\u8BEF\u8BE6\u60C5:", {
+          message: error.message,
+          stack: error.stack,
+          name: error.name
+        });
+      }
       if (!webContents.isDestroyed()) {
+        const errorData = error instanceof Error ? {
+          category: "api",
+          message: error.message || String(error),
+          cause: error.stack
+        } : {
+          category: "api",
+          message: String(error)
+        };
         webContents.send(`${channelPrefix}:progress`, {
           type: "error",
-          data: String(error)
+          data: errorData,
+          sessionId
         });
       }
+    } finally {
+      delete global.currentWebContents;
+    }
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:toolApprovalResponse`, (_event, params) => {
+    const { id, approved } = params;
+    const pending = pendingApprovals.get(id);
+    if (pending) {
+      pendingApprovals.delete(id);
+      pending.resolve(approved);
     }
   });
   import_electron.ipcMain.handle(`${channelPrefix}:cancel`, () => {
     agent.abort();
   });
-  import_electron.ipcMain.handle(`${channelPrefix}:clearAgentHistory`, () => {
-    agent.clearHistory();
+  import_electron.ipcMain.handle(`${channelPrefix}:settings:get`, async (_event, key) => {
+    return storage.getUserSetting(key, getContext());
   });
-  import_electron.ipcMain.handle(`${channelPrefix}:agentHistory`, () => {
-    return agent.getHistory();
+  import_electron.ipcMain.handle(`${channelPrefix}:settings:set`, async (_event, key, value) => {
+    await storage.setUserSetting(key, value, getContext());
+    return { success: true };
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:settings:getAll`, async () => {
+    return storage.getUserSettings(getContext());
   });
-  import_electron.ipcMain.handle(`${channelPrefix}:setWorkingDir`, (_event, dir) => {
-    agent.setWorkingDir(dir);
+  import_electron.ipcMain.handle(`${channelPrefix}:settings:delete`, async (_event, key) => {
+    await storage.deleteUserSetting(key, getContext());
+    return { success: true };
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:setCwd`, (_event, dir) => {
+    agent.setCwd(dir);
   });
   import_electron.ipcMain.handle(`${channelPrefix}:config`, () => {
     return agent.getConfig();
@@ -94,8 +188,11 @@ async function createElectronBridge(options) {
     const input = {
       id: params.id || crypto.randomUUID(),
       title: params.title || "\u65B0\u5BF9\u8BDD",
-      model: params.model || "doubao-seed-1-6-251015",
-      mode: params.mode || "agent"
+      model: params.model || import_ai_chat_core.DEFAULT_MODEL,
+      mode: params.mode || "agent",
+      webSearchEnabled: params.webSearchEnabled ?? true,
+      thinkingEnabled: params.thinkingEnabled ?? true,
+      hidden: params.hidden ?? false
     };
     return storage.createSession(input, getContext());
   });
@@ -112,17 +209,26 @@ async function createElectronBridge(options) {
   });
   import_electron.ipcMain.handle(`${channelPrefix}:messages:save`, async (_event, params) => {
     const input = {
-      id: crypto.randomUUID(),
+      id: params.id || crypto.randomUUID(),
       sessionId: params.sessionId,
       role: params.role,
       content: params.content,
-      thinking: params.thinking || null,
-      toolCalls: params.toolCalls || null,
-      searchResults: params.searchResults || null,
+      model: params.model || null,
+      mode: params.mode || null,
+      webSearchEnabled: params.webSearchEnabled ?? null,
+      thinkingEnabled: params.thinkingEnabled ?? null,
+      steps: params.steps || null,
       operationIds: params.operationIds || null
     };
     return storage.saveMessage(input, getContext());
   });
+  import_electron.ipcMain.handle(`${channelPrefix}:messages:update`, async (_event, params) => {
+    await storage.updateMessage(params.id, {
+      content: params.content,
+      steps: params.steps
+    }, getContext());
+    return { success: true };
+  });
   import_electron.ipcMain.handle(`${channelPrefix}:messages:deleteAfter`, async (_event, sessionId, timestamp) => {
     await storage.deleteMessagesAfter(sessionId, new Date(timestamp), getContext());
     return { success: true };
@@ -136,10 +242,124 @@ async function createElectronBridge(options) {
   import_electron.ipcMain.handle(`${channelPrefix}:trash:restore`, async (_event, id) => {
     return storage.restoreFromTrash?.(id, getContext());
   });
+  import_electron.ipcMain.handle(`${channelPrefix}:openExternal`, async (_event, url) => {
+    return import_electron.shell.openExternal(url);
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:fs:listDir`, async (_event, dirPath) => {
+    try {
+      const entries = fs.readdirSync(dirPath, { withFileTypes: true });
+      const files = [];
+      for (const entry of entries) {
+        if (entry.name.startsWith(".")) continue;
+        const fullPath = path.join(dirPath, entry.name);
+        try {
+          const stats = fs.statSync(fullPath);
+          files.push({
+            name: entry.name,
+            path: fullPath,
+            isDirectory: entry.isDirectory(),
+            size: stats.size,
+            modifiedAt: stats.mtime,
+            extension: entry.isDirectory() ? "" : path.extname(entry.name).toLowerCase()
+          });
+        } catch {
+        }
+      }
+      return files.sort((a, b) => {
+        if (a.isDirectory && !b.isDirectory) return -1;
+        if (!a.isDirectory && b.isDirectory) return 1;
+        return a.name.localeCompare(b.name);
+      });
+    } catch (error) {
+      console.error("[AI-Chat] \u5217\u51FA\u76EE\u5F55\u5931\u8D25:", error);
+      return [];
+    }
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:fs:exists`, async (_event, filePath) => {
+    return fs.existsSync(filePath);
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:fs:stat`, async (_event, filePath) => {
+    try {
+      const stats = fs.statSync(filePath);
+      return {
+        name: path.basename(filePath),
+        path: filePath,
+        isDirectory: stats.isDirectory(),
+        size: stats.size,
+        modifiedAt: stats.mtime,
+        extension: stats.isDirectory() ? "" : path.extname(filePath).toLowerCase()
+      };
+    } catch {
+      return null;
+    }
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:fs:readFile`, async (_event, filePath) => {
+    try {
+      return fs.readFileSync(filePath, "utf-8");
+    } catch {
+      return null;
+    }
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:fs:readFileBase64`, async (_event, filePath) => {
+    try {
+      const buffer = fs.readFileSync(filePath);
+      return buffer.toString("base64");
+    } catch {
+      return null;
+    }
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:fs:homeDir`, async () => {
+    return process.env.HOME || process.env.USERPROFILE || "/";
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:fs:resolvePath`, async (_event, inputPath) => {
+    if (inputPath.startsWith("~")) {
+      const homeDir = process.env.HOME || process.env.USERPROFILE || "/";
+      return path.join(homeDir, inputPath.slice(1));
+    }
+    return path.resolve(inputPath);
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:fs:parentDir`, async (_event, dirPath) => {
+    return path.dirname(dirPath);
+  });
+  const activeWatchers = /* @__PURE__ */ new Map();
+  import_electron.ipcMain.handle(`${channelPrefix}:fs:watchDir`, async (event, dirPath) => {
+    const webContents = event.sender;
+    if (activeWatchers.has(dirPath)) {
+      activeWatchers.get(dirPath)?.close();
+      activeWatchers.delete(dirPath);
+    }
+    try {
+      const watcher = fs.watch(dirPath, { persistent: false }, (eventType, filename) => {
+        if (!webContents.isDestroyed()) {
+          webContents.send(`${channelPrefix}:fs:dirChange`, {
+            dirPath,
+            eventType,
+            filename
+          });
+        }
+      });
+      watcher.on("error", (error) => {
+        console.error("[AI-Chat] Watch error:", error);
+        activeWatchers.delete(dirPath);
+      });
+      activeWatchers.set(dirPath, watcher);
+      return true;
+    } catch (error) {
+      console.error("[AI-Chat] Failed to watch directory:", error);
+      return false;
+    }
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:fs:unwatchDir`, async (_event, dirPath) => {
+    const watcher = activeWatchers.get(dirPath);
+    if (watcher) {
+      watcher.close();
+      activeWatchers.delete(dirPath);
+    }
+  });
   return { agent, storage };
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  AVAILABLE_MODELS,
+  MODELS,
   createElectronBridge
 });

package/dist/main/index.d.cts

@@ -1,5 +1,5 @@
-import { AgentConfig, HybridAgent } from '@huyooo/ai-chat-core';
-export { AVAILABLE_MODELS, AgentConfig, ChatMode, ChatOptions, ChatProgress, ModelProvider } from '@huyooo/ai-chat-core';
+import { AgentConfig, ToolConfigItem, HybridAgent } from '@huyooo/ai-chat-core';
+export { AgentConfig, ChatEvent, ChatMode, ChatOptions, MODELS, ProviderType } from '@huyooo/ai-chat-core';
 import { StorageContext, StorageAdapter } from '@huyooo/ai-chat-storage';
 export { MessageRecord, SessionRecord, StorageAdapter, StorageContext } from '@huyooo/ai-chat-storage';
 
@@ -10,13 +10,39 @@ export { MessageRecord, SessionRecord, StorageAdapter, StorageContext } from '@h
  * 集成 SQLite 存储
  */
 
-interface ElectronBridgeOptions extends AgentConfig {
+/** 文件信息 */
+interface FileInfo {
+    name: string;
+    path: string;
+    isDirectory: boolean;
+    size: number;
+    modifiedAt: Date;
+    extension: string;
+}
+
+interface ElectronBridgeOptions extends Omit<AgentConfig, 'tools'> {
     /** IPC channel 前缀 */
     channelPrefix?: string;
-    /** SQLite 数据库路径 */
+    /**
+     * 统一数据目录（推荐）
+     * 所有数据都存储在此目录下：
+     * - db.sqlite: 对话历史
+     * - search-data/: 搜索索引
+     * 默认: ~/.ai-chat/
+     */
+    dataDir?: string;
+    /**
+     * SQLite 数据库路径（可选，覆盖 dataDir 设置）
+     * @deprecated 推荐使用 dataDir
+     */
     storagePath?: string;
     /** 默认租户上下文 */
     defaultContext?: StorageContext;
+    /**
+     * 工具列表（Vite 插件风格）
+     * 支持：ToolPlugin、Promise<ToolPlugin>
+     */
+    tools?: ToolConfigItem[];
 }
 /**
  * 创建 Electron IPC 桥接
@@ -38,4 +64,4 @@ declare function createElectronBridge(options: ElectronBridgeOptions): Promise<{
     storage: StorageAdapter;
 }>;
 
-export { type ElectronBridgeOptions, createElectronBridge };
+export { type ElectronBridgeOptions, type FileInfo, createElectronBridge };

package/dist/main/index.d.ts

@@ -1,5 +1,5 @@
-import { AgentConfig, HybridAgent } from '@huyooo/ai-chat-core';
-export { AVAILABLE_MODELS, AgentConfig, ChatMode, ChatOptions, ChatProgress, ModelProvider } from '@huyooo/ai-chat-core';
+import { AgentConfig, ToolConfigItem, HybridAgent } from '@huyooo/ai-chat-core';
+export { AgentConfig, ChatEvent, ChatMode, ChatOptions, MODELS, ProviderType } from '@huyooo/ai-chat-core';
 import { StorageContext, StorageAdapter } from '@huyooo/ai-chat-storage';
 export { MessageRecord, SessionRecord, StorageAdapter, StorageContext } from '@huyooo/ai-chat-storage';
 
@@ -10,13 +10,39 @@ export { MessageRecord, SessionRecord, StorageAdapter, StorageContext } from '@h
  * 集成 SQLite 存储
  */
 
-interface ElectronBridgeOptions extends AgentConfig {
+/** 文件信息 */
+interface FileInfo {
+    name: string;
+    path: string;
+    isDirectory: boolean;
+    size: number;
+    modifiedAt: Date;
+    extension: string;
+}
+
+interface ElectronBridgeOptions extends Omit<AgentConfig, 'tools'> {
     /** IPC channel 前缀 */
     channelPrefix?: string;
-    /** SQLite 数据库路径 */
+    /**
+     * 统一数据目录（推荐）
+     * 所有数据都存储在此目录下：
+     * - db.sqlite: 对话历史
+     * - search-data/: 搜索索引
+     * 默认: ~/.ai-chat/
+     */
+    dataDir?: string;
+    /**
+     * SQLite 数据库路径（可选，覆盖 dataDir 设置）
+     * @deprecated 推荐使用 dataDir
+     */
     storagePath?: string;
     /** 默认租户上下文 */
     defaultContext?: StorageContext;
+    /**
+     * 工具列表（Vite 插件风格）
+     * 支持：ToolPlugin、Promise<ToolPlugin>
+     */
+    tools?: ToolConfigItem[];
 }
 /**
  * 创建 Electron IPC 桥接
@@ -38,4 +64,4 @@ declare function createElectronBridge(options: ElectronBridgeOptions): Promise<{
     storage: StorageAdapter;
 }>;
 
-export { type ElectronBridgeOptions, createElectronBridge };
+export { type ElectronBridgeOptions, type FileInfo, createElectronBridge };