/* global dbExec, getStyles, saveStyle */ /* global handleCssTransitionBug */ /* global usercssHelper openEditor */ /* global styleViaAPI */ 'use strict'; // eslint-disable-next-line no-var var browserCommands, contextMenus; // ************************************************************************* // register all listeners chrome.runtime.onMessage.addListener(onRuntimeMessage); { const [listener] = [ [webNavigationListenerChrome, CHROME], [webNavigationListenerFF, FIREFOX], [webNavigationListener, true], ].find(([, selected]) => selected); chrome.webNavigation.onBeforeNavigate.addListener(data => listener(null, data)); chrome.webNavigation.onCommitted.addListener(data => listener('styleApply', data)); chrome.webNavigation.onHistoryStateUpdated.addListener(data => listener('styleReplaceAll', data)); chrome.webNavigation.onReferenceFragmentUpdated.addListener(data => listener('styleReplaceAll', data)); if (FIREFOX) { // FF applies page CSP even to content scripts, https://bugzil.la/1267027 chrome.webNavigation.onCommitted.addListener(webNavUsercssInstallerFF, { url: [ {urlPrefix: 'https://raw.githubusercontent.com/', urlSuffix: '.user.css'}, {urlPrefix: 'https://raw.githubusercontent.com/', urlSuffix: '.user.styl'}, ] }); } } 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]()); } if (!chrome.browserAction || !['setIcon', 'setBadgeBackgroundColor', 'setBadgeText'].every(name => chrome.browserAction[name])) { window.updateIcon = () => {}; } // ************************************************************************* // set the default icon displayed after a tab is created until webNavigation kicks in prefs.subscribe(['iconset'], () => updateIcon({id: undefined}, {})); // ************************************************************************* { const onInstall = ({reason}) => { chrome.runtime.onInstalled.removeListener(onInstall); const manifest = chrome.runtime.getManifest(); // Open FAQs page once after installation to guide new users. // Do not display it in development mode. if (reason === 'install' && manifest.update_url) { // don't hardcode homepage URL, extract it from "Get Help" label translation // TODO: add a built-in tour page in the extension const getHelpHtml = chrome.i18n.getMessage('manageText').match(/]+/g); const url = (getHelpHtml[1] || '').replace(/^.+?=\s*/, '').replace(/^['"]|["']$/g, ''); if (url) { setTimeout(openURL, 100, {url}); } } // reset L10N cache on update if (reason === 'update') { localStorage.L10N = JSON.stringify({ browserUIlanguage: chrome.i18n.getUILanguage(), }); } if (!FIREFOX && chrome.declarativeContent) { chrome.declarativeContent.onPageChanged.removeRules(null, () => { chrome.declarativeContent.onPageChanged.addRules([{ conditions: [ new chrome.declarativeContent.PageStateMatcher({ pageUrl: {urlContains: ':'}, }) ], actions: [ new chrome.declarativeContent.RequestContentScript({ js: ['/content/apply.js'], allFrames: true, matchAboutBlank: true, }), ], }]); }); } }; // bind for 60 seconds max and auto-unbind if it's a normal run chrome.runtime.onInstalled.addListener(onInstall); setTimeout(onInstall, 60e3, {reason: 'unbindme'}); } // ************************************************************************* // browser commands browserCommands = { openManage() { openURL({url: 'manage.html'}); }, styleDisableAll(info) { prefs.set('disableAll', info ? info.checked : !prefs.get('disableAll')); }, }; // ************************************************************************* // 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, }, 'editor.contextDelete': { presentIf: () => !FIREFOX && prefs.get('editor.contextDelete'), title: 'editDeleteText', type: 'normal', contexts: ['editable'], documentUrlPatterns: [URLS.ownOrigin + 'edit*'], click: (info, tab) => { sendMessage({tabId: tab.id, method: 'editDeleteText'}); }, } }; if (chrome.contextMenus) { const 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; const prefValue = prefs.readOnlyValues[id]; item.title = chrome.i18n.getMessage(item.title); if (!item.type && typeof prefValue === 'boolean') { item.type = 'checkbox'; item.checked = prefValue; } if (!item.contexts) { item.contexts = ['browser_action']; } delete item.click; chrome.contextMenus.create(item, ignoreChromeError); } }; // circumvent the bug with disabling check marks in Chrome 62+ // TODO: replace 1e6 with the actual rev. number when/if the bug is fixed const toggleCheckmark = CHROME >= 3172 && CHROME <= 1e6 ? (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.readOnlyValues[id] === 'boolean'), toggleCheckmark); prefs.subscribe(keys.filter(id => contextMenus[id].presentIf), togglePresence); createContextMenus(keys); } // ************************************************************************* // [re]inject content scripts window.addEventListener('storageReady', function _() { window.removeEventListener('storageReady', _); updateIcon({id: undefined}, {}); if (FIREFOX) { queryTabs().then(tabs => tabs.forEach(tab => { if (!tab.width) { // skip lazy-loaded tabs (width = 0) that seem to start loading on message return; } const tabId = tab.id; const frameUrls = {0: tab.url}; styleViaAPI.allFrameUrls.set(tabId, frameUrls); chrome.webNavigation.getAllFrames({tabId}, frames => frames && frames.forEach(({frameId, parentFrameId, url}) => { if (frameId) { frameUrls[frameId] = url === 'about:blank' ? frameUrls[parentFrameId] : url; } })); })); return; } const NTP = 'chrome://newtab/'; const ALL_URLS = ''; const contentScripts = chrome.runtime.getManifest().content_scripts; contentScripts.push({ js: ['content/apply.js'], matches: [''], run_at: 'document_start', match_about_blank: true, all_frames: true }); // expand * as .*? const wildcardAsRegExp = (s, flags) => new RegExp( s.replace(/[{}()[\]/\\.+?^$:=!|]/g, '\\$&') .replace(/\*/g, '.*?'), flags); for (const cs of contentScripts) { cs.matches = cs.matches.map(m => ( m === ALL_URLS ? m : wildcardAsRegExp(m) )); } const injectCS = (cs, tabId) => { chrome.tabs.executeScript(tabId, { file: cs.js[0], runAt: cs.run_at, allFrames: cs.all_frames, matchAboutBlank: cs.match_about_blank, }, ignoreChromeError); }; const pingCS = (cs, {id, url}) => { const maybeInject = pong => !pong && injectCS(cs, id); cs.matches.some(match => { if ((match === ALL_URLS || url.match(match)) && (!url.startsWith('chrome') || url === NTP)) { sendMessage({method: 'ping', tabId: id}, maybeInject); return true; } }); }; queryTabs().then(tabs => tabs.forEach(tab => tab.width && contentScripts.forEach(cs => setTimeout(pingCS, 0, cs, tab)))); }); // ************************************************************************* function webNavigationListener(method, {url, tabId, frameId}) { getStyles({matchUrl: url, enabled: true, asHash: true}).then(styles => { if (method && URLS.supported(url) && tabId >= 0) { if (method === 'styleApply') { handleCssTransitionBug({tabId, frameId, url, styles}); } sendMessage({ tabId, frameId, method, // ping own page so it retrieves the styles directly styles: url.startsWith(URLS.ownOrigin) ? 'DIY' : styles, }); } // main page frame id is 0 if (frameId === 0) { updateIcon({id: tabId, url}, styles); } }); } function webNavigationListenerChrome(method, data) { const {tabId, frameId, url} = data; if (url.startsWith('https://www.google.') && url.includes('/_/chrome/newtab?')) { // Chrome 61.0.3161+ doesn't run content scripts on NTP getTab(tabId).then(tab => { data.url = tab.url === 'chrome://newtab/' ? tab.url : url; webNavigationListener(method, data); }); } else { webNavigationListener(method, data); // chrome.declarativeContent doesn't inject scripts in about:blank iframes if (method && frameId && url === 'about:blank') { chrome.tabs.executeScript(tabId, { file: '/content/apply.js', runAt: 'document_start', matchAboutBlank: true, frameId, }, ignoreChromeError); } } } function webNavigationListenerFF(method, data) { const {tabId, frameId, url} = data; //console.log(method, data); if (frameId === 0 || url !== 'about:blank') { if ((!method || method === 'styleApply') && styleViaAPI.getFrameUrl(tabId, frameId) !== url) { styleViaAPI.cache.delete(tabId); } styleViaAPI.setFrameUrl(tabId, frameId, url); webNavigationListener(method, data); return; } //const frames = styleViaAPI.allFrameUrls.get(tabId); //if (Object.keys(frames).length === 1) { // frames[frameId] = frames['0']; // webNavigationListener(method, data); // return; //} //chrome.webNavigation.getFrame({tabId, frameId}, info => { // const hasParent = !chrome.runtime.lastError && info.parentFrameId >= 0; // frames[frameId] = hasParent ? frames[info.parentFrameId] : url; // webNavigationListener(method, data); //}); } function webNavUsercssInstallerFF(data) { const {tabId} = data; // 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(tab => { if (tab.url !== 'about:blank') { usercssHelper.openInstallPage(tab, {direct: true}); } }); } function updateIcon(tab, styles) { if (tab.id < 0) { return; } if (URLS.chromeProtectsNTP && tab.url === 'chrome://newtab/') { styles = {}; } if (styles) { stylesReceived(styles); return; } getTabRealURL(tab) .then(url => getStyles({matchUrl: url, enabled: true, asHash: true})) .then(stylesReceived); function stylesReceived(styles) { let numStyles = styles.length; if (numStyles === undefined) { // for 'styles' asHash:true fake the length by counting numeric ids manually numStyles = 0; for (const id of Object.keys(styles)) { numStyles += id.match(/^\d+$/) ? 1 : 0; } } const disableAll = 'disableAll' in styles ? styles.disableAll : prefs.get('disableAll'); const postfix = disableAll ? 'x' : numStyles === 0 ? 'w' : ''; const color = prefs.get(disableAll ? 'badgeDisabled' : 'badgeNormal'); const text = prefs.get('show-badge') && numStyles ? String(numStyles) : ''; const iconset = ['', 'light/'][prefs.get('iconset')] || ''; const path = 'images/icon/' + iconset; chrome.browserAction.setIcon({ tabId: tab.id, path: { // Material Design 2016 new size is 16px 16: `${path}16${postfix}.png`, 32: `${path}32${postfix}.png`, // Chromium forks or non-chromium browsers may still use the traditional 19px 19: `${path}19${postfix}.png`, 38: `${path}38${postfix}.png`, // TODO: add Edge preferred sizes: 20, 25, 30, 40 }, }, () => { if (chrome.runtime.lastError || tab.id === undefined) { return; } // Vivaldi bug workaround: setBadgeText must follow setBadgeBackgroundColor chrome.browserAction.setBadgeBackgroundColor({color}); setTimeout(() => { getTab(tab.id).then(realTab => { // skip pre-rendered tabs if (realTab.index >= 0) { chrome.browserAction.setBadgeText({text, tabId: tab.id}); } }); }); }); } } function onRuntimeMessage(request, sender, sendResponseInternal) { const sendResponse = data => { // wrap Error object instance as {__ERROR__: message} - will be unwrapped in sendMessage if (data instanceof Error) { data = {__ERROR__: data.message}; } // prevent browser exception bug on sending a response to a closed tab tryCatch(sendResponseInternal, data); }; switch (request.method) { case 'getStyles': getStyles(request).then(sendResponse); return KEEP_CHANNEL_OPEN; case 'saveStyle': saveStyle(request).then(sendResponse); return KEEP_CHANNEL_OPEN; case 'saveUsercss': usercssHelper.save(request, true).then(sendResponse); return KEEP_CHANNEL_OPEN; case 'buildUsercss': usercssHelper.build(request, true).then(sendResponse); return KEEP_CHANNEL_OPEN; case 'healthCheck': dbExec() .then(() => sendResponse(true)) .catch(() => sendResponse(false)); return KEEP_CHANNEL_OPEN; case 'download': download(request.url) .then(sendResponse) .catch(() => sendResponse(null)); return KEEP_CHANNEL_OPEN; case 'openUsercssInstallPage': usercssHelper.openInstallPage(sender.tab, request).then(sendResponse); return KEEP_CHANNEL_OPEN; case 'closeTab': chrome.tabs.remove(request.tabId || sender.tab.id, () => { if (chrome.runtime.lastError && request.tabId !== sender.tab.id) { sendResponse(new Error(chrome.runtime.lastError.message)); } }); return KEEP_CHANNEL_OPEN; case 'openEditor': openEditor(request.id); return; } }