如何向Rust过程宏动态传递file!()获取的文件名？

你的过程宏当前强制要求输入是字符串字面量（LitStr），但file!()是一个编译期宏，它的展开结果是字符串字面量，但直接传递给my_macro时，过程宏接收到的是file!()这个未展开的TokenStream，而非最终的字符串字面量，因此解析失败。

根据你的需求场景，有两种可行方案：

方案1：修改过程宏接受任意表达式（运行时处理文件名） #

如果你的逻辑是在运行时处理文件名，不需要在编译期获取文件名的具体值，可以修改过程宏接受任意表达式，直接将表达式嵌入生成的代码中：

// lib.rs - proc_macro crate
use proc_macro::TokenStream;
use quote::quote;
use syn::parse_macro_input;
use syn::Expr;

#[proc_macro]
pub fn my_macro(input: TokenStream) -> TokenStream {
    // 解析输入为任意表达式，而非仅字符串字面量
    let input_expr = parse_macro_input!(input as Expr);

    // 生成代码：这里可以添加你的文件名处理逻辑，比如提取目录名
    let expanded = quote! {
        {
            let file_path = #input_expr;
            std::path::Path::new(file_path)
                .parent()
                .expect("Failed to get parent directory")
                .to_str()
                .expect("Path contains non-UTF-8 characters")
        }
    };

    TokenStream::from(expanded)
}

调用方式不变：

fn main() {
    let dir_name = my_macro!(file!());
    println!("Dir name: {:?}", dir_name);
}

方案2：通过宏_rules!中转，传递编译期展开后的字符串字面量 #

如果需要在编译期处理文件名（比如根据文件名生成特定代码、校验文件存在等），需要让过程宏接收到最终的字符串字面量。可以用macro_rules!作为中间层，先展开file!()再传递给过程宏：

首先在proc_macro crate中添加一个宏_rules!：

// lib.rs - proc_macro crate
#[macro_export]
macro_rules! my_macro_from_file {
    () => {
        $crate::my_macro!(file!())
    };
}

原过程宏保持不变（仍解析为LitStr）：

#[proc_macro]
pub fn my_macro(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as LitStr);
    let file_name = input.value();

    // 编译期处理文件名，比如提取目录名
    let dir_name = std::path::Path::new(&file_name)
        .parent()
        .unwrap()
        .to_str()
        .unwrap();

    let expanded = quote! {
        #dir_name
    };

    TokenStream::from(expanded)
}

调用时使用中转宏：

fn main() {
    let dir_name = my_macro_from_file!();
    println!("Dir name: {:?}", dir_name);
}

这样my_macro_from_file!()会先展开file!()为字符串字面量，再传递给my_macro，过程宏就能在编译期拿到具体的文件名进行处理。

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