violentmonkey/src/browser.js

/* global chrome */
const global = window;

function wrapAsync(func) {
  return (...args) => new Promise((resolve, reject) => {
    args.push(res => {
      const err = chrome.runtime.lastError;
      if (err) {
        console.error(args);
        reject(err);
      } else {
        resolve(res);
      }
    });
    func(...args);
  });
}
function wrapAPIs(source, meta) {
  const target = {};
  Object.keys(source).forEach(key => {
    const metaVal = meta && meta[key];
    if (metaVal) {
      const value = source[key];
      if (typeof metaVal === 'function') {
        target[key] = metaVal(value);
      } else if (typeof metaVal === 'object' && typeof value === 'object') {
        target[key] = wrapAPIs(value, metaVal);
      } else {
        target[key] = value;
      }
    }
  });
  return target;
}
const meta = {
  browserAction: true,
  i18n: true,
  notifications: {
    onClicked: true,
    onClosed: true,
    create: wrapAsync,
  },
  runtime: {
    getManifest: true,
    getURL: true,
    onMessage(onMessage) {
      function wrapListener(listener) {
        return function onChromeMessage(message, sender, sendResponse) {
          const result = listener(message, sender);
          if (result && typeof result.then === 'function') {
            result.then(data => {
              sendResponse({ data });
            }, error => {
              console.error(error);
              sendResponse({ error });
            });
            return true;
          } else if (typeof result !== 'undefined') {
            // In some browsers (e.g Chrome 56, Vivaldi), the listener in
            // popup pages are not properly cleared after closed.
            // They may send `undefined` before the real response is sent.
            sendResponse({ data: result });
          }
        };
      }
      return {
        addListener(listener) {
          return onMessage.addListener(wrapListener(listener));
        },
      };
    },
    sendMessage(sendMessage) {
      const promisifiedSendMessage = wrapAsync(sendMessage);
      return data => promisifiedSendMessage(data)
      .then(res => {
        if (res && res.error) throw res.error;
        return res && res.data;
      });
    },
  },
  tabs: {
    onUpdated: true,
    create: wrapAsync,
    get: wrapAsync,
    query: wrapAsync,
    reload: wrapAsync,
    remove: wrapAsync,
    sendMessage: wrapAsync,
    update: wrapAsync,
  },
  webRequest: true,
};
if (typeof browser === 'undefined' && typeof chrome !== 'undefined') {
  global.browser = wrapAPIs(chrome, meta);
  global.browser.__patched = true;
}