feat: 增强内容脚本与后台通信，添加 PING 响应和确保脚本就绪的机制

2026-04-11 02:01:36 +08:00
parent bac7113c0c
commit 359566131d
3 changed files with 102 additions and 41 deletions
@@ -17,10 +17,18 @@ const LOG_PREFIX = `[MultiPage:${SCRIPT_SOURCE}]`;
 const STOP_ERROR_MESSAGE = '流程已被用户停止。';
 let flowStopped = false;

-chrome.runtime.onMessage.addListener((message) => {
+chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
  if (message.type === 'STOP_FLOW') {
    flowStopped = true;
    console.warn(LOG_PREFIX, STOP_ERROR_MESSAGE);
+    return;
+  }
+
+  if (message.type === 'PING') {
+    sendResponse({
+      ok: true,
+      source: SCRIPT_SOURCE,
+    });
  }
 });

@@ -25,24 +25,32 @@

 console.log('[MultiPage:vps-panel] Content script loaded on', location.href);

-// Listen for commands from Background
-chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
-  if (message.type === 'EXECUTE_STEP') {
-    resetStopState();
-    handleStep(message.step, message.payload).then(() => {
-      sendResponse({ ok: true });
-    }).catch(err => {
-      if (isStopError(err)) {
-        log(`步骤 ${message.step}：已被用户停止。`, 'warn');
-        sendResponse({ stopped: true, error: err.message });
-        return;
-      }
-      reportError(message.step, err.message);
-      sendResponse({ error: err.message });
-    });
-    return true;
-  }
-});
+const VPS_PANEL_LISTENER_SENTINEL = 'data-multipage-vps-panel-listener';
+
+if (document.documentElement.getAttribute(VPS_PANEL_LISTENER_SENTINEL) !== '1') {
+  document.documentElement.setAttribute(VPS_PANEL_LISTENER_SENTINEL, '1');
+
+  // Listen for commands from Background
+  chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
+    if (message.type === 'EXECUTE_STEP') {
+      resetStopState();
+      handleStep(message.step, message.payload).then(() => {
+        sendResponse({ ok: true });
+      }).catch(err => {
+        if (isStopError(err)) {
+          log(`步骤 ${message.step}：已被用户停止。`, 'warn');
+          sendResponse({ stopped: true, error: err.message });
+          return;
+        }
+        reportError(message.step, err.message);
+        sendResponse({ error: err.message });
+      });
+      return true;
+    }
+  });
+} else {
+  console.log('[MultiPage:vps-panel] 消息监听已存在，跳过重复注册');
+}

 async function handleStep(step, payload) {
  switch (step) {