anchor-cli-macros 1.1.2

Macros used internally by the Anchor CLI
Documentation 
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use {
    proc_macro2::TokenStream,
    quote::{format_ident, quote},
    syn::{parse_macro_input, DataEnum, DataStruct, DeriveInput, Fields},
};

#[proc_macro_derive(AbsolutePath)]
pub fn absolute_path_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let input = parse_macro_input!(input as DeriveInput);
    let constructor = match input.data {
        syn::Data::Struct(data_struct) => derive_struct(data_struct),
        syn::Data::Enum(data_enum) => derive_enum(data_enum),
        syn::Data::Union(_) => {
            quote! { compile_error!("unions are not supported") }
        }
    };
    let name = input.ident;

    quote! {
        // Some fields/commands in the CLI are deprecated
        #[allow(deprecated)]
        impl AbsolutePath for #name {
            fn absolute(self) -> Self {
                #constructor
            }
        }
    }
    .into()
}

fn derive_struct(input: DataStruct) -> TokenStream {
    match input.fields {
        Fields::Named(fields) => {
            let members: Vec<_> = fields
                .named
                .into_iter()
                .map(|field| field.ident.expect("named field"))
                .collect();
            quote! {
                Self {
                    #(#members: self.#members.absolute()),*
                }
            }
        }
        Fields::Unnamed(fields) => {
            let members: Vec<_> = (0..fields.unnamed.len()).map(syn::Index::from).collect();
            quote! {
                Self(
                    #(self.#members.absolute()),*
                )
            }
        }
        Fields::Unit => quote! { Self },
    }
}

fn derive_enum(input: DataEnum) -> TokenStream {
    let variants = input.variants.into_iter().map(|variant| {
        let name = variant.ident;
        match variant.fields {
            Fields::Named(fields) => {
                let members: Vec<_> = fields
                    .named
                    .into_iter()
                    .map(|field| field.ident.expect("named field"))
                    .collect();
                quote! {
                    Self::#name { #(#members),* } => Self::#name {
                        #(#members: #members.absolute()),*
                    }
                }
            }
            Fields::Unnamed(fields) => {
                let members: Vec<_> = (0..fields.unnamed.len())
                    .map(|index| format_ident!("field_{index}"))
                    .collect();
                quote! {
                    Self::#name(#(#members),*) => Self::#name(
                        #(#members.absolute()),*
                    )
                }
            }
            Fields::Unit => quote! {
                Self::#name => Self::#name
            },
        }
    });
    quote! {
        match self {
            #(#variants),*
        }
    }
}