Vite 2.0 预构建源码解析

853 阅读5分钟

预优化 2.0

本篇章将讲述 Vite2.0 版本在预优化(Pre-Bundling)过程中所做的一些工作。为了更好的阅读体验以及更加完整的章节阅读。我们推荐你到原地址进行阅读,包含了 Vite1.0 - Vite2.0 后续文章的所有更新。vite-design.surge.sh/guide/2.0/o…

image.png

功能总览

Vite2.0 虽然底层代码跟 1.0 比改动很大,但总体理念和使用方式目前看起来差别不大。

Vite2.0 在底层代码的改动较大的地方大概是使用了 http + connect 模块来代替 1.0 中的直接使用 koa 框架的一些能力。并且预优化的工具也由 rollup 的 commonjs 插件替换为 esbuild
在 1.0 的使用过程中我就发现了一些 rollup 的 commonjs 插件的一些 bug,并且提了一些 issue 记录,但是后续由于忙着开发自己的 SSR 框架去了, 就没怎么跟进后续的进展。现在看到 2.0 换成了 esbuild,不仅构建速度大大提升,相应的 bug 也少了不少。
在正式阅读源码前,本来以为 Vite 只是做了模块格式 format:esm 的简单操作,但是仔细阅读之后发现 Vite 做的工作还是不少的。这里大力推荐大家阅读一下 Vite2.0 的代码无论是仓库规范还是具体编码都是非常优秀值得大家学习的,且体量不大易于调试,比 Webpack 这些巨无霸级别的工具估计连作者自己都没办法掌握所有代码的要好得多。

本地调试

调试方式与 1.0 大体没有变化,只是 2.0 的架构变成了 monorepo 的形式,当然我们不需要管其他的 package,只需要调试 Vite 即可。

$ git clone git@github.com:vitejs/vite.git
$ cd vite && yarn
$ cd packages/vite && yarn build && yarn link
$ yarn dev

然后再通过 Vite 脚手架创建一个最简单的 example 来 link Vite

$ npm init @vitejs/app my-vue-app --template vue
$ cd my-vue-app && yarn && yarn link vite
$ npx vite optimize --force

然后就可以开始愉快的调试源码了

vite esbuild 插件解析

这里我们舍去对本章内容不重要的 Vite resolve 模块的逻辑,只看 Vite 到底用 esbuild 干了什么
我们直接找到最精华的一段代码

// vite/src/node/optimizer/index.ts

const esbuildService = await ensureService()
await esbuildService.build({
  entryPoints: Object.keys(flatIdDeps),
  bundle: true,
  format: 'esm',
  external: config.optimizeDeps?.exclude,
  logLevel: 'error',
  splitting: true,
  sourcemap: true,
  outdir: cacheDir,
  treeShaking: 'ignore-annotations',
  metafile: esbuildMetaPath,
  define,
  plugins: [esbuildDepPlugin(flatIdDeps, flatIdToExports, config)]
})

这里就是最终的 Vite 调用 esbuild api 的地方,让我们来一一分析它到底干了啥。

entryPoints

首先是 entryPoints 就是打包的入口,这里我们先不用管 Vite 到底是怎么去收集模块依赖的。我们后续章节会分析。只需要知道默认的 example 中 flatIdDeps 大概长下面这样。

{
  vue: '/Users/yuuang/Desktop/my-vue-app/node_modules/vue/dist/vue.runtime.esm-bundler.js'
}

Vite 将会根据 module, jsnext:main, jsnext 这样的字段顺序来查找最终解析的文件的绝对路径。比如 Vue 的 module 字段指向 dist/vue.runtime.esm-bundler.js, 所以最终的到的对象就是上述所示。
这里取的是 Object.keys 也就是 entryPoints: ['vue'] 的形式,当你的应用还依赖其他模块时,就会是一个模块依赖数组例如 entryPoints: ['vue', 'vue-router']

bundle

bundle: true 就是将模块的依赖与模块自身打包成一个文件。这里对标的是 Webpack 的 bundle 功能而不是 tsc, babel 这种将原模块与 convert 之后的模块文件一一对应的这种工具。

external

依赖外置,不需要处理的模块。这个选项在做服务端渲染或者应用体积优化的时候经常用到。举个例子当开启了这个选项并做了一些配置时。

import * as React from 'react'

打包后的代码仍然保留这段代码,而不是将 react 的代码打包进来。

format

输出模块格式为 esm,这个没什么好说的

outdir

预优化的缓存文件夹,默认为 node_modules/.vite

plugins

esbuildDepPlugin 这个插件就是 Vite 在 esbuild 打包中最核心的逻辑了。让我们来看看他到底干了什么事情。
在分析这个插件的源码之前,我们先看 esbuild 官方给的一个最简单的插件例子,来看看如何编写 esbuild 的插件,了解一个最基本的工作流程。

