Optimized Unify more #291

1 files changed, 14 insertions, 10 deletions

diff --git a/src/pages/background/background.js b/src/pages/background/background.js
index bf992161..933172b6 100644
--- a/src/pages/background/background.js
+++ b/src/pages/background/background.js
@@ -60,15 +60,7 @@ redditHelper.setLibredditCookies();
 redditHelper.setTedditCookies();
 tiktokHelper.setProxiTokCookies();
 
-let incognitoInit = false;
-browser.tabs.onCreated.addListener(
-  tab => {
-    if (!incognitoInit && tab.incognito) {
-      browser.tabs.create({ url: browser.runtime.getURL("/pages/background/incognito.html") });
-      incognitoInit = true;
-    }
-  }
-);
+
 
 let BYPASSTABs = [];
 browser.webRequest.onBeforeRequest.addListener(
@@ -82,7 +74,6 @@ browser.webRequest.onBeforeRequest.addListener(
     }
     catch { return null; }
 
-
     let newUrl = youtubeMusicHelper.redirect(url, details.type)
     if (!newUrl) newUrl = youtubeHelper.redirect(url, details, initiator)
     if (!newUrl) newUrl = twitterHelper.redirect(url, initiator);
@@ -126,8 +117,21 @@ browser.webRequest.onBeforeRequest.addListener(
   ["blocking"]
 );
 
+
+let incognitoList = [];
+browser.tabs.onCreated.addListener(
+  tab => {
+    if (tab.incognito) {
+      incognitoList.push(tab.id);
+      if (incognitoList.length == 1) browser.tabs.create({ url: browser.runtime.getURL("/pages/background/incognito.html") });
+    }
+  }
+);
+
 browser.tabs.onRemoved.addListener(
   tabId => {
+    incognitoList.pop(tabId)
+
     const i = BYPASSTABs.indexOf(tabId);
     if (i > -1) {
       BYPASSTABs.splice(i, 1);