Add DeclarationDescriptor and ReferenceDescriptor

Fixes #142
Fixes #146

1 files changed, 54 insertions, 3 deletions

diff --git a/crates/ra_analysis/src/descriptors/mod.rs b/crates/ra_analysis/src/descriptors/mod.rs
index 0c4991757..e27f8314a 100644
--- a/crates/ra_analysis/src/descriptors/mod.rs
+++ b/crates/ra_analysis/src/descriptors/mod.rs
@@ -5,16 +5,17 @@ use std::sync::Arc;
 
 use ra_syntax::{
     SmolStr,
-    ast::{FnDefNode},
+    ast::{self, AstNode, FnDefNode},
+    TextRange
 };
 
 use crate::{
     FileId, Cancelable,
     db::SyntaxDatabase,
     descriptors::module::{ModuleTree, ModuleId, ModuleScope},
-    descriptors::function::{FnId, FnScopes},
+    descriptors::function::{FnId, FnScopes, resolve_local_name},
     input::SourceRootId,
-    syntax_ptr::SyntaxPtrDatabase,
+    syntax_ptr::{SyntaxPtrDatabase, LocalSyntaxPtr},
 };
 
 
@@ -44,3 +45,53 @@ salsa::query_group! {
         }
     }
 }
+
+#[derive(Debug)]
+pub struct ReferenceDescriptor {
+    pub range: TextRange,
+    pub name: String
+}
+
+#[derive(Debug)]
+pub struct DeclarationDescriptor<'a> {
+    pat: ast::BindPat<'a>,
+    pub range: TextRange
+}
+
+impl<'a> DeclarationDescriptor<'a> {
+    pub fn new(pat: ast::BindPat) -> DeclarationDescriptor {
+        let range = pat.syntax().range();
+
+        DeclarationDescriptor {
+            pat,
+            range
+        }
+    }
+
+    pub fn find_all_refs(&self) -> Vec<ReferenceDescriptor> {
+        let name_ptr = LocalSyntaxPtr::new(self.pat.syntax());
+
+        let fn_def = match self.pat.syntax().ancestors().find_map(ast::FnDef::cast) {
+            Some(def) => def,
+            None => return Default::default()
+        };
+
+        let fn_scopes = FnScopes::new(fn_def);
+
+        let refs : Vec<_> = fn_def.syntax().descendants()
+            .filter_map(ast::NameRef::cast)
+            .filter(|name_ref| {
+                match resolve_local_name(*name_ref, &fn_scopes) {
+                    None => false,
+                    Some(entry) => entry.ptr() == name_ptr,
+                }
+            })
+            .map(|name_ref| ReferenceDescriptor {
+                name: name_ref.syntax().text().to_string(),
+                range : name_ref.syntax().range(),
+            })
+            .collect();
+
+        refs
+    }
+}