使用JavaScript实现一个简单的编译器

在前端开发中也会或多或少接触到一些与编译相关的内容，常见的有

• 将ES6、7代码编译成ES5的代码
• 将SCSS、LESS代码转换成浏览器支持的CSS代码
• 通过uglifyjs、uglifycss等工具压缩代码
• 将TypeScript代码转换成JavaScript代码
• Vue模板语法转换成render函数、JSX语法转换成JS代码

尽管社区的工具如bable、*-loader已经帮我们完成了上面的所有工作，我们不用关心编译的过程，甚至也很少有人关注输出的代码，但是了解编译原理还是很有必要的，这篇文章主要用来记录我在学习编译原理时整理的一些笔记。

本文包含大量示例代码，跳过部分代码并不影响阅读重要，因此可酌情忽略，附：源码地址。

本文主要参考下面两篇文章

• the-super-tiny-compiler，本文实现的编译器基本参照这个项目实现，附中文讲解-用 JavaScript 写一个超小型编译器
• 怎样写一个解释器，这篇文章讲的十分好，不妨移步阅读

写在前面

在了解编译器的相关概念之前，让我们先确定一下到底是称呼"编译器"还是"解释器"。

程序主要有两种运行方式：静态编译与动态解释。静态编译的程序在执行前全部被翻译为机器码，通常将这种类型称为AOT （Ahead of time）即 “提前编译”；而解释执行的则是一句一句边翻译边运行，通常将这种类型称为JIT（Just-in-time）即“即时编译”。

从静态编译与动态解释这两种程序运行的方式，可以引申出两个工具

• 解释器：直接执行用编程语言编写的指令的程序，即JIT运行时的工具
• 编译器：把源代码转换成（翻译）低级语言的程序，即AOT运行时的工具

实际上，我认为初学时不需要纠结于这些概念，本文的主要目的是实现将一个简单的工具，它的功能是：将dart函数命名参数调用形式

sayHello(name: "shymean", msg: 'hello');
复制代码

转换为javascript函数参数调用

sayHello({name: 'shymean', msg: 'hello'})
复制代码

如果只是为了输出JavaScript代码，可以把这个工具看做是编译器，因为它将一种代码语言编译成了另外一种代码语言；如果在工具后面在添加一行eval，这个工具就变成了解释器，因为它可以通过JavaScript直接“运行”Dart代码。

所以，简而言之，这篇文章不关注我们实现的到底是编译器还是解释器，统称为编译器即可。

现在回到我们要实现的这个编译器，看起来只要加一对大括号就行了吗？不，我们将学习编译原理，并了解编译器的基本工作流程。

编译器的工作流程

大多数编译器可以分成三个阶段：解析（Parsing），转换（Transformation）以及代码生成（Code Generation）

• 解析是将源代码转换为一种更抽象的表达方式，一般将结果称为AST抽象语法树
• 转换是对AST做一些处理，让他能做到编译器预期他做到的事情
• 代码生成器接收AST转换之后的数据结构，然后把它转换成新的代码。

在vue的源码compiler实现中，可以查看模板编译相关的逻辑

export const createCompiler = createCompilerCreator(function baseCompile ( template: string, options: CompilerOptions ): CompiledResult { const ast = parse(template.trim(), options) if (options.optimize !== false) { optimize(ast, options) } const code = generate(ast, options) return { ast, render: code.render, staticRenderFns: code.staticRenderFns } }) 复制代码

可以看见其主要的工作跟上面提到的编译器工作流程基本一致

• parse，将template模板解析成ast，主要是解析标签和上面的属性、指令等内容
• optimize，主要是标记ast的静态节点，优化虚拟DOM树，方便后期diff优化算法
• generate，通过ast生成render函数

接下来我们逐个学习编译器的每一个流程。

解析获取AST

解析一般来说会分成两个阶段：词法分析和语法分析

• 词法分析主要是拆分原始代码，并将其分割成一些Token，token由一些代码语句的碎片组成，可以是数字、标签、运算符等
• 语法分析接收词法分析生成的token，然后把他们转换成一种抽象的表示，这种表示描述了代码语句中每一个片段以及他们之间的关系，这被称为AST抽象语法树，AST是一个嵌套很深的对象，用一种更容易处理的方式代表了代码本身

解析阶段最主要的目的就是从源码中获取所需的信息，但是词法分析和语法分析算是编译器中十分繁琐和无趣的"脏活"，下面我们通过两个例子来理解解析的概念和实现。

通过正则拆分文本

这是之前曾做过一个功能，将一段txt文本解析成对话小说，文本内容大致类似于下面结构(此外还有如图片、分支选项等结构，这里暂且省略)

[作者]shymean
[简介]测试测试测试测试

这里是背景旁白~

【小明】
How are you?

[小花]
I'm fine, 3Q you & you?
复制代码

需要将其解析成大致如下JSON结构，方便后续提交到服务端

