Start code_model::Module

1 files changed, 22 insertions, 1 deletions

diff --git a/crates/ra_hir/src/code_model_api.rs b/crates/ra_hir/src/code_model_api.rs
index e8b3a1fb6..63e2e34e8 100644
--- a/crates/ra_hir/src/code_model_api.rs
+++ b/crates/ra_hir/src/code_model_api.rs
@@ -1,6 +1,6 @@
 use ra_db::{CrateId, Cancelable};
 
-use crate::{Module, Name, db::HirDatabase};
+use crate::{Name, db::HirDatabase, DefId};
 
 /// hir::Crate describes a single crate. It's the main inteface with which
 /// crate's dependencies interact. Mostly, it should be just a proxy for the
@@ -24,3 +24,24 @@ impl Crate {
         self.root_module_impl(db)
     }
 }
+
+#[derive(Debug, Clone, PartialEq, Eq, Hash)]
+pub struct Module {
+    pub(crate) def_id: DefId,
+}
+
+impl Module {
+    /// Returns the crate this module is part of.
+    pub fn krate(&self, db: &impl HirDatabase) -> Cancelable<Option<Crate>> {
+        self.krate_impl(db)
+    }
+
+    pub fn crate_root(&self, db: &impl HirDatabase) -> Cancelable<Module> {
+        self.crate_root_impl(db)
+    }
+
+    /// Finds a child module with the specified name.
+    pub fn child(&self, db: &impl HirDatabase, name: &Name) -> Cancelable<Option<Module>> {
+        self.child_impl(db, name)
+    }
+}