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 });