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

use super::r#enum::generate_match_arms;
use proc_macro2::Span;
use quote::quote;
use syn::ItemEnum;

pub fn generate(mut input: ItemEnum) -> proc_macro::TokenStream {
    if input.generics.params.len() > 1 {
        let err = syn::Error::new(
            Span::call_site(),
            "Enums with multiple generic arguments are currently not supported.",
        )
        .into_compile_error();

        return proc_macro::TokenStream::from(quote!(#err));
    }

    let ident = &input.ident;

    let mut variants = &mut input.variants;
    let canonize_impl = generate_match_arms(&mut variants, &ident, &ident, true);
    let humanize_impl = generate_match_arms(&mut variants, &ident, &ident, false);

    proc_macro::TokenStream::from(quote! {
        #[automatically_derived]
        impl fadroma::prelude::Canonize for #ident<cosmwasm_std::Addr> {
            type Output = #ident<cosmwasm_std::CanonicalAddr>;

            fn canonize(self, api: &dyn cosmwasm_std::Api) -> cosmwasm_std::StdResult<Self::Output> {
                Ok(match self {
                    #canonize_impl
                })
            }
        }

        #[automatically_derived]
        impl fadroma::prelude::Humanize for #ident<cosmwasm_std::CanonicalAddr> {
            type Output = #ident<cosmwasm_std::Addr>;

            fn humanize(self, api: &dyn cosmwasm_std::Api) -> cosmwasm_std::StdResult<Self::Output> {
                Ok(match self {
                    #humanize_impl
                })
            }
        }
    })
}