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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use maplit::btreemap;
use std::collections::BTreeMap;
use crate::object::Object;
use serde::Serialize;
use crate::comment::Comment;
use crate::r#enum::member::Member;
use crate::traits::documentable::Documentable;
use crate::traits::named::Named;

#[derive(Debug, Serialize)]
pub struct Enum {
    pub path: Vec<String>,
    pub comment: Option<Comment>,
    pub option: bool,
    pub interface: bool,
    pub members: Vec<Member>,
    pub data: BTreeMap<String, Object>,
    pub cache: Cache,
}

impl Enum {

    pub fn new() -> Self {
        Self {
            path: vec![],
            comment: None,
            option: false,
            interface: false,
            members: vec![],
            data: btreemap! {},
            cache: Cache {
                member_names: vec![]
            }
        }
    }

    pub fn finalize(&mut self) {
        self.cache.member_names = self.members.iter().map(|m| m.name.clone()).collect();
    }

    pub fn members(&self) -> &Vec<Member> {
        &self.members
    }
}

#[derive(Debug, Serialize)]
pub struct Cache {
    pub member_names: Vec<String>,
}

impl Named for Enum {

    fn name(&self) -> &str {
        self.path.last().unwrap().as_str()
    }
}

impl Documentable for Enum {

    fn comment(&self) -> Option<&Comment> {
        self.comment.as_ref()
    }

    fn kind(&self) -> &'static str {
        "enum"
    }
}