rust - 如何获取传递给内部属性宏的模块内容？

Question

所以我定义了以下过程宏:

#[proc_macro_attribute]
pub fn hello(attr: TokenStream, item: TokenStream) -> TokenStream {
  println!("attr: {}", attr);
  println!("item: {}", item);
  item
}

然后我将此 proc 宏应用于非内联模块:

// In file some_mod.rs
#![hello]

fn foo() {}
fn bar() {}

编译器的输出表明传递给 proc 宏 hello 的项目是 mod some_mod;，模块中没有任何项目。但是我想对模块 some_mod 的内容做一些修改。

我发现内联模块有效:

mod some_mod {
  #![hello]
  fn foo() { }
  fo bar() { }
}

传递给我的 proc 宏的项目是 mod some_mod { fn foo() { } fn bar() { } } 。
但我会在复杂的模块层次结构中使用我的 proc 宏，我不想将所有这些模块放在一个文件中。

有什么办法可以让我的proc宏获取非内联模块的内容吗？

Answer 1

您描述的问题是由于目前在内部属性位置使用过程宏是不稳定的。有一个tracking issue对于这个问题大约 1.5 年了，this comment明确提及您当前面临的问题，将宏应用于内联模块会产生与将其应用于文件模块不同的结果。

据我所知，目前没有针对此问题的现成解决方案。

或者，由于您基本上只是修改文件的标记流，您可以尝试通过 build.rs 将您的程序 Hook 到编译过程中(您可以加载您想要的任何库构建过程)，但这将涉及手动筛选所有文件，并且必须从输入 token 流中动态生成新文件。不幸的是，我目前无法想到任何更好的选择，只要您真的需要宏将整个模块作为输入即可。