kstool-helper-generator 0.7.0

A macro help user create mpsc communications and other
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
49
50
51
52
53
54
55
56
57
58
59
60
61

use proc_macro2::{Literal, TokenStream};
use quote::quote;
use syn::{Data, DeriveInput, Ident};

use crate::basic::EnumDefinition;

pub(crate) fn parse_bit(input: DeriveInput) -> syn::Result<TokenStream> {
    //eprintln!("{:#?}", input);

    let vis = input.vis.clone();
    let ident = &input.ident;

    let Data::Enum(de) = input.data else {
        return Err(syn::Error::new_spanned(input, "Should use enum as input"));
    };

    let mut v = Vec::with_capacity(de.variants.len());
    let mut body = TokenStream::new();

    for (index, element) in de.variants.iter().enumerate() {
        //eprintln!("{element:?}");
        let ident = &element.ident;
        v.push(&element.ident);

        let index = Literal::usize_unsuffixed(index);

        //let index = index.to_string();

        body.extend(quote! {
            #ident = 1 << #index,
        });
    }

    //let mut checker = Vec::with_capacity(v.len());

    let mut impl_body = TokenStream::new();

    for ident in v {
        let name = format!(
            "is_{}",
            EnumDefinition::string_into_snake_case(&ident.to_string())
        );
        let fn_name = Ident::new(&name, ident.span());
        impl_body.extend(quote! {
            #vis fn #fn_name(&self) -> bool {
                matches!(&self, Self::#ident)
            }
        });
    }

    Ok(quote! {
        #[derive(Clone, Copy, Debug)]
        #vis enum #ident {
            #body
        }

        impl #ident {
            #impl_body
        }
    })
}