stylus/messaging.js

/* global getStyleWithNoCode, applyOnMessage, onBackgroundMessage, getStyles */
'use strict';

// keep message channel open for sendResponse in chrome.runtime.onMessage listener
const KEEP_CHANNEL_OPEN = true;
const OWN_ORIGIN = chrome.runtime.getURL('');


function notifyAllTabs(request) {
  // list all tabs including chrome-extension:// which can be ours
  if (request.codeIsUpdated === false && request.style) {
    request = Object.assign({}, request, {
      style: getStyleWithNoCode(request.style)
    });
  }
  chrome.tabs.query({}, tabs => {
    for (const tab of tabs) {
      chrome.tabs.sendMessage(tab.id, request);
      updateIcon(tab);
    }
  });
  // notify all open popups
  const reqPopup = Object.assign({}, request, {method: 'updatePopup', reason: request.method});
  chrome.runtime.sendMessage(reqPopup);
  // notify self: the message no longer is sent to the origin in new Chrome
  if (typeof applyOnMessage !== 'undefined') {
    applyOnMessage(reqPopup);
  }
  // notify self: pref changed by background page
  if (request.method == 'prefChanged' && typeof onBackgroundMessage !== 'undefined') {
    onBackgroundMessage(request);
  }
}


function refreshAllTabs() {
  return new Promise(resolve => {
    // list all tabs including chrome-extension:// which can be ours
    chrome.tabs.query({}, tabs => {
      const lastTab = tabs[tabs.length - 1];
      for (const tab of tabs) {
        getStyles({matchUrl: tab.url, enabled: true, asHash: true}, styles => {
          const message = {method: 'styleReplaceAll', styles};
          if (tab.url == location.href && typeof applyOnMessage !== 'undefined') {
            applyOnMessage(message);
          } else {
            chrome.tabs.sendMessage(tab.id, message);
          }
          updateIcon(tab, styles);
          if (tab == lastTab) {
            resolve();
          }
        });
      }
    });
  });
}


function updateIcon(tab, styles) {
  // while NTP is still loading only process the request for its main frame with a real url
  // (but when it's loaded we should process style toggle requests from popups, for example)
  if (tab.url == 'chrome://newtab/' && tab.status != 'complete') {
    return;
  }
  if (styles) {
    // check for not-yet-existing tabs e.g. omnibox instant search
    chrome.tabs.get(tab.id, () => {
      if (!chrome.runtime.lastError) {
        stylesReceived(styles);
      }
    });
    return;
  }
  getTabRealURL(tab).then(url => {
    // if we have access to this, call directly
    // (Chrome no longer sends messages to the page itself)
    const options = {method: 'getStyles', matchUrl: url, enabled: true, asHash: true};
    if (typeof getStyles != 'undefined') {
      getStyles(options, stylesReceived);
    } else {
      chrome.runtime.sendMessage(options, 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' : '';
    chrome.browserAction.setIcon({
      path: {
        // Material Design 2016 new size is 16px
        16: `/images/icon/16${postfix}.png`, 32: `/images/icon/32${postfix}.png`,
        // Chromium forks or non-chromium browsers may still use the traditional 19px
        19: `/images/icon/19${postfix}.png`, 38: `/images/icon/38${postfix}.png`,
      },
      tabId: tab.id
    }, () => {
      // if the tab was just closed an error may occur,
      // e.g. 'windowPosition' pref updated in edit.js::window.onbeforeunload
      if (!chrome.runtime.lastError) {
        const text = prefs.get('show-badge') && numStyles ? String(numStyles) : '';
        chrome.browserAction.setBadgeText({text, tabId: tab.id});
        chrome.browserAction.setBadgeBackgroundColor({
          color: prefs.get(disableAll ? 'badgeDisabled' : 'badgeNormal')
        });
      }
    });
  }
}


function getActiveTab() {
  return new Promise(resolve =>
    chrome.tabs.query({currentWindow: true, active: true}, tabs =>
      resolve(tabs[0])));
}


function getActiveTabRealURL() {
  return getActiveTab()
    .then(getTabRealURL);
}


function getTabRealURL(tab) {
  return new Promise(resolve => {
    if (tab.url != 'chrome://newtab/') {
      resolve(tab.url);
    } else {
      chrome.webNavigation.getFrame({tabId: tab.id, frameId: 0, processId: -1}, frame => {
        frame && resolve(frame.url);
      });
    }
  });
}


// opens a tab or activates the already opened one,
// reuses the New Tab page if it's focused now
function openURL({url, currentWindow = true}) {
  if (!url.includes('://')) {
    url = chrome.runtime.getURL(url);
  }
  return new Promise(resolve => {
    // [some] chromium forks don't handle their fake branded protocols
    url = url.replace(/^(opera|vivaldi)/, 'chrome');
    // API doesn't handle the hash-fragment part
    chrome.tabs.query({url: url.replace(/#.*/, ''), currentWindow}, tabs => {
      for (const tab of tabs) {
        if (tab.url == url) {
          activateTab(tab).then(resolve);
          return;
        }
      }
      getActiveTab().then(tab => (
        tab && tab.url == 'chrome://newtab/'
          ? chrome.tabs.update({url}, resolve)
          : chrome.tabs.create({url}, resolve)
      ));
    });
  });
}


function activateTab(tab) {
  return Promise.all([
    new Promise(resolve => {
      chrome.tabs.update(tab.id, {active: true}, resolve);
    }),
    new Promise(resolve => {
      chrome.windows.update(tab.windowId, {focused: true}, resolve);
    }),
  ]);
}


function stringAsRegExp(s, flags) {
  return new RegExp(s.replace(/[{}()[\]/\\.+?^$:=*!|]/g, '\\$&'), flags);
}


// expands * as .*?
function wildcardAsRegExp(s, flags) {
  return new RegExp(s.replace(/[{}()[\]/\\.+?^$:=!|]/g, '\\$&').replace(/\*/g, '.*?'), flags);
}


// isolate deoptimization trigger:
// https://github.com/petkaantonov/bluebird/wiki/Optimization-killers
// * Functions that contain object literals that contain __proto__, or get or set declarations.
const configureCommands = (() => ({
  get url() {
    return navigator.userAgent.includes('OPR')
      ? 'opera://settings/configureCommands'
      : 'chrome://extensions/configureCommands';
  },
  open: () => {
    chrome.tabs.create({
      'url': configureCommands.url
    });
  }
}))();