archive/yaklang-chrome-extension
1
0
Fork 0
mirror of https://github.com/yaklang/yaklang-chrome-extension.git synced 2026-09-22 03:10:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix inject

Browse Source
This commit is contained in:
go0p
2025-03-02 15:31:20 +08:00
parent 5dc3c4c248
commit 391f7292e7
2 changed files with 31 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files
public/background.js
+5 -31
View File
@@ -1,4 +1,4 @@
import {ActionType, WebSocketManager} from './socket.js';
import {ActionType, injectScriptAndSendMessage, WebSocketManager} from './socket.js';
console.info("Chrome Extenstion Background is loaded")
@@ -56,38 +56,12 @@ chrome.runtime.onMessage.addListener(function (msg, sender, sendResponse) {
break;
case ActionType.INJECT_SCRIPT:
(async () => {
try {
// 注入 JS 脚本
await chrome.scripting.executeScript({
target: {tabId: msg.tabId},
files: ['content.js']
});
// 发送消息
const response = await chrome.tabs.sendMessage(msg.tabId, {
type: ActionType.INJECT_SCRIPT,
value: msg.value
});
console.log("response", response);
if (response && response.action === ActionType.TO_EXTENSION_PAGE) {
await chrome.runtime.sendMessage(response);
}
} catch (err) {
console.error('Script or CSS injection failed:', err);
}
await injectScriptAndSendMessage(msg.tabId, {
type: ActionType.INJECT_SCRIPT,
value: msg.value
});
})();
break
}
})
const pageFunction = (code) => {
chrome.runtime.sendMessage({code}, response => {
if (response && response.success) {
console.log('Result:', response.result);
} else {
console.error('Error:', response.error);
}
});
}
public/socket.js
+26 -33
View File
@@ -92,49 +92,42 @@ export class WebSocketManager {
return this.socket && this.socket.readyState === WebSocket.OPEN;
}
async handleMessage(message) {
console.log("message", message)
handleMessage(message) {
message = JSON.parse(message);
if (message && message.type === "eval") {
console.log("msg ", message)
const code = message.code;
// try {
// const [tab] = await getTab();
// chrome.runtime.sendMessage({
// action: ActionType.INJECT_SCRIPT, tabId: tab.id, value: {
// mode: "CONTENT_EVAL_CODE", code: code,
// },
// });
// } catch (error) {
// console.log(error.toString())
// }
try {
(async () => {
const [tab] = await getTab();
// 注入 JS 脚本
await chrome.scripting.executeScript({
target: {tabId: tab.id},
files: ['content.js']
});
// 发送消息
const response = await chrome.tabs.sendMessage(tab.id, {
await injectScriptAndSendMessage(tab.id, {
type: ActionType.INJECT_SCRIPT,
value: {
mode: "CONTENT_EVAL_CODE", code: code,
mode: "CONTENT_EVAL_CODE", code: message.code,
}
});
console.log("response", response);
if (response && response.action === ActionType.TO_EXTENSION_PAGE) {
await chrome.runtime.sendMessage(response);
}
} catch (err) {
console.error('Script or CSS injection failed:', err);
}
})();
}
}
}
function getTab() {
const getTab = async () => {
return chrome.tabs.query({active: true, lastFocusedWindow: true})
}
export const injectScriptAndSendMessage = async (tabId, message) => {
try {
// 注入 JS 脚本
await chrome.scripting.executeScript({
target: {tabId: tabId},
files: ['content.js']
});
// 发送消息
const response = await chrome.tabs.sendMessage(tabId, message);
console.log("response", response);
if (response && response.action === ActionType.TO_EXTENSION_PAGE) {
await chrome.runtime.sendMessage(response);
}
} catch (err) {
console.error('Script or CSS injection failed:', err);
}
}