dos-uid-derive 3.0.0

dos-actors UniqueIdentifier derive macro
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
62
63

use quote::quote;
use syn::{
    parse::{Parse, ParseStream},
    Ident, LitInt, Token, Type,
};

use crate::{Expand, Expanded};

/// UID attributes
///
/// #[uid(data = <type>, port = <u32>)]
#[derive(Debug, Clone, Default)]
pub struct Attributes {
    pub data: Option<Type>,
    pub port: Option<LitInt>,
}

impl Parse for Attributes {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        let mut uid_attrs: Attributes = Default::default();
        while let Ok(key) = input.parse::<Ident>() {
            let _ = input.parse::<Token!(=)>()?;
            if key == "data" {
                uid_attrs.data = input.parse::<Type>().ok();
            }
            if key == "port" {
                uid_attrs.port = input.parse::<LitInt>().ok();
            }
            let Ok(_) = input.parse::<Token!(,)>() else {
                return Ok(uid_attrs);
            };
        }
        Ok(uid_attrs)
    }
}
impl Expand for Attributes {
    fn expand(&self, ident: &Ident) -> Expanded {
        match (&self.data, &self.port) {
            (None, None) => quote! {
                impl ::gmt_dos_clients::interface::UniqueIdentifier for #ident {
                    type DataType = Vec<f64>;
                },
            },
            (None, Some(port)) => quote! {
                impl ::gmt_dos_clients::interface::UniqueIdentifier for #ident {
                    const PORT: u32 = #port;
                    type DataType = Vec<f64>;
                },
            },
            (Some(data), None) => quote! {
                impl ::gmt_dos_clients::interface::UniqueIdentifier for #ident {
                    type DataType = #data;
                },
            },
            (Some(data), Some(port)) => quote! {
                impl ::gmt_dos_clients::interface::UniqueIdentifier for #ident {
                    const PORT: u32 = #port;
                    type DataType = #data;
                },
            },
        }
    }
}