Fixed theming for existing users

1 files changed, 47 insertions, 38 deletions

diff --git a/content-script.js b/content-script.js
index 4f02174..4e99e29 100644
--- a/content-script.js
+++ b/content-script.js
@@ -1,42 +1,51 @@
-(function() {
-    const stylesheetId = 'zeninternet-custom-styles';
-    
-    // Create or get our stylesheet element
-    function getStylesheet() {
-        let stylesheet = document.getElementById(stylesheetId);
-        if (!stylesheet) {
-            stylesheet = document.createElement('style');
-            stylesheet.id = stylesheetId;
-            stylesheet.type = 'text/css';
-            document.head.appendChild(stylesheet);
-        }
-        return stylesheet;
+(function () {
+  const stylesheetId = "zeninternet-custom-styles";
+
+  // Create or get our stylesheet element
+  function getStylesheet() {
+    let stylesheet = document.getElementById(stylesheetId);
+    if (!stylesheet) {
+      stylesheet = document.createElement("style");
+      stylesheet.id = stylesheetId;
+      stylesheet.type = "text/css";
+      document.head.appendChild(stylesheet);
     }
-    
-    // Update our stylesheet content
-    function updateStyles(css) {
-        const stylesheet = getStylesheet();
-        stylesheet.textContent = css || '';
-        console.log('ZenInternet: Styles were ' + (css ? 'updated' : 'removed'));
-    }
-    
-    // Announce content script is ready and provide current hostname
-    function announceReady() {
-        browser.runtime.sendMessage({
-            action: 'contentScriptReady',
-            hostname: window.location.hostname,
+    return stylesheet;
+  }
+
+  // Update our stylesheet content
+  function updateStyles(css) {
+    const stylesheet = getStylesheet();
+    stylesheet.textContent = css || "";
+    console.log("ZenInternet: Styles were " + (css ? "updated" : "removed"));
+  }
+
+  // Announce content script is ready and provide current hostname
+  function announceReady() {
+    try {
+      browser.runtime
+        .sendMessage({
+          action: "contentScriptReady",
+          hostname: window.location.hostname,
+        })
+        .catch((err) => {
+          // Silent fail - background might not be ready yet
+          console.log("ZenInternet: Could not announce ready state");
         });
+    } catch (e) {
+      // Fail silently
+    }
+  }
+
+  // Listen for messages from background script
+  browser.runtime.onMessage.addListener((message) => {
+    if (message.action === "applyStyles") {
+      updateStyles(message.css);
+      return Promise.resolve({ success: true });
     }
-    
-    // Listen for messages from background script
-    browser.runtime.onMessage.addListener((message) => {
-        if (message.action === 'applyStyles') {
-            updateStyles(message.css);
-            return Promise.resolve({ success: true });
-        }
-        return false;
-    });
-    
-    // Announce content script is ready on load
-    announceReady();
+    return false;
+  });
+
+  // Announce content script is ready on load
+  announceReady();
 })();