rust/crates/ra_hir_def/src/item_tree.rs

860 lines
28 KiB
Rust
Raw Normal View History

2020-03-25 09:33:01 -05:00
//! A simplified AST that only contains items.
use hir_expand::{
ast_id_map::{AstIdMap, FileAstId},
hygiene::Hygiene,
name::{name, AsName, Name},
2020-06-12 06:58:02 -05:00
HirFileId, InFile,
2020-03-25 09:33:01 -05:00
};
use ra_arena::{Arena, Idx, RawId};
2020-06-11 12:46:56 -05:00
use ra_syntax::{ast, match_ast};
2020-03-25 09:33:01 -05:00
use crate::{
attr::Attrs,
2020-06-11 12:46:56 -05:00
db::DefDatabase,
2020-06-12 06:58:02 -05:00
generics,
2020-03-25 09:33:01 -05:00
path::{path, AssociatedTypeBinding, GenericArgs, ImportAlias, ModPath, Path},
type_ref::{Mutability, TypeBound, TypeRef},
visibility::RawVisibility,
};
2020-06-12 06:58:02 -05:00
use ast::{AstNode, AttrsOwner, ModuleItemOwner, NameOwner, StructKind, TypeAscriptionOwner};
use either::Either;
2020-06-15 12:16:02 -05:00
use rustc_hash::FxHashMap;
2020-06-12 06:58:02 -05:00
use smallvec::SmallVec;
2020-03-25 09:33:01 -05:00
use std::{
ops::{Index, Range},
sync::Arc,
};
2020-06-12 06:58:02 -05:00
/// The item tree of a source file.
2020-06-11 12:46:56 -05:00
#[derive(Debug, Default, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct ItemTree {
2020-06-11 12:46:56 -05:00
top_level: Vec<ModItem>,
2020-06-15 12:16:02 -05:00
top_attrs: Attrs,
attrs: FxHashMap<ModItem, Attrs>,
empty_attrs: Attrs,
2020-03-25 09:33:01 -05:00
imports: Arena<Import>,
functions: Arena<Function>,
structs: Arena<Struct>,
fields: Arena<Field>,
unions: Arena<Union>,
enums: Arena<Enum>,
variants: Arena<Variant>,
consts: Arena<Const>,
statics: Arena<Static>,
traits: Arena<Trait>,
impls: Arena<Impl>,
type_aliases: Arena<TypeAlias>,
mods: Arena<Mod>,
macro_calls: Arena<MacroCall>,
exprs: Arena<Expr>,
}
impl ItemTree {
2020-06-11 12:46:56 -05:00
pub fn item_tree_query(db: &dyn DefDatabase, file_id: HirFileId) -> Arc<ItemTree> {
2020-06-15 12:16:02 -05:00
let _p = ra_prof::profile("item_tree_query");
2020-06-11 12:46:56 -05:00
let syntax = if let Some(node) = db.parse_or_expand(file_id) {
node
} else {
return Default::default();
};
2020-06-15 12:16:02 -05:00
let hygiene = Hygiene::new(db.upcast(), file_id);
let mut top_attrs = None;
2020-06-11 12:46:56 -05:00
let (macro_storage, file_storage);
let item_owner = match_ast! {
match syntax {
ast::MacroItems(items) => {
macro_storage = items;
&macro_storage as &dyn ModuleItemOwner
},
ast::SourceFile(file) => {
2020-06-15 12:16:02 -05:00
top_attrs = Some(Attrs::new(&file, &hygiene));
2020-06-11 12:46:56 -05:00
file_storage = file;
&file_storage
},
_ => return Default::default(),
}
};
let map = db.ast_id_map(file_id);
2020-06-15 12:16:02 -05:00
let mut ctx = Ctx {
2020-06-11 12:46:56 -05:00
tree: ItemTree::default(),
2020-06-15 12:16:02 -05:00
hygiene,
2020-06-12 06:58:02 -05:00
file: file_id,
2020-06-11 12:46:56 -05:00
source_ast_id_map: map,
body_ctx: crate::body::LowerCtx::new(db, file_id),
};
2020-06-15 12:16:02 -05:00
ctx.tree.top_attrs = top_attrs.unwrap_or_default();
2020-06-11 12:46:56 -05:00
Arc::new(ctx.lower(item_owner))
}
/// Returns an iterator over all items located at the top level of the `HirFileId` this
/// `ItemTree` was created from.
2020-06-12 16:24:26 -05:00
pub fn top_level_items(&self) -> &[ModItem] {
&self.top_level
2020-03-25 09:33:01 -05:00
}
2020-06-15 12:16:02 -05:00
/// Returns the inner attributes of the source file.
pub fn top_level_attrs(&self) -> &Attrs {
&self.top_attrs
}
pub fn attrs(&self, of: ModItem) -> &Attrs {
self.attrs.get(&of).unwrap_or(&self.empty_attrs)
}
2020-03-25 09:33:01 -05:00
}
macro_rules! impl_index {
( $($fld:ident: $t:ty),+ $(,)? ) => {
$(
impl Index<Idx<$t>> for ItemTree {
type Output = $t;
fn index(&self, index: Idx<$t>) -> &Self::Output {
&self.$fld[index]
}
}
)+
};
}
impl_index!(
imports: Import,
functions: Function,
structs: Struct,
fields: Field,
unions: Union,
enums: Enum,
variants: Variant,
consts: Const,
statics: Static,
traits: Trait,
impls: Impl,
type_aliases: TypeAlias,
mods: Mod,
macro_calls: MacroCall,
exprs: Expr,
);
2020-06-12 06:58:02 -05:00
/// A desugared `extern crate` or `use` import.
2020-06-12 16:24:26 -05:00
#[derive(Debug, Clone, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Import {
pub path: ModPath,
pub alias: Option<ImportAlias>,
pub visibility: RawVisibility,
pub is_glob: bool,
pub is_prelude: bool,
pub is_extern_crate: bool,
pub is_macro_use: bool,
}
2020-06-11 12:46:56 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Function {
pub name: Name,
pub attrs: Attrs,
pub visibility: RawVisibility,
2020-06-12 06:58:02 -05:00
pub generic_params: generics::GenericParams,
2020-03-25 09:33:01 -05:00
pub has_self_param: bool,
pub params: Vec<TypeRef>,
pub ret_type: TypeRef,
2020-06-15 12:16:02 -05:00
pub ast_id: FileAstId<ast::FnDef>,
2020-06-12 06:58:02 -05:00
// FIXME inner items
2020-03-25 09:33:01 -05:00
}
2020-06-11 12:46:56 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Struct {
pub name: Name,
pub attrs: Attrs,
pub visibility: RawVisibility,
2020-06-12 06:58:02 -05:00
pub generic_params: generics::GenericParams,
2020-03-25 09:33:01 -05:00
pub fields: Fields,
2020-06-15 12:16:02 -05:00
pub ast_id: FileAstId<ast::StructDef>,
pub kind: StructDefKind,
}
#[derive(Debug, Eq, PartialEq)]
pub enum StructDefKind {
/// `struct S { ... }` - type namespace only.
Record,
/// `struct S(...);`
Tuple,
/// `struct S;`
Unit,
2020-03-25 09:33:01 -05:00
}
2020-06-11 12:46:56 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Union {
pub name: Name,
pub attrs: Attrs,
pub visibility: RawVisibility,
2020-06-12 06:58:02 -05:00
pub generic_params: generics::GenericParams,
2020-03-25 09:33:01 -05:00
pub fields: Fields,
2020-06-15 12:16:02 -05:00
pub ast_id: FileAstId<ast::UnionDef>,
2020-03-25 09:33:01 -05:00
}
2020-06-11 12:46:56 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Enum {
pub name: Name,
pub attrs: Attrs,
pub visibility: RawVisibility,
2020-06-12 06:58:02 -05:00
pub generic_params: generics::GenericParams,
2020-03-25 09:33:01 -05:00
pub variants: Range<Idx<Variant>>,
2020-06-15 12:16:02 -05:00
pub ast_id: FileAstId<ast::EnumDef>,
2020-03-25 09:33:01 -05:00
}
2020-06-11 12:46:56 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Const {
/// const _: () = ();
pub name: Option<Name>,
pub visibility: RawVisibility,
pub type_ref: TypeRef,
2020-06-15 12:16:02 -05:00
pub ast_id: FileAstId<ast::ConstDef>,
2020-03-25 09:33:01 -05:00
}
2020-06-11 12:46:56 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Static {
pub name: Name,
pub visibility: RawVisibility,
pub type_ref: TypeRef,
2020-06-15 12:16:02 -05:00
pub ast_id: FileAstId<ast::StaticDef>,
2020-03-25 09:33:01 -05:00
}
2020-06-11 12:46:56 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Trait {
pub name: Name,
pub visibility: RawVisibility,
2020-06-12 06:58:02 -05:00
pub generic_params: generics::GenericParams,
2020-03-25 09:33:01 -05:00
pub auto: bool,
pub items: Vec<AssocItem>,
2020-06-15 12:16:02 -05:00
pub ast_id: FileAstId<ast::TraitDef>,
2020-03-25 09:33:01 -05:00
}
2020-06-11 12:46:56 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Impl {
2020-06-12 06:58:02 -05:00
pub generic_params: generics::GenericParams,
2020-03-25 09:33:01 -05:00
pub target_trait: Option<TypeRef>,
pub target_type: TypeRef,
pub is_negative: bool,
pub items: Vec<AssocItem>,
2020-06-15 12:16:02 -05:00
pub ast_id: FileAstId<ast::ImplDef>,
2020-03-25 09:33:01 -05:00
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct TypeAlias {
pub name: Name,
pub visibility: RawVisibility,
2020-06-12 06:58:02 -05:00
pub generic_params: generics::GenericParams,
2020-03-25 09:33:01 -05:00
pub type_ref: Option<TypeRef>,
2020-06-15 12:16:02 -05:00
pub ast_id: FileAstId<ast::TypeAliasDef>,
2020-03-25 09:33:01 -05:00
}
2020-06-11 12:46:56 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Mod {
pub name: Name,
pub visibility: RawVisibility,
2020-06-15 12:16:02 -05:00
pub kind: ModKind,
pub ast_id: FileAstId<ast::Module>,
2020-03-25 09:33:01 -05:00
}
2020-06-15 12:16:02 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-06-12 16:24:26 -05:00
pub enum ModKind {
2020-06-15 12:16:02 -05:00
/// `mod m { ... }`
2020-06-12 16:24:26 -05:00
Inline { items: Vec<ModItem> },
2020-06-15 12:16:02 -05:00
/// `mod m;`
2020-06-12 16:24:26 -05:00
Outline {},
}
2020-06-11 12:46:56 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct MacroCall {
2020-06-15 12:16:02 -05:00
/// For `macro_rules!` declarations, this is the name of the declared macro.
2020-03-25 09:33:01 -05:00
pub name: Option<Name>,
2020-06-15 12:16:02 -05:00
/// Path to the called macro.
2020-03-25 09:33:01 -05:00
pub path: ModPath,
2020-06-12 06:58:02 -05:00
/// Has `#[macro_export]`.
pub is_export: bool,
/// Has `#[macro_export(local_inner_macros)]`.
pub is_local_inner: bool,
/// Has `#[rustc_builtin_macro]`.
pub is_builtin: bool,
2020-03-25 09:33:01 -05:00
pub ast_id: FileAstId<ast::MacroCall>,
}
// NB: There's no `FileAstId` for `Expr`. The only case where this would be useful is for array
// lengths, but we don't do much with them yet.
2020-06-11 12:46:56 -05:00
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Expr;
2020-06-11 12:46:56 -05:00
macro_rules! impl_froms {
($e:ident { $($v:ident ($t:ty)),* $(,)? }) => {
$(
impl From<$t> for $e {
fn from(it: $t) -> $e {
$e::$v(it)
}
}
)*
}
}
2020-06-15 12:16:02 -05:00
#[derive(Debug, Copy, Clone, Eq, PartialEq, Hash)]
2020-03-25 09:33:01 -05:00
pub enum ModItem {
Import(Idx<Import>),
Function(Idx<Function>),
Struct(Idx<Struct>),
Union(Idx<Union>),
Enum(Idx<Enum>),
Const(Idx<Const>),
Static(Idx<Static>),
Trait(Idx<Trait>),
Impl(Idx<Impl>),
TypeAlias(Idx<TypeAlias>),
Mod(Idx<Mod>),
MacroCall(Idx<MacroCall>),
}
2020-06-11 12:46:56 -05:00
impl_froms!(ModItem {
Import(Idx<Import>),
Function(Idx<Function>),
Struct(Idx<Struct>),
Union(Idx<Union>),
Enum(Idx<Enum>),
Const(Idx<Const>),
Static(Idx<Static>),
Trait(Idx<Trait>),
Impl(Idx<Impl>),
TypeAlias(Idx<TypeAlias>),
Mod(Idx<Mod>),
MacroCall(Idx<MacroCall>),
});
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub enum AssocItem {
Function(Idx<Function>),
TypeAlias(Idx<TypeAlias>),
Const(Idx<Const>),
MacroCall(Idx<MacroCall>),
}
2020-06-11 12:46:56 -05:00
impl_froms!(AssocItem {
Function(Idx<Function>),
TypeAlias(Idx<TypeAlias>),
Const(Idx<Const>),
MacroCall(Idx<MacroCall>),
});
#[derive(Debug, Eq, PartialEq)]
2020-03-25 09:33:01 -05:00
pub struct Variant {
pub name: Name,
pub fields: Fields,
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum Fields {
Record(Range<Idx<Field>>),
Tuple(Range<Idx<Field>>),
Unit,
}
/// A single field of an enum variant or struct
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Field {
pub name: Name,
pub type_ref: TypeRef,
pub visibility: RawVisibility,
}
2020-06-12 06:58:02 -05:00
struct ModItems(SmallVec<[ModItem; 1]>);
impl<T> From<T> for ModItems
where
T: Into<ModItem>,
{
fn from(t: T) -> Self {
ModItems(SmallVec::from_buf([t.into(); 1]))
}
}
2020-03-25 09:33:01 -05:00
struct Ctx {
tree: ItemTree,
hygiene: Hygiene,
2020-06-12 06:58:02 -05:00
file: HirFileId,
2020-03-25 09:33:01 -05:00
source_ast_id_map: Arc<AstIdMap>,
body_ctx: crate::body::LowerCtx,
}
impl Ctx {
2020-06-11 12:46:56 -05:00
fn lower(mut self, item_owner: &dyn ModuleItemOwner) -> ItemTree {
2020-06-12 06:58:02 -05:00
self.tree.top_level = item_owner
.items()
.flat_map(|item| self.lower_mod_item(&item))
.flat_map(|items| items.0)
.collect();
2020-06-11 12:46:56 -05:00
self.tree
2020-03-25 09:33:01 -05:00
}
2020-06-12 06:58:02 -05:00
fn lower_mod_item(&mut self, item: &ast::ModuleItem) -> Option<ModItems> {
2020-06-15 12:16:02 -05:00
let attrs = Attrs::new(item, &self.hygiene);
let items = match item {
2020-03-25 09:33:01 -05:00
ast::ModuleItem::StructDef(ast) => {
2020-06-11 12:46:56 -05:00
self.lower_struct(ast).map(|data| self.tree.structs.alloc(data).into())
2020-03-25 09:33:01 -05:00
}
ast::ModuleItem::UnionDef(ast) => {
2020-06-11 12:46:56 -05:00
self.lower_union(ast).map(|data| self.tree.unions.alloc(data).into())
2020-03-25 09:33:01 -05:00
}
ast::ModuleItem::EnumDef(ast) => {
2020-06-11 12:46:56 -05:00
self.lower_enum(ast).map(|data| self.tree.enums.alloc(data).into())
2020-03-25 09:33:01 -05:00
}
ast::ModuleItem::FnDef(ast) => {
2020-06-11 12:46:56 -05:00
self.lower_function(ast).map(|data| self.tree.functions.alloc(data).into())
2020-03-25 09:33:01 -05:00
}
ast::ModuleItem::TypeAliasDef(ast) => {
2020-06-11 12:46:56 -05:00
self.lower_type_alias(ast).map(|data| self.tree.type_aliases.alloc(data).into())
2020-03-25 09:33:01 -05:00
}
ast::ModuleItem::StaticDef(ast) => {
2020-06-11 12:46:56 -05:00
self.lower_static(ast).map(|data| self.tree.statics.alloc(data).into())
2020-03-25 09:33:01 -05:00
}
ast::ModuleItem::ConstDef(ast) => {
let data = self.lower_const(ast);
2020-06-11 12:46:56 -05:00
Some(self.tree.consts.alloc(data).into())
}
ast::ModuleItem::Module(ast) => {
self.lower_module(ast).map(|data| self.tree.mods.alloc(data).into())
2020-03-25 09:33:01 -05:00
}
2020-06-12 06:58:02 -05:00
ast::ModuleItem::TraitDef(ast) => {
self.lower_trait(ast).map(|data| self.tree.traits.alloc(data).into())
}
ast::ModuleItem::ImplDef(ast) => {
self.lower_impl(ast).map(|data| self.tree.impls.alloc(data).into())
}
ast::ModuleItem::UseItem(ast) => Some(ModItems(
self.lower_use(ast)
.into_iter()
.map(|data| self.tree.imports.alloc(data).into())
.collect::<SmallVec<_>>(),
)),
ast::ModuleItem::ExternCrateItem(ast) => {
self.lower_extern_crate(ast).map(|data| self.tree.imports.alloc(data).into())
}
ast::ModuleItem::MacroCall(ast) => {
self.lower_macro_call(ast).map(|data| self.tree.macro_calls.alloc(data).into())
}
ast::ModuleItem::ExternBlock(ast) => Some(ModItems(
self.lower_extern_block(ast)
.into_iter()
.map(|item| match item {
Either::Left(func) => self.tree.functions.alloc(func).into(),
Either::Right(statik) => self.tree.statics.alloc(statik).into(),
})
.collect::<SmallVec<_>>(),
)),
2020-06-15 12:16:02 -05:00
};
if !attrs.is_empty() {
for item in items.iter().flat_map(|items| &items.0) {
self.tree.attrs.insert(*item, attrs.clone());
}
2020-06-12 06:58:02 -05:00
}
2020-06-15 12:16:02 -05:00
items
2020-06-12 06:58:02 -05:00
}
fn lower_assoc_item(&mut self, item: &ast::AssocItem) -> Option<AssocItem> {
match item {
ast::AssocItem::FnDef(ast) => {
self.lower_function(ast).map(|data| self.tree.functions.alloc(data).into())
}
ast::AssocItem::TypeAliasDef(ast) => {
self.lower_type_alias(ast).map(|data| self.tree.type_aliases.alloc(data).into())
}
ast::AssocItem::ConstDef(ast) => {
let data = self.lower_const(ast);
Some(self.tree.consts.alloc(data).into())
}
2020-03-25 09:33:01 -05:00
}
}
fn lower_struct(&mut self, strukt: &ast::StructDef) -> Option<Struct> {
let attrs = self.lower_attrs(strukt);
let visibility = self.lower_visibility(strukt);
let name = strukt.name()?.as_name();
let generic_params = self.lower_generic_params(strukt);
let fields = self.lower_fields(&strukt.kind());
2020-06-15 12:16:02 -05:00
let ast_id = self.source_ast_id_map.ast_id(strukt);
let kind = match strukt.kind() {
ast::StructKind::Record(_) => StructDefKind::Record,
ast::StructKind::Tuple(_) => StructDefKind::Tuple,
ast::StructKind::Unit => StructDefKind::Unit,
};
let res = Struct { name, attrs, visibility, generic_params, fields, ast_id, kind };
2020-03-25 09:33:01 -05:00
Some(res)
}
fn lower_fields(&mut self, strukt_kind: &ast::StructKind) -> Fields {
match strukt_kind {
ast::StructKind::Record(it) => {
let range = self.lower_record_fields(it);
Fields::Record(range)
}
ast::StructKind::Tuple(it) => {
let range = self.lower_tuple_fields(it);
Fields::Tuple(range)
}
ast::StructKind::Unit => Fields::Unit,
}
}
fn lower_record_fields(&mut self, fields: &ast::RecordFieldDefList) -> Range<Idx<Field>> {
let start = self.next_field_idx();
for field in fields.fields() {
if let Some(data) = self.lower_record_field(&field) {
2020-06-12 06:58:02 -05:00
self.tree.fields.alloc(data);
2020-03-25 09:33:01 -05:00
}
}
let end = self.next_field_idx();
start..end
}
fn lower_record_field(&self, field: &ast::RecordFieldDef) -> Option<Field> {
let name = field.name()?.as_name();
let visibility = self.lower_visibility(field);
let type_ref = self.lower_type_ref(&field.ascribed_type()?);
let res = Field { name, type_ref, visibility };
Some(res)
}
fn lower_tuple_fields(&mut self, fields: &ast::TupleFieldDefList) -> Range<Idx<Field>> {
let start = self.next_field_idx();
for (i, field) in fields.fields().enumerate() {
if let Some(data) = self.lower_tuple_field(i, &field) {
2020-06-12 06:58:02 -05:00
self.tree.fields.alloc(data);
2020-03-25 09:33:01 -05:00
}
}
let end = self.next_field_idx();
start..end
}
fn lower_tuple_field(&self, idx: usize, field: &ast::TupleFieldDef) -> Option<Field> {
let name = Name::new_tuple_field(idx);
let visibility = self.lower_visibility(field);
let type_ref = self.lower_type_ref(&field.type_ref()?);
let res = Field { name, type_ref, visibility };
Some(res)
}
fn lower_union(&mut self, union: &ast::UnionDef) -> Option<Union> {
let attrs = self.lower_attrs(union);
let visibility = self.lower_visibility(union);
let name = union.name()?.as_name();
let generic_params = self.lower_generic_params(union);
let fields = match union.record_field_def_list() {
Some(record_field_def_list) => {
self.lower_fields(&StructKind::Record(record_field_def_list))
}
None => Fields::Record(self.next_field_idx()..self.next_field_idx()),
};
2020-06-15 12:16:02 -05:00
let ast_id = self.source_ast_id_map.ast_id(union);
let res = Union { name, attrs, visibility, generic_params, fields, ast_id };
2020-03-25 09:33:01 -05:00
Some(res)
}
fn lower_enum(&mut self, enum_: &ast::EnumDef) -> Option<Enum> {
let attrs = self.lower_attrs(enum_);
let visibility = self.lower_visibility(enum_);
let name = enum_.name()?.as_name();
let generic_params = self.lower_generic_params(enum_);
let variants = match &enum_.variant_list() {
Some(variant_list) => self.lower_variants(variant_list),
None => self.next_variant_idx()..self.next_variant_idx(),
};
2020-06-15 12:16:02 -05:00
let ast_id = self.source_ast_id_map.ast_id(enum_);
let res = Enum { name, attrs, visibility, generic_params, variants, ast_id };
2020-03-25 09:33:01 -05:00
Some(res)
}
fn lower_variants(&mut self, variants: &ast::EnumVariantList) -> Range<Idx<Variant>> {
let start = self.next_variant_idx();
for variant in variants.variants() {
if let Some(data) = self.lower_variant(&variant) {
2020-06-12 06:58:02 -05:00
self.tree.variants.alloc(data);
2020-03-25 09:33:01 -05:00
}
}
let end = self.next_variant_idx();
start..end
}
fn lower_variant(&mut self, variant: &ast::EnumVariant) -> Option<Variant> {
let name = variant.name()?.as_name();
let fields = self.lower_fields(&variant.kind());
let res = Variant { name, fields };
Some(res)
}
fn lower_function(&mut self, func: &ast::FnDef) -> Option<Function> {
let attrs = self.lower_attrs(func);
let visibility = self.lower_visibility(func);
let name = func.name()?.as_name();
let generic_params = self.lower_generic_params(func);
let mut params = Vec::new();
let mut has_self_param = false;
if let Some(param_list) = func.param_list() {
if let Some(self_param) = param_list.self_param() {
let self_type = if let Some(type_ref) = self_param.ascribed_type() {
TypeRef::from_ast(&self.body_ctx, type_ref)
} else {
let self_type = TypeRef::Path(name![Self].into());
match self_param.kind() {
ast::SelfParamKind::Owned => self_type,
ast::SelfParamKind::Ref => {
TypeRef::Reference(Box::new(self_type), Mutability::Shared)
}
ast::SelfParamKind::MutRef => {
TypeRef::Reference(Box::new(self_type), Mutability::Mut)
}
}
};
params.push(self_type);
has_self_param = true;
}
for param in param_list.params() {
let type_ref = TypeRef::from_ast_opt(&self.body_ctx, param.ascribed_type());
params.push(type_ref);
}
}
let ret_type = match func.ret_type().and_then(|rt| rt.type_ref()) {
Some(type_ref) => TypeRef::from_ast(&self.body_ctx, type_ref),
_ => TypeRef::unit(),
};
let ret_type = if func.async_token().is_some() {
let future_impl = desugar_future_path(ret_type);
let ty_bound = TypeBound::Path(future_impl);
TypeRef::ImplTrait(vec![ty_bound])
} else {
ret_type
};
2020-06-15 12:16:02 -05:00
let ast_id = self.source_ast_id_map.ast_id(func);
2020-03-25 09:33:01 -05:00
let res = Function {
name,
attrs,
visibility,
generic_params,
has_self_param,
params,
ret_type,
2020-06-15 12:16:02 -05:00
ast_id,
2020-03-25 09:33:01 -05:00
};
Some(res)
}
fn lower_type_alias(&mut self, type_alias: &ast::TypeAliasDef) -> Option<TypeAlias> {
let name = type_alias.name()?.as_name();
let type_ref = type_alias.type_ref().map(|it| self.lower_type_ref(&it));
let visibility = self.lower_visibility(type_alias);
let generic_params = self.lower_generic_params(type_alias);
2020-06-15 12:16:02 -05:00
let ast_id = self.source_ast_id_map.ast_id(type_alias);
let res = TypeAlias { name, visibility, generic_params, type_ref, ast_id };
2020-03-25 09:33:01 -05:00
Some(res)
}
fn lower_static(&mut self, static_: &ast::StaticDef) -> Option<Static> {
let name = static_.name()?.as_name();
let type_ref = self.lower_type_ref_opt(static_.ascribed_type());
let visibility = self.lower_visibility(static_);
2020-06-15 12:16:02 -05:00
let ast_id = self.source_ast_id_map.ast_id(static_);
let res = Static { name, visibility, type_ref, ast_id };
2020-03-25 09:33:01 -05:00
Some(res)
}
fn lower_const(&mut self, konst: &ast::ConstDef) -> Const {
let name = konst.name().map(|it| it.as_name());
let type_ref = self.lower_type_ref_opt(konst.ascribed_type());
let visibility = self.lower_visibility(konst);
2020-06-15 12:16:02 -05:00
let ast_id = self.source_ast_id_map.ast_id(konst);
Const { name, visibility, type_ref, ast_id }
2020-03-25 09:33:01 -05:00
}
2020-06-11 12:46:56 -05:00
fn lower_module(&mut self, module: &ast::Module) -> Option<Mod> {
let name = module.name()?.as_name();
let visibility = self.lower_visibility(module);
2020-06-15 12:16:02 -05:00
let kind = if module.semicolon_token().is_some() {
ModKind::Outline {}
} else {
ModKind::Inline {
items: module
.item_list()
.map(|list| {
list.items()
.flat_map(|item| self.lower_mod_item(&item))
.flat_map(|items| items.0)
.collect()
})
.unwrap_or_default(),
}
};
let ast_id = self.source_ast_id_map.ast_id(module);
Some(Mod { name, visibility, kind, ast_id })
2020-06-11 12:46:56 -05:00
}
2020-06-12 06:58:02 -05:00
fn lower_trait(&mut self, trait_def: &ast::TraitDef) -> Option<Trait> {
let name = trait_def.name()?.as_name();
let visibility = self.lower_visibility(trait_def);
let generic_params = self.lower_generic_params(trait_def);
let auto = trait_def.auto_token().is_some();
2020-06-15 12:16:02 -05:00
let items = trait_def.item_list().map(|list| {
2020-06-12 06:58:02 -05:00
// FIXME: Does not handle macros
2020-06-15 12:16:02 -05:00
list.assoc_items().flat_map(|item| self.lower_assoc_item(&item)).collect()
2020-06-12 06:58:02 -05:00
});
2020-06-15 12:16:02 -05:00
let ast_id = self.source_ast_id_map.ast_id(trait_def);
Some(Trait {
name,
visibility,
generic_params,
auto,
items: items.unwrap_or_default(),
ast_id,
})
2020-06-12 06:58:02 -05:00
}
fn lower_impl(&mut self, impl_def: &ast::ImplDef) -> Option<Impl> {
let generic_params = self.lower_generic_params(impl_def);
let target_trait = impl_def.target_trait().map(|tr| self.lower_type_ref(&tr));
let target_type = self.lower_type_ref(&impl_def.target_type()?);
let is_negative = impl_def.excl_token().is_some();
let items = impl_def
.item_list()?
.assoc_items()
.filter_map(|item| self.lower_assoc_item(&item))
.collect();
2020-06-15 12:16:02 -05:00
let ast_id = self.source_ast_id_map.ast_id(impl_def);
Some(Impl { generic_params, target_trait, target_type, is_negative, items, ast_id })
2020-06-12 06:58:02 -05:00
}
fn lower_use(&mut self, use_item: &ast::UseItem) -> Vec<Import> {
// FIXME: cfg_attr
let is_prelude = use_item.has_atom_attr("prelude_import");
let visibility = self.lower_visibility(use_item);
// Every use item can expand to many `Import`s.
let mut imports = Vec::new();
ModPath::expand_use_item(
InFile::new(self.file, use_item.clone()),
&self.hygiene,
|path, _tree, is_glob, alias| {
imports.push(Import {
path,
alias,
visibility: visibility.clone(),
is_glob,
is_prelude,
is_extern_crate: false,
is_macro_use: false,
});
},
);
imports
}
fn lower_extern_crate(&mut self, extern_crate: &ast::ExternCrateItem) -> Option<Import> {
let path = ModPath::from_name_ref(&extern_crate.name_ref()?);
let alias = extern_crate.alias().map(|a| {
a.name().map(|it| it.as_name()).map_or(ImportAlias::Underscore, ImportAlias::Alias)
});
let visibility = self.lower_visibility(extern_crate);
// FIXME: cfg_attr
let is_macro_use = extern_crate.has_atom_attr("macro_use");
Some(Import {
path,
alias,
visibility,
is_glob: false,
is_prelude: false,
is_extern_crate: true,
is_macro_use,
})
}
fn lower_macro_call(&mut self, m: &ast::MacroCall) -> Option<MacroCall> {
let name = m.name().map(|it| it.as_name());
let attrs = Attrs::new(m, &self.hygiene);
let path = ModPath::from_src(m.path()?, &self.hygiene)?;
let ast_id = self.source_ast_id_map.ast_id(m);
// FIXME: cfg_attr
let export_attr = attrs.by_key("macro_export");
let is_export = export_attr.exists();
let is_local_inner = if is_export {
export_attr.tt_values().map(|it| &it.token_trees).flatten().any(|it| match it {
tt::TokenTree::Leaf(tt::Leaf::Ident(ident)) => {
ident.text.contains("local_inner_macros")
}
_ => false,
})
} else {
false
};
let is_builtin = attrs.by_key("rustc_builtin_macro").exists();
Some(MacroCall { name, path, is_export, is_builtin, is_local_inner, ast_id })
}
fn lower_extern_block(&mut self, block: &ast::ExternBlock) -> Vec<Either<Function, Static>> {
block.extern_item_list().map_or(Vec::new(), |list| {
list.extern_items()
.filter_map(|item| match item {
ast::ExternItem::FnDef(ast) => self.lower_function(&ast).map(Either::Left),
ast::ExternItem::StaticDef(ast) => self.lower_static(&ast).map(Either::Right),
})
.collect()
})
}
fn lower_generic_params(
&mut self,
_item: &impl ast::TypeParamsOwner,
) -> generics::GenericParams {
// TODO
generics::GenericParams { types: Arena::new(), where_predicates: Vec::new() }
2020-03-25 09:33:01 -05:00
}
fn lower_attrs(&self, item: &impl ast::AttrsOwner) -> Attrs {
Attrs::new(item, &self.hygiene)
}
fn lower_visibility(&self, item: &impl ast::VisibilityOwner) -> RawVisibility {
RawVisibility::from_ast_with_hygiene(item.visibility(), &self.hygiene)
}
fn lower_type_ref(&self, type_ref: &ast::TypeRef) -> TypeRef {
TypeRef::from_ast(&self.body_ctx, type_ref.clone())
}
fn lower_type_ref_opt(&self, type_ref: Option<ast::TypeRef>) -> TypeRef {
TypeRef::from_ast_opt(&self.body_ctx, type_ref)
}
fn next_field_idx(&self) -> Idx<Field> {
Idx::from_raw(RawId::from(self.tree.fields.len() as u32))
}
fn next_variant_idx(&self) -> Idx<Variant> {
Idx::from_raw(RawId::from(self.tree.variants.len() as u32))
}
}
fn desugar_future_path(orig: TypeRef) -> Path {
2020-06-12 06:58:02 -05:00
let path = path![core::future::Future];
2020-03-25 09:33:01 -05:00
let mut generic_args: Vec<_> = std::iter::repeat(None).take(path.segments.len() - 1).collect();
let mut last = GenericArgs::empty();
let binding =
AssociatedTypeBinding { name: name![Output], type_ref: Some(orig), bounds: Vec::new() };
last.bindings.push(binding);
generic_args.push(Some(Arc::new(last)));
Path::from_known_path(path, generic_args)
}