66 lines
2.3 KiB
JavaScript
66 lines
2.3 KiB
JavaScript
/* eslint-disable */
|
|
|
|
(() => {
|
|
// Prevent duplicate injection of the bridge itself.
|
|
if (window.__INJECT_SCRIPT_TOOL_UNIVERSAL_BRIDGE_LOADED__) return;
|
|
window.__INJECT_SCRIPT_TOOL_UNIVERSAL_BRIDGE_LOADED__ = true;
|
|
const EVENT_NAME = {
|
|
RESPONSE: 'chrome-mcp:response',
|
|
CLEANUP: 'chrome-mcp:cleanup',
|
|
EXECUTE: 'chrome-mcp:execute',
|
|
};
|
|
const pendingRequests = new Map();
|
|
|
|
const messageHandler = (request, _sender, sendResponse) => {
|
|
// --- Lifecycle Command ---
|
|
if (request.type === EVENT_NAME.CLEANUP) {
|
|
window.dispatchEvent(new CustomEvent(EVENT_NAME.CLEANUP));
|
|
// Acknowledge cleanup signal received, but don't hold the connection.
|
|
sendResponse({ success: true });
|
|
return true;
|
|
}
|
|
|
|
// --- Execution Command for MAIN world ---
|
|
if (request.targetWorld === 'MAIN') {
|
|
const requestId = `req-${Date.now()}-${Math.random()}`;
|
|
pendingRequests.set(requestId, sendResponse);
|
|
|
|
window.dispatchEvent(
|
|
new CustomEvent(EVENT_NAME.EXECUTE, {
|
|
detail: {
|
|
action: request.action,
|
|
payload: request.payload,
|
|
requestId: requestId,
|
|
},
|
|
}),
|
|
);
|
|
return true; // Async response is expected.
|
|
}
|
|
// Note: Requests for ISOLATED world are handled by the user's isolatedWorldCode script directly.
|
|
// This listener won't process them unless it's the only script in ISOLATED world.
|
|
};
|
|
|
|
chrome.runtime.onMessage.addListener(messageHandler);
|
|
|
|
// Listen for responses coming back from the MAIN world.
|
|
const responseHandler = (event) => {
|
|
const { requestId, data, error } = event.detail;
|
|
if (pendingRequests.has(requestId)) {
|
|
const sendResponse = pendingRequests.get(requestId);
|
|
sendResponse({ data, error });
|
|
pendingRequests.delete(requestId);
|
|
}
|
|
};
|
|
window.addEventListener(EVENT_NAME.RESPONSE, responseHandler);
|
|
|
|
// --- Self Cleanup ---
|
|
// When the cleanup signal arrives, this bridge must also clean itself up.
|
|
const cleanupHandler = () => {
|
|
chrome.runtime.onMessage.removeListener(messageHandler);
|
|
window.removeEventListener(EVENT_NAME.RESPONSE, responseHandler);
|
|
window.removeEventListener(EVENT_NAME.CLEANUP, cleanupHandler);
|
|
delete window.__INJECT_SCRIPT_TOOL_UNIVERSAL_BRIDGE_LOADED__;
|
|
};
|
|
window.addEventListener(EVENT_NAME.CLEANUP, cleanupHandler);
|
|
})();
|