概述

本文次要的内容是通过之前应用的prerender-spa-plugin插件的源码浏览,来看下咱们应该如何编写一个webpack的插件,同时理解下预渲染插件到底是如何实现的。

这个内容其实曾经在应用prerender-spa-plugin外面有所波及了,这一章的内容算是对之前一篇文章的补充和拓展,具体介绍下Webpack的插件机制到底是如何运行的,之前写的简略的替换的插件失效的原理到底是什么。

如果大家还没有看之前的如何应用prerender-spa-plugin插件对页面进行预渲染这篇文章,能够先去看看,理解下这个插件到底是做什么的,咱们的插件大略是什么样的。

插件源码浅析

prerender-spa-plugin是开源的,源码在GitHub下面能够看到,大家如果有趣味的话,能够本人点击看下。

首先,咱们让咱们来简略回顾下这个插件是如何应用的,这个对于咱们理解其外部结构,有肯定的帮忙。咱们就间接应用它官网文档上提供的一个例子。

const path = require('path')const PrerenderSPAPlugin = require('prerender-spa-plugin')module.exports = {  plugins: [    ...    new PrerenderSPAPlugin({      // Required - The path to the webpack-outputted app to prerender.      staticDir: path.join(__dirname, 'dist'),      // Required - Routes to render.      routes: [ '/', '/about', '/some/deep/nested/route' ],    })  ]}

从下面这个例子来看,咱们能够晓得这个插件须要初始化一个实例,而后传入对应的参数如输入的门路staticDir、须要渲染的路由routes等。

接下来,让咱们来简略介绍下他的源码构造。代码具体分块如下:

function PrerenderSPAPlugin (...args) {  ...}PrerenderSPAPlugin.prototype.apply = function (compiler) {  const afterEmit = (compilation, done) => {    ...  }    if (compiler.hooks) {    const plugin = { name: 'PrerenderSPAPlugin' }    compiler.hooks.afterEmit.tapAsync(plugin, afterEmit)  } else {    compiler.plugin('after-emit', afterEmit)  }}

整个prerender-spa-plugin的插件是由2大部分形成:

  1. 一个function函数,次要用于初始化数据获取与解决。在应用这个插件的过程中,咱们须要先进行初始化。这个函数能够用来进行一些数据的解决和解析。
  2. 一个原型上的apply函数,作为一个钩子函数,次要用于解决Webpack触发插件执行后,相干逻辑的解决。

上面,咱们就基于prerender-spa-plugin插件,来一个一个局部的看下。

初始化function函数

首先让咱们来看下初始化的function函数。这个函数次要做的是一些初始化参数获取后的解决。具体代码如下:

function PrerenderSPAPlugin (...args) {  const rendererOptions = {} // Primarily for backwards-compatibility.  this._options = {}  // Normal args object.  if (args.length === 1) {    this._options = args[0] || {}  // Backwards-compatibility with v2  } else {    console.warn("[prerender-spa-plugin] You appear to be using the v2 argument-based configuration options. It's recommended that you migrate to the clearer object-based configuration system.\nCheck the documentation for more information.")    let staticDir, routes    args.forEach(arg => {      if (typeof arg === 'string') staticDir = arg      else if (Array.isArray(arg)) routes = arg      else if (typeof arg === 'object') this._options = arg    })    staticDir ? this._options.staticDir = staticDir : null    routes ? this._options.routes = routes : null  }  // Backwards compatiblity with v2.  if (this._options.captureAfterDocumentEvent) {    console.warn('[prerender-spa-plugin] captureAfterDocumentEvent has been renamed to renderAfterDocumentEvent and should be moved to the renderer options.')    rendererOptions.renderAfterDocumentEvent = this._options.captureAfterDocumentEvent  }  if (this._options.captureAfterElementExists) {    console.warn('[prerender-spa-plugin] captureAfterElementExists has been renamed to renderAfterElementExists and should be moved to the renderer options.')    rendererOptions.renderAfterElementExists = this._options.captureAfterElementExists  }  if (this._options.captureAfterTime) {    console.warn('[prerender-spa-plugin] captureAfterTime has been renamed to renderAfterTime and should be moved to the renderer options.')    rendererOptions.renderAfterTime = this._options.captureAfterTime  }  this._options.server = this._options.server || {}  this._options.renderer = this._options.renderer || new PuppeteerRenderer(Object.assign({}, { headless: true }, rendererOptions))  if (this._options.postProcessHtml) {    console.warn('[prerender-spa-plugin] postProcessHtml should be migrated to postProcess! Consult the documentation for more information.')  }}

因为咱们的插件应用的形式是实例化后增加(即new操作符实例化后应用),所以function函数的入参次要是将一些须要的参数绑定到this对象上,这样实例化后,就能够获取相干的参数。

很多的SDK或者说插件相干的工具,因为可能承受多种类型、不同长度的入参,因而会在一开始对参数类型进行判断,确定传入的参数类型到底是哪一种。

从代码中看,目前记录的参数有输入的参数staticDir、须要渲染的路由routes。如果本人定义了renderer函数,那么也绑定存储下来。同时,这个V3版本的代码还对V2版本进行了向前兼容。

钩子apply函数

说完了初始化的function,咱们来看下最重要的apply函数。具体代码如下:

PrerenderSPAPlugin.prototype.apply = function (compiler) {  const compilerFS = compiler.outputFileSystem  // From https://github.com/ahmadnassri/mkdirp-promise/blob/master/lib/index.js  const mkdirp = function (dir, opts) {    return new Promise((resolve, reject) => {      compilerFS.mkdirp(dir, opts, (err, made) => err === null ? resolve(made) : reject(err))    })  }  const afterEmit = (compilation, done) => {    ...  }  if (compiler.hooks) {    const plugin = { name: 'PrerenderSPAPlugin' }    compiler.hooks.afterEmit.tapAsync(plugin, afterEmit)  } else {    compiler.plugin('after-emit', afterEmit)  }}

在说apply函数之前,咱们先看下apply函数接管的参数compiler对象和mkdirp这个办法,以及生命周期绑定的代码。

complier对象

整个apply办法,接管的参数只有一个complier对象,具体的内容咱们能够看webpack中对于complier对象的形容,具体的源码能够见此处。我上面简略介绍下:

complier对象是webpack提供的一个全局的对象,这个对象下面挂载了一些在插件生命周期中会应用到的性能和属性,比方options、loader、plugin等。咱们能够通过这个对象,在构建中获取webpack相干的数据。

mkdirp办法

这个办法就是将执行mkdir -p办法的函数转化成一个Promise对象。具体能够看代码下面的原文正文。因为比较简单,这里我就不过多介绍了。

生命周期绑定

在最初,钩子函数生命实现后,须要将其关联到最近的生命周期上。这个插件关联的是afterEmit这个节点,大家如果想看下整个webpack相干构建流程的生命周期,能够参考这个文档。

看完了简略的局部,上面咱们来看下最重点的钩子函数。

钩子函数

接下来,让咱们来看下这个插件中最外围的钩子函数。这个插件的关联的申明周期是afterEmit这个节点,接下来咱们来看下具体的代码。

const afterEmit = (compilation, done) => {  const PrerendererInstance = new Prerenderer(this._options);  PrerendererInstance.initialize()    .then(() => {      return PrerendererInstance.renderRoutes(this._options.routes || []);    })    // Backwards-compatibility with v2 (postprocessHTML should be migrated to postProcess)    .then((renderedRoutes) =>      this._options.postProcessHtml        ? renderedRoutes.map((renderedRoute) => {            const processed = this._options.postProcessHtml(renderedRoute);            if (typeof processed === "string") renderedRoute.html = processed;            else renderedRoute = processed;            return renderedRoute;          })        : renderedRoutes    )    // Run postProcess hooks.    .then((renderedRoutes) =>      this._options.postProcess        ? Promise.all(            renderedRoutes.map((renderedRoute) =>              this._options.postProcess(renderedRoute)            )          )        : renderedRoutes    )    // Check to ensure postProcess hooks returned the renderedRoute object properly.    .then((renderedRoutes) => {      const isValid = renderedRoutes.every((r) => typeof r === "object");      if (!isValid) {        throw new Error(          "[prerender-spa-plugin] Rendered routes are empty, did you forget to return the `context` object in postProcess?"        );      }      return renderedRoutes;    })    // Minify html files if specified in config.    .then((renderedRoutes) => {      if (!this._options.minify) return renderedRoutes;      renderedRoutes.forEach((route) => {        route.html = minify(route.html, this._options.minify);      });      return renderedRoutes;    })    // Calculate outputPath if it hasn't been set already.    .then((renderedRoutes) => {      renderedRoutes.forEach((rendered) => {        if (!rendered.outputPath) {          rendered.outputPath = path.join(            this._options.outputDir || this._options.staticDir,            rendered.route,            "index.html"          );        }      });      return renderedRoutes;    })    // Create dirs and write prerendered files.    .then((processedRoutes) => {      const promises = Promise.all(        processedRoutes.map((processedRoute) => {          return mkdirp(path.dirname(processedRoute.outputPath))            .then(() => {              return new Promise((resolve, reject) => {                compilerFS.writeFile(                  processedRoute.outputPath,                  processedRoute.html.trim(),                  (err) => {                    if (err)                      reject(                        `[prerender-spa-plugin] Unable to write rendered route to file "${processedRoute.outputPath}" \n ${err}.`                      );                    else resolve();                  }                );              });            })            .catch((err) => {              if (typeof err === "string") {                err = `[prerender-spa-plugin] Unable to create directory ${path.dirname(                  processedRoute.outputPath                )} for route ${processedRoute.route}. \n ${err}`;              }              throw err;            });        })      );      return promises;    })    .then((r) => {      PrerendererInstance.destroy();      done();    })    .catch((err) => {      PrerendererInstance.destroy();      const msg = "[prerender-spa-plugin] Unable to prerender all routes!";      console.error(msg);      compilation.errors.push(new Error(msg));      done();    });};

在这个办法中,又呈现了一个新的compilation对象。这个办法具体的介绍能够看Webpack compilation对象,具体的源码能够见此处。上面我简略介绍下:这个对象代表的是一次文件资源的构建。每次有文件变动时,就会创立一个新的对象。这个文件次要蕴含了以后资源构建和变动过程中的一些属性和信息。

另一个done参数,代表着以后插件执行完后执行下一步的一个触发器,和咱们常见的Node框架中的next()一样。

接下来,咱们来简略说下这个函数执行的逻辑:

  1. 初始化了一个Prerenderer的实例。这个实例是用于对页面进行预渲染的一个工具,具体的代码能够见GitHub。
  2. 实例初始化后,针对每一个路由,进行了一次预渲染操作。
  3. 依据拿到的预渲染相干的数据,对有效性进行查看。
  4. 如果指定了压缩,那么对预渲染数据进行相干的压缩解决。
  5. 最终将预渲染相干的数据输入到指定门路上。
  6. 销毁Prerenderer实例。

这个就是一个插件执行的残缺流程。

总结

通过prerender-spa-plugin这个插件,大家应该可能理解到咱们现行的一个插件到底是如何运行的,咱们编写一个插件须要的核心部件:

  • 一个初始化的function函数。
  • 一个原型链上的apply办法。

    - 一个钩子函数。

    - 一个绑定生命周期的代码。

有了这些货色,咱们的一个Webpack的插件就实现了。

心愿通过一个插件源码的示例,可能让大家理解下咱们日常应用的看似很简单的Webpack插件,到底是怎么实现的。

附录

  1. Webpack官网:如何编写一个插件
  2. Webpack Complier钩子
  3. Webpack Compilation对象
  4. Webpack钩子