前端开发 工程化 编译器
Babel 的原理
剑指前端 作者:剑指前端 2022-02-09 807 0

相关问题

  • Babel 是什么
  • Babel 有什么用
  • 压缩代码如何实现

回答关键点

JS 编译器 AST 插件系统

Babel 是 JavaScript 编译器:他能让开发者在开发过程中,直接使用各类方言(如 TS、Flow、JSX)或新的语法特性,而不需要考虑运行环境,因为 Babel 可以做到按需转换为低版本支持的代码;Babel 内部原理是将 JS 代码转换为 AST,对 AST 应用各种插件进行处理,最终输出编译后的 JS 代码。

知识点深入

1. AST 抽象语法树

简单定义:以树的形式来表现编程语言的语法结构。

image

利用在线 playground 调试,可以对 AST 有个直观感受:生成的树有多个节点,节点有不同的类型,不同类型节点有不同的属性。

const custom = "HZFE";  

Copy

image

AST 是源代码的高效表示,能便捷的表示大多数编程语言的结构。适用于做代码分析或转换等需求。之所以用树来进行分析或转换,是因为树能使得程序中的每一节点恰好被访问一次(前序或后序遍历)。

常见使用场景:代码压缩混淆功能可以借助 AST 来实现:分析 AST,基于各种规则进行优化(如 IF 语句优化;移除不可访问代码;移除 debugger 等),从而生成更小的 AST 树,最终输出精简的代码结果。

2. Babel 编译流程

三大步骤

image

  1. 解析阶段:Babel 默认使用 @babel/parser 将代码转换为 AST。解析一般分为两个阶段:词法分析和语法分析。

    • 词法分析:对输入的字符序列做标记化(tokenization)操作。
    • 语法分析:处理标记与标记之间的关系,最终形成一颗完整的 AST 结构。
  2. 转换阶段:Babel 使用 @babel/traverse 提供的方法对 AST 进行深度优先遍历,调用插件对关注节点的处理函数,按需对 AST 节点进行增删改操作。

  3. 生成阶段:Babel 默认使用 @babel/generator 将上一阶段处理后的 AST 转换为代码字符串。

3. Babel 插件系统

Babel 的核心模块 @babel/core,@babel/parser,@babel/traverse 和 @babel/generator 提供了完整的编译流程。而具体的转换逻辑需要插件来完成。

在使用 Babel 时,我们可通过配置文件指定 plugin 和 preset。而 preset 可以是 plugin 和 preset 以及其他配置的集合。Babel 会递归读取 preset,最终获取一个大的 plugins 数组,用于后续使用。

常见 presets

  • @babel/preset-env
  • @babel/preset-typescript
  • @babel/preset-react
  • @babel/preset-flow

最常见的 @babel/preset-env 预设,包含了一组最新浏览器已支持的 ES 语法特性,并且可以通过配置目标运行环境范围,自动按需引入插件。

编写 Babel 插件

Babel 插件的写法是借助访问者模式(Visitor Pattern)对关注的节点定义处理函数。参考一个简单 Babel 插件例子:

module.exports = function () {  
 return { pre() {}, // 在 visitor 下挂载各种感兴趣的节点类型的监听方法 visitor: { /** * 对 Identify 类型的节点进行处理 * @param {NodePath} path */ Identifier(path) { path.node.name = path.node.name.toUpperCase(); }, }, post() {}, };};  

Copy

使用该 Babel 插件的效果如下:

// input  
  
// index.js  
function hzfe() {}  
  
// .babelrc  
{  
 "plugins": ["babel-plugin-yourpluginname"]}  

Copy

// output  
function HZFE() {}  

Copy

深入 Babel 转换阶段

在转换阶段,Babel 的相关方法会获得一个插件数组变量,用于后续的操作。插件结构可参考以下接口。

interface Plugin {  
 key: string | undefined | null; post: Function | void; pre: Function | void; visitor: Object; parserOverride: Function | void; generatorOverride: Function | void; // ...}  

Copy

转换阶段,Babel 会按以下顺序执行。详细逻辑可查看源码

  1. 执行所有插件的 pre 方法。
  2. 按需执行 visitor 中的方法。
  3. 执行所有插件的 post 方法。

一般来说,写 Babel 插件主要使用到的是 visitor 对象,这个 visitor 对象中会书写对于关注的 AST 节点的处理逻辑。而上面执行顺序中的第二步所指的 visitor 对象,是整合自各插件的 visitor,最终形成一个大的 visitor 对象,大致的数据结构可参考以下接口:

// 书写插件时的 visitor 结构  
interface VisitorInPlugin {  
 [ASTNodeTypeName: string]: | Function | { enter?: Function; exit?: Function; };}  
  
// babel 最终整合的 visitor 结构  
interface VisitorInTransform {  
 [ASTNodeTypeName: string]: { // 不同插件对相同节点的处理会合并为数组 enter?: Function[]; exit?: Function[]; };}  

Copy

在对 AST 进行深度优先遍历的过程中,会创建 TraversalContext 对象来把控对 NodePath 节点的访问,访问时调用对节点所定义的处理方法,从而实现按需执行 visitor 中的方法。详细实现请看 babel-traverse 中的源码。

参考资料

  1. AST
  2. Babel-handbook
  3. estree
  4. 访问者模式
来自:Web前端工程化 | 剑指前端
发表评论

本站文章用于学习交流

新浪微博   |   QQ群1:161644793qq   |   QQ群2:98711210

网站地图   |   网站统计

Copyright 2011 - 2021 paocode.com All Rights Reversed. 浙ICP备19041980号

瞎猫内容中心