Finish move of StructField for pattern type inference

This commit is contained in:
Marcus Klaas de Vries 2019-01-18 15:38:11 +01:00 committed by Aleksey Kladov
parent 4277f420aa
commit bcbfa2cc11
3 changed files with 46 additions and 31 deletions

View File

@ -158,7 +158,7 @@ impl Module {
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct StructField {
struct_: Struct,
parent: DefId,
name: Name,
}
@ -166,8 +166,9 @@ impl StructField {
pub fn name(&self) -> &Name {
&self.name
}
pub fn ty(&self, db: &impl HirDatabase) -> Option<Ty> {
db.type_for_field(self.struct_.def_id, self.name.clone())
db.type_for_field(self.parent, self.name.clone())
}
}
@ -191,7 +192,7 @@ impl Struct {
.fields()
.iter()
.map(|it| StructField {
struct_: self.clone(),
parent: self.def_id,
name: it.name.clone(),
})
.collect()
@ -255,6 +256,17 @@ impl EnumVariant {
db.enum_variant_data(self.def_id).variant_data.clone()
}
pub fn fields(&self, db: &impl HirDatabase) -> Vec<StructField> {
self.variant_data(db)
.fields()
.iter()
.map(|it| StructField {
parent: self.def_id,
name: it.name.clone(),
})
.collect()
}
pub fn source(&self, db: &impl HirDatabase) -> (HirFileId, TreeArc<ast::EnumVariant>) {
def_id_to_ast(db, self.def_id)
}

View File

@ -872,22 +872,19 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
}
}
// TODO: add fields method for tuple like structs and variants
// TODO: and add tests!
fn resolve_fields(&self, path: Option<&Path>) -> Option<(Ty, Vec<StructField>)> {
let def_id = self.module.resolve_path(self.db, path?).take_types()?;
let def = def_id.resolve(self.db);
match def {
Def::Struct(s) => {
let fields: Vec<_> = s.fields(self.db);
let fields = s.fields(self.db);
Some((type_for_struct(self.db, s), fields))
}
// Def::EnumVariant(ev) => {
// let fields: Vec<_> = ev.variant_data(self.db).fields().to_owned();
// Some((type_for_enum_variant(self.db, ev), fields))
// }
Def::EnumVariant(ev) => {
let fields = ev.fields(self.db);
Some((type_for_enum_variant(self.db, ev), fields))
}
_ => None,
}
}

View File

@ -43,11 +43,13 @@ SOURCE_FILE@[0; 119)
FIELD_PAT_LIST@[40; 56)
L_CURLY@[40; 41)
WHITESPACE@[41; 42)
FIELD_PAT@[42; 43)
BIND_PAT@[42; 43)
NAME@[42; 43)
IDENT@[42; 43) "f"
COMMA@[43; 44)
WHITESPACE@[44; 45)
FIELD_PAT@[45; 54)
BIND_PAT@[45; 54)
REF_KW@[45; 48)
WHITESPACE@[48; 49)
@ -77,6 +79,8 @@ SOURCE_FILE@[0; 119)
FIELD_PAT_LIST@[73; 84)
L_CURLY@[73; 74)
WHITESPACE@[74; 75)
FIELD_PAT@[75; 79)
NAME@[75; 76)
IDENT@[75; 76) "h"
COLON@[76; 77)
WHITESPACE@[77; 78)
@ -106,6 +110,8 @@ SOURCE_FILE@[0; 119)
FIELD_PAT_LIST@[101; 110)
L_CURLY@[101; 102)
WHITESPACE@[102; 103)
FIELD_PAT@[103; 107)
NAME@[103; 104)
IDENT@[103; 104) "h"
COLON@[104; 105)
WHITESPACE@[105; 106)