@huyooo/ai-chat-bridge-electron 0.1.0

package/dist/main/index.cjs

@@ -0,0 +1,145 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+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,
+  createElectronBridge: () => createElectronBridge
+});
+module.exports = __toCommonJS(main_exports);
+var import_electron = require("electron");
+var import_ai_chat_core = require("@huyooo/ai-chat-core");
+var import_ai_chat_storage = require("@huyooo/ai-chat-storage");
+async function createElectronBridge(options) {
+  const {
+    channelPrefix = "ai-chat",
+    storagePath,
+    defaultContext = {},
+    ...agentConfig
+  } = options;
+  const agent = new import_ai_chat_core.HybridAgent(agentConfig);
+  const storage = await (0, import_ai_chat_storage.createStorage)({
+    type: "sqlite",
+    sqlitePath: storagePath || (0, import_ai_chat_storage.getDefaultStoragePath)()
+  });
+  const getContext = () => defaultContext;
+  import_electron.ipcMain.handle(`${channelPrefix}:models`, () => {
+    return import_ai_chat_core.AVAILABLE_MODELS;
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:send`, async (event, params) => {
+    const webContents = event.sender;
+    const { message, images, options: options2 = {}, sessionId } = params;
+    console.log("[AI-Chat] \u6536\u5230\u6D88\u606F:", { message, options: options2, images: images?.length || 0, sessionId });
+    try {
+      for await (const progress 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)
+        );
+        if (!webContents.isDestroyed()) {
+          webContents.send(`${channelPrefix}:progress`, progress);
+        }
+      }
+      console.log("[AI-Chat] \u5B8C\u6210");
+    } catch (error) {
+      console.error("[AI-Chat] \u9519\u8BEF:", error);
+      if (!webContents.isDestroyed()) {
+        webContents.send(`${channelPrefix}:progress`, {
+          type: "error",
+          data: String(error)
+        });
+      }
+    }
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:cancel`, () => {
+    agent.abort();
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:clearAgentHistory`, () => {
+    agent.clearHistory();
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:agentHistory`, () => {
+    return agent.getHistory();
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:setWorkingDir`, (_event, dir) => {
+    agent.setWorkingDir(dir);
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:config`, () => {
+    return agent.getConfig();
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:sessions:list`, async () => {
+    return storage.getSessions(getContext());
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:sessions:get`, async (_event, id) => {
+    return storage.getSession(id, getContext());
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:sessions:create`, async (_event, params) => {
+    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"
+    };
+    return storage.createSession(input, getContext());
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:sessions:update`, async (_event, id, data) => {
+    await storage.updateSession(id, data, getContext());
+    return storage.getSession(id, getContext());
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:sessions:delete`, async (_event, id) => {
+    await storage.deleteSession(id, getContext());
+    return { success: true };
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:messages:list`, async (_event, sessionId) => {
+    return storage.getMessages(sessionId, getContext());
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:messages:save`, async (_event, params) => {
+    const input = {
+      id: crypto.randomUUID(),
+      sessionId: params.sessionId,
+      role: params.role,
+      content: params.content,
+      thinking: params.thinking || null,
+      toolCalls: params.toolCalls || null,
+      searchResults: params.searchResults || null,
+      operationIds: params.operationIds || null
+    };
+    return storage.saveMessage(input, getContext());
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:messages:deleteAfter`, async (_event, sessionId, timestamp) => {
+    await storage.deleteMessagesAfter(sessionId, new Date(timestamp), getContext());
+    return { success: true };
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:operations:list`, async (_event, sessionId) => {
+    return storage.getOperations(sessionId, getContext());
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:trash:list`, async () => {
+    return storage.getTrashItems?.(getContext()) || [];
+  });
+  import_electron.ipcMain.handle(`${channelPrefix}:trash:restore`, async (_event, id) => {
+    return storage.restoreFromTrash?.(id, getContext());
+  });
+  return { agent, storage };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  AVAILABLE_MODELS,
+  createElectronBridge
+});

package/dist/main/index.d.cts

@@ -0,0 +1,41 @@
+import { AgentConfig, HybridAgent } from '@huyooo/ai-chat-core';
+export { AVAILABLE_MODELS, AgentConfig, ChatMode, ChatOptions, ChatProgress, ModelProvider } from '@huyooo/ai-chat-core';
+import { StorageContext, StorageAdapter } from '@huyooo/ai-chat-storage';
+export { MessageRecord, SessionRecord, StorageAdapter, StorageContext } from '@huyooo/ai-chat-storage';
+
+/**
+ * Electron 主进程桥接
+ *
+ * 在主进程中调用，注册 IPC handlers
+ * 集成 SQLite 存储
+ */
+
+interface ElectronBridgeOptions extends AgentConfig {
+    /** IPC channel 前缀 */
+    channelPrefix?: string;
+    /** SQLite 数据库路径 */
+    storagePath?: string;
+    /** 默认租户上下文 */
+    defaultContext?: StorageContext;
+}
+/**
+ * 创建 Electron IPC 桥接
+ *
+ * @example
+ * ```ts
+ * import { createElectronBridge } from '@huyooo/ai-chat-bridge-electron/main';
+ *
+ * app.whenReady().then(async () => {
+ *   const { agent, storage } = await createElectronBridge({
+ *     arkApiKey: 'xxx',
+ *     geminiApiKey: 'xxx',
+ *   });
+ * });
+ * ```
+ */
+declare function createElectronBridge(options: ElectronBridgeOptions): Promise<{
+    agent: HybridAgent;
+    storage: StorageAdapter;
+}>;
+
+export { type ElectronBridgeOptions, createElectronBridge };

package/dist/main/index.d.ts

@@ -0,0 +1,41 @@
+import { AgentConfig, HybridAgent } from '@huyooo/ai-chat-core';
+export { AVAILABLE_MODELS, AgentConfig, ChatMode, ChatOptions, ChatProgress, ModelProvider } from '@huyooo/ai-chat-core';
+import { StorageContext, StorageAdapter } from '@huyooo/ai-chat-storage';
+export { MessageRecord, SessionRecord, StorageAdapter, StorageContext } from '@huyooo/ai-chat-storage';
+
+/**
+ * Electron 主进程桥接
+ *
+ * 在主进程中调用，注册 IPC handlers
+ * 集成 SQLite 存储
+ */
+
+interface ElectronBridgeOptions extends AgentConfig {
+    /** IPC channel 前缀 */
+    channelPrefix?: string;
+    /** SQLite 数据库路径 */
+    storagePath?: string;
+    /** 默认租户上下文 */
+    defaultContext?: StorageContext;
+}
+/**
+ * 创建 Electron IPC 桥接
+ *
+ * @example
+ * ```ts
+ * import { createElectronBridge } from '@huyooo/ai-chat-bridge-electron/main';
+ *
+ * app.whenReady().then(async () => {
+ *   const { agent, storage } = await createElectronBridge({
+ *     arkApiKey: 'xxx',
+ *     geminiApiKey: 'xxx',
+ *   });
+ * });
+ * ```
+ */
+declare function createElectronBridge(options: ElectronBridgeOptions): Promise<{
+    agent: HybridAgent;
+    storage: StorageAdapter;
+}>;
+
+export { type ElectronBridgeOptions, createElectronBridge };

package/dist/main/index.js

@@ -0,0 +1,125 @@
+// src/main/index.ts
+import { ipcMain } from "electron";
+import {
+  HybridAgent,
+  AVAILABLE_MODELS
+} from "@huyooo/ai-chat-core";
+import {
+  createStorage,
+  getDefaultStoragePath
+} from "@huyooo/ai-chat-storage";
+async function createElectronBridge(options) {
+  const {
+    channelPrefix = "ai-chat",
+    storagePath,
+    defaultContext = {},
+    ...agentConfig
+  } = options;
+  const agent = new HybridAgent(agentConfig);
+  const storage = await createStorage({
+    type: "sqlite",
+    sqlitePath: storagePath || getDefaultStoragePath()
+  });
+  const getContext = () => defaultContext;
+  ipcMain.handle(`${channelPrefix}:models`, () => {
+    return AVAILABLE_MODELS;
+  });
+  ipcMain.handle(`${channelPrefix}:send`, async (event, params) => {
+    const webContents = event.sender;
+    const { message, images, options: options2 = {}, sessionId } = params;
+    console.log("[AI-Chat] \u6536\u5230\u6D88\u606F:", { message, options: options2, images: images?.length || 0, sessionId });
+    try {
+      for await (const progress 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)
+        );
+        if (!webContents.isDestroyed()) {
+          webContents.send(`${channelPrefix}:progress`, progress);
+        }
+      }
+      console.log("[AI-Chat] \u5B8C\u6210");
+    } catch (error) {
+      console.error("[AI-Chat] \u9519\u8BEF:", error);
+      if (!webContents.isDestroyed()) {
+        webContents.send(`${channelPrefix}:progress`, {
+          type: "error",
+          data: String(error)
+        });
+      }
+    }
+  });
+  ipcMain.handle(`${channelPrefix}:cancel`, () => {
+    agent.abort();
+  });
+  ipcMain.handle(`${channelPrefix}:clearAgentHistory`, () => {
+    agent.clearHistory();
+  });
+  ipcMain.handle(`${channelPrefix}:agentHistory`, () => {
+    return agent.getHistory();
+  });
+  ipcMain.handle(`${channelPrefix}:setWorkingDir`, (_event, dir) => {
+    agent.setWorkingDir(dir);
+  });
+  ipcMain.handle(`${channelPrefix}:config`, () => {
+    return agent.getConfig();
+  });
+  ipcMain.handle(`${channelPrefix}:sessions:list`, async () => {
+    return storage.getSessions(getContext());
+  });
+  ipcMain.handle(`${channelPrefix}:sessions:get`, async (_event, id) => {
+    return storage.getSession(id, getContext());
+  });
+  ipcMain.handle(`${channelPrefix}:sessions:create`, async (_event, params) => {
+    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"
+    };
+    return storage.createSession(input, getContext());
+  });
+  ipcMain.handle(`${channelPrefix}:sessions:update`, async (_event, id, data) => {
+    await storage.updateSession(id, data, getContext());
+    return storage.getSession(id, getContext());
+  });
+  ipcMain.handle(`${channelPrefix}:sessions:delete`, async (_event, id) => {
+    await storage.deleteSession(id, getContext());
+    return { success: true };
+  });
+  ipcMain.handle(`${channelPrefix}:messages:list`, async (_event, sessionId) => {
+    return storage.getMessages(sessionId, getContext());
+  });
+  ipcMain.handle(`${channelPrefix}:messages:save`, async (_event, params) => {
+    const input = {
+      id: crypto.randomUUID(),
+      sessionId: params.sessionId,
+      role: params.role,
+      content: params.content,
+      thinking: params.thinking || null,
+      toolCalls: params.toolCalls || null,
+      searchResults: params.searchResults || null,
+      operationIds: params.operationIds || null
+    };
+    return storage.saveMessage(input, getContext());
+  });
+  ipcMain.handle(`${channelPrefix}:messages:deleteAfter`, async (_event, sessionId, timestamp) => {
+    await storage.deleteMessagesAfter(sessionId, new Date(timestamp), getContext());
+    return { success: true };
+  });
+  ipcMain.handle(`${channelPrefix}:operations:list`, async (_event, sessionId) => {
+    return storage.getOperations(sessionId, getContext());
+  });
+  ipcMain.handle(`${channelPrefix}:trash:list`, async () => {
+    return storage.getTrashItems?.(getContext()) || [];
+  });
+  ipcMain.handle(`${channelPrefix}:trash:restore`, async (_event, id) => {
+    return storage.restoreFromTrash?.(id, getContext());
+  });
+  return { agent, storage };
+}
+export {
+  AVAILABLE_MODELS,
+  createElectronBridge
+};

package/dist/preload/index.cjs

@@ -0,0 +1,68 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/preload/index.ts
+var preload_exports = {};
+__export(preload_exports, {
+  exposeElectronBridge: () => exposeElectronBridge
+});
+module.exports = __toCommonJS(preload_exports);
+var import_electron = require("electron");
+function exposeElectronBridge(options = {}) {
+  const { channelPrefix = "ai-chat", exposeName = "aiChatBridge" } = options;
+  const bridge = {
+    // ============ Chat API ============
+    getModels: () => import_electron.ipcRenderer.invoke(`${channelPrefix}:models`),
+    send: (params) => import_electron.ipcRenderer.invoke(`${channelPrefix}:send`, params),
+    cancel: () => import_electron.ipcRenderer.invoke(`${channelPrefix}:cancel`),
+    clearAgentHistory: () => import_electron.ipcRenderer.invoke(`${channelPrefix}:clearAgentHistory`),
+    getAgentHistory: () => import_electron.ipcRenderer.invoke(`${channelPrefix}:agentHistory`),
+    setWorkingDir: (dir) => import_electron.ipcRenderer.invoke(`${channelPrefix}:setWorkingDir`, dir),
+    getConfig: () => import_electron.ipcRenderer.invoke(`${channelPrefix}:config`),
+    onProgress: (callback) => {
+      const handler = (_event, progress) => {
+        callback(progress);
+      };
+      import_electron.ipcRenderer.on(`${channelPrefix}:progress`, handler);
+      return () => {
+        import_electron.ipcRenderer.removeListener(`${channelPrefix}:progress`, handler);
+      };
+    },
+    // ============ Sessions API ============
+    getSessions: () => import_electron.ipcRenderer.invoke(`${channelPrefix}:sessions:list`),
+    getSession: (id) => import_electron.ipcRenderer.invoke(`${channelPrefix}:sessions:get`, id),
+    createSession: (params) => import_electron.ipcRenderer.invoke(`${channelPrefix}:sessions:create`, params || {}),
+    updateSession: (id, data) => import_electron.ipcRenderer.invoke(`${channelPrefix}:sessions:update`, id, data),
+    deleteSession: (id) => import_electron.ipcRenderer.invoke(`${channelPrefix}:sessions:delete`, id),
+    // ============ Messages API ============
+    getMessages: (sessionId) => import_electron.ipcRenderer.invoke(`${channelPrefix}:messages:list`, sessionId),
+    saveMessage: (params) => import_electron.ipcRenderer.invoke(`${channelPrefix}:messages:save`, params),
+    deleteMessagesAfter: (sessionId, timestamp) => import_electron.ipcRenderer.invoke(`${channelPrefix}:messages:deleteAfter`, sessionId, timestamp),
+    // ============ Operations API ============
+    getOperations: (sessionId) => import_electron.ipcRenderer.invoke(`${channelPrefix}:operations:list`, sessionId),
+    // ============ Trash API ============
+    getTrashItems: () => import_electron.ipcRenderer.invoke(`${channelPrefix}:trash:list`),
+    restoreFromTrash: (id) => import_electron.ipcRenderer.invoke(`${channelPrefix}:trash:restore`, id)
+  };
+  import_electron.contextBridge.exposeInMainWorld(exposeName, bridge);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  exposeElectronBridge
+});

package/dist/preload/index.d.cts

@@ -0,0 +1,143 @@
+/**
+ * Electron Preload 桥接
+ *
+ * 在 preload 脚本中调用，暴露 API 给渲染进程
+ */
+/** 发送消息参数 */
+interface SendMessageParams {
+    message: string;
+    images?: string[];
+    sessionId?: string;
+    options?: {
+        mode?: string;
+        model?: string;
+        provider?: string;
+        enableWebSearch?: boolean;
+        thinkingMode?: string;
+        thinkingBudget?: number;
+    };
+}
+/** 模型配置 */
+interface ModelConfig {
+    provider: string;
+    model: string;
+    displayName: string;
+    supportsTools: boolean;
+    supportsWebSearch: boolean;
+    supportedThinkingModes: string[];
+}
+/** 会话记录 */
+interface SessionRecord {
+    id: string;
+    appId: string | null;
+    userId: string | null;
+    title: string;
+    model: string;
+    mode: 'agent' | 'plan' | 'ask';
+    createdAt: Date;
+    updatedAt: Date;
+}
+/** 消息记录 */
+interface MessageRecord {
+    id: string;
+    sessionId: string;
+    role: 'user' | 'assistant';
+    content: string;
+    thinking?: string | null;
+    toolCalls?: string | null;
+    searchResults?: 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 AiChatBridge {
+    /** 获取可用模型 */
+    getModels(): Promise<ModelConfig[]>;
+    /** 发送消息 */
+    send(params: SendMessageParams): Promise<void>;
+    /** 取消当前请求 */
+    cancel(): Promise<void>;
+    /** 清空 Agent 内存历史 */
+    clearAgentHistory(): Promise<void>;
+    /** 获取 Agent 内存历史 */
+    getAgentHistory(): Promise<unknown[]>;
+    /** 设置工作目录 */
+    setWorkingDir(dir: string): Promise<void>;
+    /** 获取配置 */
+    getConfig(): Promise<unknown>;
+    /** 监听进度事件 */
+    onProgress(callback: (progress: unknown) => void): () => void;
+    /** 获取会话列表 */
+    getSessions(): Promise<SessionRecord[]>;
+    /** 获取单个会话 */
+    getSession(id: string): Promise<SessionRecord | null>;
+    /** 创建会话 */
+    createSession(params?: {
+        title?: string;
+        model?: string;
+        mode?: string;
+    }): Promise<SessionRecord>;
+    /** 更新会话 */
+    updateSession(id: string, data: {
+        title?: string;
+        model?: string;
+        mode?: string;
+    }): Promise<SessionRecord | null>;
+    /** 删除会话 */
+    deleteSession(id: string): Promise<{
+        success: boolean;
+    }>;
+    /** 获取会话消息 */
+    getMessages(sessionId: string): Promise<MessageRecord[]>;
+    /** 保存消息 */
+    saveMessage(params: {
+        sessionId: string;
+        role: 'user' | 'assistant';
+        content: string;
+        thinking?: string;
+        toolCalls?: string;
+        searchResults?: string;
+        operationIds?: string;
+    }): Promise<MessageRecord>;
+    /** 删除指定时间之后的消息（用于分叉） */
+    deleteMessagesAfter(sessionId: string, timestamp: number): Promise<{
+        success: boolean;
+    }>;
+    /** 获取操作日志 */
+    getOperations(sessionId: string): Promise<OperationRecord[]>;
+    /** 获取回收站 */
+    getTrashItems(): Promise<TrashRecord[]>;
+    /** 恢复回收站项目 */
+    restoreFromTrash(id: string): Promise<TrashRecord | undefined>;
+}
+interface ExposeOptions {
+    /** IPC channel 前缀 */
+    channelPrefix?: string;
+    /** 暴露的全局变量名 */
+    exposeName?: string;
+}
+/**
+ * 暴露 AI Chat API 给渲染进程
+ */
+declare function exposeElectronBridge(options?: ExposeOptions): void;
+
+export { type AiChatBridge, type ExposeOptions, exposeElectronBridge };