mirror of https://github.com/rust-lang/rust.git
Collect field data for structs/enum variants
This commit is contained in:
parent
4cb7b0f2af
commit
07a7285965
|
@ -1,10 +1,11 @@
|
|||
use std::sync::Arc;
|
||||
|
||||
use ra_syntax::{SmolStr, ast::{self, NameOwner}};
|
||||
use ra_syntax::{SmolStr, ast::{self, NameOwner, StructFlavor}};
|
||||
|
||||
use crate::{
|
||||
DefId, Cancelable,
|
||||
db::{HirDatabase},
|
||||
module::Module,
|
||||
ty::{Ty},
|
||||
};
|
||||
|
||||
|
@ -37,14 +38,18 @@ pub struct StructData {
|
|||
}
|
||||
|
||||
impl StructData {
|
||||
pub(crate) fn new(struct_def: ast::StructDef) -> StructData {
|
||||
pub(crate) fn new(
|
||||
db: &impl HirDatabase,
|
||||
module: &Module,
|
||||
struct_def: ast::StructDef,
|
||||
) -> Cancelable<StructData> {
|
||||
let name = struct_def
|
||||
.name()
|
||||
.map(|n| n.text())
|
||||
.unwrap_or(SmolStr::new("[error]"));
|
||||
let variant_data = VariantData::Unit; // TODO implement this
|
||||
let variant_data = VariantData::new(db, module, struct_def.flavor())?;
|
||||
let variant_data = Arc::new(variant_data);
|
||||
StructData { name, variant_data }
|
||||
Ok(StructData { name, variant_data })
|
||||
}
|
||||
|
||||
pub fn name(&self) -> &SmolStr {
|
||||
|
@ -81,13 +86,30 @@ pub struct EnumData {
|
|||
}
|
||||
|
||||
impl EnumData {
|
||||
pub(crate) fn new(enum_def: ast::EnumDef) -> Self {
|
||||
pub(crate) fn new(
|
||||
db: &impl HirDatabase,
|
||||
module: &Module,
|
||||
enum_def: ast::EnumDef,
|
||||
) -> Cancelable<Self> {
|
||||
let name = enum_def
|
||||
.name()
|
||||
.map(|n| n.text())
|
||||
.unwrap_or(SmolStr::new("[error]"));
|
||||
let variants = Vec::new(); // TODO implement this
|
||||
EnumData { name, variants }
|
||||
let variants = if let Some(evl) = enum_def.variant_list() {
|
||||
evl.variants()
|
||||
.map(|v| {
|
||||
Ok((
|
||||
v.name()
|
||||
.map(|n| n.text())
|
||||
.unwrap_or_else(|| SmolStr::new("[error]")),
|
||||
Arc::new(VariantData::new(db, module, v.flavor())?),
|
||||
))
|
||||
})
|
||||
.collect::<Cancelable<_>>()?
|
||||
} else {
|
||||
Vec::new()
|
||||
};
|
||||
Ok(EnumData { name, variants })
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -107,6 +129,39 @@ pub enum VariantData {
|
|||
}
|
||||
|
||||
impl VariantData {
|
||||
pub fn new(db: &impl HirDatabase, module: &Module, flavor: StructFlavor) -> Cancelable<Self> {
|
||||
Ok(match flavor {
|
||||
StructFlavor::Tuple(fl) => {
|
||||
let fields = fl
|
||||
.fields()
|
||||
.enumerate()
|
||||
.map(|(i, fd)| {
|
||||
Ok(StructField {
|
||||
name: SmolStr::new(i.to_string()),
|
||||
ty: Ty::new_opt(db, &module, fd.type_ref())?,
|
||||
})
|
||||
})
|
||||
.collect::<Cancelable<_>>()?;
|
||||
VariantData::Tuple(fields)
|
||||
}
|
||||
StructFlavor::Named(fl) => {
|
||||
let fields = fl
|
||||
.fields()
|
||||
.map(|fd| {
|
||||
Ok(StructField {
|
||||
name: fd
|
||||
.name()
|
||||
.map(|n| n.text())
|
||||
.unwrap_or_else(|| SmolStr::new("[error]")),
|
||||
ty: Ty::new_opt(db, &module, fd.type_ref())?,
|
||||
})
|
||||
})
|
||||
.collect::<Cancelable<_>>()?;
|
||||
VariantData::Struct(fields)
|
||||
}
|
||||
StructFlavor::Unit => VariantData::Unit,
|
||||
})
|
||||
}
|
||||
pub fn fields(&self) -> &[StructField] {
|
||||
match *self {
|
||||
VariantData::Struct(ref fields) | VariantData::Tuple(ref fields) => fields,
|
||||
|
|
|
@ -46,8 +46,7 @@ impl Function {
|
|||
}
|
||||
|
||||
pub fn module(&self, db: &impl HirDatabase) -> Cancelable<Module> {
|
||||
let loc = self.fn_id.0.loc(db);
|
||||
Module::new(db, loc.source_root_id, loc.module_id)
|
||||
self.fn_id.0.module(db)
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
@ -135,6 +135,12 @@ impl DefId {
|
|||
};
|
||||
Ok(res)
|
||||
}
|
||||
|
||||
/// For a module, returns that module; for any other def, returns the containing module.
|
||||
pub fn module(self, db: &impl HirDatabase) -> Cancelable<Module> {
|
||||
let loc = self.loc(db);
|
||||
Module::new(db, loc.source_root_id, loc.module_id)
|
||||
}
|
||||
}
|
||||
|
||||
/// Identifier of item within a specific file. This is stable over reparses, so
|
||||
|
|
|
@ -52,7 +52,12 @@ pub(super) fn struct_data(db: &impl HirDatabase, def_id: DefId) -> Cancelable<Ar
|
|||
let syntax = db.file_item(def_loc.source_item_id);
|
||||
let struct_def =
|
||||
ast::StructDef::cast(syntax.borrowed()).expect("struct def should point to StructDef node");
|
||||
Ok(Arc::new(StructData::new(struct_def.borrowed())))
|
||||
let module = def_id.module(db)?;
|
||||
Ok(Arc::new(StructData::new(
|
||||
db,
|
||||
&module,
|
||||
struct_def.borrowed(),
|
||||
)?))
|
||||
}
|
||||
|
||||
pub(super) fn enum_data(db: &impl HirDatabase, def_id: DefId) -> Cancelable<Arc<EnumData>> {
|
||||
|
@ -61,7 +66,8 @@ pub(super) fn enum_data(db: &impl HirDatabase, def_id: DefId) -> Cancelable<Arc<
|
|||
let syntax = db.file_item(def_loc.source_item_id);
|
||||
let enum_def =
|
||||
ast::EnumDef::cast(syntax.borrowed()).expect("enum def should point to EnumDef node");
|
||||
Ok(Arc::new(EnumData::new(enum_def.borrowed())))
|
||||
let module = def_id.module(db)?;
|
||||
Ok(Arc::new(EnumData::new(db, &module, enum_def.borrowed())?))
|
||||
}
|
||||
|
||||
pub(super) fn file_items(db: &impl HirDatabase, file_id: FileId) -> Arc<SourceFileItems> {
|
||||
|
|
|
@ -156,6 +156,15 @@ impl Ty {
|
|||
Ok(ty)
|
||||
}
|
||||
|
||||
pub(crate) fn new_opt(
|
||||
db: &impl HirDatabase,
|
||||
module: &Module,
|
||||
node: Option<ast::TypeRef>,
|
||||
) -> Cancelable<Self> {
|
||||
node.map(|n| Ty::new(db, module, n))
|
||||
.unwrap_or(Ok(Ty::Unknown))
|
||||
}
|
||||
|
||||
pub(crate) fn new(
|
||||
db: &impl HirDatabase,
|
||||
module: &Module,
|
||||
|
@ -534,7 +543,7 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
|
|||
Ty::Unknown
|
||||
}
|
||||
ast::Expr::StructLit(e) => {
|
||||
let (ty, variant_data) = self.resolve_variant(e.path())?;
|
||||
let (ty, _variant_data) = self.resolve_variant(e.path())?;
|
||||
if let Some(nfl) = e.named_field_list() {
|
||||
for field in nfl.fields() {
|
||||
if let Some(e) = field.expr() {
|
||||
|
|
Loading…
Reference in New Issue