fix: 注入js

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);
-            }
-            const zz = a(name, args);
-            console.log("res ", zz);
-            window.postMessage({type: "FROM_PAGE", text: zz}, "*");
-            return
+(() => {
+    if (window.contentScriptInjected) {
+        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;
-}
+})()
+