diff options
| -rw-r--r-- | extension/public/background.js | 14 | ||||
| -rw-r--r-- | extension/public/iconGrabber.js | 10 |
2 files changed, 20 insertions, 4 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(); }) diff --git a/extension/public/iconGrabber.js b/extension/public/iconGrabber.js index 63ba7e0..c463451 100644 --- a/extension/public/iconGrabber.js +++ b/extension/public/iconGrabber.js @@ -1,3 +1,11 @@ +export function getBrowser() { + if (typeof browser === "undefined") { + return chrome; + } else { + return browser; + } +} + const tagTypes = ["apple-touch-icon", "shortcut icon", "icon"] let x = Array.from(document.getElementsByTagName("link")) @@ -10,6 +18,6 @@ let x = Array.from(document.getElementsByTagName("link")) }) .map(elem => elem.href); -chrome.runtime.sendMessage([window.location.href, x[0]]).catch(() => { +getBrowser().runtime.sendMessage([window.location.href, x[0]]).catch(() => { console.log("failed to send message") })
\ No newline at end of file |