diff options
author | bors[bot] <26634292+bors[bot]@users.noreply.github.com> | 2020-01-16 16:58:13 +0000 |
---|---|---|
committer | GitHub <[email protected]> | 2020-01-16 16:58:13 +0000 |
commit | d3c4fbbbc45afc7d480185493b5ce77b5daa1747 (patch) | |
tree | aa12838a1b0c45710bb7657f6f3c34947519484c /crates/ra_hir | |
parent | e614356de10256c64e02f5d0f82cfa39e27b0f07 (diff) | |
parent | 846f11c2177f3e3a6348acb67503028180b3c1f8 (diff) |
Merge #2862
2862: Move from `from_source` to `SourceBinder` r=matklad a=matklad
bors r+
Co-authored-by: Aleksey Kladov <[email protected]>
Diffstat (limited to 'crates/ra_hir')
-rw-r--r-- | crates/ra_hir/src/from_source.rs | 265 | ||||
-rw-r--r-- | crates/ra_hir/src/lib.rs | 2 | ||||
-rw-r--r-- | crates/ra_hir/src/source_binder.rs | 252 |
3 files changed, 218 insertions, 301 deletions
diff --git a/crates/ra_hir/src/from_source.rs b/crates/ra_hir/src/from_source.rs deleted file mode 100644 index 6314be8d4..000000000 --- a/crates/ra_hir/src/from_source.rs +++ /dev/null | |||
@@ -1,265 +0,0 @@ | |||
1 | //! Finds a corresponding hir data structure for a syntax node in a specific | ||
2 | //! file. | ||
3 | |||
4 | use hir_def::{ | ||
5 | child_by_source::ChildBySource, dyn_map::DynMap, keys, keys::Key, nameres::ModuleSource, | ||
6 | ConstId, DefWithBodyId, EnumId, EnumVariantId, FunctionId, GenericDefId, ImplId, ModuleId, | ||
7 | StaticId, StructId, TraitId, TypeAliasId, UnionId, VariantId, | ||
8 | }; | ||
9 | use hir_expand::{name::AsName, AstId, MacroDefId, MacroDefKind}; | ||
10 | use ra_db::FileId; | ||
11 | use ra_prof::profile; | ||
12 | use ra_syntax::{ | ||
13 | ast::{self, AstNode, NameOwner}, | ||
14 | match_ast, SyntaxNode, | ||
15 | }; | ||
16 | |||
17 | use crate::{ | ||
18 | db::{DefDatabase, HirDatabase}, | ||
19 | Const, DefWithBody, Enum, EnumVariant, FieldSource, Function, ImplBlock, InFile, Local, | ||
20 | MacroDef, Module, Static, Struct, StructField, Trait, TypeAlias, TypeParam, Union, | ||
21 | }; | ||
22 | |||
23 | pub trait FromSource: Sized { | ||
24 | type Ast; | ||
25 | fn from_source(db: &impl DefDatabase, src: InFile<Self::Ast>) -> Option<Self>; | ||
26 | } | ||
27 | |||
28 | pub trait FromSourceByContainer: Sized { | ||
29 | type Ast: AstNode + 'static; | ||
30 | type Id: Copy + 'static; | ||
31 | const KEY: Key<Self::Ast, Self::Id>; | ||
32 | } | ||
33 | |||
34 | impl<T: FromSourceByContainer> FromSource for T | ||
35 | where | ||
36 | T: From<<T as FromSourceByContainer>::Id>, | ||
37 | { | ||
38 | type Ast = <T as FromSourceByContainer>::Ast; | ||
39 | fn from_source(db: &impl DefDatabase, src: InFile<Self::Ast>) -> Option<Self> { | ||
40 | analyze_container(db, src.as_ref().map(|it| it.syntax()))[T::KEY] | ||
41 | .get(&src) | ||
42 | .copied() | ||
43 | .map(Self::from) | ||
44 | } | ||
45 | } | ||
46 | |||
47 | macro_rules! from_source_by_container_impls { | ||
48 | ($(($hir:ident, $id:ident, $ast:path, $key:path)),* ,) => {$( | ||
49 | impl FromSourceByContainer for $hir { | ||
50 | type Ast = $ast; | ||
51 | type Id = $id; | ||
52 | const KEY: Key<Self::Ast, Self::Id> = $key; | ||
53 | } | ||
54 | )*} | ||
55 | } | ||
56 | |||
57 | from_source_by_container_impls![ | ||
58 | (Struct, StructId, ast::StructDef, keys::STRUCT), | ||
59 | (Union, UnionId, ast::UnionDef, keys::UNION), | ||
60 | (Enum, EnumId, ast::EnumDef, keys::ENUM), | ||
61 | (Trait, TraitId, ast::TraitDef, keys::TRAIT), | ||
62 | (Function, FunctionId, ast::FnDef, keys::FUNCTION), | ||
63 | (Static, StaticId, ast::StaticDef, keys::STATIC), | ||
64 | (Const, ConstId, ast::ConstDef, keys::CONST), | ||
65 | (TypeAlias, TypeAliasId, ast::TypeAliasDef, keys::TYPE_ALIAS), | ||
66 | (ImplBlock, ImplId, ast::ImplBlock, keys::IMPL), | ||
67 | ]; | ||
68 | |||
69 | impl FromSource for MacroDef { | ||
70 | type Ast = ast::MacroCall; | ||
71 | fn from_source(db: &impl DefDatabase, src: InFile<Self::Ast>) -> Option<Self> { | ||
72 | let kind = MacroDefKind::Declarative; | ||
73 | |||
74 | let module_src = ModuleSource::from_child_node(db, src.as_ref().map(|it| it.syntax())); | ||
75 | let module = Module::from_definition(db, InFile::new(src.file_id, module_src))?; | ||
76 | let krate = Some(module.krate().id); | ||
77 | |||
78 | let ast_id = Some(AstId::new(src.file_id, db.ast_id_map(src.file_id).ast_id(&src.value))); | ||
79 | |||
80 | let id: MacroDefId = MacroDefId { krate, ast_id, kind }; | ||
81 | Some(MacroDef { id }) | ||
82 | } | ||
83 | } | ||
84 | |||
85 | impl FromSource for EnumVariant { | ||
86 | type Ast = ast::EnumVariant; | ||
87 | fn from_source(db: &impl DefDatabase, src: InFile<Self::Ast>) -> Option<Self> { | ||
88 | let parent_enum = src.value.parent_enum(); | ||
89 | let src_enum = InFile { file_id: src.file_id, value: parent_enum }; | ||
90 | let parent_enum = Enum::from_source(db, src_enum)?; | ||
91 | parent_enum.id.child_by_source(db)[keys::ENUM_VARIANT] | ||
92 | .get(&src) | ||
93 | .copied() | ||
94 | .map(EnumVariant::from) | ||
95 | } | ||
96 | } | ||
97 | |||
98 | impl FromSource for StructField { | ||
99 | type Ast = FieldSource; | ||
100 | fn from_source(db: &impl DefDatabase, src: InFile<Self::Ast>) -> Option<Self> { | ||
101 | let src = src.as_ref(); | ||
102 | |||
103 | // FIXME this is buggy | ||
104 | let variant_id: VariantId = match src.value { | ||
105 | FieldSource::Named(field) => { | ||
106 | let value = field.syntax().ancestors().find_map(ast::StructDef::cast)?; | ||
107 | let src = InFile { file_id: src.file_id, value }; | ||
108 | let def = Struct::from_source(db, src)?; | ||
109 | def.id.into() | ||
110 | } | ||
111 | FieldSource::Pos(field) => { | ||
112 | let value = field.syntax().ancestors().find_map(ast::EnumVariant::cast)?; | ||
113 | let src = InFile { file_id: src.file_id, value }; | ||
114 | let def = EnumVariant::from_source(db, src)?; | ||
115 | EnumVariantId::from(def).into() | ||
116 | } | ||
117 | }; | ||
118 | |||
119 | let dyn_map = variant_id.child_by_source(db); | ||
120 | match src.value { | ||
121 | FieldSource::Pos(it) => dyn_map[keys::TUPLE_FIELD].get(&src.with_value(it.clone())), | ||
122 | FieldSource::Named(it) => dyn_map[keys::RECORD_FIELD].get(&src.with_value(it.clone())), | ||
123 | } | ||
124 | .copied() | ||
125 | .map(StructField::from) | ||
126 | } | ||
127 | } | ||
128 | |||
129 | impl Local { | ||
130 | pub fn from_source(db: &impl HirDatabase, src: InFile<ast::BindPat>) -> Option<Self> { | ||
131 | let file_id = src.file_id; | ||
132 | let parent: DefWithBody = src.value.syntax().ancestors().find_map(|it| { | ||
133 | let res = match_ast! { | ||
134 | match it { | ||
135 | ast::ConstDef(value) => { Const::from_source(db, InFile { value, file_id})?.into() }, | ||
136 | ast::StaticDef(value) => { Static::from_source(db, InFile { value, file_id})?.into() }, | ||
137 | ast::FnDef(value) => { Function::from_source(db, InFile { value, file_id})?.into() }, | ||
138 | _ => return None, | ||
139 | } | ||
140 | }; | ||
141 | Some(res) | ||
142 | })?; | ||
143 | let (_body, source_map) = db.body_with_source_map(parent.into()); | ||
144 | let src = src.map(ast::Pat::from); | ||
145 | let pat_id = source_map.node_pat(src.as_ref())?; | ||
146 | Some(Local { parent, pat_id }) | ||
147 | } | ||
148 | } | ||
149 | |||
150 | impl TypeParam { | ||
151 | pub fn from_source(db: &impl HirDatabase, src: InFile<ast::TypeParam>) -> Option<Self> { | ||
152 | let file_id = src.file_id; | ||
153 | let parent: GenericDefId = src.value.syntax().ancestors().find_map(|it| { | ||
154 | let res = match_ast! { | ||
155 | match it { | ||
156 | ast::FnDef(value) => { Function::from_source(db, InFile { value, file_id})?.id.into() }, | ||
157 | ast::StructDef(value) => { Struct::from_source(db, InFile { value, file_id})?.id.into() }, | ||
158 | ast::EnumDef(value) => { Enum::from_source(db, InFile { value, file_id})?.id.into() }, | ||
159 | ast::TraitDef(value) => { Trait::from_source(db, InFile { value, file_id})?.id.into() }, | ||
160 | ast::TypeAliasDef(value) => { TypeAlias::from_source(db, InFile { value, file_id})?.id.into() }, | ||
161 | ast::ImplBlock(value) => { ImplBlock::from_source(db, InFile { value, file_id})?.id.into() }, | ||
162 | _ => return None, | ||
163 | } | ||
164 | }; | ||
165 | Some(res) | ||
166 | })?; | ||
167 | let &id = parent.child_by_source(db)[keys::TYPE_PARAM].get(&src)?; | ||
168 | Some(TypeParam { id }) | ||
169 | } | ||
170 | } | ||
171 | |||
172 | impl Module { | ||
173 | pub fn from_declaration(db: &impl DefDatabase, src: InFile<ast::Module>) -> Option<Self> { | ||
174 | let _p = profile("Module::from_declaration"); | ||
175 | let parent_declaration = src.value.syntax().ancestors().skip(1).find_map(ast::Module::cast); | ||
176 | |||
177 | let parent_module = match parent_declaration { | ||
178 | Some(parent_declaration) => { | ||
179 | let src_parent = InFile { file_id: src.file_id, value: parent_declaration }; | ||
180 | Module::from_declaration(db, src_parent) | ||
181 | } | ||
182 | None => { | ||
183 | let source_file = db.parse(src.file_id.original_file(db)).tree(); | ||
184 | let src_parent = | ||
185 | InFile { file_id: src.file_id, value: ModuleSource::SourceFile(source_file) }; | ||
186 | Module::from_definition(db, src_parent) | ||
187 | } | ||
188 | }?; | ||
189 | |||
190 | let child_name = src.value.name()?.as_name(); | ||
191 | let def_map = db.crate_def_map(parent_module.id.krate); | ||
192 | let child_id = def_map[parent_module.id.local_id].children.get(&child_name)?; | ||
193 | Some(parent_module.with_module_id(*child_id)) | ||
194 | } | ||
195 | |||
196 | pub fn from_definition(db: &impl DefDatabase, src: InFile<ModuleSource>) -> Option<Self> { | ||
197 | let _p = profile("Module::from_definition"); | ||
198 | match src.value { | ||
199 | ModuleSource::Module(ref module) => { | ||
200 | assert!(!module.has_semi()); | ||
201 | return Module::from_declaration( | ||
202 | db, | ||
203 | InFile { file_id: src.file_id, value: module.clone() }, | ||
204 | ); | ||
205 | } | ||
206 | ModuleSource::SourceFile(_) => (), | ||
207 | }; | ||
208 | |||
209 | let original_file = src.file_id.original_file(db); | ||
210 | Module::from_file(db, original_file) | ||
211 | } | ||
212 | |||
213 | fn from_file(db: &impl DefDatabase, file: FileId) -> Option<Self> { | ||
214 | let _p = profile("Module::from_file"); | ||
215 | let (krate, local_id) = db.relevant_crates(file).iter().find_map(|&crate_id| { | ||
216 | let crate_def_map = db.crate_def_map(crate_id); | ||
217 | let local_id = crate_def_map.modules_for_file(file).next()?; | ||
218 | Some((crate_id, local_id)) | ||
219 | })?; | ||
220 | Some(Module { id: ModuleId { krate, local_id } }) | ||
221 | } | ||
222 | } | ||
223 | |||
224 | fn analyze_container(db: &impl DefDatabase, src: InFile<&SyntaxNode>) -> DynMap { | ||
225 | let _p = profile("analyze_container"); | ||
226 | return child_by_source(db, src).unwrap_or_default(); | ||
227 | |||
228 | fn child_by_source(db: &impl DefDatabase, src: InFile<&SyntaxNode>) -> Option<DynMap> { | ||
229 | for container in src.value.ancestors().skip(1) { | ||
230 | let res = match_ast! { | ||
231 | match container { | ||
232 | ast::TraitDef(it) => { | ||
233 | let def = Trait::from_source(db, src.with_value(it))?; | ||
234 | def.id.child_by_source(db) | ||
235 | }, | ||
236 | ast::ImplBlock(it) => { | ||
237 | let def = ImplBlock::from_source(db, src.with_value(it))?; | ||
238 | def.id.child_by_source(db) | ||
239 | }, | ||
240 | ast::FnDef(it) => { | ||
241 | let def = Function::from_source(db, src.with_value(it))?; | ||
242 | DefWithBodyId::from(def.id) | ||
243 | .child_by_source(db) | ||
244 | }, | ||
245 | ast::StaticDef(it) => { | ||
246 | let def = Static::from_source(db, src.with_value(it))?; | ||
247 | DefWithBodyId::from(def.id) | ||
248 | .child_by_source(db) | ||
249 | }, | ||
250 | ast::ConstDef(it) => { | ||
251 | let def = Const::from_source(db, src.with_value(it))?; | ||
252 | DefWithBodyId::from(def.id) | ||
253 | .child_by_source(db) | ||
254 | }, | ||
255 | _ => { continue }, | ||
256 | } | ||
257 | }; | ||
258 | return Some(res); | ||
259 | } | ||
260 | |||
261 | let module_source = ModuleSource::from_child_node(db, src); | ||
262 | let c = Module::from_definition(db, src.with_value(module_source))?; | ||
263 | Some(c.id.child_by_source(db)) | ||
264 | } | ||
265 | } | ||
diff --git a/crates/ra_hir/src/lib.rs b/crates/ra_hir/src/lib.rs index a2350573c..e1c7b7a20 100644 --- a/crates/ra_hir/src/lib.rs +++ b/crates/ra_hir/src/lib.rs | |||
@@ -36,7 +36,6 @@ mod from_id; | |||
36 | mod code_model; | 36 | mod code_model; |
37 | 37 | ||
38 | mod has_source; | 38 | mod has_source; |
39 | mod from_source; | ||
40 | 39 | ||
41 | pub use crate::{ | 40 | pub use crate::{ |
42 | code_model::{ | 41 | code_model::{ |
@@ -45,7 +44,6 @@ pub use crate::{ | |||
45 | MacroDef, Module, ModuleDef, ScopeDef, Static, Struct, StructField, Trait, Type, TypeAlias, | 44 | MacroDef, Module, ModuleDef, ScopeDef, Static, Struct, StructField, Trait, Type, TypeAlias, |
46 | TypeParam, Union, VariantDef, | 45 | TypeParam, Union, VariantDef, |
47 | }, | 46 | }, |
48 | from_source::FromSource, | ||
49 | has_source::HasSource, | 47 | has_source::HasSource, |
50 | source_analyzer::{PathResolution, ScopeEntryWithSyntax, SourceAnalyzer}, | 48 | source_analyzer::{PathResolution, ScopeEntryWithSyntax, SourceAnalyzer}, |
51 | source_binder::SourceBinder, | 49 | source_binder::SourceBinder, |
diff --git a/crates/ra_hir/src/source_binder.rs b/crates/ra_hir/src/source_binder.rs index 00541dbe1..f3150f578 100644 --- a/crates/ra_hir/src/source_binder.rs +++ b/crates/ra_hir/src/source_binder.rs | |||
@@ -1,22 +1,24 @@ | |||
1 | //! `SourceBinder` should be the main entry point for getting info about source code. | 1 | //! `SourceBinder` is the main entry point for getting info about source code. |
2 | //! It's main task is to map source syntax trees to hir-level IDs. | 2 | //! It's main task is to map source syntax trees to hir-level IDs. |
3 | //! | ||
4 | //! It is intended to subsume `FromSource` and `SourceAnalyzer`. | ||
5 | 3 | ||
6 | use hir_def::{ | 4 | use hir_def::{ |
7 | child_by_source::ChildBySource, | 5 | child_by_source::ChildBySource, |
8 | dyn_map::DynMap, | 6 | dyn_map::DynMap, |
9 | keys::{self, Key}, | 7 | keys::{self, Key}, |
10 | resolver::{HasResolver, Resolver}, | 8 | resolver::{HasResolver, Resolver}, |
11 | ConstId, DefWithBodyId, EnumId, EnumVariantId, FunctionId, ImplId, ModuleId, StaticId, | 9 | ConstId, DefWithBodyId, EnumId, EnumVariantId, FunctionId, GenericDefId, ImplId, ModuleId, |
12 | StructFieldId, StructId, TraitId, TypeAliasId, UnionId, VariantId, | 10 | StaticId, StructFieldId, StructId, TraitId, TypeAliasId, UnionId, VariantId, |
13 | }; | 11 | }; |
14 | use hir_expand::InFile; | 12 | use hir_expand::{name::AsName, AstId, InFile, MacroDefId, MacroDefKind}; |
15 | use ra_prof::profile; | 13 | use ra_prof::profile; |
16 | use ra_syntax::{ast, match_ast, AstNode, SyntaxNode, TextUnit}; | 14 | use ra_syntax::{ |
15 | ast::{self, NameOwner}, | ||
16 | match_ast, AstNode, SyntaxNode, TextUnit, | ||
17 | }; | ||
17 | use rustc_hash::FxHashMap; | 18 | use rustc_hash::FxHashMap; |
18 | 19 | ||
19 | use crate::{db::HirDatabase, ModuleSource, SourceAnalyzer}; | 20 | use crate::{db::HirDatabase, Local, Module, SourceAnalyzer, TypeParam}; |
21 | use ra_db::FileId; | ||
20 | 22 | ||
21 | pub struct SourceBinder<'a, DB> { | 23 | pub struct SourceBinder<'a, DB> { |
22 | pub db: &'a DB, | 24 | pub db: &'a DB, |
@@ -48,32 +50,27 @@ impl<DB: HirDatabase> SourceBinder<'_, DB> { | |||
48 | ChildContainer::ModuleId(it) => it.resolver(self.db), | 50 | ChildContainer::ModuleId(it) => it.resolver(self.db), |
49 | ChildContainer::EnumId(it) => it.resolver(self.db), | 51 | ChildContainer::EnumId(it) => it.resolver(self.db), |
50 | ChildContainer::VariantId(it) => it.resolver(self.db), | 52 | ChildContainer::VariantId(it) => it.resolver(self.db), |
53 | ChildContainer::GenericDefId(it) => it.resolver(self.db), | ||
51 | }; | 54 | }; |
52 | SourceAnalyzer::new_for_resolver(resolver, src) | 55 | SourceAnalyzer::new_for_resolver(resolver, src) |
53 | } | 56 | } |
54 | 57 | ||
55 | pub fn to_def<D, T>(&mut self, src: InFile<T>) -> Option<D> | 58 | pub fn to_def<T: ToDef>(&mut self, src: InFile<T>) -> Option<T::Def> { |
56 | where | 59 | T::to_def(self, src) |
57 | D: From<T::ID>, | 60 | } |
58 | T: ToId, | 61 | |
59 | { | 62 | pub fn to_module_def(&mut self, file: FileId) -> Option<Module> { |
60 | let id: T::ID = self.to_id(src)?; | 63 | let _p = profile("SourceBinder::to_module_def"); |
61 | Some(id.into()) | 64 | let (krate, local_id) = self.db.relevant_crates(file).iter().find_map(|&crate_id| { |
65 | let crate_def_map = self.db.crate_def_map(crate_id); | ||
66 | let local_id = crate_def_map.modules_for_file(file).next()?; | ||
67 | Some((crate_id, local_id)) | ||
68 | })?; | ||
69 | Some(Module { id: ModuleId { krate, local_id } }) | ||
62 | } | 70 | } |
63 | 71 | ||
64 | fn to_id<T: ToId>(&mut self, src: InFile<T>) -> Option<T::ID> { | 72 | fn to_id<T: ToId>(&mut self, src: InFile<T>) -> Option<T::ID> { |
65 | let container = self.find_container(src.as_ref().map(|it| it.syntax()))?; | 73 | T::to_id(self, src) |
66 | let db = self.db; | ||
67 | let dyn_map = | ||
68 | &*self.child_by_source_cache.entry(container).or_insert_with(|| match container { | ||
69 | ChildContainer::DefWithBodyId(it) => it.child_by_source(db), | ||
70 | ChildContainer::ModuleId(it) => it.child_by_source(db), | ||
71 | ChildContainer::TraitId(it) => it.child_by_source(db), | ||
72 | ChildContainer::ImplId(it) => it.child_by_source(db), | ||
73 | ChildContainer::EnumId(it) => it.child_by_source(db), | ||
74 | ChildContainer::VariantId(it) => it.child_by_source(db), | ||
75 | }); | ||
76 | dyn_map[T::KEY].get(&src).copied() | ||
77 | } | 74 | } |
78 | 75 | ||
79 | fn find_container(&mut self, src: InFile<&SyntaxNode>) -> Option<ChildContainer> { | 76 | fn find_container(&mut self, src: InFile<&SyntaxNode>) -> Option<ChildContainer> { |
@@ -112,19 +109,75 @@ impl<DB: HirDatabase> SourceBinder<'_, DB> { | |||
112 | let def: UnionId = self.to_id(container.with_value(it))?; | 109 | let def: UnionId = self.to_id(container.with_value(it))?; |
113 | VariantId::from(def).into() | 110 | VariantId::from(def).into() |
114 | }, | 111 | }, |
115 | // FIXME: handle out-of-line modules here | 112 | ast::Module(it) => { |
113 | let def: ModuleId = self.to_id(container.with_value(it))?; | ||
114 | def.into() | ||
115 | }, | ||
116 | _ => { continue }, | 116 | _ => { continue }, |
117 | } | 117 | } |
118 | }; | 118 | }; |
119 | return Some(res); | 119 | return Some(res); |
120 | } | 120 | } |
121 | 121 | ||
122 | let module_source = ModuleSource::from_child_node(self.db, src); | 122 | let c = self.to_module_def(src.file_id.original_file(self.db))?; |
123 | let c = crate::Module::from_definition(self.db, src.with_value(module_source))?; | ||
124 | Some(c.id.into()) | 123 | Some(c.id.into()) |
125 | } | 124 | } |
125 | |||
126 | fn child_by_source(&mut self, container: ChildContainer) -> &DynMap { | ||
127 | let db = self.db; | ||
128 | self.child_by_source_cache.entry(container).or_insert_with(|| match container { | ||
129 | ChildContainer::DefWithBodyId(it) => it.child_by_source(db), | ||
130 | ChildContainer::ModuleId(it) => it.child_by_source(db), | ||
131 | ChildContainer::TraitId(it) => it.child_by_source(db), | ||
132 | ChildContainer::ImplId(it) => it.child_by_source(db), | ||
133 | ChildContainer::EnumId(it) => it.child_by_source(db), | ||
134 | ChildContainer::VariantId(it) => it.child_by_source(db), | ||
135 | ChildContainer::GenericDefId(it) => it.child_by_source(db), | ||
136 | }) | ||
137 | } | ||
126 | } | 138 | } |
127 | 139 | ||
140 | pub trait ToId: Sized { | ||
141 | type ID: Sized + Copy + 'static; | ||
142 | fn to_id<DB: HirDatabase>(sb: &mut SourceBinder<'_, DB>, src: InFile<Self>) | ||
143 | -> Option<Self::ID>; | ||
144 | } | ||
145 | |||
146 | pub trait ToDef: Sized + AstNode + 'static { | ||
147 | type Def; | ||
148 | fn to_def<DB: HirDatabase>( | ||
149 | sb: &mut SourceBinder<'_, DB>, | ||
150 | src: InFile<Self>, | ||
151 | ) -> Option<Self::Def>; | ||
152 | } | ||
153 | |||
154 | macro_rules! to_def_impls { | ||
155 | ($(($def:path, $ast:path)),* ,) => {$( | ||
156 | impl ToDef for $ast { | ||
157 | type Def = $def; | ||
158 | fn to_def<DB: HirDatabase>(sb: &mut SourceBinder<'_, DB>, src: InFile<Self>) | ||
159 | -> Option<Self::Def> | ||
160 | { sb.to_id(src).map(Into::into) } | ||
161 | } | ||
162 | )*} | ||
163 | } | ||
164 | |||
165 | to_def_impls![ | ||
166 | (crate::Module, ast::Module), | ||
167 | (crate::Struct, ast::StructDef), | ||
168 | (crate::Enum, ast::EnumDef), | ||
169 | (crate::Union, ast::UnionDef), | ||
170 | (crate::Trait, ast::TraitDef), | ||
171 | (crate::ImplBlock, ast::ImplBlock), | ||
172 | (crate::TypeAlias, ast::TypeAliasDef), | ||
173 | (crate::Const, ast::ConstDef), | ||
174 | (crate::Static, ast::StaticDef), | ||
175 | (crate::Function, ast::FnDef), | ||
176 | (crate::StructField, ast::RecordFieldDef), | ||
177 | (crate::EnumVariant, ast::EnumVariant), | ||
178 | (crate::MacroDef, ast::MacroCall), // this one is dubious, not all calls are macros | ||
179 | ]; | ||
180 | |||
128 | #[derive(Clone, Copy, PartialEq, Eq, Hash, Debug)] | 181 | #[derive(Clone, Copy, PartialEq, Eq, Hash, Debug)] |
129 | enum ChildContainer { | 182 | enum ChildContainer { |
130 | DefWithBodyId(DefWithBodyId), | 183 | DefWithBodyId(DefWithBodyId), |
@@ -133,6 +186,9 @@ enum ChildContainer { | |||
133 | ImplId(ImplId), | 186 | ImplId(ImplId), |
134 | EnumId(EnumId), | 187 | EnumId(EnumId), |
135 | VariantId(VariantId), | 188 | VariantId(VariantId), |
189 | /// XXX: this might be the same def as, for example an `EnumId`. However, | ||
190 | /// here the children generic parameters, and not, eg enum variants. | ||
191 | GenericDefId(GenericDefId), | ||
136 | } | 192 | } |
137 | impl_froms! { | 193 | impl_froms! { |
138 | ChildContainer: | 194 | ChildContainer: |
@@ -142,23 +198,46 @@ impl_froms! { | |||
142 | ImplId, | 198 | ImplId, |
143 | EnumId, | 199 | EnumId, |
144 | VariantId, | 200 | VariantId, |
201 | GenericDefId | ||
145 | } | 202 | } |
146 | 203 | ||
147 | pub trait ToId: Sized + AstNode + 'static { | 204 | pub trait ToIdByKey: Sized + AstNode + 'static { |
148 | type ID: Sized + Copy + 'static; | 205 | type ID: Sized + Copy + 'static; |
149 | const KEY: Key<Self, Self::ID>; | 206 | const KEY: Key<Self, Self::ID>; |
150 | } | 207 | } |
151 | 208 | ||
152 | macro_rules! to_id_impls { | 209 | impl<T: ToIdByKey> ToId for T { |
210 | type ID = <T as ToIdByKey>::ID; | ||
211 | fn to_id<DB: HirDatabase>( | ||
212 | sb: &mut SourceBinder<'_, DB>, | ||
213 | src: InFile<Self>, | ||
214 | ) -> Option<Self::ID> { | ||
215 | let container = sb.find_container(src.as_ref().map(|it| it.syntax()))?; | ||
216 | let db = sb.db; | ||
217 | let dyn_map = | ||
218 | &*sb.child_by_source_cache.entry(container).or_insert_with(|| match container { | ||
219 | ChildContainer::DefWithBodyId(it) => it.child_by_source(db), | ||
220 | ChildContainer::ModuleId(it) => it.child_by_source(db), | ||
221 | ChildContainer::TraitId(it) => it.child_by_source(db), | ||
222 | ChildContainer::ImplId(it) => it.child_by_source(db), | ||
223 | ChildContainer::EnumId(it) => it.child_by_source(db), | ||
224 | ChildContainer::VariantId(it) => it.child_by_source(db), | ||
225 | ChildContainer::GenericDefId(it) => it.child_by_source(db), | ||
226 | }); | ||
227 | dyn_map[T::KEY].get(&src).copied() | ||
228 | } | ||
229 | } | ||
230 | |||
231 | macro_rules! to_id_key_impls { | ||
153 | ($(($id:ident, $ast:path, $key:path)),* ,) => {$( | 232 | ($(($id:ident, $ast:path, $key:path)),* ,) => {$( |
154 | impl ToId for $ast { | 233 | impl ToIdByKey for $ast { |
155 | type ID = $id; | 234 | type ID = $id; |
156 | const KEY: Key<Self, Self::ID> = $key; | 235 | const KEY: Key<Self, Self::ID> = $key; |
157 | } | 236 | } |
158 | )*} | 237 | )*} |
159 | } | 238 | } |
160 | 239 | ||
161 | to_id_impls![ | 240 | to_id_key_impls![ |
162 | (StructId, ast::StructDef, keys::STRUCT), | 241 | (StructId, ast::StructDef, keys::STRUCT), |
163 | (UnionId, ast::UnionDef, keys::UNION), | 242 | (UnionId, ast::UnionDef, keys::UNION), |
164 | (EnumId, ast::EnumDef, keys::ENUM), | 243 | (EnumId, ast::EnumDef, keys::ENUM), |
@@ -171,3 +250,108 @@ to_id_impls![ | |||
171 | (StructFieldId, ast::RecordFieldDef, keys::RECORD_FIELD), | 250 | (StructFieldId, ast::RecordFieldDef, keys::RECORD_FIELD), |
172 | (EnumVariantId, ast::EnumVariant, keys::ENUM_VARIANT), | 251 | (EnumVariantId, ast::EnumVariant, keys::ENUM_VARIANT), |
173 | ]; | 252 | ]; |
253 | |||
254 | // FIXME: use DynMap as well? | ||
255 | impl ToId for ast::MacroCall { | ||
256 | type ID = MacroDefId; | ||
257 | fn to_id<DB: HirDatabase>( | ||
258 | sb: &mut SourceBinder<'_, DB>, | ||
259 | src: InFile<Self>, | ||
260 | ) -> Option<Self::ID> { | ||
261 | let kind = MacroDefKind::Declarative; | ||
262 | |||
263 | let krate = sb.to_module_def(src.file_id.original_file(sb.db))?.id.krate; | ||
264 | |||
265 | let ast_id = | ||
266 | Some(AstId::new(src.file_id, sb.db.ast_id_map(src.file_id).ast_id(&src.value))); | ||
267 | |||
268 | Some(MacroDefId { krate: Some(krate), ast_id, kind }) | ||
269 | } | ||
270 | } | ||
271 | |||
272 | impl ToDef for ast::BindPat { | ||
273 | type Def = Local; | ||
274 | |||
275 | fn to_def<DB: HirDatabase>(sb: &mut SourceBinder<'_, DB>, src: InFile<Self>) -> Option<Local> { | ||
276 | let file_id = src.file_id; | ||
277 | let parent: DefWithBodyId = src.value.syntax().ancestors().find_map(|it| { | ||
278 | let res = match_ast! { | ||
279 | match it { | ||
280 | ast::ConstDef(value) => { sb.to_id(InFile { value, file_id})?.into() }, | ||
281 | ast::StaticDef(value) => { sb.to_id(InFile { value, file_id})?.into() }, | ||
282 | ast::FnDef(value) => { sb.to_id(InFile { value, file_id})?.into() }, | ||
283 | _ => return None, | ||
284 | } | ||
285 | }; | ||
286 | Some(res) | ||
287 | })?; | ||
288 | let (_body, source_map) = sb.db.body_with_source_map(parent); | ||
289 | let src = src.map(ast::Pat::from); | ||
290 | let pat_id = source_map.node_pat(src.as_ref())?; | ||
291 | Some(Local { parent: parent.into(), pat_id }) | ||
292 | } | ||
293 | } | ||
294 | |||
295 | impl ToDef for ast::TypeParam { | ||
296 | type Def = TypeParam; | ||
297 | |||
298 | fn to_def<DB: HirDatabase>( | ||
299 | sb: &mut SourceBinder<'_, DB>, | ||
300 | src: InFile<ast::TypeParam>, | ||
301 | ) -> Option<TypeParam> { | ||
302 | let mut sb = SourceBinder::new(sb.db); | ||
303 | let file_id = src.file_id; | ||
304 | let parent: GenericDefId = src.value.syntax().ancestors().find_map(|it| { | ||
305 | let res = match_ast! { | ||
306 | match it { | ||
307 | ast::FnDef(value) => { sb.to_id(InFile { value, file_id})?.into() }, | ||
308 | ast::StructDef(value) => { sb.to_id(InFile { value, file_id})?.into() }, | ||
309 | ast::EnumDef(value) => { sb.to_id(InFile { value, file_id})?.into() }, | ||
310 | ast::TraitDef(value) => { sb.to_id(InFile { value, file_id})?.into() }, | ||
311 | ast::TypeAliasDef(value) => { sb.to_id(InFile { value, file_id})?.into() }, | ||
312 | ast::ImplBlock(value) => { sb.to_id(InFile { value, file_id})?.into() }, | ||
313 | _ => return None, | ||
314 | } | ||
315 | }; | ||
316 | Some(res) | ||
317 | })?; | ||
318 | let &id = sb.child_by_source(parent.into())[keys::TYPE_PARAM].get(&src)?; | ||
319 | Some(TypeParam { id }) | ||
320 | } | ||
321 | } | ||
322 | |||
323 | impl ToId for ast::Module { | ||
324 | type ID = ModuleId; | ||
325 | |||
326 | fn to_id<DB: HirDatabase>( | ||
327 | sb: &mut SourceBinder<'_, DB>, | ||
328 | src: InFile<ast::Module>, | ||
329 | ) -> Option<ModuleId> { | ||
330 | { | ||
331 | let _p = profile("ast::Module::to_def"); | ||
332 | let parent_declaration = src | ||
333 | .as_ref() | ||
334 | .map(|it| it.syntax()) | ||
335 | .cloned() | ||
336 | .ancestors_with_macros(sb.db) | ||
337 | .skip(1) | ||
338 | .find_map(|it| { | ||
339 | let m = ast::Module::cast(it.value.clone())?; | ||
340 | Some(it.with_value(m)) | ||
341 | }); | ||
342 | |||
343 | let parent_module = match parent_declaration { | ||
344 | Some(parent_declaration) => sb.to_id(parent_declaration)?, | ||
345 | None => { | ||
346 | let file_id = src.file_id.original_file(sb.db); | ||
347 | sb.to_module_def(file_id)?.id | ||
348 | } | ||
349 | }; | ||
350 | |||
351 | let child_name = src.value.name()?.as_name(); | ||
352 | let def_map = sb.db.crate_def_map(parent_module.krate); | ||
353 | let child_id = *def_map[parent_module.local_id].children.get(&child_name)?; | ||
354 | Some(ModuleId { krate: parent_module.krate, local_id: child_id }) | ||
355 | } | ||
356 | } | ||
357 | } | ||