auto-delegate-impl 0.1.2

Auto delegate allows you that automatic impl of traits and delegate their handling to child members.
Documentation 
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

use proc_macro::TokenStream;

use proc_macro2::Span;
use syn::{ItemEnum, ItemStruct};

use crate::derive::r#enum::try_expand_derive_enum;
use crate::derive::r#struct::try_expand_derive_delegate_struct;

mod by_fields;
mod r#struct;
mod r#enum;

pub fn expand_derive_delegate(input: TokenStream) -> proc_macro2::TokenStream {
    if let Ok(item_struct) = syn::parse::<ItemStruct>(input.clone()) {
        return expand_struct(item_struct);
    }

    if let Ok(item_enum) = syn::parse::<ItemEnum>(input) {
        return expand_enum(item_enum);
    }


    syn::Error::new(Span::call_site(), "Inherits must be Enum or Structs.").to_compile_error()
}


fn expand_struct(item_struct: ItemStruct) -> proc_macro2::TokenStream {
    try_expand_derive_delegate_struct(item_struct)
        .unwrap_or_else(|error| error.to_compile_error())
}


fn expand_enum(item_enum: ItemEnum) -> proc_macro2::TokenStream {
    try_expand_derive_enum(&item_enum)
        .unwrap_or_else(|error| error.to_compile_error())
}