这里将讨论如何根据JavaScript AST生成渲染函数的代码，即代码生成。代码生成本质上是字符串拼接的艺术。需要访问JavaScript AST中的节点，为每一种类型的节点生成相符的 JavaScript代码。
本节，将实现 generate函数来完成代码生成的任务。
function compile(template) {
//模板 AST
const ast = parse(template)
//将模板 AST转换为JavaScript AST
transform(ast)
// 代码生成
const code = generate(ast.jsNode)
return code
}
在这里，代码生成也需要上下文对象。
function generate(node){
const context = {
// 存储最终生成的渲染函数代码
code: '',
// 在生成代码时，通过调用 push 函数完成代码的拼接
push(code){
context.code += code
}
}
// 该方法完成代码生成的工作
genNode(node,context)
// 返回渲染函数代码
return context.code
}
另外，为了让最终生成的代码具有较强的可读性，要考虑生成代码的格式，这就需要扩展context对象，为其增加用来完成换行和缩进的工具函数，如下面代码所示：
function generate(node){
const context = {
code: '',
push(code){
context.code += code
},
// 当前缩进的级别，初始值为 0，即没有缩进
currentIndent: 0,
// 该函数用来换行，即在代码字符串的后面追加 \n 字符
// 另外，换行时应该保留缩进，所以还要追加 currentIndent * 2 个空格字符
newline(){
context.code += '\n'+` `.repeat(context.currentIndent)
},
// 用来缩进，即让 currentIndent 自增后，调用换行函数
indent(){
context.currentIndent++
context.newline()
},
// 取消缩进
deIndent(){
context.currentIndent--
context.newline()
},
}
genNode(node, context)
return context.code
}
有了这些基础能力之后，就可以开始编写 genNode 函数来完成代码生成的工作了。代码生成的原理其实很简单，只需要匹配各种类型的 JavaScript AST节点，并调用对应的生成函数即可，如下面的代码所示:
function genNode(node,context){
switch(node.type){
case 'FunctionDecl':
genFunctionDecl(node,context)
break
case 'ReturnStatement':
genReturnStatement(node,context)
break
case 'CallExpression':
genCallExpression(node,context)
break
case 'StringLiteral':
genStringLiteral(node,context)
break
case 'ArrayExpression':
genArrayExpression(node,context)
break
}
}
如果后续需要增加节点类型，只需要再genNode函数中添加相应的处理分支即可
接下来逐步完善代码生成工作，首先实现函数生命语句的代码生成，即genFunctionDecl函数，如下面的代码所示：
function genFunctionDecl(node,context){
// 从context 对象中取出工具函数
const {push, indent, deIndent} = context
push(`function ${node.id.name}`)
push(`(`)
//调用 genNodeList 为函数的参数生成代码
genNodeList(node.params, context)
push(`)`)
push(`{`)
indent()
//为函数体生成代码，这里递归地调用了 genNode 函数
node.body.forEach(n=>genNode(n,context))
deIndent()
push(`}`)
}
genFunctionDecl函数用来为函数声明类型的节点生成对应的 JavaScript代码。以声明节点为例，它最终生成的代码将会是:
function render(){
...函数体
}
genNodeList函数的实现如下：
function genNodeList(nodes, context){
const {push} = context
for(let i=0;i<nodes.length;i++){
const node = nodes[i]
genNode(node,context)
if(i<nodes.length - 1){
push(',')
}
}
}
这里要注意的一点是，每处理完一个节点，需要在生成的代码后面拼接逗号。
实际上,genArrayExpression函数就利用了这个特点来实现对数组表达式的代码生成，如下面的代码所示：
function genArrayExpression(node, context){
const {push} = context
// 追加方括号
push(`[`)
genNodeList(node.elements, context)
// 补全方括号
push(`]`)
}
对于 genFunctionDecl 函数，另外需要注意的是，由于函数体本身也是一个节点数组，所以需要遍历它并递归地调用 genNode 函数生成代码。
对于ReturnStatement和StringLiteral类型的节点来说，为它们生成代码很简单，如下所示:
function genReturnStatement(node, context){
const {push} = context
push(`return `)
genNode(node.return, context)
}
function genStringLiteral(node, context){
const {push} = context
// 对于字符串字面量，只需要追加与 node.value 对应的字符串即可
genNode(`'${node.value}'`)
}
最后，只剩下genCallExpression 函数了，实现如下:
function genCallExpression(node, context){
const {push} = context
// 取得被调用函数名称和参数列表
const {callee, arguments: args} = node
// 生成函数调用代码
push(`${callee.name}`)
genNodeList(args,context)
push(`)`)
}
配合上述生成器函数的实现，就能得到符合语气的渲染函数代码，用下面的代码测试
const ast = parse(`<div><p>Vue</p><p>Template</p></div>`)
transform(ast)
const code = generate(ast.jsNode)
最终得到的代码字符串如下：
function render (){
return h('div',[h('p','Vue'), h('p','Template')])
}
到此这篇关于Vue编译器实现代码生成方法介绍的文章就介绍到这了,更多相关Vue代码生成内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！
