Update extension icon when adding/removing custom setting

Author: Pythack

background.js

@@ -96,6 +96,10 @@ browser.runtime.onMessage.addListener((message, sender) => {
             restoreOptions(tab); // Apply new settings in each tab
           });
         });
+      case "update-action-icon": // When the user adds a new custom setting
+        browser.tabs.query({ active: true, currentWindow: true }, tabs => {
+          update_action_icon({ tabId: tabs[0].id })// Update icon
+        });
         break;
       // case "display-notification": // When the user saves the settings (from popup)
       //     browser.notifications.create(message.notificationName, message.notification);

popup.js

@@ -69,6 +69,9 @@ function addCustom() {
 		select.appendChild(optionToAdd); // Append new option to select
 		select.value = custom_url;
 		select.dispatchEvent(new Event('change'));
+		browser.runtime.sendMessage({ // Send message to background script asking to update the action icon
+			request: "update-action-icon"
+		});
 	}, onError);
 };
 
@@ -98,6 +101,9 @@ function removeCustom() {
 			document.querySelector('div#textShadowOptions').style.display = "none";
 		}
 		updatePreview(); // Update preview
+		browser.runtime.sendMessage({ // Send message to background script asking to update the action icon
+			request: "update-action-icon"
+		});
 	}, onError);
 };