1 files changed, 61 insertions, 0 deletions

diff --git a/crates/ra_ide_api/src/references/search_scope.rs b/crates/ra_ide_api/src/references/search_scope.rs
new file mode 100644
index 000000000..557ee7b57
--- /dev/null
+++ b/crates/ra_ide_api/src/references/search_scope.rs
@@ -0,0 +1,61 @@
+use hir::{DefWithBody, HasSource, ModuleSource};
+use ra_db::{FileId, SourceDatabase};
+use ra_syntax::{AstNode, TextRange};
+
+use crate::db::RootDatabase;
+
+use super::{Definition, NameKind};
+
+pub(crate) struct SearchScope {
+    pub scope: Vec<(FileId, Option<TextRange>)>,
+}
+
+impl Definition {
+    pub fn scope(&self, db: &RootDatabase) -> SearchScope {
+        let module_src = self.container.definition_source(db);
+        let file_id = module_src.file_id.original_file(db);
+
+        if let NameKind::Pat((def, _)) = self.item {
+            let range = match def {
+                DefWithBody::Function(f) => f.source(db).ast.syntax().text_range(),
+                DefWithBody::Const(c) => c.source(db).ast.syntax().text_range(),
+                DefWithBody::Static(s) => s.source(db).ast.syntax().text_range(),
+            };
+            return SearchScope { scope: vec![(file_id, Some(range))] };
+        }
+
+        if let Some(ref vis) = self.visibility {
+            let source_root_id = db.file_source_root(file_id);
+            let source_root = db.source_root(source_root_id);
+            let mut files = source_root.walk().map(|id| (id.into(), None)).collect::<Vec<_>>();
+
+            if vis.syntax().to_string().as_str() == "pub(crate)" {
+                return SearchScope { scope: files };
+            }
+            if vis.syntax().to_string().as_str() == "pub" {
+                let krate = self.container.krate(db).unwrap();
+                let crate_graph = db.crate_graph();
+
+                for crate_id in crate_graph.iter() {
+                    let mut crate_deps = crate_graph.dependencies(crate_id);
+
+                    if crate_deps.any(|dep| dep.crate_id() == krate.crate_id()) {
+                        let root_file = crate_graph.crate_root(crate_id);
+                        let source_root_id = db.file_source_root(root_file);
+                        let source_root = db.source_root(source_root_id);
+                        files.extend(source_root.walk().map(|id| (id.into(), None)));
+                    }
+                }
+
+                return SearchScope { scope: files };
+            }
+            // FIXME: "pub(super)", "pub(in path)"
+        }
+
+        let range = match module_src.ast {
+            ModuleSource::Module(m) => Some(m.syntax().text_range()),
+            ModuleSource::SourceFile(_) => None,
+        };
+        SearchScope { scope: vec![(file_id, range)] }
+    }
+}