以图搜图 添加新引擎

Author: xushengfeng

Diff for: src/main/main.ts

@@ -1362,6 +1362,12 @@ async function createMainWindow(op: MainWinType) {
         // 确保切换到index时能传递window_name
         mainWindow.webContents.send("text", windowName, op);
 
+        if (op.type === "image" && op.arg0 === "ai") {
+            createBrowser(windowName, "http://ai-v.netlify.app").then((c) => {
+                c.executeJavaScript(`setImg("${op.content}")`);
+            });
+        }
+
         if (mainWindow.html) {
             mainWindow.webContents.send("html", mainWindow.html);
         }
@@ -1528,6 +1534,11 @@ async function createBrowser(windowName: number, url: string) {
         rendererPath(searchView.webContents, "browser_bg.html", { query: { type: "certificate-error" } });
         if (dev) searchView.webContents.openDevTools();
     });
+    return new Promise((resolve: (x: Electron.WebContents) => void) => {
+        searchView.webContents.on("did-finish-load", () => {
+            resolve(searchView.webContents);
+        });
+    });
 }
 /**
  * 标签页事件

Diff for: src/renderer/capture.html

@@ -44,6 +44,7 @@
         <div id="tool_search" title="以图搜图">
             <img src="./assets/icons/search.svg" class="icon">
             <select name="识图引擎" id="识图引擎" class="side_select">
+                <option value="ai">AI</option>
                 <option value="baidu">百度</option>
                 <option value="yandex">Yandex</option>
                 <option value="google">Google</option>

Diff for: src/renderer/clip/clip_window.ts

@@ -686,7 +686,7 @@ document.getElementById("tool_search").title = `以图搜图 - ${识图引擎.va
 function runSearch() {
     const type = 识图引擎.value;
     getClipPhoto("png").then((c: HTMLCanvasElement) => {
-        ipcRenderer.send("clip_main_b", "search", [c.toDataURL().replace(/^data:image\/\w+;base64,/, ""), type]);
+        ipcRenderer.send("clip_main_b", "search", [c.toDataURL(), type]);
     });
     tool.close();
 }

Diff for: src/renderer/editor/editor.ts

@@ -1578,6 +1578,7 @@ window.onbeforeunload = () => {
 /************************************以图搜图 */
 
 function searchImg(img: string, type: "baidu" | "yandex" | "google", callback: Function) {
+    img = img.replace(/^data:image\/\w+;base64,/, "");
     switch (type) {
         case "baidu":
             baidu(img, (err, url) => {