如何在Rust过程宏中获取调用者位置以读取相对路径文件？

要实现过程宏读取调用者目录下的文件，核心是利用Rust过程宏提供的Span API——它能在编译时获取宏调用位置的源文件信息，进而解析出绝对路径。以下是具体实现方案：

核心思路 #

过程宏运行在编译阶段，Span可以获取宏调用所在的源文件绝对路径，以此为基础拼接目标文件的路径，再读取文件内容生成常量。

具体实现步骤 #

1. 解析宏输入并获取调用位置的Span #

在过程宏入口函数中，先解析传入的文件名参数，然后通过参数的Span获取调用者的源文件路径：

use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, LitStr, punctuated::Punctuated, Token};

#[proc_macro]
pub fn my_proc_macro(input: TokenStream) -> TokenStream {
    // 解析逗号分隔的文件名列表
    let filenames = parse_macro_input!(input as Punctuated<LitStr, Token![,]>);
    
    // 获取宏调用位置的Span（取第一个参数的Span即可代表调用位置）
    let caller_span = filenames.first().expect("No filenames provided").span();
    let caller_file_path = caller_span.source_file().path();
    
    // 获取调用者源文件所在的目录
    let caller_dir = caller_file_path.parent()
        .expect("Failed to get parent directory of caller file");

2. 构建目标文件路径并读取内容 #

基于调用者目录拼接目标文件路径，读取文件内容后生成对应的常量：

let mut constants = Vec::new();
    
    for (idx, filename_lit) in filenames.iter().enumerate() {
        let base_name = filename_lit.value();
        let target_path = caller_dir.join(format!("{}.txt", base_name));
        
        // 读取文件内容（编译时执行，注意处理错误）
        let file_content = std::fs::read_to_string(&target_path)
            .unwrap_or_else(|err| panic!("Failed to read {}: {}", target_path.display(), err));
        
        // 生成常量标识符，比如FILE1、FILE2
        let const_ident = syn::Ident::new(&format!("FILE{}", idx + 1), filename_lit.span());
        
        constants.push(quote! {
            pub const #const_ident: &str = #file_content;
        });
    }

3. 返回生成的TokenStream #

将所有生成的常量组合成最终的TokenStream返回：

quote! {
        #(#constants)*
    }.into()
}

关键注意事项 #

• 绝对路径可靠性：Span::source_file().path()返回的是编译时解析的绝对路径，无论调用者是本地 crate 还是作为依赖引入的 crate，都能正确获取。
• 错误处理：实际使用中建议替换unwrap/panic为syn提供的编译错误提示（比如syn::Error::new_spanned），让开发者看到更友好的编译期错误。
• 跨平台兼容性：使用std::path::Path的方法拼接路径，避免手动拼接字符串导致的跨平台问题。
• 为什么其他方法不行：#[track_caller]是运行时特性，无法在编译阶段的过程宏中使用；include_str!和file!是编译器内置宏，展开时机晚于过程宏执行，过程宏无法提前利用它们的结果读取文件。

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