diff options
Diffstat (limited to 'crates/hir_def/src/item_tree/lower.rs')
-rw-r--r-- | crates/hir_def/src/item_tree/lower.rs | 705 |
1 files changed, 705 insertions, 0 deletions
diff --git a/crates/hir_def/src/item_tree/lower.rs b/crates/hir_def/src/item_tree/lower.rs new file mode 100644 index 000000000..391ab5d39 --- /dev/null +++ b/crates/hir_def/src/item_tree/lower.rs | |||
@@ -0,0 +1,705 @@ | |||
1 | //! AST -> `ItemTree` lowering code. | ||
2 | |||
3 | use std::{collections::hash_map::Entry, mem, sync::Arc}; | ||
4 | |||
5 | use arena::map::ArenaMap; | ||
6 | use hir_expand::{ast_id_map::AstIdMap, hygiene::Hygiene, HirFileId}; | ||
7 | use smallvec::SmallVec; | ||
8 | use syntax::{ | ||
9 | ast::{self, ModuleItemOwner}, | ||
10 | SyntaxNode, | ||
11 | }; | ||
12 | |||
13 | use crate::{ | ||
14 | attr::Attrs, | ||
15 | generics::{GenericParams, TypeParamData, TypeParamProvenance}, | ||
16 | }; | ||
17 | |||
18 | use super::*; | ||
19 | |||
20 | fn id<N: ItemTreeNode>(index: Idx<N>) -> FileItemTreeId<N> { | ||
21 | FileItemTreeId { index, _p: PhantomData } | ||
22 | } | ||
23 | |||
24 | struct ModItems(SmallVec<[ModItem; 1]>); | ||
25 | |||
26 | impl<T> From<T> for ModItems | ||
27 | where | ||
28 | T: Into<ModItem>, | ||
29 | { | ||
30 | fn from(t: T) -> Self { | ||
31 | ModItems(SmallVec::from_buf([t.into(); 1])) | ||
32 | } | ||
33 | } | ||
34 | |||
35 | pub(super) struct Ctx { | ||
36 | tree: ItemTree, | ||
37 | hygiene: Hygiene, | ||
38 | file: HirFileId, | ||
39 | source_ast_id_map: Arc<AstIdMap>, | ||
40 | body_ctx: crate::body::LowerCtx, | ||
41 | inner_items: Vec<ModItem>, | ||
42 | forced_visibility: Option<RawVisibilityId>, | ||
43 | } | ||
44 | |||
45 | impl Ctx { | ||
46 | pub(super) fn new(db: &dyn DefDatabase, hygiene: Hygiene, file: HirFileId) -> Self { | ||
47 | Self { | ||
48 | tree: ItemTree::empty(), | ||
49 | hygiene, | ||
50 | file, | ||
51 | source_ast_id_map: db.ast_id_map(file), | ||
52 | body_ctx: crate::body::LowerCtx::new(db, file), | ||
53 | inner_items: Vec::new(), | ||
54 | forced_visibility: None, | ||
55 | } | ||
56 | } | ||
57 | |||
58 | pub(super) fn lower_module_items(mut self, item_owner: &dyn ModuleItemOwner) -> ItemTree { | ||
59 | self.tree.top_level = item_owner | ||
60 | .items() | ||
61 | .flat_map(|item| self.lower_mod_item(&item, false)) | ||
62 | .flat_map(|items| items.0) | ||
63 | .collect(); | ||
64 | self.tree | ||
65 | } | ||
66 | |||
67 | pub(super) fn lower_inner_items(mut self, within: &SyntaxNode) -> ItemTree { | ||
68 | self.collect_inner_items(within); | ||
69 | self.tree | ||
70 | } | ||
71 | |||
72 | fn data(&mut self) -> &mut ItemTreeData { | ||
73 | self.tree.data_mut() | ||
74 | } | ||
75 | |||
76 | fn lower_mod_item(&mut self, item: &ast::Item, inner: bool) -> Option<ModItems> { | ||
77 | assert!(inner || self.inner_items.is_empty()); | ||
78 | |||
79 | // Collect inner items for 1-to-1-lowered items. | ||
80 | match item { | ||
81 | ast::Item::Struct(_) | ||
82 | | ast::Item::Union(_) | ||
83 | | ast::Item::Enum(_) | ||
84 | | ast::Item::Fn(_) | ||
85 | | ast::Item::TypeAlias(_) | ||
86 | | ast::Item::Const(_) | ||
87 | | ast::Item::Static(_) | ||
88 | | ast::Item::MacroCall(_) => { | ||
89 | // Skip this if we're already collecting inner items. We'll descend into all nodes | ||
90 | // already. | ||
91 | if !inner { | ||
92 | self.collect_inner_items(item.syntax()); | ||
93 | } | ||
94 | } | ||
95 | |||
96 | // These are handled in their respective `lower_X` method (since we can't just blindly | ||
97 | // walk them). | ||
98 | ast::Item::Trait(_) | ast::Item::Impl(_) | ast::Item::ExternBlock(_) => {} | ||
99 | |||
100 | // These don't have inner items. | ||
101 | ast::Item::Module(_) | ast::Item::ExternCrate(_) | ast::Item::Use(_) => {} | ||
102 | }; | ||
103 | |||
104 | let attrs = Attrs::new(item, &self.hygiene); | ||
105 | let items = match item { | ||
106 | ast::Item::Struct(ast) => self.lower_struct(ast).map(Into::into), | ||
107 | ast::Item::Union(ast) => self.lower_union(ast).map(Into::into), | ||
108 | ast::Item::Enum(ast) => self.lower_enum(ast).map(Into::into), | ||
109 | ast::Item::Fn(ast) => self.lower_function(ast).map(Into::into), | ||
110 | ast::Item::TypeAlias(ast) => self.lower_type_alias(ast).map(Into::into), | ||
111 | ast::Item::Static(ast) => self.lower_static(ast).map(Into::into), | ||
112 | ast::Item::Const(ast) => Some(self.lower_const(ast).into()), | ||
113 | ast::Item::Module(ast) => self.lower_module(ast).map(Into::into), | ||
114 | ast::Item::Trait(ast) => self.lower_trait(ast).map(Into::into), | ||
115 | ast::Item::Impl(ast) => self.lower_impl(ast).map(Into::into), | ||
116 | ast::Item::Use(ast) => Some(ModItems( | ||
117 | self.lower_use(ast).into_iter().map(Into::into).collect::<SmallVec<_>>(), | ||
118 | )), | ||
119 | ast::Item::ExternCrate(ast) => self.lower_extern_crate(ast).map(Into::into), | ||
120 | ast::Item::MacroCall(ast) => self.lower_macro_call(ast).map(Into::into), | ||
121 | ast::Item::ExternBlock(ast) => { | ||
122 | Some(ModItems(self.lower_extern_block(ast).into_iter().collect::<SmallVec<_>>())) | ||
123 | } | ||
124 | }; | ||
125 | |||
126 | if !attrs.is_empty() { | ||
127 | for item in items.iter().flat_map(|items| &items.0) { | ||
128 | self.add_attrs((*item).into(), attrs.clone()); | ||
129 | } | ||
130 | } | ||
131 | |||
132 | items | ||
133 | } | ||
134 | |||
135 | fn add_attrs(&mut self, item: AttrOwner, attrs: Attrs) { | ||
136 | match self.tree.attrs.entry(item) { | ||
137 | Entry::Occupied(mut entry) => { | ||
138 | *entry.get_mut() = entry.get().merge(attrs); | ||
139 | } | ||
140 | Entry::Vacant(entry) => { | ||
141 | entry.insert(attrs); | ||
142 | } | ||
143 | } | ||
144 | } | ||
145 | |||
146 | fn collect_inner_items(&mut self, container: &SyntaxNode) { | ||
147 | let forced_vis = self.forced_visibility.take(); | ||
148 | let mut inner_items = mem::take(&mut self.tree.inner_items); | ||
149 | inner_items.extend(container.descendants().skip(1).filter_map(ast::Item::cast).filter_map( | ||
150 | |item| { | ||
151 | let ast_id = self.source_ast_id_map.ast_id(&item); | ||
152 | Some((ast_id, self.lower_mod_item(&item, true)?.0)) | ||
153 | }, | ||
154 | )); | ||
155 | self.tree.inner_items = inner_items; | ||
156 | self.forced_visibility = forced_vis; | ||
157 | } | ||
158 | |||
159 | fn lower_assoc_item(&mut self, item: &ast::AssocItem) -> Option<AssocItem> { | ||
160 | match item { | ||
161 | ast::AssocItem::Fn(ast) => self.lower_function(ast).map(Into::into), | ||
162 | ast::AssocItem::TypeAlias(ast) => self.lower_type_alias(ast).map(Into::into), | ||
163 | ast::AssocItem::Const(ast) => Some(self.lower_const(ast).into()), | ||
164 | ast::AssocItem::MacroCall(ast) => self.lower_macro_call(ast).map(Into::into), | ||
165 | } | ||
166 | } | ||
167 | |||
168 | fn lower_struct(&mut self, strukt: &ast::Struct) -> Option<FileItemTreeId<Struct>> { | ||
169 | let visibility = self.lower_visibility(strukt); | ||
170 | let name = strukt.name()?.as_name(); | ||
171 | let generic_params = self.lower_generic_params(GenericsOwner::Struct, strukt); | ||
172 | let fields = self.lower_fields(&strukt.kind()); | ||
173 | let ast_id = self.source_ast_id_map.ast_id(strukt); | ||
174 | let kind = match strukt.kind() { | ||
175 | ast::StructKind::Record(_) => StructDefKind::Record, | ||
176 | ast::StructKind::Tuple(_) => StructDefKind::Tuple, | ||
177 | ast::StructKind::Unit => StructDefKind::Unit, | ||
178 | }; | ||
179 | let res = Struct { name, visibility, generic_params, fields, ast_id, kind }; | ||
180 | Some(id(self.data().structs.alloc(res))) | ||
181 | } | ||
182 | |||
183 | fn lower_fields(&mut self, strukt_kind: &ast::StructKind) -> Fields { | ||
184 | match strukt_kind { | ||
185 | ast::StructKind::Record(it) => { | ||
186 | let range = self.lower_record_fields(it); | ||
187 | Fields::Record(range) | ||
188 | } | ||
189 | ast::StructKind::Tuple(it) => { | ||
190 | let range = self.lower_tuple_fields(it); | ||
191 | Fields::Tuple(range) | ||
192 | } | ||
193 | ast::StructKind::Unit => Fields::Unit, | ||
194 | } | ||
195 | } | ||
196 | |||
197 | fn lower_record_fields(&mut self, fields: &ast::RecordFieldList) -> IdRange<Field> { | ||
198 | let start = self.next_field_idx(); | ||
199 | for field in fields.fields() { | ||
200 | if let Some(data) = self.lower_record_field(&field) { | ||
201 | let idx = self.data().fields.alloc(data); | ||
202 | self.add_attrs(idx.into(), Attrs::new(&field, &self.hygiene)); | ||
203 | } | ||
204 | } | ||
205 | let end = self.next_field_idx(); | ||
206 | IdRange::new(start..end) | ||
207 | } | ||
208 | |||
209 | fn lower_record_field(&mut self, field: &ast::RecordField) -> Option<Field> { | ||
210 | let name = field.name()?.as_name(); | ||
211 | let visibility = self.lower_visibility(field); | ||
212 | let type_ref = self.lower_type_ref_opt(field.ty()); | ||
213 | let res = Field { name, type_ref, visibility }; | ||
214 | Some(res) | ||
215 | } | ||
216 | |||
217 | fn lower_tuple_fields(&mut self, fields: &ast::TupleFieldList) -> IdRange<Field> { | ||
218 | let start = self.next_field_idx(); | ||
219 | for (i, field) in fields.fields().enumerate() { | ||
220 | let data = self.lower_tuple_field(i, &field); | ||
221 | let idx = self.data().fields.alloc(data); | ||
222 | self.add_attrs(idx.into(), Attrs::new(&field, &self.hygiene)); | ||
223 | } | ||
224 | let end = self.next_field_idx(); | ||
225 | IdRange::new(start..end) | ||
226 | } | ||
227 | |||
228 | fn lower_tuple_field(&mut self, idx: usize, field: &ast::TupleField) -> Field { | ||
229 | let name = Name::new_tuple_field(idx); | ||
230 | let visibility = self.lower_visibility(field); | ||
231 | let type_ref = self.lower_type_ref_opt(field.ty()); | ||
232 | let res = Field { name, type_ref, visibility }; | ||
233 | res | ||
234 | } | ||
235 | |||
236 | fn lower_union(&mut self, union: &ast::Union) -> Option<FileItemTreeId<Union>> { | ||
237 | let visibility = self.lower_visibility(union); | ||
238 | let name = union.name()?.as_name(); | ||
239 | let generic_params = self.lower_generic_params(GenericsOwner::Union, union); | ||
240 | let fields = match union.record_field_list() { | ||
241 | Some(record_field_list) => self.lower_fields(&StructKind::Record(record_field_list)), | ||
242 | None => Fields::Record(IdRange::new(self.next_field_idx()..self.next_field_idx())), | ||
243 | }; | ||
244 | let ast_id = self.source_ast_id_map.ast_id(union); | ||
245 | let res = Union { name, visibility, generic_params, fields, ast_id }; | ||
246 | Some(id(self.data().unions.alloc(res))) | ||
247 | } | ||
248 | |||
249 | fn lower_enum(&mut self, enum_: &ast::Enum) -> Option<FileItemTreeId<Enum>> { | ||
250 | let visibility = self.lower_visibility(enum_); | ||
251 | let name = enum_.name()?.as_name(); | ||
252 | let generic_params = self.lower_generic_params(GenericsOwner::Enum, enum_); | ||
253 | let variants = match &enum_.variant_list() { | ||
254 | Some(variant_list) => self.lower_variants(variant_list), | ||
255 | None => IdRange::new(self.next_variant_idx()..self.next_variant_idx()), | ||
256 | }; | ||
257 | let ast_id = self.source_ast_id_map.ast_id(enum_); | ||
258 | let res = Enum { name, visibility, generic_params, variants, ast_id }; | ||
259 | Some(id(self.data().enums.alloc(res))) | ||
260 | } | ||
261 | |||
262 | fn lower_variants(&mut self, variants: &ast::VariantList) -> IdRange<Variant> { | ||
263 | let start = self.next_variant_idx(); | ||
264 | for variant in variants.variants() { | ||
265 | if let Some(data) = self.lower_variant(&variant) { | ||
266 | let idx = self.data().variants.alloc(data); | ||
267 | self.add_attrs(idx.into(), Attrs::new(&variant, &self.hygiene)); | ||
268 | } | ||
269 | } | ||
270 | let end = self.next_variant_idx(); | ||
271 | IdRange::new(start..end) | ||
272 | } | ||
273 | |||
274 | fn lower_variant(&mut self, variant: &ast::Variant) -> Option<Variant> { | ||
275 | let name = variant.name()?.as_name(); | ||
276 | let fields = self.lower_fields(&variant.kind()); | ||
277 | let res = Variant { name, fields }; | ||
278 | Some(res) | ||
279 | } | ||
280 | |||
281 | fn lower_function(&mut self, func: &ast::Fn) -> Option<FileItemTreeId<Function>> { | ||
282 | let visibility = self.lower_visibility(func); | ||
283 | let name = func.name()?.as_name(); | ||
284 | |||
285 | let mut params = Vec::new(); | ||
286 | let mut has_self_param = false; | ||
287 | if let Some(param_list) = func.param_list() { | ||
288 | if let Some(self_param) = param_list.self_param() { | ||
289 | let self_type = match self_param.ty() { | ||
290 | Some(type_ref) => TypeRef::from_ast(&self.body_ctx, type_ref), | ||
291 | None => { | ||
292 | let self_type = TypeRef::Path(name![Self].into()); | ||
293 | match self_param.kind() { | ||
294 | ast::SelfParamKind::Owned => self_type, | ||
295 | ast::SelfParamKind::Ref => { | ||
296 | TypeRef::Reference(Box::new(self_type), Mutability::Shared) | ||
297 | } | ||
298 | ast::SelfParamKind::MutRef => { | ||
299 | TypeRef::Reference(Box::new(self_type), Mutability::Mut) | ||
300 | } | ||
301 | } | ||
302 | } | ||
303 | }; | ||
304 | params.push(self_type); | ||
305 | has_self_param = true; | ||
306 | } | ||
307 | for param in param_list.params() { | ||
308 | let type_ref = TypeRef::from_ast_opt(&self.body_ctx, param.ty()); | ||
309 | params.push(type_ref); | ||
310 | } | ||
311 | } | ||
312 | |||
313 | let mut is_varargs = false; | ||
314 | if let Some(params) = func.param_list() { | ||
315 | if let Some(last) = params.params().last() { | ||
316 | is_varargs = last.dotdotdot_token().is_some(); | ||
317 | } | ||
318 | } | ||
319 | |||
320 | let ret_type = match func.ret_type().and_then(|rt| rt.ty()) { | ||
321 | Some(type_ref) => TypeRef::from_ast(&self.body_ctx, type_ref), | ||
322 | _ => TypeRef::unit(), | ||
323 | }; | ||
324 | |||
325 | let ret_type = if func.async_token().is_some() { | ||
326 | let future_impl = desugar_future_path(ret_type); | ||
327 | let ty_bound = TypeBound::Path(future_impl); | ||
328 | TypeRef::ImplTrait(vec![ty_bound]) | ||
329 | } else { | ||
330 | ret_type | ||
331 | }; | ||
332 | |||
333 | let ast_id = self.source_ast_id_map.ast_id(func); | ||
334 | let mut res = Function { | ||
335 | name, | ||
336 | visibility, | ||
337 | generic_params: GenericParamsId::EMPTY, | ||
338 | has_self_param, | ||
339 | is_unsafe: func.unsafe_token().is_some(), | ||
340 | params: params.into_boxed_slice(), | ||
341 | is_varargs, | ||
342 | ret_type, | ||
343 | ast_id, | ||
344 | }; | ||
345 | res.generic_params = self.lower_generic_params(GenericsOwner::Function(&res), func); | ||
346 | |||
347 | Some(id(self.data().functions.alloc(res))) | ||
348 | } | ||
349 | |||
350 | fn lower_type_alias( | ||
351 | &mut self, | ||
352 | type_alias: &ast::TypeAlias, | ||
353 | ) -> Option<FileItemTreeId<TypeAlias>> { | ||
354 | let name = type_alias.name()?.as_name(); | ||
355 | let type_ref = type_alias.ty().map(|it| self.lower_type_ref(&it)); | ||
356 | let visibility = self.lower_visibility(type_alias); | ||
357 | let bounds = self.lower_type_bounds(type_alias); | ||
358 | let generic_params = self.lower_generic_params(GenericsOwner::TypeAlias, type_alias); | ||
359 | let ast_id = self.source_ast_id_map.ast_id(type_alias); | ||
360 | let res = TypeAlias { | ||
361 | name, | ||
362 | visibility, | ||
363 | bounds: bounds.into_boxed_slice(), | ||
364 | generic_params, | ||
365 | type_ref, | ||
366 | ast_id, | ||
367 | }; | ||
368 | Some(id(self.data().type_aliases.alloc(res))) | ||
369 | } | ||
370 | |||
371 | fn lower_static(&mut self, static_: &ast::Static) -> Option<FileItemTreeId<Static>> { | ||
372 | let name = static_.name()?.as_name(); | ||
373 | let type_ref = self.lower_type_ref_opt(static_.ty()); | ||
374 | let visibility = self.lower_visibility(static_); | ||
375 | let mutable = static_.mut_token().is_some(); | ||
376 | let ast_id = self.source_ast_id_map.ast_id(static_); | ||
377 | let res = Static { name, visibility, mutable, type_ref, ast_id }; | ||
378 | Some(id(self.data().statics.alloc(res))) | ||
379 | } | ||
380 | |||
381 | fn lower_const(&mut self, konst: &ast::Const) -> FileItemTreeId<Const> { | ||
382 | let name = konst.name().map(|it| it.as_name()); | ||
383 | let type_ref = self.lower_type_ref_opt(konst.ty()); | ||
384 | let visibility = self.lower_visibility(konst); | ||
385 | let ast_id = self.source_ast_id_map.ast_id(konst); | ||
386 | let res = Const { name, visibility, type_ref, ast_id }; | ||
387 | id(self.data().consts.alloc(res)) | ||
388 | } | ||
389 | |||
390 | fn lower_module(&mut self, module: &ast::Module) -> Option<FileItemTreeId<Mod>> { | ||
391 | let name = module.name()?.as_name(); | ||
392 | let visibility = self.lower_visibility(module); | ||
393 | let kind = if module.semicolon_token().is_some() { | ||
394 | ModKind::Outline {} | ||
395 | } else { | ||
396 | ModKind::Inline { | ||
397 | items: module | ||
398 | .item_list() | ||
399 | .map(|list| { | ||
400 | list.items() | ||
401 | .flat_map(|item| self.lower_mod_item(&item, false)) | ||
402 | .flat_map(|items| items.0) | ||
403 | .collect() | ||
404 | }) | ||
405 | .unwrap_or_else(|| { | ||
406 | mark::hit!(name_res_works_for_broken_modules); | ||
407 | Box::new([]) as Box<[_]> | ||
408 | }), | ||
409 | } | ||
410 | }; | ||
411 | let ast_id = self.source_ast_id_map.ast_id(module); | ||
412 | let res = Mod { name, visibility, kind, ast_id }; | ||
413 | Some(id(self.data().mods.alloc(res))) | ||
414 | } | ||
415 | |||
416 | fn lower_trait(&mut self, trait_def: &ast::Trait) -> Option<FileItemTreeId<Trait>> { | ||
417 | let name = trait_def.name()?.as_name(); | ||
418 | let visibility = self.lower_visibility(trait_def); | ||
419 | let generic_params = | ||
420 | self.lower_generic_params_and_inner_items(GenericsOwner::Trait(trait_def), trait_def); | ||
421 | let auto = trait_def.auto_token().is_some(); | ||
422 | let items = trait_def.assoc_item_list().map(|list| { | ||
423 | self.with_inherited_visibility(visibility, |this| { | ||
424 | list.assoc_items() | ||
425 | .filter_map(|item| { | ||
426 | let attrs = Attrs::new(&item, &this.hygiene); | ||
427 | this.collect_inner_items(item.syntax()); | ||
428 | this.lower_assoc_item(&item).map(|item| { | ||
429 | this.add_attrs(ModItem::from(item).into(), attrs); | ||
430 | item | ||
431 | }) | ||
432 | }) | ||
433 | .collect() | ||
434 | }) | ||
435 | }); | ||
436 | let ast_id = self.source_ast_id_map.ast_id(trait_def); | ||
437 | let res = Trait { | ||
438 | name, | ||
439 | visibility, | ||
440 | generic_params, | ||
441 | auto, | ||
442 | items: items.unwrap_or_default(), | ||
443 | ast_id, | ||
444 | }; | ||
445 | Some(id(self.data().traits.alloc(res))) | ||
446 | } | ||
447 | |||
448 | fn lower_impl(&mut self, impl_def: &ast::Impl) -> Option<FileItemTreeId<Impl>> { | ||
449 | let generic_params = | ||
450 | self.lower_generic_params_and_inner_items(GenericsOwner::Impl, impl_def); | ||
451 | let target_trait = impl_def.trait_().map(|tr| self.lower_type_ref(&tr)); | ||
452 | let target_type = self.lower_type_ref(&impl_def.self_ty()?); | ||
453 | let is_negative = impl_def.excl_token().is_some(); | ||
454 | |||
455 | // We cannot use `assoc_items()` here as that does not include macro calls. | ||
456 | let items = impl_def | ||
457 | .assoc_item_list() | ||
458 | .into_iter() | ||
459 | .flat_map(|it| it.assoc_items()) | ||
460 | .filter_map(|item| { | ||
461 | self.collect_inner_items(item.syntax()); | ||
462 | let assoc = self.lower_assoc_item(&item)?; | ||
463 | let attrs = Attrs::new(&item, &self.hygiene); | ||
464 | self.add_attrs(ModItem::from(assoc).into(), attrs); | ||
465 | Some(assoc) | ||
466 | }) | ||
467 | .collect(); | ||
468 | let ast_id = self.source_ast_id_map.ast_id(impl_def); | ||
469 | let res = Impl { generic_params, target_trait, target_type, is_negative, items, ast_id }; | ||
470 | Some(id(self.data().impls.alloc(res))) | ||
471 | } | ||
472 | |||
473 | fn lower_use(&mut self, use_item: &ast::Use) -> Vec<FileItemTreeId<Import>> { | ||
474 | // FIXME: cfg_attr | ||
475 | let is_prelude = use_item.has_atom_attr("prelude_import"); | ||
476 | let visibility = self.lower_visibility(use_item); | ||
477 | let ast_id = self.source_ast_id_map.ast_id(use_item); | ||
478 | |||
479 | // Every use item can expand to many `Import`s. | ||
480 | let mut imports = Vec::new(); | ||
481 | let tree = self.tree.data_mut(); | ||
482 | ModPath::expand_use_item( | ||
483 | InFile::new(self.file, use_item.clone()), | ||
484 | &self.hygiene, | ||
485 | |path, _tree, is_glob, alias| { | ||
486 | imports.push(id(tree.imports.alloc(Import { | ||
487 | path, | ||
488 | alias, | ||
489 | visibility, | ||
490 | is_glob, | ||
491 | is_prelude, | ||
492 | ast_id, | ||
493 | }))); | ||
494 | }, | ||
495 | ); | ||
496 | |||
497 | imports | ||
498 | } | ||
499 | |||
500 | fn lower_extern_crate( | ||
501 | &mut self, | ||
502 | extern_crate: &ast::ExternCrate, | ||
503 | ) -> Option<FileItemTreeId<ExternCrate>> { | ||
504 | let path = ModPath::from_name_ref(&extern_crate.name_ref()?); | ||
505 | let alias = extern_crate.rename().map(|a| { | ||
506 | a.name().map(|it| it.as_name()).map_or(ImportAlias::Underscore, ImportAlias::Alias) | ||
507 | }); | ||
508 | let visibility = self.lower_visibility(extern_crate); | ||
509 | let ast_id = self.source_ast_id_map.ast_id(extern_crate); | ||
510 | // FIXME: cfg_attr | ||
511 | let is_macro_use = extern_crate.has_atom_attr("macro_use"); | ||
512 | |||
513 | let res = ExternCrate { path, alias, visibility, is_macro_use, ast_id }; | ||
514 | Some(id(self.data().extern_crates.alloc(res))) | ||
515 | } | ||
516 | |||
517 | fn lower_macro_call(&mut self, m: &ast::MacroCall) -> Option<FileItemTreeId<MacroCall>> { | ||
518 | let name = m.name().map(|it| it.as_name()); | ||
519 | let attrs = Attrs::new(m, &self.hygiene); | ||
520 | let path = ModPath::from_src(m.path()?, &self.hygiene)?; | ||
521 | |||
522 | let ast_id = self.source_ast_id_map.ast_id(m); | ||
523 | |||
524 | // FIXME: cfg_attr | ||
525 | let export_attr = attrs.by_key("macro_export"); | ||
526 | |||
527 | let is_export = export_attr.exists(); | ||
528 | let is_local_inner = if is_export { | ||
529 | export_attr.tt_values().map(|it| &it.token_trees).flatten().any(|it| match it { | ||
530 | tt::TokenTree::Leaf(tt::Leaf::Ident(ident)) => { | ||
531 | ident.text.contains("local_inner_macros") | ||
532 | } | ||
533 | _ => false, | ||
534 | }) | ||
535 | } else { | ||
536 | false | ||
537 | }; | ||
538 | |||
539 | let is_builtin = attrs.by_key("rustc_builtin_macro").exists(); | ||
540 | let res = MacroCall { name, path, is_export, is_builtin, is_local_inner, ast_id }; | ||
541 | Some(id(self.data().macro_calls.alloc(res))) | ||
542 | } | ||
543 | |||
544 | fn lower_extern_block(&mut self, block: &ast::ExternBlock) -> Vec<ModItem> { | ||
545 | block.extern_item_list().map_or(Vec::new(), |list| { | ||
546 | list.extern_items() | ||
547 | .filter_map(|item| { | ||
548 | self.collect_inner_items(item.syntax()); | ||
549 | let attrs = Attrs::new(&item, &self.hygiene); | ||
550 | let id: ModItem = match item { | ||
551 | ast::ExternItem::Fn(ast) => { | ||
552 | let func = self.lower_function(&ast)?; | ||
553 | self.data().functions[func.index].is_unsafe = true; | ||
554 | func.into() | ||
555 | } | ||
556 | ast::ExternItem::Static(ast) => { | ||
557 | let statik = self.lower_static(&ast)?; | ||
558 | statik.into() | ||
559 | } | ||
560 | ast::ExternItem::MacroCall(_) => return None, | ||
561 | }; | ||
562 | self.add_attrs(id.into(), attrs); | ||
563 | Some(id) | ||
564 | }) | ||
565 | .collect() | ||
566 | }) | ||
567 | } | ||
568 | |||
569 | /// Lowers generics defined on `node` and collects inner items defined within. | ||
570 | fn lower_generic_params_and_inner_items( | ||
571 | &mut self, | ||
572 | owner: GenericsOwner<'_>, | ||
573 | node: &impl ast::GenericParamsOwner, | ||
574 | ) -> GenericParamsId { | ||
575 | // Generics are part of item headers and may contain inner items we need to collect. | ||
576 | if let Some(params) = node.generic_param_list() { | ||
577 | self.collect_inner_items(params.syntax()); | ||
578 | } | ||
579 | if let Some(clause) = node.where_clause() { | ||
580 | self.collect_inner_items(clause.syntax()); | ||
581 | } | ||
582 | |||
583 | self.lower_generic_params(owner, node) | ||
584 | } | ||
585 | |||
586 | fn lower_generic_params( | ||
587 | &mut self, | ||
588 | owner: GenericsOwner<'_>, | ||
589 | node: &impl ast::GenericParamsOwner, | ||
590 | ) -> GenericParamsId { | ||
591 | let mut sm = &mut ArenaMap::default(); | ||
592 | let mut generics = GenericParams::default(); | ||
593 | match owner { | ||
594 | GenericsOwner::Function(func) => { | ||
595 | generics.fill(&self.body_ctx, sm, node); | ||
596 | // lower `impl Trait` in arguments | ||
597 | for param in &*func.params { | ||
598 | generics.fill_implicit_impl_trait_args(param); | ||
599 | } | ||
600 | } | ||
601 | GenericsOwner::Struct | ||
602 | | GenericsOwner::Enum | ||
603 | | GenericsOwner::Union | ||
604 | | GenericsOwner::TypeAlias => { | ||
605 | generics.fill(&self.body_ctx, sm, node); | ||
606 | } | ||
607 | GenericsOwner::Trait(trait_def) => { | ||
608 | // traits get the Self type as an implicit first type parameter | ||
609 | let self_param_id = generics.types.alloc(TypeParamData { | ||
610 | name: Some(name![Self]), | ||
611 | default: None, | ||
612 | provenance: TypeParamProvenance::TraitSelf, | ||
613 | }); | ||
614 | sm.insert(self_param_id, Either::Left(trait_def.clone())); | ||
615 | // add super traits as bounds on Self | ||
616 | // i.e., trait Foo: Bar is equivalent to trait Foo where Self: Bar | ||
617 | let self_param = TypeRef::Path(name![Self].into()); | ||
618 | generics.fill_bounds(&self.body_ctx, trait_def, self_param); | ||
619 | |||
620 | generics.fill(&self.body_ctx, &mut sm, node); | ||
621 | } | ||
622 | GenericsOwner::Impl => { | ||
623 | // Note that we don't add `Self` here: in `impl`s, `Self` is not a | ||
624 | // type-parameter, but rather is a type-alias for impl's target | ||
625 | // type, so this is handled by the resolver. | ||
626 | generics.fill(&self.body_ctx, &mut sm, node); | ||
627 | } | ||
628 | } | ||
629 | |||
630 | self.data().generics.alloc(generics) | ||
631 | } | ||
632 | |||
633 | fn lower_type_bounds(&mut self, node: &impl ast::TypeBoundsOwner) -> Vec<TypeBound> { | ||
634 | match node.type_bound_list() { | ||
635 | Some(bound_list) => { | ||
636 | bound_list.bounds().map(|it| TypeBound::from_ast(&self.body_ctx, it)).collect() | ||
637 | } | ||
638 | None => Vec::new(), | ||
639 | } | ||
640 | } | ||
641 | |||
642 | fn lower_visibility(&mut self, item: &impl ast::VisibilityOwner) -> RawVisibilityId { | ||
643 | let vis = match self.forced_visibility { | ||
644 | Some(vis) => return vis, | ||
645 | None => RawVisibility::from_ast_with_hygiene(item.visibility(), &self.hygiene), | ||
646 | }; | ||
647 | |||
648 | self.data().vis.alloc(vis) | ||
649 | } | ||
650 | |||
651 | fn lower_type_ref(&self, type_ref: &ast::Type) -> TypeRef { | ||
652 | TypeRef::from_ast(&self.body_ctx, type_ref.clone()) | ||
653 | } | ||
654 | fn lower_type_ref_opt(&self, type_ref: Option<ast::Type>) -> TypeRef { | ||
655 | type_ref.map(|ty| self.lower_type_ref(&ty)).unwrap_or(TypeRef::Error) | ||
656 | } | ||
657 | |||
658 | /// Forces the visibility `vis` to be used for all items lowered during execution of `f`. | ||
659 | fn with_inherited_visibility<R>( | ||
660 | &mut self, | ||
661 | vis: RawVisibilityId, | ||
662 | f: impl FnOnce(&mut Self) -> R, | ||
663 | ) -> R { | ||
664 | let old = mem::replace(&mut self.forced_visibility, Some(vis)); | ||
665 | let res = f(self); | ||
666 | self.forced_visibility = old; | ||
667 | res | ||
668 | } | ||
669 | |||
670 | fn next_field_idx(&self) -> Idx<Field> { | ||
671 | Idx::from_raw(RawId::from( | ||
672 | self.tree.data.as_ref().map_or(0, |data| data.fields.len() as u32), | ||
673 | )) | ||
674 | } | ||
675 | fn next_variant_idx(&self) -> Idx<Variant> { | ||
676 | Idx::from_raw(RawId::from( | ||
677 | self.tree.data.as_ref().map_or(0, |data| data.variants.len() as u32), | ||
678 | )) | ||
679 | } | ||
680 | } | ||
681 | |||
682 | fn desugar_future_path(orig: TypeRef) -> Path { | ||
683 | let path = path![core::future::Future]; | ||
684 | let mut generic_args: Vec<_> = std::iter::repeat(None).take(path.segments.len() - 1).collect(); | ||
685 | let mut last = GenericArgs::empty(); | ||
686 | let binding = | ||
687 | AssociatedTypeBinding { name: name![Output], type_ref: Some(orig), bounds: Vec::new() }; | ||
688 | last.bindings.push(binding); | ||
689 | generic_args.push(Some(Arc::new(last))); | ||
690 | |||
691 | Path::from_known_path(path, generic_args) | ||
692 | } | ||
693 | |||
694 | enum GenericsOwner<'a> { | ||
695 | /// We need access to the partially-lowered `Function` for lowering `impl Trait` in argument | ||
696 | /// position. | ||
697 | Function(&'a Function), | ||
698 | Struct, | ||
699 | Enum, | ||
700 | Union, | ||
701 | /// The `TraitDef` is needed to fill the source map for the implicit `Self` parameter. | ||
702 | Trait(&'a ast::Trait), | ||
703 | TypeAlias, | ||
704 | Impl, | ||
705 | } | ||