2018-12-24 21:00:14 +01:00
|
|
|
use std::sync::Arc;
|
|
|
|
|
2018-12-28 19:34:58 +01:00
|
|
|
use ra_syntax::ast::{self, NameOwner, StructFlavor};
|
2018-12-24 19:07:48 +01:00
|
|
|
|
|
|
|
use crate::{
|
2019-01-08 13:27:00 +01:00
|
|
|
DefId, Name, AsName, Struct, Enum, VariantData, StructField,
|
2018-12-25 21:40:33 +01:00
|
|
|
type_ref::TypeRef,
|
2018-12-24 19:07:48 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
impl Struct {
|
|
|
|
pub(crate) fn new(def_id: DefId) -> Self {
|
|
|
|
Struct { def_id }
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
|
|
pub struct StructData {
|
2019-01-08 13:19:37 +01:00
|
|
|
pub(crate) name: Option<Name>,
|
|
|
|
pub(crate) variant_data: Arc<VariantData>,
|
2018-12-24 19:07:48 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
impl StructData {
|
2019-01-08 09:28:42 +01:00
|
|
|
pub(crate) fn new(struct_def: &ast::StructDef) -> StructData {
|
2018-12-28 19:34:58 +01:00
|
|
|
let name = struct_def.name().map(|n| n.as_name());
|
2018-12-25 21:40:33 +01:00
|
|
|
let variant_data = VariantData::new(struct_def.flavor());
|
2018-12-24 21:00:14 +01:00
|
|
|
let variant_data = Arc::new(variant_data);
|
2018-12-25 21:40:33 +01:00
|
|
|
StructData { name, variant_data }
|
2018-12-24 19:07:48 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Enum {
|
|
|
|
pub(crate) fn new(def_id: DefId) -> Self {
|
|
|
|
Enum { def_id }
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
|
|
pub struct EnumData {
|
2019-01-08 13:22:57 +01:00
|
|
|
pub(crate) name: Option<Name>,
|
|
|
|
pub(crate) variants: Vec<(Name, Arc<VariantData>)>,
|
2018-12-24 19:07:48 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
impl EnumData {
|
2019-01-08 09:28:42 +01:00
|
|
|
pub(crate) fn new(enum_def: &ast::EnumDef) -> Self {
|
2018-12-28 19:34:58 +01:00
|
|
|
let name = enum_def.name().map(|n| n.as_name());
|
2018-12-25 13:31:30 +01:00
|
|
|
let variants = if let Some(evl) = enum_def.variant_list() {
|
|
|
|
evl.variants()
|
|
|
|
.map(|v| {
|
2018-12-25 21:40:33 +01:00
|
|
|
(
|
2018-12-28 19:34:58 +01:00
|
|
|
v.name().map(|n| n.as_name()).unwrap_or_else(Name::missing),
|
2018-12-25 21:40:33 +01:00
|
|
|
Arc::new(VariantData::new(v.flavor())),
|
|
|
|
)
|
2018-12-25 13:31:30 +01:00
|
|
|
})
|
2018-12-25 21:40:33 +01:00
|
|
|
.collect()
|
2018-12-25 13:31:30 +01:00
|
|
|
} else {
|
|
|
|
Vec::new()
|
|
|
|
};
|
2018-12-25 21:40:33 +01:00
|
|
|
EnumData { name, variants }
|
2018-12-24 19:07:48 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl VariantData {
|
2019-01-08 13:32:27 +01:00
|
|
|
pub(crate) fn new(flavor: StructFlavor) -> Self {
|
2018-12-25 21:40:33 +01:00
|
|
|
match flavor {
|
2018-12-25 13:31:30 +01:00
|
|
|
StructFlavor::Tuple(fl) => {
|
|
|
|
let fields = fl
|
|
|
|
.fields()
|
|
|
|
.enumerate()
|
2018-12-25 21:40:33 +01:00
|
|
|
.map(|(i, fd)| StructField {
|
2018-12-28 19:34:58 +01:00
|
|
|
name: Name::tuple_field_name(i),
|
2018-12-25 21:40:33 +01:00
|
|
|
type_ref: TypeRef::from_ast_opt(fd.type_ref()),
|
2018-12-25 13:31:30 +01:00
|
|
|
})
|
2018-12-25 21:40:33 +01:00
|
|
|
.collect();
|
2018-12-25 13:31:30 +01:00
|
|
|
VariantData::Tuple(fields)
|
|
|
|
}
|
|
|
|
StructFlavor::Named(fl) => {
|
|
|
|
let fields = fl
|
|
|
|
.fields()
|
2018-12-25 21:40:33 +01:00
|
|
|
.map(|fd| StructField {
|
2018-12-28 19:34:58 +01:00
|
|
|
name: fd.name().map(|n| n.as_name()).unwrap_or_else(Name::missing),
|
2018-12-25 21:40:33 +01:00
|
|
|
type_ref: TypeRef::from_ast_opt(fd.type_ref()),
|
2018-12-25 13:31:30 +01:00
|
|
|
})
|
2018-12-25 21:40:33 +01:00
|
|
|
.collect();
|
2018-12-25 13:31:30 +01:00
|
|
|
VariantData::Struct(fields)
|
|
|
|
}
|
|
|
|
StructFlavor::Unit => VariantData::Unit,
|
2018-12-25 21:40:33 +01:00
|
|
|
}
|
2018-12-25 13:31:30 +01:00
|
|
|
}
|
2018-12-25 13:54:38 +01:00
|
|
|
|
2018-12-28 19:34:58 +01:00
|
|
|
pub(crate) fn get_field_type_ref(&self, field_name: &Name) -> Option<&TypeRef> {
|
2018-12-25 15:15:40 +01:00
|
|
|
self.fields()
|
|
|
|
.iter()
|
2019-01-08 13:27:00 +01:00
|
|
|
.find(|f| f.name() == field_name)
|
|
|
|
.map(|f| f.type_ref())
|
2018-12-25 13:54:38 +01:00
|
|
|
}
|
2018-12-24 19:07:48 +01:00
|
|
|
}
|