introduce EnumId

1 files changed, 19 insertions, 0 deletions

diff --git a/crates/ra_hir/src/ids.rs b/crates/ra_hir/src/ids.rs
index 2791149dd..530fdf5cd 100644
--- a/crates/ra_hir/src/ids.rs
+++ b/crates/ra_hir/src/ids.rs
@@ -15,6 +15,7 @@ pub struct HirInterner {
     macros: LocationIntener<MacroCallLoc, MacroCallId>,
     fns: LocationIntener<FunctionLoc, FunctionId>,
     structs: LocationIntener<StructLoc, StructId>,
+    enums: LocationIntener<EnumLoc, EnumId>,
 }
 
 impl HirInterner {
@@ -213,6 +214,24 @@ impl StructLoc {
     }
 }
 
+#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
+pub struct EnumId(RawId);
+impl_arena_id!(EnumId);
+
+pub(crate) type EnumLoc = ItemLoc<ast::EnumDef>;
+
+impl EnumId {
+    pub(crate) fn loc(self, db: &impl AsRef<HirInterner>) -> EnumLoc {
+        db.as_ref().enums.id2loc(self)
+    }
+}
+
+impl EnumLoc {
+    pub(crate) fn id(&self, db: &impl AsRef<HirInterner>) -> EnumId {
+        db.as_ref().enums.loc2id(&self)
+    }
+}
+
 /// Def's are a core concept of hir. A `Def` is an Item (function, module, etc)
 /// in a specific module.
 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]