80行代碼寫一個Webpack插件并發(fā)布到npm

更新時間：2021年05月24日 11:30:05 作者：Yuan-yiming

最近在學習 Webpack 相關的原理，本文用80行代碼寫一個Webpack插件并發(fā)布到npm，具有一定的參考價值，感興趣的小伙伴們可以參考一下

1. 前言

最近在學習 Webpack 相關的原理，以前只知道 Webpack 的配置方法，但并不知道其內部流程，經(jīng)過一輪的學習，感覺獲益良多，為了鞏固學習的內容，我決定嘗試自己動手寫一個插件。

這個插件實現(xiàn)的功能比較簡單：

默認清除 js 代碼中的 console.log 的打印輸出；
可通過傳入配置，實現(xiàn)移除 console 的其它方法，如 console.warn、console.error 等；

2. Webpack 的構建流程以及 plugin 的原理

2.1 Webpack 構建流程

Webpack 的主要構建流程，可以分為三個階段:

初始化階段：啟動構建，讀取與合并配置參數(shù)，加載 Plugin，實例化 Compiler。
編譯階段：從 Entry 發(fā)出，針對每個 Module 串行調用對應的 Loader 去翻譯文件內容，再找到該 Module 依賴的 Module，遞歸地進行編譯處理。
生成階段：對編譯后的 Module 組合成 Chunk，把 Chunk 轉換成文件，輸出到文件系統(tǒng)。

如果 Webpack 打包生產(chǎn)環(huán)境文件時，只會執(zhí)行一次構建，以上階段會按順序執(zhí)行一遍。但是在開啟監(jiān)聽模式時，如開發(fā)環(huán)境，Webpack 會持續(xù)的進行構建。

2.2 plugin 原理

Webpack 插件通常是一個帶有 apply 函數(shù)的類，其中 constructor 可以接收傳入的配置項。插件被安裝時，apply 函數(shù)會被調用一次，并接收 Compiler 對象，然后我們可以在 Compiler 對象上監(jiān)聽不同的事件鉤子，從而進行插件功能的開發(fā)。

// 定義一個插件
class MyPlugin {
  // 構造函數(shù)，接收插件的配置項 options 
  constructor(options) {
    // 獲取配置項，初始化插件
  }

  // 插件安裝時會調用 apply，并傳入 compiler
  apply(compiler) {
    // 獲取 comolier 獨享，可以監(jiān)聽事件鉤子
    // 功能開發(fā) ... 
  }
}

2.3 compiler 和 compilation 對象

在開發(fā) Plugin 過程中最常用的兩個對象就是 Compiler 和 Compilation：

Compiler 對象在 Webpack 啟動時被實例化，該對象包含了 Webpack 環(huán)境所有的配置信息，包括 options、loaders、plugins 等。在整個 Webpack 構建過程中，Compiler 對象是全局唯一的，它提供了很多事件鉤子回調供插件使用。
Compilation 對象包含了當前的模塊資源、編譯生成資源、變化的文件等。Compilation 對象在 Webpack 構建過程中并不是唯一的，如果在開發(fā)模式下 Webpack 開啟了文件檢測功能，每當文件變化時，Webpack 會重新構建，此時會生成一個新的 Compilation 對象。Compilation 對象也提供了很多事件回調供插件做擴展。

3. 插件開發(fā)

3.1 項目目錄

該插件實現(xiàn)的功能比較簡單，文件目錄也不復雜。首先新建一個空文件夾 remove-console-Webpack-plugin，并在該文件夾目錄下運行 npm init，根據(jù)提示來填寫 package.json 相關信息。然后再新建一個 src 文件夾，插件主要代碼就放在 src/index.js 里面。如果你需要把項目放到 github 上，最好也添加一下 .gitignore、README.md 等文件。

// remove-console-Webpack-plugin
├─src
│  └─index.js  
├─.gitignore
├─package.json
└─README.md

3.2 插件代碼

插件代碼邏輯也并不復雜，主要有幾點：

在構造函數(shù)中接收配置參數(shù)，并對參數(shù)進行合并，得到需要清除的 console 函數(shù), 存放在 removed 數(shù)組中；
在 apply 函數(shù)中監(jiān)聽 compiler.hook.compilation 鉤子，該鉤子觸發(fā)后，拿到 compilation 后進一步監(jiān)聽它的鉤子，這里 Webpack4 和 Webpack5 的鉤子不一樣，需要做兼容；
定義 assetsHandler 方法來處理 js 文件，利用正則表達式清除 removed 中包括的 console 函數(shù)；

class RemoveConsoleWebpackPlugin {
  // 構造函數(shù)接受配置參數(shù)
  constructor(options) {
    let include = options && options.include;
    let removed = ['log']; // 默認清除的方法

    if (include) {
      if (!Array.isArray(include)) {
        console.error('options.include must be an Array.');
      } else if (include.includes('*')) {
        // 傳入 * 表示清除所有 console 的方法
        removed = Object.keys(console).filter(fn => {
          return typeof console[fn] === 'function';
        })
      } else {
        removed = include; // 根據(jù)傳入配置覆蓋
      }
    }

    this.removed = removed;
  }

  // Webpack 會調用插件實例的 apply 方法，并傳入compiler 對象
  apply(compiler) {
    // js 資源代碼處理函數(shù)
    let assetsHandler = (assets, compilation) => {
      let removedStr = this.removed.reduce((a, b) => (a + '|' + b));

      let reDict = {
        1: [RegExp(`\\.console\\.(${removedStr})\\(\\)`, 'g'), ''],
        2: [RegExp(`\\.console\\.(${removedStr})\\(`, 'g'), ';('],
        3: [RegExp(`console\\.(${removedStr})\\(\\)`, 'g'), ''],
        4: [RegExp(`console\\.(${removedStr})\\(`, 'g'), '(']
      }

      Object.entries(assets).forEach(([filename, source]) => {
        // 匹配js文件
        if (/\.js$/.test(filename)) {
          // 處理前文件內容
          let outputContent = source.source();

          Object.keys(reDict).forEach(i => {
            let [re, s] = reDict[i];
            outputContent = outputContent.replace(re, s);
          })

          compilation.assets[filename] = {
            // 返回文件內容
            source: () => {
              return outputContent
            },
            // 返回文件大小
            size: () => {
              return Buffer.byteLength(outputContent, 'utf8')
            }
          }
        }
      })
    }

    /**
     * 通過 compiler.hooks.compilation.tap 監(jiān)聽事件
     * 在回調方法中獲取到 compilation 對象
     */
    compiler.hooks.compilation.tap('RemoveConsoleWebpackPlugin',
      compilation => {
        // Webpack 5
        if (compilation.hooks.processAssets) {
          compilation.hooks.processAssets.tap(
            { name: 'RemoveConsoleWebpackPlugin' },
            assets => assetsHandler(assets, compilation)
          );
        } else if (compilation.hooks.optimizeAssets) {
          // Webpack 4
          compilation.hooks.optimizeAssets.tap(
            'RemoveConsoleWebpackPlugin', 
            assets => assetsHandler(assets, compilation)
          );
        }
      })
  }
}

// export Plugin
module.exports = RemoveConsoleWebpackPlugin;