1mod gen;
2mod input;
3mod validate;
4
5use gen::*;
6use input::*;
7use validate::*;
8
9use darling::{Error, FromDeriveInput};
10use proc_macro::TokenStream;
11
12#[proc_macro_derive(EnumPtr, attributes(enum_ptr))]
13pub fn enum_ptr(input: TokenStream) -> TokenStream {
14 let input = syn::parse_macro_input!(input as syn::DeriveInput);
15 match enum_ptr_inner(&input) {
16 Ok(output) => output,
17 Err(err) => err.write_errors().into(),
18 }
19}
20
21fn enum_ptr_inner(input: &syn::DeriveInput) -> Result<TokenStream, Error> {
22 let input = Input::from_derive_input(input)?;
23
24 validate_input(&input)?;
25
26 let mut output = gen_basic(&input);
27 if let Some(conf) = input.borrow.clone() {
28 output.extend(gen_borrow(&input, &conf.unwrap_or_default()));
29 }
30 if let Some(conf) = input.borrow_mut.clone() {
31 output.extend(gen_borrow_mut(&input, &conf.unwrap_or_default()));
32 }
33
34 Ok(output)
35}