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
extern crate proc_macro; extern crate quote; extern crate syn; use proc_macro::TokenStream; use quote::quote; use syn::{parse_macro_input, DeriveInput, Ident}; #[proc_macro_derive(SerializableModel)] pub fn model(item: TokenStream) -> TokenStream { let input = parse_macro_input!(item as DeriveInput); let name = input.ident; let tokens = quote! { impl #name { pub fn from_value(value: serde_yaml::Value) -> Option<Box<dyn ReportableModel>> { match serde_yaml::from_value::<Self>(value) { Ok(model) => Some(Box::new(model)), Err(_) => None } } } impl SerializableModel for #name { fn get_type(&self) -> &'static str { stringify!(#name) } fn serialize(&self) -> serde_yaml::Value { serde_yaml::to_value(self).unwrap_or(serde_yaml::Value::Null) } } }; tokens.into() } #[proc_macro] pub fn register(item: TokenStream) -> TokenStream { let name = parse_macro_input!(item as Ident); let tokens = quote! { sim::models::model_factory::register( stringify!(#name), #name::from_value as sim::models::model_factory::ModelConstructor ); }; tokens.into() }