flat: 暂存

src/preload/index.ts

@@ -1,22 +1,30 @@
-import { contextBridge } from 'electron'
-import { electronAPI } from '@electron-toolkit/preload'
+import { contextBridge, ipcRenderer } from "electron";
+import { electronAPI } from "@electron-toolkit/preload";
 
-// Custom APIs for renderer
-const api = {}
+const api = {
+    installOllama: () => ipcRenderer.invoke("install-ollama-and-model"),
+
+    // 监听进度 (Send/On)
+    onInstallProgress: (callback) => {
+        ipcRenderer.on("install-progress", (_event, value) => callback(value));
+    },
+
+    // 移除监听器
+    removeInstallProgressListeners: () => {
+        ipcRenderer.removeAllListeners("install-progress");
+    },
+};
 
-// Use `contextBridge` APIs to expose Electron APIs to
-// renderer only if context isolation is enabled, otherwise
-// just add to the DOM global.
 if (process.contextIsolated) {
-  try {
-    contextBridge.exposeInMainWorld('electron', electronAPI)
-    contextBridge.exposeInMainWorld('api', api)
-  } catch (error) {
-    console.error(error)
-  }
+    try {
+        contextBridge.exposeInMainWorld("electron", electronAPI);
+        contextBridge.exposeInMainWorld("api", api);
+    } catch (error) {
+        console.error(error);
+    }
 } else {
-  // @ts-ignore (define in dts)
-  window.electron = electronAPI
-  // @ts-ignore (define in dts)
-  window.api = api
+    // @ts-ignore (define in dts)
+    window.electron = electronAPI;
+    // @ts-ignore (define in dts)
+    window.api = api;
 }