itroduce FunctionDescriptor

2 files changed, 30 insertions, 4 deletions

diff --git a/crates/ra_analysis/src/hir/function/mod.rs b/crates/ra_analysis/src/hir/function/mod.rs
index 86eee5e93..c8af6bc21 100644
--- a/crates/ra_analysis/src/hir/function/mod.rs
+++ b/crates/ra_analysis/src/hir/function/mod.rs
@@ -1,7 +1,10 @@
 pub(super) mod imp;
 mod scope;
 
-use std::cmp::{max, min};
+use std::{
+    cmp::{max, min},
+    sync::Arc,
+};
 
 use ra_syntax::{
     ast::{self, AstNode, DocCommentsOwner, NameOwner},
@@ -9,6 +12,7 @@ use ra_syntax::{
 };
 
 use crate::{
+    hir::HirDatabase,
     syntax_ptr::SyntaxPtr, FileId,
     loc2id::IdDatabase,
 };
@@ -23,6 +27,25 @@ impl FnId {
     }
 }
 
+pub(crate) struct FunctionDescriptor {
+    fn_id: FnId,
+}
+
+impl FunctionDescriptor {
+    pub(crate) fn guess_from_source(
+        db: &impl HirDatabase,
+        file_id: FileId,
+        fn_def: ast::FnDef,
+    ) -> FunctionDescriptor {
+        let fn_id = FnId::get(db, file_id, fn_def);
+        FunctionDescriptor { fn_id }
+    }
+
+    pub(crate) fn scope(&self, db: &impl HirDatabase) -> Arc<FnScopes> {
+        db.fn_scopes(self.fn_id)
+    }
+}
+
 #[derive(Debug, Clone)]
 pub struct FnDescriptor {
     pub name: String,
diff --git a/crates/ra_analysis/src/hir/mod.rs b/crates/ra_analysis/src/hir/mod.rs
index 232a8558b..edeaeb8e6 100644
--- a/crates/ra_analysis/src/hir/mod.rs
+++ b/crates/ra_analysis/src/hir/mod.rs
@@ -21,7 +21,7 @@ use crate::{
     db::SyntaxDatabase,
     hir::function::{resolve_local_name, FnId, FnScopes},
     hir::module::{
-        ModuleId, ModuleTree, ModuleSource, ModuleDescriptor,
+        ModuleId, ModuleTree, ModuleSource,
         nameres::{ItemMap, InputModuleItems, FileItems}
     },
     input::SourceRootId,
@@ -30,8 +30,11 @@ use crate::{
     Cancelable,
 };
 
-pub(crate) use self::path::{Path, PathKind};
-pub(crate) use self::module::nameres::FileItemId;
+pub(crate) use self::{
+    path::{Path, PathKind},
+    module::{ModuleDescriptor, nameres::FileItemId},
+    function::FunctionDescriptor,
+};
 
 salsa::query_group! {
 pub(crate) trait HirDatabase: SyntaxDatabase + IdDatabase {