typhoon-context-macro 0.1.0

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
48

use {
    crate::{injector::LifetimeInjector, remover::AttributeRemover},
    syn::{parse::Parse, spanned::Spanned, visit_mut::VisitMut, Item, ItemStruct},
    typhoon_syn::{constraints::CONSTRAINT_IDENT_STR, Arguments, InstructionAccount},
};

pub struct ParsingContext {
    pub item_struct: ItemStruct,
    pub accounts: Vec<InstructionAccount>,
    pub args: Option<Arguments>,
}

impl Parse for ParsingContext {
    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()?;

                let accounts: Vec<InstructionAccount> = item_struct
                    .fields
                    .iter()
                    .map(InstructionAccount::try_from)
                    .collect::<syn::Result<_>>()?;

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

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