PB-33297 Extension update available should store the state if user si…

…gned-in
passbolt · May 7, 2024 · 7deb6f1 · 7deb6f1
1 parent fe56c6d
commit 7deb6f1
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/all/background_page/service/extension/onExtensionUpdateAvailableService.js b/src/all/background_page/service/extension/onExtensionUpdateAvailableService.js
@@ -20,14 +20,16 @@ import WebIntegrationPagemod from "../../pagemod/webIntegrationPagemod";
 import WorkerService from "../worker/workerService";
 import PublicWebsiteSignInPagemod from "../../pagemod/publicWebsiteSignInPagemod";
 
+const PASSBOLT_EXTENSION_UPDATE = "passboltExtensionUpdate";
+
 class OnExtensionUpdateAvailableService {
   /**
    * Execute the OnExtensionUpdateAvailableService process
    * @returns {Promise<void>}
    */
   static async exec() {
     if (await isUserAuthenticated()) {
-      this.shouldReload = true;
+      await browser.storage.session.set({[PASSBOLT_EXTENSION_UPDATE]: true});
     } else {
       await this.cleanAndReload();
     }
@@ -47,8 +49,9 @@ class OnExtensionUpdateAvailableService {
    * It triggers a runtime reload if an extension update was available while the user was signed in.
    */
   static async handleUserLoggedOut() {
-    if (this.shouldReload) {
-      this.shouldReload = false;
+    const shouldUpdate = await browser.storage.session.get(PASSBOLT_EXTENSION_UPDATE);
+    if (shouldUpdate && shouldUpdate[PASSBOLT_EXTENSION_UPDATE]) {
+      await browser.storage.session.remove(PASSBOLT_EXTENSION_UPDATE);
       await this.cleanAndReload();
     }
   }