From a3cbdb69085c72db5e2a17a321e9e4cb078c780a Mon Sep 17 00:00:00 2001
From: BPower0036 <80090789+BPower0036@users.noreply.github.com>
Date: Thu, 5 May 2022 11:43:44 +0000
Subject: [PATCH] Update messageHandler.js

---
 src/js/background/messageHandler.js | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/js/background/messageHandler.js b/src/js/background/messageHandler.js
index b748916..6e5417a 100644
--- a/src/js/background/messageHandler.js
+++ b/src/js/background/messageHandler.js
@@ -20,6 +20,9 @@ const messageHandler = {
       case "resetSync":
         response = sync.resetSync();
         break;
+      case "resetPageAction":
+        response = assignManager.resetPageAction();
+        break;
       case "deleteContainer":
         response = backgroundLogic.deleteContainer(m.message.userContextId);
         break;
@@ -196,6 +199,17 @@ const messageHandler = {
       });
     }, {urls: ["<all_urls>"], types: ["main_frame"]});
 
+    browser.tabs.onUpdated.addListener((tabId) => {
+      // check if the page action is enabled right away to avoid flashing
+      browser.storage.local.get({ pageActionEnabled: true }).then(({ pageActionEnabled }) => {
+        if (pageActionEnabled) {
+          browser.pageAction.show(tabId);
+        }
+      }).catch(e => {
+        throw e;
+      });
+    }, { properties: ["status"] });
+
     browser.tabs.onCreated.addListener((tab) => {
       // lets remember the last tab created so we can close it if it looks like a redirect
       this.lastCreatedTab = tab;