@lzwme/feps-webpack-plugin/lib/FEPSPlugin.js

This plug-in is used for function execution performance statistics. It calculates the execution time by injecting statistical code and finds slow functions.

/*
 * @Author: lzw
 * @Date: 2021-12-13 09:44:19
 * @LastEditors: lzw
 * @LastEditTime: 2022-01-07 11:00:07
 * @Description:
 */
// @ts-check

const FEPSDependency = require('./FEPSDependency');
const comm = require('./common');
const PLUGIN_NAME = comm.PLUGIN_NAME;

let totalFiles = 0;

module.exports = class FEPSPlugin {
  options = comm.getDefaultOptions();
  stats = { total: 0, matchFiles: 0 };
  /**
   * @param {comm.FepsOptions} options
   */
  constructor(options = null) {
    if (options && options !== this.options) {
      Object.keys(this.options).forEach(key => {
        if (options[key] != null) this.options[key] = options[key];
      });
      this.options.timeLimit = Math.max(10, +this.options.timeLimit || 0);
      this.options.minFnCodeLength = Math.max(100, +this.options.minFnCodeLength || 0);
    }
  }
  /**
   * @param {import('webpack').Compiler} compiler
   */
  apply(compiler) {
    if (this.options.disabled === true) return;

    compiler.hooks.compilation.tap(PLUGIN_NAME, (compilation) => {
      // @ts-ignore
      compilation.dependencyTemplates.set(FEPSDependency, new FEPSDependency.Template());
    });

    compiler.hooks.normalModuleFactory.tap(PLUGIN_NAME, factory => {
      factory.hooks.parser.for('javascript/auto').tap(PLUGIN_NAME, parser => this.handler(parser, compiler.name));
    });
  }
  /**
   * @param {comm.JavascriptParser} parser
   * @param {string} desc
   */
  handler(parser, desc) {
    comm.astHandler(parser, this.options, (isMatch, ast, relativePath) => {
      totalFiles++;
      this.stats.total++;

      if (isMatch) {
        this.stats.matchFiles++;
        parser.state.module.ast = ast;
        parser.state.module.addDependency(new FEPSDependency(relativePath, this.options, parser.state.module));
      }

      if (this.options.debug) {
        const { gray, green, yellow, yellowBright, cyan } = comm.getColors();
        console.log(
          gray(`[${PLUGIN_NAME}]`+ (desc ? `[${desc}]` : '')),
          isMatch ? green('[matched]') : yellow('[ignored]'),
          yellowBright(`${this.stats.matchFiles}/${this.stats.total}`),
          'totalFiles:',
          totalFiles,
          cyan(relativePath)
        );
      }
    });
  }
};