From 8cdd8257f0ebf53d97cf8fa4a3a406a7a636515d Mon Sep 17 00:00:00 2001
From: suziwen <suziwen1@gmail.com>
Date: Wed, 14 Aug 2024 09:47:41 +0800
Subject: [PATCH] Optimize firefox startup

---
 .../overlay/x-background.js                         | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/omega-target-chromium-extension/overlay/x-background.js b/omega-target-chromium-extension/overlay/x-background.js
index 6c2297e..0ee60c9 100644
--- a/omega-target-chromium-extension/overlay/x-background.js
+++ b/omega-target-chromium-extension/overlay/x-background.js
@@ -23,17 +23,26 @@ import "./js/background.js" // zeroBackground
 const isFirefox = !!globalThis.localStorage
 
 function detectPrivateMode(cb) {
-    var db,
+    var db, tempMode,
     on = cb.bind(null, true),
     off = cb.bind(null, false)
   if (isFirefox) {
-    db = indexedDB.open("zeroOmega-test"), db.onerror = on, db.onsuccess = off
+    // in private mode, localStorage will be erased when browser restart
+    tempMode = localStorage.getItem('zeroOmega.isPrivateMode')
+    if (tempMode) {
+      tempMode == 'true' ? on() : off()
+    } else {
+      db = indexedDB.open("zeroOmega-test"), db.onerror = on, db.onsuccess = off
+    }
   } else {
     off()
   }
 }
 
 detectPrivateMode(function (isPrivateMode) {
+  if (isFirefox) {
+    localStorage.setItem('zeroOmega.isPrivateMode', isPrivateMode ? 'true' : 'false')
+  }
 
   if (isPrivateMode && isFirefox) {
     // fake indexedDB