{ author: 'shymean',
  abstract: '测试测试测试测试',
  nodes:
   [ { type: 10, content: '这里是背景旁白~' },
     { name: '小明', content: 'How are you?', type: 0 },
     { name: '小花', content: 'I\'m fine, 3Q you & you?', type: 0 } ] }
复制代码

由于整个文档结构都是按约定编写的，因此可以通过正则等方式提取关键信息。接下来是整个功能的简单实现，通过这个功能，可以大致了解文本拆分的逻辑

function parser(input) {
    let [header, ...content] = input.split(/\r?\n\s*?\r?\n/); let { author, abstract } = parseHeader(header); let body = getNode(content); return { author, abstract, nodes: body }; // 拆分头部信息 function parseHeader(head) { let res = null; let total = []; const reHeadInfo = /[\[【［].*?[\]】］](.*)\r?\n?/g; while ((res = reHeadInfo.exec(head))) { total.push(res[1]); } let [author, abstract] = total; return { author, abstract }; } // 获取节点列表 function parseContent(content) { const reNode = /[\[【［](.*?)[\]】］].*(?=\r?\n)\r?\n([^]*)/; return content.map(item => { let res = reNode.exec(item); if (res) { // 对话 let name = res[1].trim(); let chatContent = res[2].trim(); return { name: name, content: chatContent, type: 0 }; } else { content = item.trim(); return { type: 10, content }; } }); } } 复制代码

可以看见，我们可以使用编译器语言支持的正则来进行拆分，Hexo解析markdown文本并输出静态html文件，大致也是同样的原理。当然，直接遍历源码文本也是可以的，接下来我们来实现简单的词法分析

通过遍历字符串拆分文本

以下面一段简单的JS代码为例

var a = 100; 
var b = "hello";
复制代码

我们需要把这段代码里面的两个变量声明语句拆分出来，这里我们采用遍历源码文本的方法来实现

function isReserveWords(name) {
    const reserveWords = ["var"]; // 假设这里只有var一个关键字 return reserveWords.includes(name); } function tokenizer(input) { let tokens = []; let current = 0; while (current < input.length) { let char = input[current]; if (char === "=" || char===';') { tokens.push({ type: "symbol", value: char }); current++; continue; } // 去除空格 let WHITESPACE = /\s/; if (WHITESPACE.test(char)) { current++; continue; } let NUMBERS = /[0-9]/; if (NUMBERS.test(char)) { let value = ""; while (NUMBERS.test(char)) { value += char; char = input[++current]; } tokens.push({ type: "number", value }); continue; } // 解析字符串，这里只处理了双引号的字符串 if (char === '"') { let value = ""; char = input[++current]; while (char !== '"') { value += char; char = input[++current]; } current++; tokens.push({ type: "string", value }); continue; } // 处理函数、变量名，除开上面的字面量后，只有保留字和函数变量名了 let LETTERS = /[a-z]/i; if (LETTERS.test(char)) { let value = ""; while (LETTERS.test(char)) { value += char; char = input[++current]; } if (isReserveWords(value)) { tokens.push({ type: "reserveWords", value }); } else { tokens.push({ type: "name", value }); } continue; } throw new TypeError("I dont know what this character is: " + char); } return tokens; } let code = `var a = 100; var b = "hello";`; let tokens = tokenizer(code); console.log(tokens); // 从源码中解析得到tokens数组 复制代码

可见对于某个符号或者某个连续字符串，都是由我们在解析过程中自己定义的，语言越复杂，词法分析时需要判断的情况越多~

将token转换成AST

这里引用了怎样写一个解释器这篇文章里面的图。

我们可以把(* (+ 1 2) (+ 3 4))这样的表达式拆分成一个tokens数组，然后我们可以遍历这个数组，将其转换成一个AST。下面是该表达式对应的AST结构

 

 

此外还可以参考这个在线示例,将tokens转换成AST是一个比较复杂的过程，在本文实现的编译器中有比较粗略的实现，这里不再赘述，其主要逻辑是通过遍历tokens数组，将不同的token转换成对应类型的节点，然后构造成AST树结构。

遍历AST

虽然AST不一定是二叉树，但是了解二叉树的遍历方式还有有一些帮助的。二叉树常见的两种遍历次序是：深度优先和广度优先

• 在广度优先中，先处理节点的子节点，然后遍历下一层

• 在深度优先中，根据根节点访问顺序的不同又分为了

  • 先序遍历，根节点首先被访问，然后先序遍历左子树，最后先序遍历右子树
  • 中序遍历，先中序遍历左子树，然后访问跟根节点，最后中序遍历右子树
  • 后序遍历，先后序遍历左子树，接着后序遍历右子树，最后访问根节点

// 先序遍历
// 先访问根节点，然后是左子树，最后是右子树
// 所谓访问就是获取节点所保存的数据，
// 如果将树叶当作左右节点都为空的根节点，则访问指的就是访问每层的根节点。
function preorder(node) { if (node) { console.log(node.data); preorder(node.left); preorder(node.right); } } // 中序遍历 // 先遍历左子树，然后是根节点，最后是右子树 function inorder(node) { if (node) { inorder(node.left); console.log(node.data); inorder(node.right); } } // 后序遍历 // 先访问左子树，然后是右子树，最后是根节点 function postorder(node) { if (node) { postorder(node.left); postorder(node.right); console.log(node.data); } } 复制代码

遍历AST一般采用的方式是从根节点开始，通过深度优先遍历AST，将不同类型的节点保存在目标语言的对象中，方便后续的代码生成。

同样以Vue的源码为例，在optimizer中，从根节点开始，遍历所有节点，并将对应类型的节点标记为static。

export function optimize (root: ?ASTElement, options: CompilerOptions) { if (!root) return isStaticKey = genStaticKeysCached(options.staticKeys || '') isPlatformReservedTag = options.isReservedTag || no // first pass: mark all non-static nodes. markStatic(root) // second pass: mark static roots. markStaticRoots(root, false) } 复制代码

获取到AST，且掌握了对它的遍历方法之后，我们就可以对它进行接下来的操作了。

转换

解析完成后的下一步是转换，它只是把 AST 拿过来然后对它做一些修改。它可以在同种语言下作 AST，也可以把 AST 翻译成全新的语言。

如果解析阶段获取的AST能满足后续的需求，那么转换步骤也许并不是必须的。

代码生成

代码生成阶段主要是根据转换阶段的AST结果来输出目标代码，代码生成器知道如何打印AST获取的各种节点，然后递归调用自身，直到所有节点都被打印在一个很长的字符串中，最后输出目标代码。

因此代码生成阶段最主要的工作，仍旧是遍历AST，并根据每个节点的类型，输出合适的代码。

代码实现

现在我们大致了解了编译器解析、转换和代码生成这三个流程，接下来让我们来一步一步地实现前文提到的编译器。

解析

首先我们通过遍历源码(一个很长的字符串)

sayHello(userName:getUsername("shymean"), msg:"hello");
复制代码

将其拆分到一个token数组里面

[ { type: 'name', value: 'sayHello' },
  { type: 'paren', value: '(' }, { type: 'name', value: 'userName' }, { type: 'colon', value: ':' }, { type: 'name', value: 'getUsername' }, { type: 'paren', value: '(' }, { type: 'string', value: 'shymean' }, { type: 'paren', value: ')' }, { type: 'comma', value: ',' }, { type: 'name', value: 'msg' }, { type: 'colon', value: ':' }, { type: 'string', value: 'hello' }, { type: 'paren', value: ')' } 复制代码

接着遍历这个token数组，将其解析成对应的ast

{
    type: "Program",
    body: [
        {
            type: "CallExpression", name: "sayHello", params: [ { type: "NameParam", name: "userName", value: { type: "CallExpression", name: "getUsername", params: [{ type: "StringLiteral", value: "shymean" }] } }, { type: "NameParam", name: "msg", value: { type: "StringLiteral", value: "hello" } } ] } ] } 复制代码

拆分tokens和将tokens的具体实现可以在源码中查到。

转换

看起来我们的简易编译器并不需要生成新的AST，因为他们的函数调用、参数都是类似的，唯一的区别在于dart的命名参数在JS中需要通过对象参数来实现。因此转换这一步被我们简单地略过了，在the-super-tiny-compiler这个项目中，实现了转换功能，将原始的ast转换成更符合JavaScript语义的AST，如函数的calle、arguments等属性

代码生成

在我们的编译器中，这一步的实现是非常简单的：对于NameParam类型的节点，我们会在参数列表首尾拼接{}就大功告成了

function codeGenerator(node, index, nodeList) {
    switch (node.type) { // ...其他类型 case "NameParam": let str = ""; // 第一个参数前拼接'{' if (index === 0) { str += "{"; } str += node.name + ":" + codeGenerator(node.value); // 最后一个参数后拼接 '}' if (index === nodeList.length - 1) { str += "}"; } return str; } } 复制代码

将上面的流程汇总，然后就可以得到一个非常简单的编译器了

function compiler(input) {
    let tokens = tokenizer(input); let ast = parser(tokens); let output = codeGenerator(ast); return output; } let input = `sayHello(userName:getUsername("shymean"), msg:"hello");`; let code = compiler(input); // sayHello({userName:getUsername("shymean"), msg:"hello"}) 复制代码

小结

本文先了解了编译器的基本工作流程，然后介绍了词法分析和语法分析的基本实现，接着介绍了遍历AST的方法，最后将各个阶段结合起来，实现了一个十分建议的编译器。

虽然实现的编译器看起来只是添加了一对{}，实际上却大致展示一个编译器的工作流程

• 词法分析和语法分析，将源代码转换成AST
• 根据AST输出新的代码，如果目标代码是比较底层的中间代码或机器代码，工作可能会更加繁琐

当然仅仅了解这一些东西是万万不够的，不过万事开头难，了解到编译器的雏形和基本概念之后，进一步的学习才会更有动力，继续学习吧~

作者：橙红年代
链接：https://juejin.im/post/5ceff22af265da1b6b1cbb90