impl_opaque_macro/
lib.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
mod attr;
mod expand;
mod gen;

use attr::Attr;
use expand::FieldExpander;
use gen::Gen;
use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, ImplItem, ImplItemFn, ItemImpl};

#[proc_macro_attribute]
pub fn opaque(attr: TokenStream, item: TokenStream) -> TokenStream {
    let attr: Attr = parse_macro_input!(attr);
    let mut block: ItemImpl = parse_macro_input!(item);

    let mut expander = FieldExpander::new();
    for item in &mut block.items {
        if let ImplItem::Fn(ImplItemFn { block, .. }) = item {
            expander.expand(block);
        }
    }

    let gen = Gen {
        struct_attrs: block.attrs.drain(..).collect(),
        attr,
        ty: *block.self_ty.clone(),
        generics: block.generics.clone(),
        fields: expander.fields,
    };

    TokenStream::from(quote!(
        #gen
        #block
    ))
}