diff options
-rw-r--r-- | crates/ra_ide/src/imports_locator.rs | 6 | ||||
-rw-r--r-- | crates/ra_ide/src/references.rs | 5 | ||||
-rw-r--r-- | crates/ra_ide/src/references/classify.rs | 112 | ||||
-rw-r--r-- | crates/ra_ide/src/references/name_definition.rs | 85 | ||||
-rw-r--r-- | crates/ra_ide/src/references/search_scope.rs | 102 | ||||
-rw-r--r-- | crates/ra_ide_db/src/defs.rs | 194 | ||||
-rw-r--r-- | crates/ra_ide_db/src/lib.rs | 1 |
7 files changed, 252 insertions, 253 deletions
diff --git a/crates/ra_ide/src/imports_locator.rs b/crates/ra_ide/src/imports_locator.rs index 23bf43b07..0dca0c86c 100644 --- a/crates/ra_ide/src/imports_locator.rs +++ b/crates/ra_ide/src/imports_locator.rs | |||
@@ -4,16 +4,14 @@ | |||
4 | use hir::{db::HirDatabase, ModuleDef, SourceBinder}; | 4 | use hir::{db::HirDatabase, ModuleDef, SourceBinder}; |
5 | use ra_assists::ImportsLocator; | 5 | use ra_assists::ImportsLocator; |
6 | use ra_ide_db::{ | 6 | use ra_ide_db::{ |
7 | defs::NameKind, | ||
7 | symbol_index::{self, FileSymbol}, | 8 | symbol_index::{self, FileSymbol}, |
8 | RootDatabase, | 9 | RootDatabase, |
9 | }; | 10 | }; |
10 | use ra_prof::profile; | 11 | use ra_prof::profile; |
11 | use ra_syntax::{ast, AstNode, SyntaxKind::NAME}; | 12 | use ra_syntax::{ast, AstNode, SyntaxKind::NAME}; |
12 | 13 | ||
13 | use crate::{ | 14 | use crate::{references::classify_name, Query}; |
14 | references::{classify_name, NameKind}, | ||
15 | Query, | ||
16 | }; | ||
17 | 15 | ||
18 | pub(crate) struct ImportsLocatorIde<'a> { | 16 | pub(crate) struct ImportsLocatorIde<'a> { |
19 | source_binder: SourceBinder<'a, RootDatabase>, | 17 | source_binder: SourceBinder<'a, RootDatabase>, |
diff --git a/crates/ra_ide/src/references.rs b/crates/ra_ide/src/references.rs index b47f8bcd9..c215040f4 100644 --- a/crates/ra_ide/src/references.rs +++ b/crates/ra_ide/src/references.rs | |||
@@ -10,7 +10,6 @@ | |||
10 | //! resolved to the search element definition, we get a reference. | 10 | //! resolved to the search element definition, we get a reference. |
11 | 11 | ||
12 | mod classify; | 12 | mod classify; |
13 | mod name_definition; | ||
14 | mod rename; | 13 | mod rename; |
15 | mod search_scope; | 14 | mod search_scope; |
16 | 15 | ||
@@ -29,9 +28,9 @@ use crate::{display::ToNav, FilePosition, FileRange, NavigationTarget, RangeInfo | |||
29 | 28 | ||
30 | pub(crate) use self::{ | 29 | pub(crate) use self::{ |
31 | classify::{classify_name, classify_name_ref}, | 30 | classify::{classify_name, classify_name_ref}, |
32 | name_definition::{NameDefinition, NameKind}, | ||
33 | rename::rename, | 31 | rename::rename, |
34 | }; | 32 | }; |
33 | pub(crate) use ra_ide_db::defs::{NameDefinition, NameKind}; | ||
35 | 34 | ||
36 | pub use self::search_scope::SearchScope; | 35 | pub use self::search_scope::SearchScope; |
37 | 36 | ||
@@ -137,7 +136,7 @@ pub(crate) fn find_all_refs( | |||
137 | }; | 136 | }; |
138 | 137 | ||
139 | let search_scope = { | 138 | let search_scope = { |
140 | let base = def.search_scope(db); | 139 | let base = SearchScope::for_def(&def, db); |
141 | match search_scope { | 140 | match search_scope { |
142 | None => base, | 141 | None => base, |
143 | Some(scope) => base.intersection(&scope), | 142 | Some(scope) => base.intersection(&scope), |
diff --git a/crates/ra_ide/src/references/classify.rs b/crates/ra_ide/src/references/classify.rs index 758ea4e8b..0326fd379 100644 --- a/crates/ra_ide/src/references/classify.rs +++ b/crates/ra_ide/src/references/classify.rs | |||
@@ -2,119 +2,13 @@ | |||
2 | 2 | ||
3 | use hir::{InFile, PathResolution, SourceBinder}; | 3 | use hir::{InFile, PathResolution, SourceBinder}; |
4 | use ra_prof::profile; | 4 | use ra_prof::profile; |
5 | use ra_syntax::{ast, match_ast, AstNode}; | 5 | use ra_syntax::{ast, AstNode}; |
6 | use test_utils::tested_by; | 6 | use test_utils::tested_by; |
7 | 7 | ||
8 | use super::{ | 8 | use super::{NameDefinition, NameKind}; |
9 | name_definition::{from_assoc_item, from_module_def, from_struct_field}, | ||
10 | NameDefinition, NameKind, | ||
11 | }; | ||
12 | use ra_ide_db::RootDatabase; | 9 | use ra_ide_db::RootDatabase; |
13 | 10 | ||
14 | pub(crate) fn classify_name( | 11 | pub use ra_ide_db::defs::{classify_name, from_assoc_item, from_module_def, from_struct_field}; |
15 | sb: &mut SourceBinder<RootDatabase>, | ||
16 | name: InFile<&ast::Name>, | ||
17 | ) -> Option<NameDefinition> { | ||
18 | let _p = profile("classify_name"); | ||
19 | let parent = name.value.syntax().parent()?; | ||
20 | |||
21 | match_ast! { | ||
22 | match parent { | ||
23 | ast::BindPat(it) => { | ||
24 | let src = name.with_value(it); | ||
25 | let local = sb.to_def(src)?; | ||
26 | Some(NameDefinition { | ||
27 | visibility: None, | ||
28 | container: local.module(sb.db), | ||
29 | kind: NameKind::Local(local), | ||
30 | }) | ||
31 | }, | ||
32 | ast::RecordFieldDef(it) => { | ||
33 | let src = name.with_value(it); | ||
34 | let field: hir::StructField = sb.to_def(src)?; | ||
35 | Some(from_struct_field(sb.db, field)) | ||
36 | }, | ||
37 | ast::Module(it) => { | ||
38 | let def = sb.to_def(name.with_value(it))?; | ||
39 | Some(from_module_def(sb.db, def.into(), None)) | ||
40 | }, | ||
41 | ast::StructDef(it) => { | ||
42 | let src = name.with_value(it); | ||
43 | let def: hir::Struct = sb.to_def(src)?; | ||
44 | Some(from_module_def(sb.db, def.into(), None)) | ||
45 | }, | ||
46 | ast::EnumDef(it) => { | ||
47 | let src = name.with_value(it); | ||
48 | let def: hir::Enum = sb.to_def(src)?; | ||
49 | Some(from_module_def(sb.db, def.into(), None)) | ||
50 | }, | ||
51 | ast::TraitDef(it) => { | ||
52 | let src = name.with_value(it); | ||
53 | let def: hir::Trait = sb.to_def(src)?; | ||
54 | Some(from_module_def(sb.db, def.into(), None)) | ||
55 | }, | ||
56 | ast::StaticDef(it) => { | ||
57 | let src = name.with_value(it); | ||
58 | let def: hir::Static = sb.to_def(src)?; | ||
59 | Some(from_module_def(sb.db, def.into(), None)) | ||
60 | }, | ||
61 | ast::EnumVariant(it) => { | ||
62 | let src = name.with_value(it); | ||
63 | let def: hir::EnumVariant = sb.to_def(src)?; | ||
64 | Some(from_module_def(sb.db, def.into(), None)) | ||
65 | }, | ||
66 | ast::FnDef(it) => { | ||
67 | let src = name.with_value(it); | ||
68 | let def: hir::Function = sb.to_def(src)?; | ||
69 | if parent.parent().and_then(ast::ItemList::cast).is_some() { | ||
70 | Some(from_assoc_item(sb.db, def.into())) | ||
71 | } else { | ||
72 | Some(from_module_def(sb.db, def.into(), None)) | ||
73 | } | ||
74 | }, | ||
75 | ast::ConstDef(it) => { | ||
76 | let src = name.with_value(it); | ||
77 | let def: hir::Const = sb.to_def(src)?; | ||
78 | if parent.parent().and_then(ast::ItemList::cast).is_some() { | ||
79 | Some(from_assoc_item(sb.db, def.into())) | ||
80 | } else { | ||
81 | Some(from_module_def(sb.db, def.into(), None)) | ||
82 | } | ||
83 | }, | ||
84 | ast::TypeAliasDef(it) => { | ||
85 | let src = name.with_value(it); | ||
86 | let def: hir::TypeAlias = sb.to_def(src)?; | ||
87 | if parent.parent().and_then(ast::ItemList::cast).is_some() { | ||
88 | Some(from_assoc_item(sb.db, def.into())) | ||
89 | } else { | ||
90 | Some(from_module_def(sb.db, def.into(), None)) | ||
91 | } | ||
92 | }, | ||
93 | ast::MacroCall(it) => { | ||
94 | let src = name.with_value(it); | ||
95 | let def = sb.to_def(src.clone())?; | ||
96 | |||
97 | let module = sb.to_module_def(src.file_id.original_file(sb.db))?; | ||
98 | |||
99 | Some(NameDefinition { | ||
100 | visibility: None, | ||
101 | container: module, | ||
102 | kind: NameKind::Macro(def), | ||
103 | }) | ||
104 | }, | ||
105 | ast::TypeParam(it) => { | ||
106 | let src = name.with_value(it); | ||
107 | let def = sb.to_def(src)?; | ||
108 | Some(NameDefinition { | ||
109 | visibility: None, | ||
110 | container: def.module(sb.db), | ||
111 | kind: NameKind::TypeParam(def), | ||
112 | }) | ||
113 | }, | ||
114 | _ => None, | ||
115 | } | ||
116 | } | ||
117 | } | ||
118 | 12 | ||
119 | pub(crate) fn classify_name_ref( | 13 | pub(crate) fn classify_name_ref( |
120 | sb: &mut SourceBinder<RootDatabase>, | 14 | sb: &mut SourceBinder<RootDatabase>, |
diff --git a/crates/ra_ide/src/references/name_definition.rs b/crates/ra_ide/src/references/name_definition.rs deleted file mode 100644 index 71565e6d3..000000000 --- a/crates/ra_ide/src/references/name_definition.rs +++ /dev/null | |||
@@ -1,85 +0,0 @@ | |||
1 | //! `NameDefinition` keeps information about the element we want to search references for. | ||
2 | //! The element is represented by `NameKind`. It's located inside some `container` and | ||
3 | //! has a `visibility`, which defines a search scope. | ||
4 | //! Note that the reference search is possible for not all of the classified items. | ||
5 | |||
6 | use hir::{ | ||
7 | Adt, AssocItem, HasSource, ImplBlock, Local, MacroDef, Module, ModuleDef, StructField, | ||
8 | TypeParam, VariantDef, | ||
9 | }; | ||
10 | use ra_syntax::{ast, ast::VisibilityOwner}; | ||
11 | |||
12 | use ra_ide_db::RootDatabase; | ||
13 | |||
14 | #[derive(Debug, PartialEq, Eq)] | ||
15 | pub enum NameKind { | ||
16 | Macro(MacroDef), | ||
17 | Field(StructField), | ||
18 | AssocItem(AssocItem), | ||
19 | Def(ModuleDef), | ||
20 | SelfType(ImplBlock), | ||
21 | Local(Local), | ||
22 | TypeParam(TypeParam), | ||
23 | } | ||
24 | |||
25 | #[derive(PartialEq, Eq)] | ||
26 | pub(crate) struct NameDefinition { | ||
27 | pub visibility: Option<ast::Visibility>, | ||
28 | /// FIXME: this doesn't really make sense. For example, builtin types don't | ||
29 | /// really have a module. | ||
30 | pub container: Module, | ||
31 | pub kind: NameKind, | ||
32 | } | ||
33 | |||
34 | pub(super) fn from_assoc_item(db: &RootDatabase, item: AssocItem) -> NameDefinition { | ||
35 | let container = item.module(db); | ||
36 | let visibility = match item { | ||
37 | AssocItem::Function(f) => f.source(db).value.visibility(), | ||
38 | AssocItem::Const(c) => c.source(db).value.visibility(), | ||
39 | AssocItem::TypeAlias(a) => a.source(db).value.visibility(), | ||
40 | }; | ||
41 | let kind = NameKind::AssocItem(item); | ||
42 | NameDefinition { kind, container, visibility } | ||
43 | } | ||
44 | |||
45 | pub(super) fn from_struct_field(db: &RootDatabase, field: StructField) -> NameDefinition { | ||
46 | let kind = NameKind::Field(field); | ||
47 | let parent = field.parent_def(db); | ||
48 | let container = parent.module(db); | ||
49 | let visibility = match parent { | ||
50 | VariantDef::Struct(s) => s.source(db).value.visibility(), | ||
51 | VariantDef::Union(e) => e.source(db).value.visibility(), | ||
52 | VariantDef::EnumVariant(e) => e.source(db).value.parent_enum().visibility(), | ||
53 | }; | ||
54 | NameDefinition { kind, container, visibility } | ||
55 | } | ||
56 | |||
57 | pub(super) fn from_module_def( | ||
58 | db: &RootDatabase, | ||
59 | def: ModuleDef, | ||
60 | module: Option<Module>, | ||
61 | ) -> NameDefinition { | ||
62 | let kind = NameKind::Def(def); | ||
63 | let (container, visibility) = match def { | ||
64 | ModuleDef::Module(it) => { | ||
65 | let container = it.parent(db).or_else(|| Some(it)).unwrap(); | ||
66 | let visibility = it.declaration_source(db).and_then(|s| s.value.visibility()); | ||
67 | (container, visibility) | ||
68 | } | ||
69 | ModuleDef::EnumVariant(it) => { | ||
70 | let container = it.module(db); | ||
71 | let visibility = it.source(db).value.parent_enum().visibility(); | ||
72 | (container, visibility) | ||
73 | } | ||
74 | ModuleDef::Function(it) => (it.module(db), it.source(db).value.visibility()), | ||
75 | ModuleDef::Const(it) => (it.module(db), it.source(db).value.visibility()), | ||
76 | ModuleDef::Static(it) => (it.module(db), it.source(db).value.visibility()), | ||
77 | ModuleDef::Trait(it) => (it.module(db), it.source(db).value.visibility()), | ||
78 | ModuleDef::TypeAlias(it) => (it.module(db), it.source(db).value.visibility()), | ||
79 | ModuleDef::Adt(Adt::Struct(it)) => (it.module(db), it.source(db).value.visibility()), | ||
80 | ModuleDef::Adt(Adt::Union(it)) => (it.module(db), it.source(db).value.visibility()), | ||
81 | ModuleDef::Adt(Adt::Enum(it)) => (it.module(db), it.source(db).value.visibility()), | ||
82 | ModuleDef::BuiltinType(..) => (module.unwrap(), None), | ||
83 | }; | ||
84 | NameDefinition { kind, container, visibility } | ||
85 | } | ||
diff --git a/crates/ra_ide/src/references/search_scope.rs b/crates/ra_ide/src/references/search_scope.rs index 97c65c2cd..279f57be0 100644 --- a/crates/ra_ide/src/references/search_scope.rs +++ b/crates/ra_ide/src/references/search_scope.rs | |||
@@ -19,59 +19,13 @@ pub struct SearchScope { | |||
19 | } | 19 | } |
20 | 20 | ||
21 | impl SearchScope { | 21 | impl SearchScope { |
22 | fn new(entries: FxHashMap<FileId, Option<TextRange>>) -> SearchScope { | 22 | pub(crate) fn for_def(def: &NameDefinition, db: &RootDatabase) -> SearchScope { |
23 | SearchScope { entries } | ||
24 | } | ||
25 | pub fn single_file(file: FileId) -> SearchScope { | ||
26 | SearchScope::new(std::iter::once((file, None)).collect()) | ||
27 | } | ||
28 | pub(crate) fn intersection(&self, other: &SearchScope) -> SearchScope { | ||
29 | let (mut small, mut large) = (&self.entries, &other.entries); | ||
30 | if small.len() > large.len() { | ||
31 | mem::swap(&mut small, &mut large) | ||
32 | } | ||
33 | |||
34 | let res = small | ||
35 | .iter() | ||
36 | .filter_map(|(file_id, r1)| { | ||
37 | let r2 = large.get(file_id)?; | ||
38 | let r = intersect_ranges(*r1, *r2)?; | ||
39 | Some((*file_id, r)) | ||
40 | }) | ||
41 | .collect(); | ||
42 | return SearchScope::new(res); | ||
43 | |||
44 | fn intersect_ranges( | ||
45 | r1: Option<TextRange>, | ||
46 | r2: Option<TextRange>, | ||
47 | ) -> Option<Option<TextRange>> { | ||
48 | match (r1, r2) { | ||
49 | (None, r) | (r, None) => Some(r), | ||
50 | (Some(r1), Some(r2)) => { | ||
51 | let r = r1.intersection(&r2)?; | ||
52 | Some(Some(r)) | ||
53 | } | ||
54 | } | ||
55 | } | ||
56 | } | ||
57 | } | ||
58 | |||
59 | impl IntoIterator for SearchScope { | ||
60 | type Item = (FileId, Option<TextRange>); | ||
61 | type IntoIter = std::collections::hash_map::IntoIter<FileId, Option<TextRange>>; | ||
62 | fn into_iter(self) -> Self::IntoIter { | ||
63 | self.entries.into_iter() | ||
64 | } | ||
65 | } | ||
66 | |||
67 | impl NameDefinition { | ||
68 | pub(crate) fn search_scope(&self, db: &RootDatabase) -> SearchScope { | ||
69 | let _p = profile("search_scope"); | 23 | let _p = profile("search_scope"); |
70 | 24 | ||
71 | let module_src = self.container.definition_source(db); | 25 | let module_src = def.container.definition_source(db); |
72 | let file_id = module_src.file_id.original_file(db); | 26 | let file_id = module_src.file_id.original_file(db); |
73 | 27 | ||
74 | if let NameKind::Local(var) = self.kind { | 28 | if let NameKind::Local(var) = def.kind { |
75 | let range = match var.parent(db) { | 29 | let range = match var.parent(db) { |
76 | DefWithBody::Function(f) => f.source(db).value.syntax().text_range(), | 30 | DefWithBody::Function(f) => f.source(db).value.syntax().text_range(), |
77 | DefWithBody::Const(c) => c.source(db).value.syntax().text_range(), | 31 | DefWithBody::Const(c) => c.source(db).value.syntax().text_range(), |
@@ -82,10 +36,10 @@ impl NameDefinition { | |||
82 | return SearchScope::new(res); | 36 | return SearchScope::new(res); |
83 | } | 37 | } |
84 | 38 | ||
85 | let vis = self.visibility.as_ref().map(|v| v.syntax().to_string()).unwrap_or_default(); | 39 | let vis = def.visibility.as_ref().map(|v| v.syntax().to_string()).unwrap_or_default(); |
86 | 40 | ||
87 | if vis.as_str() == "pub(super)" { | 41 | if vis.as_str() == "pub(super)" { |
88 | if let Some(parent_module) = self.container.parent(db) { | 42 | if let Some(parent_module) = def.container.parent(db) { |
89 | let mut res = FxHashMap::default(); | 43 | let mut res = FxHashMap::default(); |
90 | let parent_src = parent_module.definition_source(db); | 44 | let parent_src = parent_module.definition_source(db); |
91 | let file_id = parent_src.file_id.original_file(db); | 45 | let file_id = parent_src.file_id.original_file(db); |
@@ -118,7 +72,7 @@ impl NameDefinition { | |||
118 | return SearchScope::new(res); | 72 | return SearchScope::new(res); |
119 | } | 73 | } |
120 | if vis.as_str() == "pub" { | 74 | if vis.as_str() == "pub" { |
121 | let krate = self.container.krate(); | 75 | let krate = def.container.krate(); |
122 | for rev_dep in krate.reverse_dependencies(db) { | 76 | for rev_dep in krate.reverse_dependencies(db) { |
123 | let root_file = rev_dep.root_file(db); | 77 | let root_file = rev_dep.root_file(db); |
124 | let source_root_id = db.file_source_root(root_file); | 78 | let source_root_id = db.file_source_root(root_file); |
@@ -137,4 +91,48 @@ impl NameDefinition { | |||
137 | res.insert(file_id, range); | 91 | res.insert(file_id, range); |
138 | SearchScope::new(res) | 92 | SearchScope::new(res) |
139 | } | 93 | } |
94 | |||
95 | fn new(entries: FxHashMap<FileId, Option<TextRange>>) -> SearchScope { | ||
96 | SearchScope { entries } | ||
97 | } | ||
98 | pub fn single_file(file: FileId) -> SearchScope { | ||
99 | SearchScope::new(std::iter::once((file, None)).collect()) | ||
100 | } | ||
101 | pub(crate) fn intersection(&self, other: &SearchScope) -> SearchScope { | ||
102 | let (mut small, mut large) = (&self.entries, &other.entries); | ||
103 | if small.len() > large.len() { | ||
104 | mem::swap(&mut small, &mut large) | ||
105 | } | ||
106 | |||
107 | let res = small | ||
108 | .iter() | ||
109 | .filter_map(|(file_id, r1)| { | ||
110 | let r2 = large.get(file_id)?; | ||
111 | let r = intersect_ranges(*r1, *r2)?; | ||
112 | Some((*file_id, r)) | ||
113 | }) | ||
114 | .collect(); | ||
115 | return SearchScope::new(res); | ||
116 | |||
117 | fn intersect_ranges( | ||
118 | r1: Option<TextRange>, | ||
119 | r2: Option<TextRange>, | ||
120 | ) -> Option<Option<TextRange>> { | ||
121 | match (r1, r2) { | ||
122 | (None, r) | (r, None) => Some(r), | ||
123 | (Some(r1), Some(r2)) => { | ||
124 | let r = r1.intersection(&r2)?; | ||
125 | Some(Some(r)) | ||
126 | } | ||
127 | } | ||
128 | } | ||
129 | } | ||
130 | } | ||
131 | |||
132 | impl IntoIterator for SearchScope { | ||
133 | type Item = (FileId, Option<TextRange>); | ||
134 | type IntoIter = std::collections::hash_map::IntoIter<FileId, Option<TextRange>>; | ||
135 | fn into_iter(self) -> Self::IntoIter { | ||
136 | self.entries.into_iter() | ||
137 | } | ||
140 | } | 138 | } |
diff --git a/crates/ra_ide_db/src/defs.rs b/crates/ra_ide_db/src/defs.rs new file mode 100644 index 000000000..1c983fb60 --- /dev/null +++ b/crates/ra_ide_db/src/defs.rs | |||
@@ -0,0 +1,194 @@ | |||
1 | //! `NameDefinition` keeps information about the element we want to search references for. | ||
2 | //! The element is represented by `NameKind`. It's located inside some `container` and | ||
3 | //! has a `visibility`, which defines a search scope. | ||
4 | //! Note that the reference search is possible for not all of the classified items. | ||
5 | |||
6 | use hir::{ | ||
7 | Adt, AssocItem, HasSource, ImplBlock, InFile, Local, MacroDef, Module, ModuleDef, SourceBinder, | ||
8 | StructField, TypeParam, VariantDef, | ||
9 | }; | ||
10 | use ra_prof::profile; | ||
11 | use ra_syntax::{ | ||
12 | ast::{self, AstNode, VisibilityOwner}, | ||
13 | match_ast, | ||
14 | }; | ||
15 | |||
16 | use crate::RootDatabase; | ||
17 | |||
18 | #[derive(Debug, PartialEq, Eq)] | ||
19 | pub enum NameKind { | ||
20 | Macro(MacroDef), | ||
21 | Field(StructField), | ||
22 | AssocItem(AssocItem), | ||
23 | Def(ModuleDef), | ||
24 | SelfType(ImplBlock), | ||
25 | Local(Local), | ||
26 | TypeParam(TypeParam), | ||
27 | } | ||
28 | |||
29 | #[derive(PartialEq, Eq)] | ||
30 | pub struct NameDefinition { | ||
31 | pub visibility: Option<ast::Visibility>, | ||
32 | /// FIXME: this doesn't really make sense. For example, builtin types don't | ||
33 | /// really have a module. | ||
34 | pub container: Module, | ||
35 | pub kind: NameKind, | ||
36 | } | ||
37 | |||
38 | pub fn classify_name( | ||
39 | sb: &mut SourceBinder<RootDatabase>, | ||
40 | name: InFile<&ast::Name>, | ||
41 | ) -> Option<NameDefinition> { | ||
42 | let _p = profile("classify_name"); | ||
43 | let parent = name.value.syntax().parent()?; | ||
44 | |||
45 | match_ast! { | ||
46 | match parent { | ||
47 | ast::BindPat(it) => { | ||
48 | let src = name.with_value(it); | ||
49 | let local = sb.to_def(src)?; | ||
50 | Some(NameDefinition { | ||
51 | visibility: None, | ||
52 | container: local.module(sb.db), | ||
53 | kind: NameKind::Local(local), | ||
54 | }) | ||
55 | }, | ||
56 | ast::RecordFieldDef(it) => { | ||
57 | let src = name.with_value(it); | ||
58 | let field: hir::StructField = sb.to_def(src)?; | ||
59 | Some(from_struct_field(sb.db, field)) | ||
60 | }, | ||
61 | ast::Module(it) => { | ||
62 | let def = sb.to_def(name.with_value(it))?; | ||
63 | Some(from_module_def(sb.db, def.into(), None)) | ||
64 | }, | ||
65 | ast::StructDef(it) => { | ||
66 | let src = name.with_value(it); | ||
67 | let def: hir::Struct = sb.to_def(src)?; | ||
68 | Some(from_module_def(sb.db, def.into(), None)) | ||
69 | }, | ||
70 | ast::EnumDef(it) => { | ||
71 | let src = name.with_value(it); | ||
72 | let def: hir::Enum = sb.to_def(src)?; | ||
73 | Some(from_module_def(sb.db, def.into(), None)) | ||
74 | }, | ||
75 | ast::TraitDef(it) => { | ||
76 | let src = name.with_value(it); | ||
77 | let def: hir::Trait = sb.to_def(src)?; | ||
78 | Some(from_module_def(sb.db, def.into(), None)) | ||
79 | }, | ||
80 | ast::StaticDef(it) => { | ||
81 | let src = name.with_value(it); | ||
82 | let def: hir::Static = sb.to_def(src)?; | ||
83 | Some(from_module_def(sb.db, def.into(), None)) | ||
84 | }, | ||
85 | ast::EnumVariant(it) => { | ||
86 | let src = name.with_value(it); | ||
87 | let def: hir::EnumVariant = sb.to_def(src)?; | ||
88 | Some(from_module_def(sb.db, def.into(), None)) | ||
89 | }, | ||
90 | ast::FnDef(it) => { | ||
91 | let src = name.with_value(it); | ||
92 | let def: hir::Function = sb.to_def(src)?; | ||
93 | if parent.parent().and_then(ast::ItemList::cast).is_some() { | ||
94 | Some(from_assoc_item(sb.db, def.into())) | ||
95 | } else { | ||
96 | Some(from_module_def(sb.db, def.into(), None)) | ||
97 | } | ||
98 | }, | ||
99 | ast::ConstDef(it) => { | ||
100 | let src = name.with_value(it); | ||
101 | let def: hir::Const = sb.to_def(src)?; | ||
102 | if parent.parent().and_then(ast::ItemList::cast).is_some() { | ||
103 | Some(from_assoc_item(sb.db, def.into())) | ||
104 | } else { | ||
105 | Some(from_module_def(sb.db, def.into(), None)) | ||
106 | } | ||
107 | }, | ||
108 | ast::TypeAliasDef(it) => { | ||
109 | let src = name.with_value(it); | ||
110 | let def: hir::TypeAlias = sb.to_def(src)?; | ||
111 | if parent.parent().and_then(ast::ItemList::cast).is_some() { | ||
112 | Some(from_assoc_item(sb.db, def.into())) | ||
113 | } else { | ||
114 | Some(from_module_def(sb.db, def.into(), None)) | ||
115 | } | ||
116 | }, | ||
117 | ast::MacroCall(it) => { | ||
118 | let src = name.with_value(it); | ||
119 | let def = sb.to_def(src.clone())?; | ||
120 | |||
121 | let module = sb.to_module_def(src.file_id.original_file(sb.db))?; | ||
122 | |||
123 | Some(NameDefinition { | ||
124 | visibility: None, | ||
125 | container: module, | ||
126 | kind: NameKind::Macro(def), | ||
127 | }) | ||
128 | }, | ||
129 | ast::TypeParam(it) => { | ||
130 | let src = name.with_value(it); | ||
131 | let def = sb.to_def(src)?; | ||
132 | Some(NameDefinition { | ||
133 | visibility: None, | ||
134 | container: def.module(sb.db), | ||
135 | kind: NameKind::TypeParam(def), | ||
136 | }) | ||
137 | }, | ||
138 | _ => None, | ||
139 | } | ||
140 | } | ||
141 | } | ||
142 | |||
143 | pub fn from_assoc_item(db: &RootDatabase, item: AssocItem) -> NameDefinition { | ||
144 | let container = item.module(db); | ||
145 | let visibility = match item { | ||
146 | AssocItem::Function(f) => f.source(db).value.visibility(), | ||
147 | AssocItem::Const(c) => c.source(db).value.visibility(), | ||
148 | AssocItem::TypeAlias(a) => a.source(db).value.visibility(), | ||
149 | }; | ||
150 | let kind = NameKind::AssocItem(item); | ||
151 | NameDefinition { kind, container, visibility } | ||
152 | } | ||
153 | |||
154 | pub fn from_struct_field(db: &RootDatabase, field: StructField) -> NameDefinition { | ||
155 | let kind = NameKind::Field(field); | ||
156 | let parent = field.parent_def(db); | ||
157 | let container = parent.module(db); | ||
158 | let visibility = match parent { | ||
159 | VariantDef::Struct(s) => s.source(db).value.visibility(), | ||
160 | VariantDef::Union(e) => e.source(db).value.visibility(), | ||
161 | VariantDef::EnumVariant(e) => e.source(db).value.parent_enum().visibility(), | ||
162 | }; | ||
163 | NameDefinition { kind, container, visibility } | ||
164 | } | ||
165 | |||
166 | pub fn from_module_def( | ||
167 | db: &RootDatabase, | ||
168 | def: ModuleDef, | ||
169 | module: Option<Module>, | ||
170 | ) -> NameDefinition { | ||
171 | let kind = NameKind::Def(def); | ||
172 | let (container, visibility) = match def { | ||
173 | ModuleDef::Module(it) => { | ||
174 | let container = it.parent(db).or_else(|| Some(it)).unwrap(); | ||
175 | let visibility = it.declaration_source(db).and_then(|s| s.value.visibility()); | ||
176 | (container, visibility) | ||
177 | } | ||
178 | ModuleDef::EnumVariant(it) => { | ||
179 | let container = it.module(db); | ||
180 | let visibility = it.source(db).value.parent_enum().visibility(); | ||
181 | (container, visibility) | ||
182 | } | ||
183 | ModuleDef::Function(it) => (it.module(db), it.source(db).value.visibility()), | ||
184 | ModuleDef::Const(it) => (it.module(db), it.source(db).value.visibility()), | ||
185 | ModuleDef::Static(it) => (it.module(db), it.source(db).value.visibility()), | ||
186 | ModuleDef::Trait(it) => (it.module(db), it.source(db).value.visibility()), | ||
187 | ModuleDef::TypeAlias(it) => (it.module(db), it.source(db).value.visibility()), | ||
188 | ModuleDef::Adt(Adt::Struct(it)) => (it.module(db), it.source(db).value.visibility()), | ||
189 | ModuleDef::Adt(Adt::Union(it)) => (it.module(db), it.source(db).value.visibility()), | ||
190 | ModuleDef::Adt(Adt::Enum(it)) => (it.module(db), it.source(db).value.visibility()), | ||
191 | ModuleDef::BuiltinType(..) => (module.unwrap(), None), | ||
192 | }; | ||
193 | NameDefinition { kind, container, visibility } | ||
194 | } | ||
diff --git a/crates/ra_ide_db/src/lib.rs b/crates/ra_ide_db/src/lib.rs index e922d1e5f..0715dfc66 100644 --- a/crates/ra_ide_db/src/lib.rs +++ b/crates/ra_ide_db/src/lib.rs | |||
@@ -7,6 +7,7 @@ pub mod line_index_utils; | |||
7 | pub mod feature_flags; | 7 | pub mod feature_flags; |
8 | pub mod symbol_index; | 8 | pub mod symbol_index; |
9 | pub mod change; | 9 | pub mod change; |
10 | pub mod defs; | ||
10 | mod wasm_shims; | 11 | mod wasm_shims; |
11 | 12 | ||
12 | use std::sync::Arc; | 13 | use std::sync::Arc; |