fix(scripts): wrong background script performance

wanhose · Jul 1, 2021 · 5d5d295 · 5d5d295
1 parent b54a460
commit 5d5d295
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/scripts/background.js b/scripts/background.js
@@ -209,7 +209,14 @@ const updateState = (tabId, state) => {
 
 chrome.runtime.onMessage.addListener((request, sender, responseCallback) => {
   const hasPermission = !sender.frameId || sender.frameId === 0;
-  const tabId = sender.tab.id;
+  let tabId = sender.tab ? sender.tab.id : undefined;
+
+  if (!tabId) {
+    chrome.tabs.query(
+      { active: true, currentWindow: true },
+      (tabs) => (tabId = tabs[0] ? tabs[0].id : 0)
+    );
+  }
 
   switch (request.type) {
     case "DISABLE_ICON":