From 0d0c487275103ad224821a5d772a31ba0b3d1c43 Mon Sep 17 00:00:00 2001
From: BPower0036 <80090789+BPower0036@users.noreply.github.com>
Date: Thu, 5 May 2022 11:42:27 +0000
Subject: [PATCH] Update assignManager.js

---
 src/js/background/assignManager.js | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/src/js/background/assignManager.js b/src/js/background/assignManager.js
index d8ff37c..fb641f4 100644
--- a/src/js/background/assignManager.js
+++ b/src/js/background/assignManager.js
@@ -67,6 +67,11 @@ window.assignManager = {
       return !!syncEnabled;
     },
 
+    async getPageActionEnabled() {
+      const { pageActionEnabled } = await browser.storage.local.get({ pageActionEnabled: true });
+      return !!pageActionEnabled;
+    },
+
     async getReplaceTabEnabled() {
       const { replaceTabEnabled } = await browser.storage.local.get("replaceTabEnabled");
       return !!replaceTabEnabled;
@@ -541,6 +546,19 @@ window.assignManager = {
     }
   },
 
+  async resetPageAction() {
+    const pageActionEnabled = await this.storageArea.getPageActionEnabled();
+    const tabs = await browser.tabs.query({});
+    const res = tabs.map((tab) => {
+      if (pageActionEnabled) {
+        return browser.pageAction.show(tab.id);
+      } else {
+        return browser.pageAction.hide(tab.id);
+      }
+    });
+    await Promise.all(res);
+  },
+
   contextualIdentityCreated(changeInfo) {
     browser.contextMenus.create({
       parentId: assignManager.OPEN_IN_CONTAINER,