feat(chatbot-extension): a Chrome extension that can be using for chat with AI on any website (#2235)

* feat(chatbot-extension): a Chrome extension that can be using for chat with AI on any website * fix: 插件支持语音输入 feat：chatbot支持切换 * fix: 切换chatbot后，自动隐藏bot列表
2024-08-06 10:53:01 +08:00
parent e36d9d794f
commit f35ba8e5a7
11 changed files with 1465 additions and 0 deletions
--- a/projects/app/public/chrome_extension/fastgpt_agent/src/background.js
+++ b/projects/app/public/chrome_extension/fastgpt_agent/src/background.js
@@ -0,0 +1,51 @@
+let requestInterceptor = null;
+
+chrome.runtime.onMessage.addListener(function (message, sender, sendResponse) {
+    if (message.action === "startRequestInterception") {
+        const botSrc = message.chatbotSrc;
+        console.log("src", botSrc);
+        const urlObj = new URL(botSrc);
+        const domain = urlObj.host;
+        const searchParams = urlObj.searchParams;
+        const frameShareId = searchParams.get('shareId') || '';
+        let frameChatId = searchParams.get('chatId') || '';
+
+        // 移除已有的拦截器（如果存在）
+        if (requestInterceptor) {
+            chrome.webRequest.onBeforeRequest.removeListener(requestInterceptor);
+        }
+
+        requestInterceptor = function (details) {
+            if (details.frameId !== -1
+                && details.url.includes(domain)
+                && details.url.includes("chat/completions")) {
+                console.log("Intercepted request from chatbot-iframe:", details);
+                if (details.requestBody.raw) {
+                    let decoder = new TextDecoder("utf-8");
+                    let postData = decoder.decode(new Uint8Array(details.requestBody.raw[0].bytes));
+                    try {
+                        let postDataObj = JSON.parse(postData);
+                        let shareId = postDataObj.shareId;
+                        let chatId = postDataObj.chatId;
+
+                        if (frameChatId !== chatId && frameShareId === shareId) {
+                            chrome.storage.local.set({["shareId"]: shareId});
+                            chrome.storage.local.set({["chatId"]: chatId});
+                            frameChatId = chatId;
+                            console.log(`Stored shareId: ${shareId} and chatId: ${chatId} to localStorage.`);
+                        }
+                    } catch (error) {
+                        console.error("Error parsing postData:", error);
+                    }
+                }
+            }
+            return {};
+        };
+
+        chrome.webRequest.onBeforeRequest.addListener(
+            requestInterceptor,
+            { urls: ["<all_urls>"] },
+            ["requestBody"]
+        );
+    }
+});