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 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
extern crate proc_macro; use crate::proc_macro::TokenStream; use quote::quote; use syn; #[proc_macro_derive(Deserialize)] pub fn deserialize_macro_derive(input: TokenStream) -> TokenStream { let ast = syn::parse(input).unwrap(); impl_deserialize_macro(&ast) } fn impl_deserialize_macro(ast: &syn::DeriveInput) -> TokenStream { let name = &ast.ident; let generics = &ast.generics; let mut field_list = quote! {}; let mut body = quote! {}; match &ast.data { syn::Data::Struct(data) => { for field in data.fields.iter() { let field_type = &field.ty; match &field.ident { Some(field_name) => { body = quote! { #body let #field_name = match <#field_type>::deserialize(de) { Ok(v) => v, Err(e) => { return Err(ensicoin_serializer::Error::Message(format!( "Error in reading {} {}: {}", stringify!(#name), stringify!(#field_name), e ))); } }; }; field_list = quote! {#field_list #field_name,}; } None => panic!("Can't derive unamed field in {}", name), } } } _ => panic!("Can only derive struts, {} is invalid", name), }; let gen = quote! { impl #generics Deserialize for #name #generics { fn deserialize( de: &mut ensicoin_serializer::Deserializer, ) -> ensicoin_serializer::Result<Self> { #body Ok(#name {#field_list }) } } }; gen.into() } #[proc_macro_derive(Serialize)] pub fn serialize_macro_derive(input: TokenStream) -> TokenStream { let ast = syn::parse(input).unwrap(); impl_serialize_macro(&ast) } fn impl_serialize_macro(ast: &syn::DeriveInput) -> TokenStream { let name = &ast.ident; let generics = &ast.generics; let mut body = quote! {}; match &ast.data { syn::Data::Struct(data) => { for field in data.fields.iter() { match &field.ident { Some(field_name) => { body = quote! { #body v.extend_from_slice(&self.#field_name.serialize()); } } None => panic!("Can't derive unamed field in {}", name), } } } _ => panic!("Can only derive structs, {} is invalid", name), } let gen = quote! { impl #generics Serialize for #name #generics { fn serialize(&self) -> bytes::Bytes { let mut v = bytes::Bytes::new(); #body v } } }; gen.into() }