fix: 注入js

public/background.js

@@ -49,6 +49,22 @@ chrome.runtime.onMessage.addListener(function (msg, sender, sendResponse) {
                 }
             });
             break;
+        case ActionType.INJECTSCRIPT:
+            chrome.scripting.executeScript({
+                target: { tabId: msg.tabId },
+                files: ['content.js']
+            }).then(() => {
+                chrome.tabs.sendMessage(msg.tabId, { type: 'INJECT_CODE', value: msg.value });
+            }).catch(err => {
+                console.error('Script injection failed:', err);
+            });
+            break
+        case ActionType.SENDRESFROMPAGE:
+            chrome.runtime.sendMessage({
+                action: ActionType.RESTOEVALINTAB,
+                result: msg.result
+            })
+            break
         case ActionType.ECHO:
             console.log("Echo ", msg.result)
     }
@@ -64,33 +80,3 @@ const pageFunction = (code) => {
         }
     });
 }
-
-// chrome.runtime.onInstalled.addListener(() => {
-//     chrome.scripting.registerContentScripts([{
-//         id: 'myScript',
-//         matches: ['<all_urls>'],  // 根据需要调整匹配模式
-//         js: ['content2.js'],
-//         // world: 'MAIN',
-//         // runAt: 'document_start'
-//     }], (result) => {
-//         if (chrome.runtime.lastError) {
-//             console.error(`Error registering script: ${chrome.runtime.lastError.message}`);
-//         } else {
-//             console.log('Script registered', result);
-//         }
-//     });
-// });
-
-chrome.runtime.onConnect.addListener(function (port) {
-    console.assert(port.name === "content-to-ex");
-    if (!port.hasListener) {
-        port.onMessage.addListener(function (msg) {
-            console.log("[content-to-ex]", msg)
-        });
-        port.onDisconnect.addListener(function () {
-            console.error("[content-to-ex] Disconnected from port.");
-        });
-        // port.postMessage({action: "TEST POST MESSAGE"});
-        port.hasListener = true;
-    }
-});

public/content.js

