1 files changed, 84 insertions, 0 deletions

diff --git a/src/parser/clang/entities.rs b/src/parser/clang/entities.rs
index 8da85d9..0e9305b 100644
--- a/src/parser/clang/entities.rs
+++ b/src/parser/clang/entities.rs
@@ -16,6 +16,8 @@ pub(super) enum ClangEntityKind {
     Struct(StructKind),
     Function(FunctionKind),
     Typedef,
+    Enum,
+    EnumConstant,
 }
 
 #[derive(Debug, Clone, Copy, PartialEq, Eq)]
@@ -47,6 +49,8 @@ impl ClangEntityKind {
             ClangEntityKind::Function(FunctionKind::Function) => "function",
             ClangEntityKind::Function(FunctionKind::Method) => "method",
             ClangEntityKind::Typedef => "typedef",
+            ClangEntityKind::Enum => "enum",
+            ClangEntityKind::EnumConstant => "enum-constant",
         }
     }
 
@@ -60,6 +64,8 @@ impl ClangEntityKind {
             ClangEntityKind::Function(FunctionKind::Function) => "functions",
             ClangEntityKind::Function(FunctionKind::Method) => "methods",
             ClangEntityKind::Typedef => "typedefs",
+            ClangEntityKind::Enum => "enums",
+            ClangEntityKind::EnumConstant => "enum-constants",
         }
     }
 }
@@ -89,6 +95,8 @@ impl TryFrom<EntityKind> for ClangEntityKind {
             EntityKind::TypedefDecl
             | EntityKind::TypeAliasDecl
             | EntityKind::TypeAliasTemplateDecl => ClangEntityKind::Typedef,
+            EntityKind::EnumDecl => ClangEntityKind::Enum,
+            EntityKind::EnumConstantDecl => ClangEntityKind::EnumConstant,
             kind => return Err(TryFromEntityKindError(format!("{:?}", kind))),
         })
     }
@@ -118,6 +126,9 @@ pub(super) trait ClangEntity {
     fn get_member_typedefs(&mut self) -> Option<&mut BTreeMap<Usr, Described<Typedef>>> {
         None
     }
+    fn get_member_enums(&mut self) -> Option<&mut BTreeMap<Usr, Described<Enum>>> {
+        None
+    }
 }
 
 /*
@@ -178,6 +189,15 @@ where
     }
 }
 
+impl<U> NamespaceParentManipulation<Enum> for U
+where
+    U: ClangEntity,
+{
+    fn get_members_mut(&mut self) -> Option<&mut BTreeMap<Usr, Described<Enum>>> {
+        self.get_member_enums()
+    }
+}
+
 fn entity_id(usr: Usr) -> EntityId {
     // This is a somewhat ugly workaround, because Pandoc parses markdown identifiers as alphanum
     // or one of "-_:.", which means Pandoc can't parse libclang's USRs in title ids and related.
@@ -236,6 +256,7 @@ pub(super) struct Namespace {
     pub(super) member_structs: BTreeMap<Usr, Described<Struct>>,
     pub(super) member_functions: BTreeMap<Usr, Described<Function>>,
     pub(super) member_typedefs: BTreeMap<Usr, Described<Typedef>>,
+    pub(super) member_enums: BTreeMap<Usr, Described<Enum>>,
 }
 
 impl ClangEntity for Namespace {
@@ -258,6 +279,9 @@ impl ClangEntity for Namespace {
     fn get_member_typedefs(&mut self) -> Option<&mut BTreeMap<Usr, Described<Typedef>>> {
         Some(&mut self.member_typedefs)
     }
+    fn get_member_enums(&mut self) -> Option<&mut BTreeMap<Usr, Described<Enum>>> {
+        Some(&mut self.member_enums)
+    }
 }
 
 /*
@@ -279,6 +303,8 @@ impl From<Namespace> for PartialEntity {
         append_children(&mut children, entity.member_structs);
         append_children(&mut children, entity.member_functions);
         append_children(&mut children, entity.member_typedefs);
+        append_children(&mut children, entity.member_enums);
+
         PartialEntity {
             properties: Default::default(),
             children,
@@ -328,6 +354,7 @@ pub(super) struct Struct {
     pub(super) member_structs: BTreeMap<Usr, Described<Struct>>,
     pub(super) member_functions: BTreeMap<Usr, Described<Function>>,
     pub(super) member_typedefs: BTreeMap<Usr, Described<Typedef>>,
+    pub(super) member_enums: BTreeMap<Usr, Described<Enum>>,
 }
 
 impl ClangEntity for Struct {
@@ -347,6 +374,9 @@ impl ClangEntity for Struct {
     fn get_member_typedefs(&mut self) -> Option<&mut BTreeMap<Usr, Described<Typedef>>> {
         Some(&mut self.member_typedefs)
     }
+    fn get_member_enums(&mut self) -> Option<&mut BTreeMap<Usr, Described<Enum>>> {
+        Some(&mut self.member_enums)
+    }
 }
 
 impl From<Struct> for PartialEntity {
@@ -356,6 +386,7 @@ impl From<Struct> for PartialEntity {
         append_children(&mut children, entity.member_structs);
         append_children(&mut children, entity.member_functions);
         append_children(&mut children, entity.member_typedefs);
+        append_children(&mut children, entity.member_enums);
 
         PartialEntity {
             properties: Default::default(),
@@ -455,3 +486,56 @@ impl From<Typedef> for PartialEntity {
         }
     }
 }
+
+#[derive(Debug, Clone)]
+pub(super) struct Enum {
+    pub(super) underlying_type: String,
+    pub(super) constants: BTreeMap<Usr, Described<EnumConstant>>,
+}
+
+impl ClangEntity for Enum {
+    fn kind(&self) -> ClangEntityKind {
+        ClangEntityKind::Enum
+    }
+}
+
+impl From<Enum> for PartialEntity {
+    fn from(entity: Enum) -> Self {
+        let mut properties = Map::with_capacity(1);
+        properties.insert(
+            String::from("underlying-type"),
+            JSONValue::String(entity.underlying_type),
+        );
+
+        let mut children = Default::default();
+        append_children(&mut children, entity.constants);
+
+        PartialEntity {
+            properties,
+            children,
+        }
+    }
+}
+
+#[derive(Debug, Clone)]
+pub(super) struct EnumConstant {
+    pub(super) value: String,
+}
+
+impl ClangEntity for EnumConstant {
+    fn kind(&self) -> ClangEntityKind {
+        ClangEntityKind::EnumConstant
+    }
+}
+
+impl From<EnumConstant> for PartialEntity {
+    fn from(entity: EnumConstant) -> Self {
+        let mut properties = Map::with_capacity(1);
+        properties.insert(String::from("value"), JSONValue::String(entity.value));
+
+        PartialEntity {
+            properties,
+            children: Default::default(),
+        }
+    }
+}