educe/trait_handlers/deref_mut/
mod.rs

1mod models;
2
3mod deref_mut_enum;
4mod deref_mut_struct;
5
6use deref_mut_enum::DerefMutEnumHandler;
7use deref_mut_struct::DerefMutStructHandler;
8use proc_macro2::TokenStream;
9use syn::{Data, DeriveInput, Meta};
10
11use super::TraitHandler;
12use crate::{panic, Trait};
13
14pub struct DerefMutHandler;
15
16impl TraitHandler for DerefMutHandler {
17    fn trait_meta_handler(
18        ast: &DeriveInput,
19        tokens: &mut TokenStream,
20        traits: &[Trait],
21        meta: &Meta,
22    ) {
23        match ast.data {
24            Data::Struct(_) => {
25                DerefMutStructHandler::trait_meta_handler(ast, tokens, traits, meta);
26            },
27            Data::Enum(_) => {
28                DerefMutEnumHandler::trait_meta_handler(ast, tokens, traits, meta);
29            },
30            Data::Union(_) => panic::trait_not_support_union(Trait::DerefMut),
31        }
32    }
33}