From 741385738bbab2b19b0292250c5ef1a6653772d9 Mon Sep 17 00:00:00 2001
From: Oksana Melnik <oksana@dedact.nl>
Date: Sun, 3 Nov 2019 09:30:37 +0100
Subject: [PATCH] save confirm page url to the local storage as a backup

---
 src/js/background/assignManager.js | 2 ++
 src/js/confirm-page.js             | 9 ++++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/js/background/assignManager.js b/src/js/background/assignManager.js
index d5aca77..7c9e7ac 100644
--- a/src/js/background/assignManager.js
+++ b/src/js/background/assignManager.js
@@ -490,6 +490,8 @@ const assignManager = {
         currentCookieStoreId = backgroundLogic.cookieStoreId(currentUserContextId);
         confirmUrl += `&currentCookieStoreId=${currentCookieStoreId}`;
       }
+
+      browser.storage.local.set({ "confirmUrl": confirmUrl});
       browser.tabs.create({
         url: confirmUrl,
         cookieStoreId: currentCookieStoreId,
diff --git a/src/js/confirm-page.js b/src/js/confirm-page.js
index 8a14a7b..4056237 100644
--- a/src/js/confirm-page.js
+++ b/src/js/confirm-page.js
@@ -1,5 +1,12 @@
 async function load() {
-  const searchParams = new URL(window.location).searchParams;
+
+  let searchParams = new URL(window.location).searchParams;
+
+  if (!searchParams.get("url")) {
+    const savedUrl = await browser.storage.local.get("confirmUrl");
+    searchParams = new URL(savedUrl.confirmUrl).searchParams;
+  }
+
   const redirectUrl = searchParams.get("url");
   const cookieStoreId = searchParams.get("cookieStoreId");
   const currentCookieStoreId = searchParams.get("currentCookieStoreId");