Version 2 improvements with UI and functionality

chrome/background.js

@@ -1,6 +1,7 @@
 // Background service worker for Chrome extension
 chrome.runtime.onInstalled.addListener(() => {
   console.log('GreasyBoii extension installed');
+  refreshBadge();
 });
 
 // Listen for messages from content script
@@ -11,7 +12,7 @@ chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
     });
     return true; // Keep channel open for async response
   }
-  
+
   // Execute JavaScript via scripting API to bypass CSP
   if (request.action === 'executeScript') {
     if (!sender.tab || !sender.tab.id) {
@@ -48,4 +49,40 @@ chrome.runtime.onMessage.addListener((request, sender, sendResponse) => {
 
     return true; // Keep channel open for async response
   }
+
+  // The popup closes the instant the user clicks the page to pick an element, so the
+  // content script can't hand the selector straight back to it — relay it through
+  // storage instead; the popup picks it up next time it opens (see popup.js).
+  if (request.action === 'pickerResult') {
+    chrome.storage.local.set({ pendingPickResult: request.selector });
+  } else if (request.action === 'pickerCancelled') {
+    chrome.storage.local.remove(['pickerPending', 'pendingPickResult']);
+  }
 });
+
+// Master enable/disable keyboard shortcut (see manifest.json "commands")
+chrome.commands.onCommand.addListener((command) => {
+  if (command === 'toggle-greasyboii') {
+    chrome.storage.sync.get(['masterEnabled'], (result) => {
+      const enabled = result.masterEnabled !== false;
+      chrome.storage.sync.set({ masterEnabled: !enabled });
+      // badge refresh happens via the storage.onChanged listener below
+    });
+  }
+});
+
+// Keep the toolbar badge in sync whenever masterEnabled changes, from any source
+// (the shortcut above, the popup's toggle switch, or a rules import)
+chrome.storage.onChanged.addListener((changes, area) => {
+  if (area === 'sync' && changes.masterEnabled) {
+    refreshBadge();
+  }
+});
+
+function refreshBadge() {
+  chrome.storage.sync.get(['masterEnabled'], (result) => {
+    const enabled = result.masterEnabled !== false;
+    chrome.action.setBadgeText({ text: enabled ? '' : 'OFF' });
+    chrome.action.setBadgeBackgroundColor({ color: '#dc3545' });
+  });
+}