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

mod arc;
mod r#box;
mod r#mut;
mod rc;
mod r#ref;

#[derive(Debug, PartialEq, Eq, Hash)]
pub enum Derive {
    Box,
    Ref,
    Mut,
    Rc,
    Arc,
}

impl Derive {
    pub fn from_str(s: &str) -> Option<Self> {
        match s {
            "Box" => Some(Derive::Box),
            "Ref" => Some(Derive::Ref),
            "Mut" => Some(Derive::Mut),
            "Rc" => Some(Derive::Rc),
            "Arc" => Some(Derive::Arc),
            _ => None,
        }
    }

    pub fn from_path(p: &syn::Path) -> Option<Self> {
        p.segments
            .first()
            .and_then(|s| Self::from_str(&s.ident.to_string()))
    }

    pub fn defer_trait_methods(&self, trait_: &syn::ItemTrait) -> syn::Result<syn::ItemImpl> {
        match self {
            Derive::Box => self::r#box::derive(trait_),
            Derive::Ref => self::r#ref::derive(trait_),
            Derive::Mut => self::r#mut::derive(trait_),
            Derive::Rc => self::rc::derive(trait_),
            Derive::Arc => self::arc::derive(trait_),
        }
    }
}