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

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 for this problem for about 1.5 years now and this comment 明确提到了您当前面临的问题，将宏应用于内联模块会产生与将其应用于文件模块不同的结果。

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

或者，由于您实际上只是在修改文件的标记流，您可以尝试通过 build.rs 将您的程序挂钩到编译过程中（您可以加载构建过程所需的任何库）但这将涉及手动筛选所有文件，并且必须从输入令牌流中动态生成新文件。不幸的是，我可以 - 目前 - 想不出任何更好的选择，只要你真的需要宏将整个模块作为输入。

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

How to get the content of a module passed to an inner attribute macro?

rust

rust-proc-macros