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 std::sync::Arc;
|
||||||
|
|
||||||
use ra_syntax::{SmolStr, ast::{self, NameOwner}};
|
use ra_syntax::{SmolStr, ast::{self, NameOwner, StructFlavor}};
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
DefId, Cancelable,
|
DefId, Cancelable,
|
||||||
db::{HirDatabase},
|
db::{HirDatabase},
|
||||||
|
module::Module,
|
||||||
ty::{Ty},
|
ty::{Ty},
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -37,14 +38,18 @@ pub struct StructData {
|
||||||
}
|
}
|
||||||
|
|
||||||
impl 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
|
let name = struct_def
|
||||||
.name()
|
.name()
|
||||||
.map(|n| n.text())
|
.map(|n| n.text())
|
||||||
.unwrap_or(SmolStr::new("[error]"));
|
.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);
|
let variant_data = Arc::new(variant_data);
|
||||||
StructData { name, variant_data }
|
Ok(StructData { name, variant_data })
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn name(&self) -> &SmolStr {
|
pub fn name(&self) -> &SmolStr {
|
||||||
|
@ -81,13 +86,30 @@ pub struct EnumData {
|
||||||
}
|
}
|
||||||
|
|
||||||
impl 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
|
let name = enum_def
|
||||||
.name()
|
.name()
|
||||||
.map(|n| n.text())
|
.map(|n| n.text())
|
||||||
.unwrap_or(SmolStr::new("[error]"));
|
.unwrap_or(SmolStr::new("[error]"));
|
||||||
let variants = Vec::new(); // TODO implement this
|
let variants = if let Some(evl) = enum_def.variant_list() {
|
||||||
EnumData { name, variants }
|
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 {
|
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] {
|
pub fn fields(&self) -> &[StructField] {
|
||||||
match *self {
|
match *self {
|
||||||
VariantData::Struct(ref fields) | VariantData::Tuple(ref fields) => fields,
|
VariantData::Struct(ref fields) | VariantData::Tuple(ref fields) => fields,
|
||||||
|
|
|
@ -46,8 +46,7 @@ impl Function {
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn module(&self, db: &impl HirDatabase) -> Cancelable<Module> {
|
pub fn module(&self, db: &impl HirDatabase) -> Cancelable<Module> {
|
||||||
let loc = self.fn_id.0.loc(db);
|
self.fn_id.0.module(db)
|
||||||
Module::new(db, loc.source_root_id, loc.module_id)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -135,6 +135,12 @@ impl DefId {
|
||||||
};
|
};
|
||||||
Ok(res)
|
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
|
/// 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 syntax = db.file_item(def_loc.source_item_id);
|
||||||
let struct_def =
|
let struct_def =
|
||||||
ast::StructDef::cast(syntax.borrowed()).expect("struct def should point to StructDef node");
|
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>> {
|
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 syntax = db.file_item(def_loc.source_item_id);
|
||||||
let enum_def =
|
let enum_def =
|
||||||
ast::EnumDef::cast(syntax.borrowed()).expect("enum def should point to EnumDef node");
|
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> {
|
pub(super) fn file_items(db: &impl HirDatabase, file_id: FileId) -> Arc<SourceFileItems> {
|
||||||
|
|
|
@ -156,6 +156,15 @@ impl Ty {
|
||||||
Ok(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(
|
pub(crate) fn new(
|
||||||
db: &impl HirDatabase,
|
db: &impl HirDatabase,
|
||||||
module: &Module,
|
module: &Module,
|
||||||
|
@ -534,7 +543,7 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
|
||||||
Ty::Unknown
|
Ty::Unknown
|
||||||
}
|
}
|
||||||
ast::Expr::StructLit(e) => {
|
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() {
|
if let Some(nfl) = e.named_field_list() {
|
||||||
for field in nfl.fields() {
|
for field in nfl.fields() {
|
||||||
if let Some(e) = field.expr() {
|
if let Some(e) = field.expr() {
|
||||||
|
|
Loading…
Reference in New Issue