示例:通过 rustc_interface 进行类型检查

rustc_interface 允许您在编译的各个阶段与 Rust 代码交互。

获取表达式的类型

// 完整程序见 https://github.com/rust-lang/rustc-dev-guide/blob/master/examples/rustc-driver-interacting-with-the-ast.rs 。
let config = rustc_interface::Config {
    input: config::Input::Str {
        name: source_map::FileName::Custom("main.rs".to_string()),
        input: "fn main() { let message = \"Hello, world!\"; println!(\"{}\", message); }"
            .to_string(),
    },
    /* 其他配置 */
};
rustc_interface::run_compiler(config, |compiler| {
    compiler.enter(|queries| {
        // 分析 crate 并检查光标下的类型。
        queries.global_ctxt().unwrap().take().enter(|tcx| {
            // 每次编译包含一个单独的 crate 。
            let krate = tcx.hir().krate();
            // 遍历 crate 中的顶层项,寻找 main 函数。
            for (_, item) in &krate.items {
                // 使用模式匹配在 main 函数中查找特定节点。
                if let rustc_hir::ItemKind::Fn(_, _, body_id) = item.kind {
                    let expr = &tcx.hir().body(body_id).value;
                    if let rustc_hir::ExprKind::Block(block, _) = expr.kind {
                        if let rustc_hir::StmtKind::Local(local) = block.stmts[0].kind {
                            if let Some(expr) = local.init {
                                let hir_id = expr.hir_id; // hir_id 标识字符串 "Hello, world!"
                                let def_id = tcx.hir().local_def_id(item.hir_id); // def_id 标识 main 函数
                                let ty = tcx.typeck(def_id).node_type(hir_id);
                                println!("{:?}: {:?}", expr, ty); // 打印出 expr(HirId { owner: DefIndex(3), local_id: 4 }: "Hello, world!"): &'static str
                            }
                        }
                    }
                }
            }
        })
    });
});