1 files changed, 11 insertions, 3 deletions
diff --git a/extension/public/background.js b/extension/public/background.js
index 06c75eb..15c50ba 100644
--- a/extension/public/background.js
+++ b/extension/public/background.js
@@ -1,10 +1,18 @@
-chrome.runtime.onMessage.addListener(async (request, sender, sendResponse) => {
+function getBrowser() {
+    if (typeof browser === "undefined") {
+        return chrome;
+    } else {
+        return browser;
+    }
+}
+
+getBrowser().runtime.onMessage.addListener(async (request, sender, sendResponse) => {
     console.log("received message", request)
     let [url, icon] = request;
-    let bmk = (await chrome.bookmarks.search({url : url}));
+    let bmk = (await getBrowser().bookmarks.search({url : url}));
     console.log(bmk)
     if (bmk && bmk.length > 0) {
-        await chrome.storage.local.set({["icon-"+bmk[0].id]: icon});
+        await getBrowser().storage.local.set({["icon-"+bmk[0].id]: icon});
     }
     sendResponse();
 })