typhoon-context-macro 0.1.0-alpha.12

TODO
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

use {
    crate::{accounts::Account, injector::LifetimeInjector, remover::AttributeRemover},
    syn::{parse::Parse, spanned::Spanned, visit_mut::VisitMut, Item, ItemStruct},
    typhoon_syn::arguments::Arguments,
};

pub struct Context {
    pub item_struct: ItemStruct,
    pub accounts: Vec<Account>,
    pub args: Option<Arguments>,
}

impl Parse for Context {
    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
        let mut item: Item = input.parse()?;
        LifetimeInjector.visit_item_mut(&mut item);

        match item {
            Item::Struct(mut item_struct) => {
                let args = item_struct
                    .attrs
                    .iter()
                    .find(|attr| attr.meta.path().is_ident("args"))
                    .map(Arguments::try_from)
                    .transpose()?;

                AttributeRemover::new("args").visit_item_struct_mut(&mut item_struct);

                let accounts = item_struct
                    .fields
                    .iter_mut()
                    .map(Account::try_from)
                    .collect::<Result<Vec<Account>, syn::Error>>()?;

                Ok(Context {
                    item_struct,
                    accounts,
                    args,
                })
            }
            _ => Err(syn::Error::new(
                item.span(),
                "#[context] is only implemented for struct",
            )),
        }
    }
}