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}