背景

1. 不知道webpack插件是怎麼回事，除了官方的文檔外，還有一個很直觀的方式，就是看源碼。
2. 看源碼是一個挖寶的行動，也是一次冒險，我們可以找一些代碼量不是很大的源碼
3. 比如webpack插件，我們就可以通過BannerPlugin源碼，來看下官方是如何實現一個插件的
4. 希望對各位同學有所幫助，必要時可以通過源碼進行一門技術的學習，加深理解

閑言少敘，直接上代碼

https://github.com/webpack/webpack/blob/main/lib/BannerPlugin.js

配合文檔api
https://webpack.docschina.org/api/compilation-object/#updateasset

代碼分析已添加中文註釋

/*
	MIT License http://www.opensource.org/licenses/mit-license.php
	Author Tobias Koppers @sokra
*/

"use strict";

const { ConcatSource } = require("webpack-sources");
const Compilation = require("./Compilation");
const ModuleFilenameHelpers = require("./ModuleFilenameHelpers");
const Template = require("./Template");
const createSchemaValidation = require("./util/create-schema-validation");

/** @typedef {import("../declarations/plugins/BannerPlugin").BannerPluginArgument} BannerPluginArgument */
/** @typedef {import("../declarations/plugins/BannerPlugin").BannerPluginOptions} BannerPluginOptions */
/** @typedef {import("./Compiler")} Compiler */

// 創建一個驗證
const validate = createSchemaValidation(
  require("../schemas/plugins/BannerPlugin.check.js"),
  () => require("../schemas/plugins/BannerPlugin.json"),
  {
    name: "Banner Plugin",
    baseDataPath: "options",
  }
);

//包裝Banner文字
const wrapComment = (str) => {
  if (!str.includes("\n")) {
    return Template.toComment(str);
  }
  return `/*!\n * ${str
    .replace(/\*\//g, "* /")
    .split("\n")
    .join("\n * ")
    .replace(/\s+\n/g, "\n")
    .trimRight()}\n */`;
};

//插件類
class BannerPlugin {
  /**
   * @param {BannerPluginArgument} options options object
   * 初始化插件配置
   */
  constructor(options) {
    if (typeof options === "string" || typeof options === "function") {
      options = {
        banner: options,
      };
    }

    validate(options);

    this.options = options;

    const bannerOption = options.banner;
    if (typeof bannerOption === "function") {
      const getBanner = bannerOption;
      this.banner = this.options.raw
        ? getBanner
        : (data) => wrapComment(getBanner(data));
    } else {
      const banner = this.options.raw
        ? bannerOption
        : wrapComment(bannerOption);
      this.banner = () => banner;
    }
  }

  /**
   * Apply the plugin
   * @param {Compiler} compiler the compiler instance
   * @returns {void}
   * 插件主方法
   */
  apply(compiler) {
    const options = this.options;
    const banner = this.banner;
    const matchObject = ModuleFilenameHelpers.matchObject.bind(
      undefined,
      options
    );
    //創建一個Map，處理如果添加過的文件，不在添加
    const cache = new WeakMap();

    compiler.hooks.compilation.tap("BannerPlugin", (compilation) => {
      //處理Assets的hook
      compilation.hooks.processAssets.tap(
        {
          name: "BannerPlugin",
          //PROCESS_ASSETS_STAGE_ADDITIONS — 為現有的 asset 添加額外的內容，例如 banner 或初始代碼。
          stage: Compilation.PROCESS_ASSETS_STAGE_ADDITIONS,
        },
        () => {
          //遍歷當前編譯對象的chunks
          for (const chunk of compilation.chunks) {
            //如果配置標識只處理入口，但是當前chunk不是入口，直接進入下一次迴圈
            if (options.entryOnly && !chunk.canBeInitial()) {
              continue;
            }
            //否則，遍歷chunk下的文件
            for (const file of chunk.files) {
              //根據配置匹配文件是否滿足要求，如果不滿足，直接進入下一次迴圈，處理下一個文件
              if (!matchObject(file)) {
                continue;
              }

              //否則，
              const data = {
                chunk,
                filename: file,
              };

              //獲取插值路徑？https://webpack.docschina.org/api/compilation-object/#getpath
              const comment = compilation.getPath(banner, data);

              //修改Asset，https://webpack.docschina.org/api/compilation-object/#updateasset
              compilation.updateAsset(file, (old) => {
                //從緩存中獲取
                let cached = cache.get(old);
                //如果緩存不存在 或者緩存的comment 不等於當前的comment
                if (!cached || cached.comment !== comment) {
                  //源文件追加到頭部或者尾部
                  const source = options.footer
                    ? new ConcatSource(old, "\n", comment)
                    : new ConcatSource(comment, "\n", old);
                  //創建對象加到緩存
                  cache.set(old, { source, comment });

                  //返回修改後的源
                  return source;
                }
                //返回緩存中的源
                return cached.source;
              });
            }
          }
        }
      );
    });
  }
}

module.exports = BannerPlugin;

總結

1. 查看源碼，查看源碼，查看源碼
2. WeakMap可以深入瞭解下，應該是避免對象不釋放導致記憶體問題。
3. 插件里用到的很多工具方法可以繼續深入，一遍自己開發插件時可以參考