用于在函数开头注入代码的过程属性宏

Question

我正在尝试构建一个简单的属性，它将在函数的开头注入一个 let 绑定，因此结果将是：

#[foo]
fn bar(){
    // start injected code
    let x = 0;
    // end injected code

    ...
}

我已经做到了：

use proc_macro::TokenStream;
use quote::{quote, ToTokens};
use syn::{parse_macro_input, ItemFn, Item, Stmt};



#[proc_macro_attribute]
pub fn hello(attr: TokenStream, stream: TokenStream) -> TokenStream {
    let input = parse_macro_input!(stream as ItemFn);

    let block = input.block;
    let res = quote! {let x = 1;};

    // doesn't work, can't convert TokenStream into Stmt
    block.stmts.insert(0, res.into());
    TokenStream::from(input.into_token_stream())
}

但是，我在处理这些物品时遇到了麻烦。例如，blockis 类型，现在一个语句由 ( , , )Vec<Stmt>片段组成，当尝试处理这些片段时，我只是迷失了方向。我觉得我缺少一些关于如何处理这些部分的信息，但是查看提供的示例并没有帮助，而且很多在线资源都已经过时了。LocalItem ExprSemitrace-vars

我还尝试了一种非常愚蠢的方法，即创建一个ItemFnusing 引用，解析它并Stmt从中获取，但是由于TokenStream实际上来自两个不同的板条箱，我收到了另一个错误，proc_macro并且proc_macro2：

#[proc_macro_attribute]
pub fn hello(attr: TokenStream, stream: TokenStream) -> TokenStream {
    let input = parse_macro_input!(stream as ItemFn);


    let res = quote! {fn () { let x = 1; } };
    let res = parse_macro_input!(res as ItemFn);

    let block = input.block;
    block.stmts.insert(0, res.block.stmts[0].clone());
    TokenStream::from(input.into_token_stream())
}

error[E0308]: mismatched types
  --> bsub-procedural/src/lib.rs:13:15
   |
13 |     let res = parse_macro_input!(res as ItemFn);
   |               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ expected struct `proc_macro::TokenStream`, found struct `proc_macro2::TokenStream`

运行于

[dependencies]
syn = { version = "1.0.67", features=["full", "fold", "printing"] }
quote = "1.0.9"

Answer 1

在某些情况下，“愚蠢的方法”实际上是一个非常有效的解决方案。要解析 a proc_macro2::TokenStream，请使用syn::parse2而不是宏。

解析看起来肯定是浪费了，因为结果立即再次转换为令牌流。但对于小事情来说还是可以的。另一种方法是将函数的所有部分单独转换为标记，例如：

let ItemFn { attrs, vis, sig, block } = input;
let stmts = &block.stmts;
quote! {
    #(#attrs)* #vis #sig {
        let x = 1; // <- your extra code
        #(#stmts)*
    }
}