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
extern crate proc_macro; mod base; mod convert; mod serde; use quote::quote; use syn::ItemEnum; use syn::{parse_macro_input, parse_quote, Path}; #[proc_macro_attribute] pub fn int_enum( args: proc_macro::TokenStream, input: proc_macro::TokenStream, ) -> proc_macro::TokenStream { let mut input = parse_macro_input!(input as ItemEnum); base::add_missing_debug(&mut input.attrs); let enum_ = input.ident.clone(); let int_ = match syn::parse::<base::IntType>(args) { Ok(int_) => int_.ty, Err(err) => { let err = err.to_compile_error(); return quote!(#input #err).into(); } }; let crate_ = base::crate_name(); let core: Path = parse_quote!(#crate_::__core); let base_impl = base::base_impl(&input, &crate_, &core, &enum_, &int_); let serde_impls = serde::serde_impls(&crate_, &core, &enum_, &int_); let convert_impls = convert::convert_impls(&crate_, &core, &enum_, &int_); proc_macro::TokenStream::from(quote! { #input #base_impl #serde_impls #convert_impls }) }