let envPlugin = {
  name: 'env',
  setup(build) {
    build.onResolve({ filter: /^env$/ }, args => ({
      path: args.path,
      namespace: 'env-ns',
    }))
    build.onLoad({ filter: /.*/, namespace: 'env-ns' }, () => ({
      contents: JSON.stringify(process.env),
      loader: 'json',
    }))
  },
}

require('esbuild').build({
  entryPoints: ['app.js'],
  bundle: true,
  outfile: 'out.js',
  plugins: [envPlugin],
}).catch(() => process.exit(1))

这里我们编写了一个名字为 env 的插件。它干了什么事情呢,比如我们有下面的这一段源代码

import { PATH } from 'env'
console.log(`PATH is ${PATH}`)

esbuildonResolve 阶段通过正则匹配到了 env 这个我们想 import 的模块,并且把它交给了一个名称为 env-ns 的流程做最终的处理。在 env-ns 中,我们将当前的 process.env 环境变量 stringifyjson 字符串的形式返回给了 contents。也就是 env 这个模块,最终返回的就是 process.env 的值

ok,了解完 esbuild 插件的一个基本的规范之后,我们再来看 esbuildDepPlugin 的内容。
同样我们摒除暂时不需要关系的模块 resolve 逻辑,只看核心逻辑

特定文件 external

第一个处理是对特定格式文件的 external 处理,因为这些文件 esbuild 要么无法处理要么不应该由它来处理,Vite 自身会有另外的专门针对这些类型文件的处理逻辑。


const externalTypes = [
  'css',
  // supported pre-processor types
  'less',
  'sass',
  'scss',
  'styl',
  'stylus',
  'postcss',
  // known SFC types
  'vue',
  'svelte',
  // JSX/TSX may be configured to be compiled differently from how esbuild
  // handles it by default, so exclude them as well
  'jsx',
  'tsx',
  ...KNOWN_ASSET_TYPES
]
export const KNOWN_ASSET_TYPES = [
  // images
  'png',
  'jpe?g',
  'gif',
  'svg',
  'ico',
  'webp',
  'avif',

  // media
  'mp4',
  'webm',
  'ogg',
  'mp3',
  'wav',
  'flac',
  'aac',

  // fonts
  'woff2?',
  'eot',
  'ttf',
  'otf',

  // other
  'wasm'
]
 build.onResolve(
  {
    filter: new RegExp(`\\.(` + externalTypes.join('|') + `)(\\?.*)?$`)
  },
  async ({ path: id, importer, kind }) => {
    const resolved = await resolve(id, importer, kind)
    if (resolved) {
      return {
        path: resolved,
        external: true
      }
    }
  }
)

区分入口模块和依赖模块

Vite 对入口模块和依赖模块使用了不同的处理规则,入口模块指的是我们直接 import 的模块或者通过 include 制定的模块。而依赖模块则是入口模块自身的依赖也就是 dependencies 这里可以看到如果是入口模块,我们则交给 namedep 的流程去继续处理,且我们只返回一个 bare name,裸的模块名。比如 vue, vue-router 原始的名称。而非入口模块,我们则直接返回一个模块入口文件的绝对路径。

function resolveEntry(id: string, isEntry: boolean) {
  const flatId = flattenId(id)
  if (flatId in qualified) {
    return isEntry
      ? {
          path: flatId,
          namespace: 'dep'
        }
      : {
          path: path.resolve(qualified[flatId])
        }
  }
}

build.onResolve(
  { filter: /^[\w@][^:]/ },
  async ({ path: id, importer, kind }) => {
    const isEntry = !importer
    // ensure esbuild uses our resolved entires
    let entry
    // if this is an entry, return entry namespace resolve result
    if ((entry = resolveEntry(id, isEntry))){
      return entry
    }
  }
)

dep 处理

这块的工作基本上是预优化的核心内容。这里 Vite 只干了一件事情,就是生成了一个代理模块来导出原模块的原始 id。 举个例子,上面我们提到了 Vite 会把入口模块交给 namespace 为 dep 的流程去做进一步的处理。且只传递给了一个最原始的 Bare id。 Vite 在这里用 esbuild 提供的 parse 词法分析逻辑来分析入口模块的 import, export 信息 当入口模块即没有 import 关键字 也没有 export 关键字时,我们认为它是一个 cjs 模块。生成的代理模块的格式如下

contents += `export default require("${relativePath}");`

当入口模块使用 export default 进行导出时,我们生成的代理模块的格式如下

contents += `import d from "${relativePath}";export default d;`