@@ -1,18 +1,30 @@
-console.log("Loaded content.js")
-
-if (!window.hasAddedMessageListener) {
-    window.addEventListener("message", (event) => {
-        if (event.data?.yakitex?.msg && (event.data.yakitex.msg.type === "TEST")) {
-            const name = event.data.yakitex.msg.fn_name;
-            const args = event.data.yakitex.msg.args;
-            const a = (fn, args) => {
-                return window[fn](args);
+(() => {
+    if (window.contentScriptInjected) {
+        return;
     }
-            const zz = a(name, args);
-            console.log("res ", zz);
-            window.postMessage({type: "FROM_PAGE", text: zz}, "*");
-            return
+    window.contentScriptInjected = true;
+    chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
+        if (request.type === 'INJECT_CODE') {
+            const injectedScriptURL = chrome.runtime.getURL('inject.js');
+            const script = document.createElement('script');
+            script.src = injectedScriptURL;
+            script.onload = () => {
+                window.postMessage({ type: 'CALL_FUNCTION', value: request.value }, '*');
+                script.remove();
+            };
+            (document.head || document.documentElement).appendChild(script);
+            window.addEventListener('message', function onMessage(event) {
+                if (event.source !== window || event.data.type !== 'FROM_PAGE') {
+                    return;
+                }
+                window.removeEventListener('message', onMessage);
+                // Send the result to the background script
+                chrome.runtime.sendMessage({ action: 'SEND_RES_FROM_PAGE', result: event.data.result });
+                sendResponse({ result: event.data.result });
+            });
+            // Return true to indicate that the response will be sent asynchronously
+            return true;
         }
     });
-    window.hasAddedMessageListener = true;
-}
+})()
+

public/inject.js

@@ -1,63 +1,17 @@
-function injectScript(src, id, message) {
-    return new Promise((resolve) => {
-        // Inject a script tag into the page to access methods of the window object
-        const script = document.createElement('script')
+(() => {
+  if (window.injectedMessageListener) {
+    return;
+  }
+  window.injectedMessageListener = true;
 
-        script.onload = () => {
-            const onMessage = ({ data }) => {
-                console.log(data)
-                if (!data.yakitex || !data.yakitex.msg) {
-                    return
+  window.addEventListener('message', function onMessage(event) {
+    if (event.source !== window || event.data.type !== 'CALL_FUNCTION') {
+      return;
     }
 
-                window.removeEventListener('message', onMessage)
-
-                // resolve(data.yakitex.msg)
-
-                script.remove()
-            }
-
-            window.addEventListener('message', onMessage)
-
-            window.postMessage({
-                yakitex: message,
-            })
-        }
-
-        script.setAttribute('src', chrome.runtime.getURL(src))
-
-        document.body.appendChild(script)
-    })
-}
-
-if (!window.hasAddedMessageListener2) {
-    window.addEventListener("message", (event) => {
-        if (event.data.type && (event.data.type === "FROM_PAGE")) {
-            const port = chrome.runtime.connect({ name: "content-to-ex" });
-            port.postMessage(event.data.text);
-            port.disconnect();
-            return
-        }
-    })
-    window.hasAddedMessageListener2 = true;
-}
-
-
-function connectAddListenerFun(port) {
-    console.assert(port.name === "ex-to-content");
-    if (!port.hasListener) {
-        port.onMessage.addListener(function (msg) {
-            console.log(111111111111111);
-            injectScript('content.js', 'sender', { yakitex: "123", id: "yakit", msg: msg })
+    const value = event.data.value;
+    const result = window[value.fn_name](value.args);
+    window.postMessage({ type: 'FROM_PAGE', result: result }, '*');
   });
+})();
 
-        port.onDisconnect.addListener(function () {
-            console.error("[ex-to-content] Disconnected from port.");
-        });
-        port.hasListener = true
-    }
-}
-
-if (!chrome.runtime.onConnect.hasListener(connectAddListenerFun)) {
-    chrome.runtime.onConnect.addListener(connectAddListenerFun);
-}

public/manifest.json

@@ -35,7 +35,7 @@
 
   "web_accessible_resources": [
     {
-      "resources": ["content.js"],
+      "resources": ["inject.js"],
       "matches": ["<all_urls>"],
       "use_dynamic_url": true
     }

public/socket.js

@@ -5,6 +5,9 @@ export const ActionType = {
     PROXYSTATUS: 'proxystatus',
     SETPROXY: 'setproxy',
     CLEARPROXY: 'clearproxy',
+    INJECTSCRIPT: 'INJECT_SCRIPT',
+    SENDRESFROMPAGE: "SEND_RES_FROM_PAGE",
+    RESTOEVALINTAB: "RES_TO_EVALINTAB",
     EVAL: 'eval',
     ECHO: 'echo',
 }

src/components/EvalInTab.tsx

@@ -2,13 +2,12 @@ import React, {useEffect, useState} from "react";
 import { Button } from "antd";
 import TextArea from "antd/lib/input/TextArea";
 import { wsc } from "@network/chrome";
+import { ActionType } from "../../public/socket";
 
-interface EvalInTabProps {
-}
+interface EvalInTabProps {}
 
 export const EvalInTab: React.FC<EvalInTabProps> = () => {
   const [inputData, setInputData] = useState("");
-    const [port, setPort] = useState<chrome.runtime.Port>()
 
   // useEffect(() => {
   //     // const onConnectListener = (port: chrome.runtime.Port) => {
@@ -30,29 +29,22 @@ export const EvalInTab: React.FC<EvalInTabProps> = () => {
   // }, [inputData]);
 
   useEffect(() => {
-        const injectFun = async () => {
-            const tabId = await wsc.getTabId();
-            chrome.scripting.executeScript({
-                target: {tabId: tabId},
-                files: ['inject.js'],
-            }).then(() => {
-                const newPort = chrome.tabs.connect(tabId, {name: "ex-to-content"})
-                setPort(newPort)
-                newPort.onMessage.addListener(function (msg) {
-                    console.log(msg)
-                })
-            })
+    wsc.onWSCMessage((message) => {
+      if (message.action === ActionType.RESTOEVALINTAB) {
+        console.log("res:", message.result);
       }
-        injectFun()
-        return () => {
-            if (port) {
-                port.disconnect()
-            }
-        }
-    }, [])
+    });
+  }, []);
 
   const handleClick = async () => {
-        port.postMessage({type: "TEST", fn_name: "Encrypt", args: inputData});
+    try {
+      const tabId = await wsc.getTabId();
+      chrome.runtime.sendMessage({
+        action: ActionType.INJECTSCRIPT,
+        tabId: tabId,
+        value: { fn_name: "Encrypt", args: inputData },
+      });
+    } catch (error) {}
   };
 
   const handleInputChange = (event: React.ChangeEvent<HTMLTextAreaElement>) => {
@@ -69,4 +61,4 @@ export const EvalInTab: React.FC<EvalInTabProps> = () => {
       <Button onClick={handleClick}>eval in tab me</Button>
     </div>
   );
-}
+};