Apq
/
violentmonkey
mirror of https://github.com/violentmonkey/violentmonkey.git


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144
							const { modifyWebpackConfig, shallowMerge, defaultOptions } = require('@gera2ld/plaid');
const { isProd } = require('@gera2ld/plaid/util');
const fs = require('fs');
const webpack = require('webpack');
const WrapperWebpackPlugin = require('wrapper-webpack-plugin');
const HTMLInlineCSSWebpackPlugin = isProd && require('html-inline-css-webpack-plugin').default;
const projectConfig = require('./plaid.conf');
const mergedConfig = shallowMerge(defaultOptions, projectConfig);

const INIT_FUNC_NAME = 'VMInitInjection';
// Copied from gulpfile.js: strip alphabetic suffix
const VM_VER = require('../package.json').version.replace(/-[^.]*/, '');

const pickEnvs = (items) => {
  return Object.assign({}, ...items.map(x => ({
    [`process.env.${x.key}`]: JSON.stringify(
      'val' in x ? x.val
        : process.env[x.key] ?? x.def
    ),
  })));
};

const definitions = new webpack.DefinePlugin({
  ...pickEnvs([
    { key: 'DEBUG', def: false },
    { key: 'VM_VER', val: VM_VER },
    { key: 'SYNC_GOOGLE_CLIENT_ID' },
    { key: 'SYNC_GOOGLE_CLIENT_SECRET' },
    { key: 'SYNC_ONEDRIVE_CLIENT_ID' },
    { key: 'SYNC_ONEDRIVE_CLIENT_SECRET' },
  ]),
  'process.env.INIT_FUNC_NAME': JSON.stringify(INIT_FUNC_NAME),
});

// avoid running webpack bootstrap in a potentially hacked environment
// after documentElement was replaced which triggered reinjection of content scripts
const skipReinjectionHeader = `if (window['${INIT_FUNC_NAME}'] !== 1)`;
// {entryName: path}
const entryGlobals = {
  common: './src/common/safe-globals.js',
  injected: './src/injected/safe-injected-globals.js',
};

/**
 * Adds a watcher for files in entryGlobals to properly recompile the project on changes.
 */
const addWrapper = (config, name, callback) => {
  if (!callback) { callback = name; name = ''; }
  const globals = Object.entries(entryGlobals).filter(([key]) => name === key || !name);
  const dirs = globals.map(([key]) => key).join('|');
  config.module.rules.push({
    test: new RegExp(`/(${dirs})/index\\.js$`.replace(/\//g, /[/\\]/.source)),
    use: [{
      loader: './scripts/fake-dep-loader.js',
      options: {
        files: globals.map(([, path]) => path),
      },
    }],
  });
  const reader = () => (
    globals.map(([, path]) => (
      fs.readFileSync(path, { encoding: 'utf8' })
      .replace(/export\s+(?=const\s)/g, '')
    ))
  ).join('\n');
  config.plugins.push(new WrapperWebpackPlugin(callback(reader)));
};

const modify = (page, entry, init) => modifyWebpackConfig(
  (config) => {
    config.node = {
      process: false,
      setImmediate: false,
    };
    config.plugins.push(definitions);
    if (!entry) init = page;
    if (init) init(config);
    return config;
  }, {
    projectConfig: {
      ...mergedConfig,
      ...entry && { pages: { [page]: { entry }} },
    },
  },
);

module.exports = Promise.all([
  modify((config) => {
    addWrapper(config, 'common', getGlobals => ({
      header: () => `{ ${getGlobals()}`,
      footer: '}',
      test: /^(?!injected|public).*\.js$/,
    }));
    /* Embedding as <style> to ensure uiTheme option doesn't cause FOUC.
     * Note that in production build there's no <head> in html but document.head is still
     * auto-created per the specification so our styles will be placed correctly anyway. */
    if (isProd) config.plugins.push(new HTMLInlineCSSWebpackPlugin({
      replace: {
        target: '<body>',
        position: 'before',
      },
    }));
    config.plugins.push(new class ListBackgroundScripts {
      apply(compiler) {
        compiler.hooks.afterEmit.tap(this.constructor.name, compilation => {
          const dist = compilation.outputOptions.path;
          const path = `${dist}/manifest.json`;
          const manifest = JSON.parse(fs.readFileSync(path, {encoding: 'utf8'}));
          const bgId = 'background/index';
          const bgEntry = compilation.entrypoints.get(bgId);
          const scripts = bgEntry.chunks.map(c => c.files[0]);
          if (`${manifest.background.scripts}` !== `${scripts}`) {
            manifest.background.scripts = scripts;
            fs.writeFileSync(path,
              JSON.stringify(manifest, null, isProd ? 0 : 2),
              {encoding: 'utf8'});
          }
          try {
            fs.unlinkSync(`${dist}/${bgId}.html`);
          } catch (e) {}
        });
      }
    });
  }),
  modify('injected', './src/injected', (config) => {
    addWrapper(config, getGlobals => ({
      header: () => `${skipReinjectionHeader} { ${getGlobals()}`,
      footer: '}',
    }));
  }),
  modify('injected-web', './src/injected/web', (config) => {
    config.output.libraryTarget = 'commonjs2';
    addWrapper(config, getGlobals => ({
      header: () => `${skipReinjectionHeader}
        window['${INIT_FUNC_NAME}'] = function () {
          var module = { exports: {} };
          ${getGlobals()}`,
      footer: `
          module = module.exports;
          return module.__esModule ? module.default : module;
        };0;`,
    }));
  }),
]);