mirror of
https://github.com/yaklang/yaklang-chrome-extension.git
synced 2026-09-26 21:21:53 +08:00
fix: 注入js
This commit is contained in:
+14
-60
@@ -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.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 })
|
||||
});
|
||||
|
||||
port.onDisconnect.addListener(function () {
|
||||
console.error("[ex-to-content] Disconnected from port.");
|
||||
});
|
||||
port.hasListener = true
|
||||
window.addEventListener('message', function onMessage(event) {
|
||||
if (event.source !== window || event.data.type !== 'CALL_FUNCTION') {
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
if (!chrome.runtime.onConnect.hasListener(connectAddListenerFun)) {
|
||||
chrome.runtime.onConnect.addListener(connectAddListenerFun);
|
||||
}
|
||||
const value = event.data.value;
|
||||
const result = window[value.fn_name](value.args);
|
||||
window.postMessage({ type: 'FROM_PAGE', result: result }, '*');
|
||||
});
|
||||
})();
|
||||
|
||||
|
||||
Reference in New Issue
Block a user