rtic_syntax/parse/
init.rs

1use proc_macro2::TokenStream as TokenStream2;
2
3use syn::{parse, ItemFn};
4
5use crate::{
6    ast::{Init, InitArgs},
7    parse::util,
8};
9
10impl InitArgs {
11    pub(crate) fn parse(tokens: TokenStream2) -> parse::Result<Self> {
12        crate::parse::init_args(tokens)
13    }
14}
15
16impl Init {
17    pub(crate) fn parse(args: InitArgs, item: ItemFn) -> parse::Result<Self> {
18        let valid_signature = util::check_fn_signature(&item) && item.sig.inputs.len() == 1;
19
20        let span = item.sig.ident.span();
21
22        let name = item.sig.ident.to_string();
23
24        if valid_signature {
25            if let Ok((user_shared_struct, user_local_struct)) =
26                util::type_is_init_return(&item.sig.output, &name)
27            {
28                if let Some((context, Ok(rest))) = util::parse_inputs(item.sig.inputs, &name) {
29                    if rest.is_empty() {
30                        return Ok(Init {
31                            args,
32                            attrs: item.attrs,
33                            context,
34                            name: item.sig.ident,
35                            stmts: item.block.stmts,
36                            user_shared_struct,
37                            user_local_struct,
38                        });
39                    }
40                }
41            }
42        }
43
44        Err(parse::Error::new(
45            span,
46            format!(
47                "the `#[init]` function must have signature `fn({}::Context) -> (Shared resources struct, Local resources struct, {0}::Monotonics)`",
48                name
49            ),
50        ))
51    }
52}