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

use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, Data, DeriveInput};

#[proc_macro_derive(Serialize)]
pub fn derive_serialize(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as DeriveInput);
    let name = input.ident;

    let out = match input.data {
        Data::Struct(s) => {
            let fields = s.fields.into_iter().map(|field| field.ident.unwrap());
            quote! {
                impl ::sbs_api::Serialize for #name {
                    fn serialize(&self, sbi: &mut ::serialize_internal::SBI) {
                        #(
                            Serialize::serialize(&self.#fields, sbi);
                        )*
                    }
                }
            }
        },
        _ => todo!()
    };

    out.into()
}

#[proc_macro_derive(DeSerialize)]
pub fn derive_deserialize(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as DeriveInput);
    let name = input.ident;

    let out = match input.data {
        Data::Struct(s) => {
            let fields = s.fields.clone().into_iter().map(|field| field.ident.unwrap());
            let types = s.fields.clone().into_iter().map(|field| field.ty);
            quote! {
                impl ::sbs_api::DeSerialize for #name {
                    fn deserialize(sbi: &mut SBI, offset: &mut usize) -> Result<Self, ()> where Self: Sized {
                        Ok(Self {
                            #(  
                                #fields: #types ::deserialize(sbi, offset)?,
                            )*
                        })
                    }
                }
            }
        },
        _ => todo!()
    };

    out.into()
}