From ca57b55514cc111210f5069afc6434603f919d94 Mon Sep 17 00:00:00 2001
From: wanhose <hello@wanhose.dev>
Date: Tue, 27 Feb 2024 19:43:22 +0100
Subject: [PATCH] feat(browser-extension): add SET_BADGE handler to the
 background script and reset action badge text when disabling icon

---
 packages/browser-extension/src/scripts/background.js | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/packages/browser-extension/src/scripts/background.js b/packages/browser-extension/src/scripts/background.js
index ea4357d..7a47897 100644
--- a/packages/browser-extension/src/scripts/background.js
+++ b/packages/browser-extension/src/scripts/background.js
@@ -107,6 +107,7 @@ chrome.runtime.onMessage.addListener((message, sender, callback) => {
     case 'DISABLE_ICON':
       if (isPage && tabId) {
         chrome.action.setIcon({ path: '/assets/icons/disabled.png', tabId }, suppressLastError);
+        chrome.action.setBadgeText({ tabId, text: '' });
       }
       break;
     case 'ENABLE_ICON':
@@ -158,6 +159,12 @@ chrome.runtime.onMessage.addListener((message, sender, callback) => {
         report(message, sender.tab);
       }
       break;
+    case 'SET_BADGE':
+      if (tabId) {
+        chrome.action.setBadgeBackgroundColor({ color: '#6b7280' });
+        chrome.action.setBadgeText({ tabId, text: message.value });
+      }
+      break;
     case 'SET_HOSTNAME_STATE':
       if (hostname && message.state.enabled === false) {
         storage.set({ [hostname]: message.state });