Rust ECMAScript代码生成库swc_ecma_quote的使用，支持AST节点快速构建与代码转换

Rust ECMAScript代码生成库swc_ecma_quote的使用，支持AST节点快速构建与代码转换

安装

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

cargo add swc_ecma_quote

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

swc_ecma_quote = "22.0.0"

替代说明

现在不再直接导入swc_ecma_quote，而是使用swc_core提供的等效功能swc_core::quote和swc_core::quote_expr。

完整示例

以下是使用swc_ecma_quote构建AST节点和进行代码转换的完整示例：

use swc_ecma_ast::*;
use swc_ecma_quote::quote;
use swc_ecma_visit::{Fold, FoldWith};

// 创建一个简单的转换器，将所有标识符转换为大写
pub struct UppercaseIdentifiers;

impl Fold for UppercaseIdentifiers {
    fn fold_ident(&mut self, ident: Ident) -> Ident {
        Ident {
            sym: ident.sym.to_ascii_uppercase().into(),
            span: ident.span,
            optional: ident.optional,
        }
    }
}

fn main() {
    // 使用quote宏快速构建AST节点
    let ast = quote!(
        "function test() { return foo + bar; }" as Module
    );
    
    // 应用转换器
    let transformed = ast.fold_with(&mut UppercaseIdentifiers);
    
    // 打印转换后的代码
    println!("Transformed: {:?}", transformed);
}

另一个更复杂的示例，展示如何构建函数声明：

use swc_ecma_ast::*;
use swc_ecma_quote::quote;

fn main() {
    // 构建一个函数声明AST节点
    let func = quote!(
        "function add(a, b) { return a + b; }" as Function
    );
    
    // 构建一个包含该函数的模块
    let module = Module {
        span: Default::default(),
        body: vec![
            ModuleItem::Stmt(Stmt::Decl(Decl::Fn(func)))
        ],
        shebang: None,
    };
    
    println!("Generated module: {:?}", module);
}

使用swc_core的替代方案

use swc_core::ecma::ast::*;
use swc_core::quote;

fn main() {
    // 使用swc_core的quote替代swc_ecma_quote
    let expr = quote!("1 + 2" as Expr);
    
    // 构建一个包含表达式的语句
    let stmt = Stmt::Expr(ExprStmt {
        span: Default::default(),
        expr: Box::new(expr),
    });
    
    println!("Generated statement: {:?}", stmt);
}

这些示例展示了如何使用swc_ecma_quote或swc_core::quote快速构建AST节点并进行代码转换。


1 回复

Rust ECMAScript代码生成库swc_ecma_quote的使用

介绍

swc_ecma_quote是SWC编译器工具链中的一个库，专门用于简化ECMAScript(JavaScript)抽象语法树(AST)节点的构建和代码转换。它提供了类似Rust中quote宏的功能，允许开发者通过类似JavaScript代码的语法快速构建AST节点，而无需手动构造复杂的AST结构。

主要特点

  1. 提供类似JSX的语法来构建AST节点
  2. 支持代码片段的直接引用和变量插值
  3. 简化AST操作和代码转换流程
  4. 与SWC生态系统的其他工具无缝集成

完整示例代码

use swc_ecma_quote::quote;
use swc_ecma_ast::{Stmt, Decl, Expr, Ident};

fn main() {
    // 基本示例 - 创建const声明
    let const_stmt = quote!("const PI = 3.14;" as Stmt);
    println!("Created const statement: {:?}", const_stmt);

    // 变量声明示例
    let var_stmt: Stmt = quote!("let count = 0;" as Stmt);
    println!("Variable declaration: {:?}", var_stmt);

    // 使用变量插值
    let user_age = 25;
    let age_stmt = quote!("let age = $user_age;" as Stmt, user_age: Expr = user_age.into());
    println!("Age statement with interpolation: {:?}", age_stmt);

    // 函数声明
    let add_fn = quote!(
        "function add(a, b) { return a + b; }" as Stmt
    );
    println!("Function declaration: {:?}", add_fn);

    // 条件语句
    let condition = quote!(
        "if (score > 90) { grade = 'A'; } else { grade = 'B'; }" as Stmt
    );
    println!("Conditional statement: {:?}", condition);

    // 循环语句
    let loop_example = quote!(
        "for (let i = 0; i < 5; i++) { console.log(i); }" as Stmt
    );
    println!("For loop statement: {:?}", loop_example);

    // 对象表达式
    let user_obj = quote!(
        "{ name: 'Alice', email: 'alice@example.com' }" as Expr
    );
    println!("Object expression: {:?}", user_obj);

    // 数组表达式
    let numbers = quote!(
        "[1, 2, 3, 4, 5]" as Expr
    );
    println!("Array expression: {:?}", numbers);

    // 代码转换示例 - let转const
    let let_stmt = quote!("let username = 'john_doe';" as Stmt);
    let const_stmt = match let_stmt {
        Stmt::Decl(Decl::Var(var_decl)) => {
            quote!("const $ident = $init;" as Stmt,
                ident: Ident = var_decl.decls[0].name.clone(),
                init: Expr = var_decl.decls[0].init.clone().unwrap()
            )
        }
        _ => let_stmt,
    };
    println!("Converted to const: {:?}", const_stmt);

    // 模板字符串
    let user = "Bob";
    let greeting = quote!(
        "`Welcome, ${user}!`" as Expr,
        user: Expr = user.into()
    );
    println!("Template string: {:?}", greeting);

    // 类声明
    let class_def = quote!(
        "class User {
            constructor(name) { this.name = name; }
            sayHello() { return `Hello, ${this.name}!`; }
        }" as Stmt
    );
    println!("Class definition: {:?}", class_def);
}

注意事项

  1. 引用的代码片段必须是有效的JavaScript语法
  2. 变量插值需要确保类型匹配
  3. 对于复杂的AST操作，可能需要结合swc_ecma_ast的其他功能
  4. 生成的AST节点可以进一步用SWC的其他工具进行处理或转换

swc_ecma_quote极大地简化了在Rust中操作JavaScript AST的过程，使得代码生成和转换更加直观和高效。

回到顶部