Vue编译器实现代码生成方法介绍

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
}

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
}

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
	}
}

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(`}`)
}

function render(){
	...函数体
}

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(',')
		}
	}
}

function genArrayExpression(node, context){
	const {push} = context
	// 追加方括号
	push(`[`)
	genNodeList(node.elements, context)
	// 补全方括号
	push(`]`)
}

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}'`)
}

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')])
}