Rust代码生成与AST操作库swc_ecma_quote_macros的使用，高效实现JavaScript语法树宏转换

安装

在项目目录中运行以下Cargo命令：

cargo add swc_ecma_quote_macros

或者将以下行添加到Cargo.toml中：

swc_ecma_quote_macros = "22.0.0"

基本使用示例

swc_ecma_quote_macros提供了quote!宏，可以方便地生成JavaScript语法树节点。以下是一个基本示例：

use swc_ecma_quote_macros::quote;
use swc_ecma_ast::*;

fn create_function() -> Function {
    quote!("function add(a, b) { return a + b; }" as Function)
}

完整示例

下面是一个更完整的示例，展示了如何使用swc_ecma_quote_macros库来转换JavaScript代码：

use swc_ecma_quote_macros::quote;
use swc_ecma_ast::*;
use swc_common::sync::Lrc;
use swc_common::SourceMap;
use swc_ecma_codegen::{Emitter, Config};

fn main() {
    // 创建AST节点
    let function = quote!("function add(a, b) { return a + b; }" as Function);
    
    // 创建模块，包含我们的函数
    let module = Module {
        span: Default::default(),
        body: vec![
            ModuleItem::Stmt(Stmt::Decl(Decl::Fn(FnDecl {
                ident: Ident::new("add".into(), Default::default()),
                declare: false,
                function: Box::new(function),
            }))),
        ],
        shebang: None,
    };
    
    // 设置代码生成器
    let cm: Lrc<SourceMap> = Default::default();
    let mut buf = vec![];
    {
        let mut emitter = Emitter {
            cfg: Config {
                ..Default::default()
            },
            cm: cm.clone(),
            wr: Box::new(swc_ecma_codegen::text_writer::JsWriter::new(
                cm.clone(),
                "\n",
                &mut buf,
                None,
            )),
            comments: None,
        };
        
        emitter.emit_module(&module).unwrap();
    }
    
    // 打印生成的代码
    println!("{}", String::from_utf8(buf).unwrap());
}

动态生成AST节点

swc_ecma_quote_macros还支持动态插入变量到生成的AST中：

use swc_ecma_quote_macros::quote;
use swc_ecma_ast::*;

fn create_function_with_params(params: Vec<Pat>) -> Function {
    let body = BlockStmt {
        span: Default::default(),
        stmts: vec![
            Stmt::Return(ReturnStmt {
                span: Default::default(),
                arg: Some(Box::new(Expr::Bin(BinExpr {
                    span: Default::default(),
                    op: BinaryOp::Add,
                    left: Box::new(Expr::Ident(Ident::new("a".into(), Default::default()))),
                    right: Box::new(Expr::Ident(Ident::new("b".into(), Default::default()))),
                }))),
            }),
        ],
    };
    
    quote!("function add($params) $body" as Function, params = params, body = body)
}

宏转换示例

以下示例展示了如何将简单的函数调用转换为更复杂的表达式：

use swc_ecma_quote_macros::quote;
use swc_ecma_ast::*;
use swc_common::DUMMY_SP;

fn transform_call_expr(call: CallExpr) -> Expr {
    match &call.callee {
        Callee::Expr(expr) => {
            if let Expr::Ident(ident) = &**expr {
                if ident.sym == "log" {
                    // 将简单的log调用转换为console.log
                    return Expr::Call(CallExpr {
                        callee: quote!("console.log" as Expr).into(),
                        args: call.args,
                        type_args: call.type_args,
                        span: DUMMY_SP,
                    });
                }
            }
        }
        _ => {}
    }
    
    Expr::Call(call)
}

完整示例Demo

以下是一个完整的代码转换示例，展示了如何使用swc_ecma_quote_macros来创建和转换JavaScript AST：

use swc_ecma_quote_macros::quote;
use swc_ecma_ast::*;
use swc_common::sync::Lrc;
use swc_common::SourceMap;
use swc_ecma_codegen::{Emitter, Config};
use swc_ecma_visit::{VisitMut, VisitMutWith};

// 自定义访问器，用于转换AST
struct LogToConsoleLog;

impl VisitMut for LogToConsoleLog {
    fn visit_mut_call_expr(&mut self, call: &mut CallExpr) {
        // 先访问子节点
        call.visit_mut_children_with(self);
        
        // 转换log调用为console.log
        if let Callee::Expr(expr) = &call.callee {
            if let Expr::Ident(ident) = &**expr {
                if ident.sym == "log" {
                    *call = CallExpr {
                        callee: quote!("console.log" as Expr).into(),
                        args: call.args.clone(),
                        type_args: call.type_args.clone(),
                        span: call.span,
                    };
                }
            }
        }
    }
}

fn main() {
    // 创建带有log调用的函数AST
    let mut module = Module {
        span: Default::default(),
        body: vec![
            ModuleItem::Stmt(Stmt::Decl(Decl::Fn(FnDecl {
                ident: Ident::new("test".into(), Default::default()),
                declare: false,
                function: Box::new(quote!(
                    "function test() {
                        log('Hello');
                        log('World');
                    }" as Function
                )),
            }))),
        ],
        shebang: None,
    };
    
    // 应用转换
    module.visit_mut_with(&mut LogToConsoleLog);
    
    // 设置代码生成器
    let cm: Lrc<SourceMap> = Default::default();
    let mut buf = vec![];
    {
        let mut emitter = Emitter {
            cfg: Config {
                ..Default::default()
            },
            cm: cm.clone(),
            wr: Box::new(swc_ecma_codegen::text_writer::JsWriter::new(
                cm.clone(),
                "\n",
                &mut buf,
                None,
            )),
            comments: None,
        };
        
        emitter.emit_module(&module).unwrap();
    }
    
    // 打印转换后的代码
    println!("转换后的代码:\n{}", String::from_utf8(buf).unwrap());
}

这个示例演示了：

1. 使用quote!宏创建JavaScript AST
2. 自定义访问器实现AST转换
3. 将log调用转换为console.log调用
4. 将修改后的AST重新生成为JavaScript代码

输出结果应该是：

转换后的代码:
function test() {
    console.log('Hello');
    console.log('World');
}