penum 0.1.30

Make enum conform to a given pattern
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
37
38
39
40
41
42
43
44
45
46
47
48

use syn::{
    parse_quote,
    punctuated::{IntoIter, Iter, Punctuated},
    Fields, FieldsNamed, FieldsUnnamed,
};

use crate::factory::{PatComposite, PatFieldKind};

impl From<&Fields> for PatComposite {
    fn from(value: &Fields) -> Self {
        match value {
            Fields::Named(FieldsNamed { named, brace_token }) => PatComposite::Named {
                parameters: parse_quote!(#named),
                delimiter: *brace_token,
            },
            Fields::Unnamed(FieldsUnnamed {
                unnamed,
                paren_token,
            }) => PatComposite::Unnamed {
                parameters: parse_quote!(#unnamed),
                delimiter: *paren_token,
            },
            Fields::Unit => PatComposite::Unit,
        }
    }
}

impl IntoIterator for PatComposite {
    type Item = PatFieldKind;
    type IntoIter = IntoIter<PatFieldKind>;

    fn into_iter(self) -> Self::IntoIter {
        match self {
            PatComposite::Named { parameters, .. } => parameters.into_iter(),
            PatComposite::Unnamed { parameters, .. } => parameters.into_iter(),
            _ => Punctuated::<PatFieldKind, ()>::new().into_iter(),
        }
    }
}

impl<'a> IntoIterator for &'a PatComposite {
    type Item = &'a PatFieldKind;
    type IntoIter = Iter<'a, PatFieldKind>;

    fn into_iter(self) -> Self::IntoIter {
        self.iter()
    }
}