当入口模块存在 ReExports 时,比如 export * from './xxx.js' 或者 export 关键字出现的次数大于1,或者不存在 export default的时候生成的代理模块的格式如下 这也是大多数符合标准的模块最终处理完成的格式。

contents += `\nexport * from "${relativePath}"`

以 Vue 为例,当我们处理完之后。执行 import Vue from 'vue' 时,'vue' 实际返回的 contents 是 export * from "./node_modules/vue/dist/vue.runtime.esm-bundler.js"

通过注释,我们可以看出这样做的目的有两个

  • 使 esbuild 最终输出符合期望的结构
  • 如果不分离代理模块和真实模块,esbuild 可能会重复打包相同模块

对于第一个原因,我自己调试的时候测试了一下,如果不交给 dep 去处理,生成的最终格式是什么样的。

  • 当我们使用 dep 时,生成的最终文件在 cache 目录是 node_modules/vite/vue.js
  • 不使用 dep 时,生成的最终文件在 cache 目录是 node_modules/vite/vue.runtime.esm-bundler.js,此时应用会提示找不到 Vue 文件。

且当我们入口模块超过一个时,例如存在 ['vue', 'vue-router'] 的时候,不使用 dep 的话将会生成带有文件夹的结构。

$ tree node_modules/.vite
node_modules/.vite
├── _esbuild.json
├── _metadata.json
├── vue
│   └── dist
│       └── vue.runtime.esm-bundler.js
└── vue-router
    └── dist
        └── vue-router.esm.js

于是我猜想 Vite 这么干是为了方便上层统一处理,否则生成文件名称文件结构都是不一定的会增加处理的难度。
对于第二个原因,按照注释所说是因为真实的模块可能会通过相对引用的方式被导入会造成重复打包,我尝试了几种测试用例没能够复现,可能是对这块具体意思的理解不准确。希望看到这篇文章的人有兴趣可以去分析一下并且提交 PR 来更新本文档。

 // For entry files, we'll read it ourselves and construct a proxy module
// to retain the entry's raw id instead of file path so that esbuild
// outputs desired output file structure.
// It is necessary to do the re-exporting to separate the virtual proxy
// module from the actual module since the actual module may get
// referenced via relative imports - if we don't separate the proxy and
// the actual module, esbuild will create duplicated copies of the same
// module!
const root = path.resolve(config.root)
build.onLoad({ filter: /.*/, namespace: 'dep' }, ({ path: id }) => {
  const entryFile = qualified[id]
  let relativePath = normalizePath(path.relative(root, entryFile))
  if (!relativePath.startsWith('.')) {
    relativePath = `./${relativePath}`
  }

  let contents = ''
  const data = exportsData[id]
  const [imports, exports] = data
  if (!imports.length && !exports.length) {
    // cjs
    contents += `export default require("${relativePath}");`
  } else {
    if (exports.includes('default')) {
      contents += `import d from "${relativePath}";export default d;`
    }
    if (
      data.hasReExports ||
      exports.length > 1 ||
      exports[0] !== 'default'
    ) {
      contents += `\nexport * from "${relativePath}"`
    }
  }

  let ext = path.extname(entryFile).slice(1)
  if (ext === 'mjs') ext = 'js'

  return {
    loader: ext as Loader,
    contents,
    resolveDir: root
  }
})

esm 工具

上面分析完了 Vite2.0 的 optimize逻辑
由于我们小组打算做一个基于 esm 的类似于 codesandbox 的应用,在模块处理的过程中我们希望使用 Vite 的成熟能力,所以这里我把 Vite 的 optimize 功能简单的拆分了一下写了一个 esm-optimize模块。做到项目无关,同时不需要创建 vite.config 配置文件。根据入参找到具体的模块进行更新,方便上层业务根据实际情况决定要如何使用 optimize 能力。

使用方式

我们提供 cli 的形式或者模块的形式直接导入使用

$ npm i -g esm-optimize
$ esm react vue # 优化 react vue 模块
$ esm react vue --force # 删除 cache 目录 强制重新 optimize
$ esm react vue --config # 显示最终生成的 config

以模块的形式使用

import { optimize } from 'esm-optimize'

await optimize({
  root: string, // 默认为 cwd 
  optimizeCacheDir: string, // 默认的 vite 缓存文件夹为 `node_modules/.vite`
  optimizeDeps: {
    include: [] // 要处理的 模块
  },
  force: true // 强制优化
})

SSR 框架

最后在这里推荐一下我写的 SSR框架 在最新的 v5.0 版本中,同时支持 React 和 Vue 的服务端渲染框架,且提供一键以 Serverless 的形式发布上云的功能。我们可以非常有自信说它是地球上最先进的ssr框架。并且会在最近集成 Vue3 + Vite + SSR 的最佳实践。欢迎关注使用。