/* global download prefs openURL FIREFOX CHROME VIVALDI debounce URLS ignoreChromeError getTab styleManager msg navigatorUtil iconUtil workerUtil contentScripts sync findExistTab createTab activateTab isTabReplaceable getActiveTab */ 'use strict'; // eslint-disable-next-line no-var var backgroundWorker = workerUtil.createWorker({ url: '/background/background-worker.js' }); window.API_METHODS = Object.assign(window.API_METHODS || {}, { deleteStyle: styleManager.deleteStyle, editSave: styleManager.editSave, findStyle: styleManager.findStyle, getAllStyles: styleManager.getAllStyles, // used by importer getSectionsByUrl: styleManager.getSectionsByUrl, getStyle: styleManager.get, getStylesByUrl: styleManager.getStylesByUrl, importStyle: styleManager.importStyle, importManyStyles: styleManager.importMany, installStyle: styleManager.installStyle, styleExists: styleManager.styleExists, toggleStyle: styleManager.toggleStyle, addInclusion: styleManager.addInclusion, removeInclusion: styleManager.removeInclusion, addExclusion: styleManager.addExclusion, removeExclusion: styleManager.removeExclusion, getTabUrlPrefix() { const {url} = this.sender.tab; if (url.startsWith(URLS.ownOrigin)) { return 'stylus'; } return url.match(/^([\w-]+:\/+[^/#]+)/)[1]; }, download(msg) { delete msg.method; return download(msg.url, msg); }, parseCss({code}) { return backgroundWorker.parseMozFormat({code}); }, getPrefs: prefs.getAll, openEditor, updateIconBadge(count) { // TODO: remove once our manifest's minimum_chrome_version is 50+ // Chrome 49 doesn't report own extension pages in webNavigation apparently // so we do a force update which doesn't use the cache. if (CHROME && CHROME < 2661 && this.sender.tab.url.startsWith(URLS.ownOrigin)) { updateIconBadgeForce(this.sender.tab.id, count); } else { updateIconBadge(this.sender.tab.id, count); } return true; }, // exposed for stuff that requires followup sendMessage() like popup::openSettings // that would fail otherwise if another extension forced the tab to open // in the foreground thus auto-closing the popup (in Chrome) openURL, optionsCustomizeHotkeys() { return browser.runtime.openOptionsPage() .then(() => new Promise(resolve => setTimeout(resolve, 100))) .then(() => msg.broadcastExtension({method: 'optionsCustomizeHotkeys'})); }, syncStart: sync.start, syncStop: sync.stop, syncNow: sync.syncNow, getSyncStatus: sync.getStatus, syncLogin: sync.login, openManage }); // eslint-disable-next-line no-var var browserCommands, contextMenus; // ************************************************************************* // register all listeners msg.on(onRuntimeMessage); navigatorUtil.onUrlChange(({tabId, frameId}, type) => { if (type === 'committed') { // styles would be updated when content script is injected. return; } msg.sendTab(tabId, {method: 'urlChanged'}, {frameId}) .catch(msg.ignoreError); }); if (FIREFOX) { // FF applies page CSP even to content scripts, https://bugzil.la/1267027 navigatorUtil.onCommitted(webNavUsercssInstallerFF, { url: [ {pathSuffix: '.user.css'}, {pathSuffix: '.user.styl'}, ] }); // FF misses some about:blank iframes so we inject our content script explicitly navigatorUtil.onDOMContentLoaded(webNavIframeHelperFF, { url: [ {urlEquals: 'about:blank'}, ] }); } if (chrome.contextMenus) { chrome.contextMenus.onClicked.addListener((info, tab) => contextMenus[info.menuItemId].click(info, tab)); } if (chrome.commands) { // Not available in Firefox - https://bugzilla.mozilla.org/show_bug.cgi?id=1240350 chrome.commands.onCommand.addListener(command => browserCommands[command]()); } const tabIcons = new Map(); chrome.tabs.onRemoved.addListener(tabId => tabIcons.delete(tabId)); chrome.tabs.onReplaced.addListener((added, removed) => tabIcons.delete(removed)); prefs.subscribe([ 'disableAll', 'badgeDisabled', 'badgeNormal', ], () => debounce(refreshIconBadgeColor)); prefs.subscribe([ 'show-badge' ], () => debounce(refreshAllIconsBadgeText)); prefs.subscribe([ 'disableAll', 'iconset', ], () => debounce(refreshAllIcons)); prefs.initializing.then(() => { refreshIconBadgeColor(); refreshAllIconsBadgeText(); refreshAllIcons(); }); navigatorUtil.onUrlChange(({tabId, frameId, transitionQualifiers}, type) => { if (type === 'committed' && !frameId) { // it seems that the tab icon would be reset by navigation. We // invalidate the cache here so it would be refreshed by `apply.js`. tabIcons.delete(tabId); // however, if the tab was swapped in by forward/backward buttons, // `apply.js` doesn't notify the background to update the icon, // so we have to refresh it manually. if (transitionQualifiers.includes('forward_back')) { msg.sendTab(tabId, {method: 'updateCount'}).catch(msg.ignoreError); } } }); // ************************************************************************* chrome.runtime.onInstalled.addListener(({reason}) => { // save install type: "admin", "development", "normal", "sideload" or "other" // "normal" = addon installed from webstore chrome.management.getSelf(info => { localStorage.installType = info.installType; if (reason === 'install' && info.installType === 'development' && chrome.contextMenus) { createContextMenus(['reload']); } }); if (reason !== 'update') return; // translations may change localStorage.L10N = JSON.stringify({ browserUIlanguage: chrome.i18n.getUILanguage(), }); // themes may change delete localStorage.codeMirrorThemes; }); // ************************************************************************* // browser commands browserCommands = { openManage, openOptions: () => openManage({options: true}), styleDisableAll(info) { prefs.set('disableAll', info ? info.checked : !prefs.get('disableAll')); }, reload: () => chrome.runtime.reload(), }; // ************************************************************************* // context menus contextMenus = { 'show-badge': { title: 'menuShowBadge', click: info => prefs.set(info.menuItemId, info.checked), }, 'disableAll': { title: 'disableAllStyles', click: browserCommands.styleDisableAll, }, 'open-manager': { title: 'openStylesManager', click: browserCommands.openManage, }, 'open-options': { title: 'openOptions', click: browserCommands.openOptions, }, 'reload': { presentIf: () => localStorage.installType === 'development', title: 'reload', click: browserCommands.reload, }, 'editor.contextDelete': { presentIf: () => !FIREFOX && prefs.get('editor.contextDelete'), title: 'editDeleteText', type: 'normal', contexts: ['editable'], documentUrlPatterns: [URLS.ownOrigin + 'edit*'], click: (info, tab) => { msg.sendTab(tab.id, {method: 'editDeleteText'}, undefined, 'extension'); }, } }; function createContextMenus(ids) { for (const id of ids) { let item = contextMenus[id]; if (item.presentIf && !item.presentIf()) { continue; } item = Object.assign({id}, item); delete item.presentIf; item.title = chrome.i18n.getMessage(item.title); if (!item.type && typeof prefs.defaults[id] === 'boolean') { item.type = 'checkbox'; item.checked = prefs.get(id); } if (!item.contexts) { item.contexts = ['browser_action']; } delete item.click; chrome.contextMenus.create(item, ignoreChromeError); } } if (chrome.contextMenus) { // circumvent the bug with disabling check marks in Chrome 62-64 const toggleCheckmark = CHROME >= 3172 && CHROME <= 3288 ? (id => chrome.contextMenus.remove(id, () => createContextMenus([id]) + ignoreChromeError())) : ((id, checked) => chrome.contextMenus.update(id, {checked}, ignoreChromeError)); const togglePresence = (id, checked) => { if (checked) { createContextMenus([id]); } else { chrome.contextMenus.remove(id, ignoreChromeError); } }; const keys = Object.keys(contextMenus); prefs.subscribe(keys.filter(id => typeof prefs.defaults[id] === 'boolean'), toggleCheckmark); prefs.subscribe(keys.filter(id => contextMenus[id].presentIf), togglePresence); createContextMenus(keys); } // reinject content scripts when the extension is reloaded/updated. Firefox // would handle this automatically. if (!FIREFOX) { setTimeout(contentScripts.injectToAllTabs, 0); } // register hotkeys if (FIREFOX && browser.commands && browser.commands.update) { const hotkeyPrefs = Object.keys(prefs.defaults).filter(k => k.startsWith('hotkey.')); prefs.subscribe(hotkeyPrefs, (name, value) => { try { name = name.split('.')[1]; if (value.trim()) { browser.commands.update({name, shortcut: value}); } else { browser.commands.reset(name); } } catch (e) {} }); } msg.broadcastTab({method: 'backgroundReady'}); function webNavUsercssInstallerFF(data) { const {tabId} = data; Promise.all([ msg.sendTab(tabId, {method: 'ping'}) .catch(() => false), // we need tab index to open the installer next to the original one // and also to skip the double-invocation in FF which assigns tab url later getTab(tabId), ]).then(([pong, tab]) => { if (pong !== true && tab.url !== 'about:blank') { window.API_METHODS.openUsercssInstallPage({direct: true}, {tab}); } }); } function webNavIframeHelperFF({tabId, frameId}) { if (!frameId) return; msg.sendTab(tabId, {method: 'ping'}, {frameId}) .catch(() => false) .then(pong => { if (pong) return; // insert apply.js to iframe const files = chrome.runtime.getManifest().content_scripts[0].js; for (const file of files) { chrome.tabs.executeScript(tabId, { frameId, file, matchAboutBlank: true, }, ignoreChromeError); } }); } function updateIconBadge(tabId, count) { let tabIcon = tabIcons.get(tabId); if (!tabIcon) tabIcons.set(tabId, (tabIcon = {})); if (tabIcon.count === count) { return; } const oldCount = tabIcon.count; tabIcon.count = count; refreshIconBadgeText(tabId, tabIcon); if (Boolean(oldCount) !== Boolean(count)) { refreshIcon(tabId, tabIcon); } } function updateIconBadgeForce(tabId, count) { refreshIconBadgeText(tabId, {count}); refreshIcon(tabId, {count}); } function refreshIconBadgeText(tabId, icon) { iconUtil.setBadgeText({ text: prefs.get('show-badge') && icon.count ? String(icon.count) : '', tabId }); } function refreshIcon(tabId, icon) { const disableAll = prefs.get('disableAll'); const iconset = prefs.get('iconset') === 1 ? 'light/' : ''; const postfix = disableAll ? 'x' : !icon.count ? 'w' : ''; const iconType = iconset + postfix; if (icon.iconType === iconType) { return; } icon.iconType = iconset + postfix; const sizes = FIREFOX || CHROME >= 2883 && !VIVALDI ? [16, 32] : [19, 38]; iconUtil.setIcon({ path: sizes.reduce( (obj, size) => { obj[size] = `/images/icon/${iconset}${size}${postfix}.png`; return obj; }, {} ), tabId }); } function refreshIconBadgeColor() { const color = prefs.get(prefs.get('disableAll') ? 'badgeDisabled' : 'badgeNormal'); iconUtil.setBadgeBackgroundColor({ color }); } function refreshAllIcons() { for (const [tabId, icon] of tabIcons) { refreshIcon(tabId, icon); } refreshIcon(null, {}); // default icon } function refreshAllIconsBadgeText() { for (const [tabId, icon] of tabIcons) { refreshIconBadgeText(tabId, icon); } } function onRuntimeMessage(msg, sender) { if (msg.method !== 'invokeAPI') { return; } const fn = window.API_METHODS[msg.name]; if (!fn) { throw new Error(`unknown API: ${msg.name}`); } const context = {msg, sender}; return fn.apply(context, msg.args); } function openEditor(params) { /* Open the editor. Activate if it is already opened params: { id?: Number, domain?: String, 'url-prefix'?: String } */ const searchParams = new URLSearchParams(); for (const key in params) { searchParams.set(key, params[key]); } const search = searchParams.toString(); return openURL({ url: 'edit.html' + (search && `?${search}`), newWindow: prefs.get('openEditInWindow'), windowPosition: prefs.get('windowPosition'), currentWindow: null }); } function openManage({options = false, search} = {}) { let url = chrome.runtime.getURL('manage.html'); if (search) { url += `?search=${encodeURIComponent(search)}`; } if (options) { url += '#stylus-options'; } return findExistTab({ url, currentWindow: null, ignoreHash: true, ignoreSearch: true }) .then(tab => { if (tab) { return Promise.all([ activateTab(tab), tab.url !== url && msg.sendTab(tab.id, {method: 'pushState', url}) .catch(console.error) ]); } return getActiveTab().then(tab => { if (isTabReplaceable(tab, url)) { return activateTab(tab, {url}); } return createTab({url}); }); }); }