如何让内部属性宏获取非内联模块的内容？

这个问题的核心在于Rust编译器处理非内联模块（单独文件的模块）的方式：当你在父模块中声明mod some_mod;时，编译器会将该模块视为一个独立的编译单元，此时附加在模块文件顶部的#![hello]属性只会让过程宏接收到mod some_mod;这个声明，而看不到模块内部的内容——因为模块内容还没被合并到当前上下文里。

不过我们可以通过读取模块文件内容并手动解析AST的方式来解决这个问题，下面是具体的实现方案：

步骤1：添加依赖 #

首先在你的Cargo.toml中添加syn、quote和proc-macro2，这些库帮助我们解析和生成Rust代码：

[dependencies]
syn = { version = "2.0", features = ["full", "parsing", "span"] }
quote = "1.0"
proc-macro2 = "1.0"

步骤2：修改过程宏实现 #

我们需要修改hello宏，让它能识别非内联模块，读取对应的文件内容，解析成AST后再进行修改：

use proc_macro::TokenStream;
use syn::{parse_macro_input, ItemMod, File};
use quote::quote;
use std::fs;
use std::path::{Path, PathBuf};

#[proc_macro_attribute]
pub fn hello(_attr: TokenStream, item: TokenStream) -> TokenStream {
    // 解析输入的模块声明
    let mut module = parse_macro_input!(item as ItemMod);

    // 处理非内联模块（没有内嵌内容的模块）
    if module.content.is_none() {
        // 获取当前模块声明所在的文件路径
        let parent_file_path = module.ident.span().source_file().path();
        // 计算模块文件的路径：比如父文件是src/parent/mod.rs，模块是some_mod，路径就是src/parent/some_mod.rs
        let module_file_name = format!("{}.rs", module.ident);
        let module_file_path = parent_file_path.parent().unwrap().join(module_file_name);

        // 读取模块文件内容
        let content = fs::read_to_string(&module_file_path)
            .unwrap_or_else(|e| panic!("Failed to read module file {}: {}", module_file_path.display(), e));

        // 解析文件内容为Rust AST
        let module_content = parse_macro_input!(content as File);

        // 将解析后的内容设置到模块中
        module.content = Some((module.brace_token, module_content.items));
    }

    // 这里可以对模块内容进行任意修改，比如添加一个自动生成的函数
    if let Some((_, ref mut items)) = module.content {
        items.push(syn::parse_quote! {
            /// 由过程宏自动生成的函数
            pub fn generated_by_hello_macro() {
                println!("Hello from the generated function in non-inline module!");
            }
        });
    }

    // 生成修改后的代码
    quote!(#module).into()
}

步骤3：正确使用宏 #

现在不需要在模块文件中写#![hello]了，而是在父模块的mod声明上应用宏：

// 在main.rs或者父模块文件中
#[hello]
mod some_mod;

关键注意事项 #

• 路径处理：上面的代码通过Span::source_file()获取父模块的文件路径，再拼接模块文件名，能正确处理嵌套模块的情况（比如src/a/b.rs这种层级）。
• 编译依赖：因为宏在编译时读取模块文件，所以当模块文件内容修改时，编译器会自动重新编译相关代码，符合Rust的编译规则。
• 错误处理：示例中用了unwrap和panic，实际项目中可以根据需求换成更优雅的错误处理（比如返回编译错误）。

为什么内联模块能正常工作？ #

内联模块mod some_mod { ... }的内容直接写在父模块里，编译器会将整个模块的AST传递给过程宏，所以宏能直接拿到所有内部项——这和非内联模块的编译流程差异导致了最初的问题。

内容的提问来源于stack exchange，提问作者amlo