sandstone_derive/
lib.rs

1use proc_macro::TokenStream;
2
3use quote::quote;
4use syn::{Data, DeriveInput, Fields, Ident, parse_macro_input};
5use syn::__private::Span;
6
7/// Derive the `McSerialize` trait for a struct. This implies that all fields of the struct also
8/// implement `McSerialize`.
9#[proc_macro_derive(McSerialize)]
10pub fn derive_mc_serialize(input: TokenStream) -> TokenStream {
11	let input = parse_macro_input!(input as DeriveInput);
12	let name = &input.ident;
13	let fields = match &input.data {
14		Data::Struct(data) => match &data.fields {
15			Fields::Named(fields) => fields.named.iter().map(|field| {
16				let field_name = field.ident.as_ref().unwrap();
17				quote! {
18					self.#field_name.mc_serialize(serializer)?;
19				}
20			}).collect(),
21			Fields::Unnamed(fields) => fields.unnamed.iter().enumerate().map(|(i, _field)| {
22				let field_name = Ident::new(&format!("__{}", i), Span::call_site());
23				quote! {
24					self.#field_name.mc_serialize(serializer)?;
25				}
26			}).collect(),
27			Fields::Unit => vec![],
28		},
29		Data::Enum(_) => panic!("Enums are not supported"),
30		Data::Union(_) => panic!("Unions are not supported"),
31	};
32	let expanded = quote! {
33		impl McSerialize for #name {
34			fn mc_serialize(&self, serializer: &mut McSerializer) -> Result<(), SerializingErr> {
35				#(#fields)*
36				Ok(())
37			}
38		}
39	};
40	TokenStream::from(expanded)
41}
42
43#[proc_macro_derive(McDeserialize)]
44pub fn derive_mc_deserialize(input: TokenStream) -> TokenStream {
45	let input = parse_macro_input!(input as DeriveInput);
46	let name = &input.ident;
47	let fields = match &input.data {
48		Data::Struct(data) => match &data.fields {
49			Fields::Named(fields) => fields.named.iter().map(|field| {
50				let field_name = field.ident.as_ref().unwrap();
51				let field_type = &field.ty;
52				quote! {
53					let #field_name = <#field_type>::mc_deserialize(deserializer)?;
54				}
55			}).collect(),
56			Fields::Unnamed(fields) => fields.unnamed.iter().enumerate().map(|(i, field)| {
57				let field_name = Ident::new(&format!("__{}", i), Span::call_site());
58				let field_type = &field.ty;
59				quote! {
60					let #field_name = <#field_type>::mc_deserialize(deserializer)?;
61				}
62			}).collect(),
63			Fields::Unit => vec![],
64		},
65		Data::Enum(_) => panic!("Enums are not supported"),
66		Data::Union(_) => panic!("Unions are not supported"),
67	};
68	let expanded = quote! {
69		impl McDeserialize for #name {
70			fn mc_deserialize<'a>(deserializer: &'a mut McDeserializer) -> SerializingResult<'a, Self> {
71				#(#fields)*
72				Ok(Self {
73					#(
74						#fields
75					)*
76				})
77			}
78		}
79	};
80	TokenStream::from(expanded)
81}
sandstone_derive/lib.rs

sandstone_derive/
lib.rs
