prevent close browser last tab

Author: hoang.tran12

popup/tabs.js

@@ -484,6 +484,7 @@ const tabs = [
       s.magnify_image,
       s.auto_redirectLargestImageSrc,
       s.remove_tracking_in_url,
+      s.prevent_closeBrowser_lastTab,
       s.chongLuaDao,
       s.shortenURL,
       s.unshorten,

scripts/_test.js

@@ -270,15 +270,17 @@ export default {
       document.head.appendChild(favicon);
 
       function updateFavicon() {
+        requestAnimationFrame(updateFavicon);
         let img = canvas.toDataURL();
         favicon.setAttribute("href", img);
       }
 
       setInterval(() => {
         context.clearRect(0, 0, canvas.width, canvas.height);
         context.drawImage(video, 0, 0, canvas.width, canvas.height);
-        updateFavicon();
-      }, 500);
+      }, 1000 / 30);
+
+      updateFavicon();
     },
   },
 
@@ -292,6 +294,45 @@ export default {
       console.log(res);
     },
   },
+
+  backgroundScript: {
+    tabs: {
+      onRemoved: (details, context) => {
+        const [tabId, removeInfo] = details;
+
+        // check if there is any tab left
+        chrome.tabs.query({}, (tabs) => {
+          let tabCached = context.getCache("tabs");
+
+          // case tabs was opened before extension installed/reloaded
+          if (!tabCached) {
+            tabCached = {};
+            tabs.forEach((tab) => {
+              tabCached[tab.id] = tab;
+            });
+            context.setCache("tabs", tabCached);
+          }
+
+          console.log("tabCached", tabCached);
+          if (
+            !tabCached[tabId] ||
+            tabCached[tabId]?.url?.endsWith?.("://newtab/")
+          )
+            return;
+
+          // create a new tab
+          if (tabs.length == 0) {
+            chrome.tabs.create({ url: "chrome://newtab/" });
+          }
+        });
+      },
+      onUpdated: (details, context) => {
+        console.log("onUpdated", details);
+        const [tabId, changeInfo, tab] = details;
+        context.setCache("tabs." + tabId, tab);
+      },
+    },
+  },
 };
 
 const backup = () => {

scripts/background-scripts/background_script.js

@@ -26,26 +26,42 @@ const GLOBAL = {
   runScriptsTab,
   runScriptsBackground,
   checkWillRun,
-  getAllCache,
   getCache,
-  addCache,
+  setCache,
   removeCache,
 };
 
-function getAllCache() {
-  return CACHED;
-}
-
-function getCache(key) {
-  return CACHED[key];
+// keyChain = "" => get all cached
+function getCache(keyChain, defaultValue = null) {
+  let value = CACHED;
+  keyChain?.split(".")?.forEach((key) => {
+    value = value?.[key] ?? null;
+  });
+  return value ?? defaultValue;
 }
 
-function addCache(key, value) {
-  CACHED[key] = value;
+function setCache(keyChain, value) {
+  let keys = keyChain?.split(".");
+  let obj = CACHED;
+  keys?.forEach((key, i) => {
+    if (i === keys.length - 1) {
+      obj[key] = value;
+    } else {
+      obj = obj[key] ?? (obj[key] = {});
+    }
+  });
 }
 
-function removeCache(key) {
-  delete CACHED[key];
+function removeCache(keyChain) {
+  let keys = keyChain?.split(".");
+  let obj = CACHED;
+  keys?.forEach((key, i) => {
+    if (i === keys.length - 1) {
+      delete obj[key];
+    } else {
+      obj = obj[key];
+    }
+  });
 }
 
 function cacheActiveScriptIds() {
@@ -342,8 +358,9 @@ function listenTabs() {
     "onUpdated",
     // "onZoomChange",
   ].forEach((event) => {
-    chrome.tabs[event].addListener((details) => {
-      runScriptsBackground("tabs." + event, details);
+    chrome.tabs[event].addListener((...details) => {
+      // these events will fired in all scripts active scripts (no need to call checkWillRun)
+      runScriptsBackground("tabs." + event, null, details);
     });
   });
 }

scripts/chongLuaDao.js

@@ -126,7 +126,7 @@ async function saveBgCache(context) {
       whiteList: await context.utils.Storage.get(KEYS.whiteList, []),
     };
     console.log("cached chongLuaDao", cached);
-    context.addCache("chongLuaDao", cached);
+    context.setCache("chongLuaDao", cached);
     return cached;
   } catch (e) {
     console.log("cache chongLuaDao FAIL ", e);

scripts/index.js

@@ -164,6 +164,7 @@ import remove_tracking_in_url from "./remove_tracking_in_url.js";
 import auto_lockWebsite from "./auto_lockWebsite.js";
 import showFps_v2 from "./showFps_v2.js";
 import chongLuaDao from "./chongLuaDao.js";
+import prevent_closeBrowser_lastTab from "./prevent_closeBrowser_lastTab.js";
 
 // inject badges
 const allScripts = {
@@ -344,6 +345,10 @@ const allScripts = {
   auto_lockWebsite: addBadge(auto_lockWebsite, BADGES.new),
   showFps_v2: addBadge(showFps_v2, BADGES.new),
   chongLuaDao: addBadge(chongLuaDao, BADGES.new),
+  prevent_closeBrowser_lastTab: addBadge(
+    prevent_closeBrowser_lastTab,
+    BADGES.new
+  ),
 };
 
 // alert(Object.keys(allScripts).length);