serde/serde_codegen/src/de.rs

1148 lines
33 KiB
Rust
Raw Normal View History

2015-05-18 01:14:38 -05:00
use aster;
2015-03-14 15:09:37 -05:00
use syntax::ast::{
2015-05-18 01:14:38 -05:00
self,
EnumDef,
2015-03-14 15:09:37 -05:00
Ident,
Item,
MetaItem,
2015-03-14 15:09:37 -05:00
};
2015-03-15 16:31:05 -05:00
use syntax::codemap::Span;
2015-05-18 01:14:38 -05:00
use syntax::ext::base::{Annotatable, ExtCtxt};
2015-03-14 15:09:37 -05:00
use syntax::ext::build::AstBuilder;
use syntax::parse::token::InternedString;
2015-03-14 15:09:37 -05:00
use syntax::ptr::P;
use attr;
use error::Error;
2015-03-14 15:09:37 -05:00
pub fn expand_derive_deserialize(
cx: &mut ExtCtxt,
span: Span,
2015-05-18 01:14:38 -05:00
meta_item: &MetaItem,
annotatable: &Annotatable,
2015-05-18 01:14:38 -05:00
push: &mut FnMut(Annotatable)
2015-03-14 15:09:37 -05:00
) {
let item = match *annotatable {
Annotatable::Item(ref item) => item,
2015-05-18 01:14:38 -05:00
_ => {
cx.span_err(
meta_item.span,
"`#[derive(Deserialize)]` may only be applied to structs and enums");
2015-05-18 01:14:38 -05:00
return;
}
};
2015-03-14 15:09:37 -05:00
let builder = aster::AstBuilder::new().span(span);
let generics = match item.node {
2016-02-12 23:06:47 -06:00
ast::ItemKind::Struct(_, ref generics) => generics,
ast::ItemKind::Enum(_, ref generics) => generics,
_ => {
cx.span_err(
meta_item.span,
"`#[derive(Deserialize)]` may only be applied to structs and enums");
return;
}
2015-03-14 15:09:37 -05:00
};
let impl_generics = builder.from_generics(generics.clone())
2015-03-14 15:09:37 -05:00
.add_ty_param_bound(
builder.path().global().ids(&["serde", "de", "Deserialize"]).build()
)
.build();
let ty = builder.ty().path()
.segment(item.ident).with_generics(impl_generics.clone()).build()
2015-03-14 15:09:37 -05:00
.build();
let body = match deserialize_body(cx, &builder, &item, &impl_generics, ty.clone()) {
Ok(body) => body,
Err(Error) => {
// An error occured, but it should have been reported already.
return;
}
};
2015-03-14 15:09:37 -05:00
let where_clause = &impl_generics.where_clause;
2015-03-14 15:09:37 -05:00
let impl_item = quote_item!(cx,
impl $impl_generics ::serde::de::Deserialize for $ty $where_clause {
fn deserialize<__D>(deserializer: &mut __D) -> ::std::result::Result<$ty, __D::Error>
where __D: ::serde::de::Deserializer,
{
$body
}
}
).unwrap();
2015-05-18 01:14:38 -05:00
push(Annotatable::Item(impl_item))
}
fn deserialize_body(
cx: &ExtCtxt,
builder: &aster::AstBuilder,
item: &Item,
impl_generics: &ast::Generics,
ty: P<ast::Ty>,
) -> Result<P<ast::Expr>, Error> {
let container_attrs = try!(attr::ContainerAttrs::from_item(cx, item));
match item.node {
2016-02-12 23:06:47 -06:00
ast::ItemKind::Struct(ref variant_data, _) => {
deserialize_item_struct(
2015-03-14 15:09:37 -05:00
cx,
builder,
item,
impl_generics,
ty,
item.span,
variant_data,
&container_attrs,
2015-03-14 15:09:37 -05:00
)
}
2016-02-12 23:06:47 -06:00
ast::ItemKind::Enum(ref enum_def, _) => {
deserialize_item_enum(
2015-03-14 15:09:37 -05:00
cx,
builder,
item.ident,
impl_generics,
ty,
2015-03-14 15:09:37 -05:00
enum_def,
&container_attrs,
2015-03-14 15:09:37 -05:00
)
}
_ => {
cx.span_bug(item.span,
"expected ItemStruct or ItemEnum in #[derive(Deserialize)]")
}
2015-03-14 15:09:37 -05:00
}
}
fn deserialize_item_struct(
cx: &ExtCtxt,
builder: &aster::AstBuilder,
item: &Item,
impl_generics: &ast::Generics,
ty: P<ast::Ty>,
span: Span,
variant_data: &ast::VariantData,
container_attrs: &attr::ContainerAttrs,
) -> Result<P<ast::Expr>, Error> {
match *variant_data {
ast::VariantData::Unit(_) => {
deserialize_unit_struct(
cx,
item.ident,
container_attrs,
)
}
ast::VariantData::Tuple(ref fields, _) if fields.len() == 1 => {
deserialize_newtype_struct(
cx,
&builder,
item.ident,
impl_generics,
ty,
container_attrs,
)
}
ast::VariantData::Tuple(ref fields, _) => {
2016-04-10 21:54:54 -05:00
if fields.iter().any(|field| field.ident.is_some()) {
cx.span_bug(span, "tuple struct has named fields")
}
deserialize_tuple_struct(
cx,
&builder,
item.ident,
impl_generics,
ty,
fields.len(),
container_attrs,
)
}
ast::VariantData::Struct(ref fields, _) => {
2016-04-10 21:54:54 -05:00
if fields.iter().any(|field| field.ident.is_none()) {
cx.span_bug(span, "struct has unnamed fields")
}
deserialize_struct(
cx,
&builder,
item.ident,
impl_generics,
ty,
fields,
container_attrs,
)
}
}
}
2015-03-14 15:09:37 -05:00
// Build `__Visitor<A, B, ...>(PhantomData<A>, PhantomData<B>, ...)`
fn deserialize_visitor(
builder: &aster::AstBuilder,
trait_generics: &ast::Generics,
forward_ty_params: Vec<ast::TyParam>,
forward_tys: Vec<P<ast::Ty>>
) -> Result<(P<ast::Item>, P<ast::Ty>, P<ast::Expr>, ast::Generics), Error> {
if trait_generics.ty_params.is_empty() && forward_tys.is_empty() {
Ok((
2015-05-07 16:47:46 -05:00
builder.item().tuple_struct("__Visitor").build(),
builder.ty().id("__Visitor"),
2015-03-14 15:09:37 -05:00
builder.expr().id("__Visitor"),
trait_generics.clone(),
))
2015-03-14 15:09:37 -05:00
} else {
let placeholders : Vec<_> = trait_generics.ty_params.iter()
2015-05-14 16:29:14 -05:00
.map(|t| builder.ty().id(t.ident))
.collect();
let mut trait_generics = trait_generics.clone();
let mut ty_params = forward_ty_params.clone();
ty_params.extend(trait_generics.ty_params.into_vec());
trait_generics.ty_params = P::from_vec(ty_params);
Ok((
2015-03-14 15:09:37 -05:00
builder.item().tuple_struct("__Visitor")
.generics().with(trait_generics.clone()).build()
.with_tys({
let lifetimes = trait_generics.lifetimes.iter()
.map(|lifetime_def| {
builder.ty()
.phantom_data()
.ref_().lifetime(lifetime_def.lifetime.name)
.ty()
.unit()
});
let ty_params = trait_generics.ty_params.iter()
.map(|ty_param| {
builder.ty()
.phantom_data()
.id(ty_param.ident)
});
lifetimes.chain(ty_params)
})
2015-03-14 15:09:37 -05:00
.build(),
builder.ty().path()
.segment("__Visitor").with_generics(trait_generics.clone()).build()
.build(),
builder.expr().call()
.path().segment("__Visitor")
.with_tys(forward_tys)
.with_tys(placeholders)
.build().build()
.with_args({
let len = trait_generics.lifetimes.len() + trait_generics.ty_params.len();
(0 .. len).map(|_| builder.expr().phantom_data())
})
2015-03-14 15:09:37 -05:00
.build(),
trait_generics,
))
2015-03-14 15:09:37 -05:00
}
}
fn deserializer_ty_param(builder: &aster::AstBuilder) -> ast::TyParam {
builder.ty_param("__D")
.trait_bound(builder.path()
2015-05-01 16:59:49 -05:00
.global()
.segment("serde").build()
.segment("de").build()
.id("Deserializer")
.build())
.build()
.build()
}
fn deserializer_ty_arg(builder: &aster::AstBuilder) -> P<ast::Ty>{
builder.ty().id("__D")
}
2015-03-14 17:20:00 -05:00
fn deserialize_unit_struct(
2015-03-14 15:09:37 -05:00
cx: &ExtCtxt,
type_ident: Ident,
container_attrs: &attr::ContainerAttrs,
) -> Result<P<ast::Expr>, Error> {
let type_name = container_attrs.name().deserialize_name_expr();
2015-03-14 15:09:37 -05:00
Ok(quote_expr!(cx, {
2015-03-14 15:09:37 -05:00
struct __Visitor;
impl ::serde::de::Visitor for __Visitor {
type Value = $type_ident;
#[inline]
fn visit_unit<E>(&mut self) -> ::std::result::Result<$type_ident, E>
2015-03-14 15:09:37 -05:00
where E: ::serde::de::Error,
{
Ok($type_ident)
2015-03-14 15:09:37 -05:00
}
#[inline]
fn visit_seq<V>(&mut self, mut visitor: V) -> ::std::result::Result<$type_ident, V::Error>
2015-03-14 15:09:37 -05:00
where V: ::serde::de::SeqVisitor,
{
try!(visitor.end());
self.visit_unit()
}
}
deserializer.deserialize_unit_struct($type_name, __Visitor)
}))
2015-03-14 15:09:37 -05:00
}
fn deserialize_newtype_struct(
cx: &ExtCtxt,
builder: &aster::AstBuilder,
type_ident: Ident,
impl_generics: &ast::Generics,
ty: P<ast::Ty>,
container_attrs: &attr::ContainerAttrs,
) -> Result<P<ast::Expr>, Error> {
let where_clause = &impl_generics.where_clause;
let (visitor_item, visitor_ty, visitor_expr, visitor_generics) = try!(deserialize_visitor(
builder,
impl_generics,
vec![deserializer_ty_param(builder)],
vec![deserializer_ty_arg(builder)],
));
let visit_seq_expr = deserialize_seq(
cx,
builder,
builder.path().id(type_ident).build(),
1,
);
let type_name = container_attrs.name().deserialize_name_expr();
Ok(quote_expr!(cx, {
$visitor_item
impl $visitor_generics ::serde::de::Visitor for $visitor_ty $where_clause {
type Value = $ty;
#[inline]
fn visit_newtype_struct<D>(&mut self, deserializer: &mut D) -> ::std::result::Result<Self::Value, D::Error>
where D: ::serde::de::Deserializer,
{
let value = try!(::serde::de::Deserialize::deserialize(deserializer));
Ok($type_ident(value))
}
#[inline]
fn visit_seq<__V>(&mut self, mut visitor: __V) -> ::std::result::Result<$ty, __V::Error>
where __V: ::serde::de::SeqVisitor,
{
$visit_seq_expr
}
}
deserializer.deserialize_newtype_struct($type_name, $visitor_expr)
}))
}
2015-03-14 17:20:00 -05:00
fn deserialize_tuple_struct(
2015-03-14 15:09:37 -05:00
cx: &ExtCtxt,
builder: &aster::AstBuilder,
type_ident: Ident,
impl_generics: &ast::Generics,
ty: P<ast::Ty>,
fields: usize,
container_attrs: &attr::ContainerAttrs,
) -> Result<P<ast::Expr>, Error> {
let where_clause = &impl_generics.where_clause;
2015-03-14 15:09:37 -05:00
let (visitor_item, visitor_ty, visitor_expr, visitor_generics) = try!(deserialize_visitor(
builder,
impl_generics,
vec![deserializer_ty_param(builder)],
vec![deserializer_ty_arg(builder)],
));
2015-03-14 17:20:00 -05:00
let visit_seq_expr = deserialize_seq(
2015-03-14 15:09:37 -05:00
cx,
builder,
builder.path().id(type_ident).build(),
fields,
2015-03-14 15:09:37 -05:00
);
let type_name = container_attrs.name().deserialize_name_expr();
2015-03-14 15:09:37 -05:00
Ok(quote_expr!(cx, {
2015-03-14 15:09:37 -05:00
$visitor_item
impl $visitor_generics ::serde::de::Visitor for $visitor_ty $where_clause {
type Value = $ty;
2015-03-14 15:09:37 -05:00
2015-07-30 11:31:16 -05:00
#[inline]
fn visit_seq<__V>(&mut self, mut visitor: __V) -> ::std::result::Result<$ty, __V::Error>
2015-03-14 15:09:37 -05:00
where __V: ::serde::de::SeqVisitor,
{
$visit_seq_expr
}
}
deserializer.deserialize_tuple_struct($type_name, $fields, $visitor_expr)
}))
2015-03-14 15:09:37 -05:00
}
2015-03-14 17:20:00 -05:00
fn deserialize_seq(
2015-03-14 15:09:37 -05:00
cx: &ExtCtxt,
builder: &aster::AstBuilder,
struct_path: ast::Path,
fields: usize,
2015-03-14 15:09:37 -05:00
) -> P<ast::Expr> {
let let_values: Vec<ast::Stmt> = (0 .. fields)
.map(|i| {
let name = builder.id(format!("__field{}", i));
2015-03-14 15:09:37 -05:00
quote_stmt!(cx,
let $name = match try!(visitor.visit()) {
2015-05-14 16:28:44 -05:00
Some(value) => { value },
2015-03-14 15:09:37 -05:00
None => {
return Err(::serde::de::Error::end_of_stream());
2015-03-14 15:09:37 -05:00
}
};
2015-03-19 09:35:07 -05:00
).unwrap()
2015-03-14 15:09:37 -05:00
})
.collect();
let result = builder.expr().call()
.build_path(struct_path)
.with_args((0 .. fields).map(|i| builder.expr().id(format!("__field{}", i))))
2015-03-14 15:09:37 -05:00
.build();
quote_expr!(cx, {
$let_values
try!(visitor.end());
Ok($result)
})
}
fn deserialize_struct_as_seq(
cx: &ExtCtxt,
builder: &aster::AstBuilder,
struct_path: ast::Path,
2016-03-07 01:27:12 -06:00
fields: &[(&ast::StructField, attr::FieldAttrs)],
) -> Result<P<ast::Expr>, Error> {
2016-03-07 01:27:12 -06:00
let let_values: Vec<_> = fields.iter()
.enumerate()
.map(|(i, &(_, ref attrs))| {
let name = builder.id(format!("__field{}", i));
2016-03-07 01:27:12 -06:00
if attrs.skip_deserializing_field() {
let default = builder.expr().default();
quote_stmt!(cx,
let $name = $default;
).unwrap()
} else {
quote_stmt!(cx,
let $name = match try!(visitor.visit()) {
Some(value) => { value },
None => {
return Err(::serde::de::Error::end_of_stream());
}
};
).unwrap()
}
})
.collect();
let result = builder.expr().struct_path(struct_path)
.with_id_exprs(
fields.iter()
.enumerate()
2016-03-07 01:27:12 -06:00
.map(|(i, &(field, _))| {
(
2016-04-10 21:54:54 -05:00
match field.ident {
Some(name) => name.clone(),
None => {
cx.span_bug(field.span, "struct contains unnamed fields")
}
},
builder.expr().id(format!("__field{}", i)),
)
})
)
.build();
Ok(quote_expr!(cx, {
$let_values
try!(visitor.end());
Ok($result)
}))
}
fn deserialize_struct(
2015-03-14 15:09:37 -05:00
cx: &ExtCtxt,
builder: &aster::AstBuilder,
type_ident: Ident,
impl_generics: &ast::Generics,
ty: P<ast::Ty>,
fields: &[ast::StructField],
container_attrs: &attr::ContainerAttrs,
) -> Result<P<ast::Expr>, Error> {
let where_clause = &impl_generics.where_clause;
let (visitor_item, visitor_ty, visitor_expr, visitor_generics) = try!(deserialize_visitor(
builder,
&impl_generics,
vec![deserializer_ty_param(builder)],
vec![deserializer_ty_arg(builder)],
));
2015-03-14 15:09:37 -05:00
let type_path = builder.path().id(type_ident).build();
2016-03-07 01:27:12 -06:00
let fields_with_attrs = try!(fields_with_attrs(cx, impl_generics, &ty, fields, false));
let visit_seq_expr = try!(deserialize_struct_as_seq(
cx,
builder,
type_path.clone(),
2016-03-07 01:27:12 -06:00
&fields_with_attrs,
));
let (field_visitor, fields_stmt, visit_map_expr) = try!(deserialize_struct_visitor(
2015-03-14 15:09:37 -05:00
cx,
builder,
type_path.clone(),
2016-03-07 01:27:12 -06:00
&fields_with_attrs,
container_attrs,
));
2015-03-14 15:09:37 -05:00
let type_name = container_attrs.name().deserialize_name_expr();
2015-03-14 15:09:37 -05:00
Ok(quote_expr!(cx, {
2015-03-14 23:47:23 -05:00
$field_visitor
2015-03-14 15:09:37 -05:00
$visitor_item
impl $visitor_generics ::serde::de::Visitor for $visitor_ty $where_clause {
type Value = $ty;
2015-03-14 15:09:37 -05:00
#[inline]
fn visit_seq<__V>(&mut self, mut visitor: __V) -> ::std::result::Result<$ty, __V::Error>
where __V: ::serde::de::SeqVisitor,
{
$visit_seq_expr
}
2015-03-14 15:09:37 -05:00
#[inline]
fn visit_map<__V>(&mut self, mut visitor: __V) -> ::std::result::Result<$ty, __V::Error>
2015-03-14 15:09:37 -05:00
where __V: ::serde::de::MapVisitor,
{
$visit_map_expr
}
}
$fields_stmt
deserializer.deserialize_struct($type_name, FIELDS, $visitor_expr)
}))
2015-03-14 15:09:37 -05:00
}
fn deserialize_item_enum(
2015-03-14 15:09:37 -05:00
cx: &ExtCtxt,
builder: &aster::AstBuilder,
type_ident: Ident,
impl_generics: &ast::Generics,
ty: P<ast::Ty>,
2015-03-14 15:09:37 -05:00
enum_def: &EnumDef,
container_attrs: &attr::ContainerAttrs
) -> Result<P<ast::Expr>, Error> {
let where_clause = &impl_generics.where_clause;
2015-03-14 15:09:37 -05:00
let type_name = container_attrs.name().deserialize_name_expr();
2015-03-14 15:09:37 -05:00
let variant_visitor = deserialize_field_visitor(
cx,
builder,
2016-02-08 10:03:46 -06:00
try!(
enum_def.variants.iter()
.map(|variant| {
let attrs = try!(attr::VariantAttrs::from_variant(cx, variant));
Ok(attrs.name().deserialize_name())
2016-02-08 10:03:46 -06:00
})
.collect()
),
container_attrs,
true,
);
2016-02-12 23:06:47 -06:00
let variants_expr = builder.expr().ref_().slice()
.with_exprs(
enum_def.variants.iter()
.map(|variant| {
builder.expr().str(variant.node.name)
})
)
.build();
let variants_stmt = quote_stmt!(cx,
const VARIANTS: &'static [&'static str] = $variants_expr;
).unwrap();
2016-04-10 21:54:54 -05:00
let ignored_arm = if container_attrs.deny_unknown_fields() {
None
2016-04-10 21:54:54 -05:00
} else {
Some(quote_arm!(cx, __Field::__ignore => { Err(::serde::de::Error::end_of_stream()) }))
};
2015-03-14 15:09:37 -05:00
// Match arms to extract a variant from a string
let mut variant_arms = vec![];
for (i, variant) in enum_def.variants.iter().enumerate() {
2016-02-18 11:27:41 -06:00
let variant_name = builder.pat().path()
.id("__Field").id(format!("__field{}", i))
.build();
let expr = try!(deserialize_variant(
cx,
builder,
type_ident,
impl_generics,
ty.clone(),
variant,
container_attrs,
));
let arm = quote_arm!(cx, $variant_name => { $expr });
variant_arms.push(arm);
}
variant_arms.extend(ignored_arm.into_iter());
let (visitor_item, visitor_ty, visitor_expr, visitor_generics) = try!(deserialize_visitor(
builder,
impl_generics,
vec![deserializer_ty_param(builder)],
vec![deserializer_ty_arg(builder)],
));
Ok(quote_expr!(cx, {
$variant_visitor
2015-03-14 17:20:00 -05:00
$visitor_item
2015-03-14 15:09:37 -05:00
impl $visitor_generics ::serde::de::EnumVisitor for $visitor_ty $where_clause {
type Value = $ty;
2015-03-14 15:09:37 -05:00
fn visit<__V>(&mut self, mut visitor: __V) -> ::std::result::Result<$ty, __V::Error>
where __V: ::serde::de::VariantVisitor,
2015-03-14 15:09:37 -05:00
{
match try!(visitor.visit_variant()) {
2015-03-14 15:09:37 -05:00
$variant_arms
}
}
}
$variants_stmt
deserializer.deserialize_enum($type_name, VARIANTS, $visitor_expr)
}))
2015-03-14 15:09:37 -05:00
}
fn deserialize_variant(
2015-03-14 15:09:37 -05:00
cx: &ExtCtxt,
builder: &aster::AstBuilder,
type_ident: Ident,
generics: &ast::Generics,
ty: P<ast::Ty>,
variant: &ast::Variant,
container_attrs: &attr::ContainerAttrs,
) -> Result<P<ast::Expr>, Error> {
let variant_ident = variant.node.name;
match variant.node.data {
ast::VariantData::Unit(_) => {
Ok(quote_expr!(cx, {
try!(visitor.visit_unit());
Ok($type_ident::$variant_ident)
}))
2015-03-14 15:09:37 -05:00
}
ast::VariantData::Tuple(ref args, _) if args.len() == 1 => {
Ok(quote_expr!(cx, {
2015-07-29 17:41:37 -05:00
let val = try!(visitor.visit_newtype());
Ok($type_ident::$variant_ident(val))
}))
}
ast::VariantData::Tuple(ref fields, _) => {
deserialize_tuple_variant(
2015-03-14 15:09:37 -05:00
cx,
builder,
type_ident,
variant_ident,
generics,
ty,
fields.len(),
)
2015-03-14 15:09:37 -05:00
}
ast::VariantData::Struct(ref fields, _) => {
deserialize_struct_variant(
2015-03-14 15:09:37 -05:00
cx,
builder,
type_ident,
variant_ident,
generics,
ty,
fields,
container_attrs,
)
2015-03-14 15:09:37 -05:00
}
}
}
fn deserialize_tuple_variant(
2015-03-14 15:09:37 -05:00
cx: &ExtCtxt,
builder: &aster::AstBuilder,
type_ident: ast::Ident,
variant_ident: ast::Ident,
generics: &ast::Generics,
ty: P<ast::Ty>,
fields: usize,
) -> Result<P<ast::Expr>, Error> {
let where_clause = &generics.where_clause;
2015-03-14 15:09:37 -05:00
let (visitor_item, visitor_ty, visitor_expr, visitor_generics) = try!(deserialize_visitor(
builder,
generics,
vec![deserializer_ty_param(builder)],
vec![deserializer_ty_arg(builder)],
));
2015-03-14 15:09:37 -05:00
2015-03-14 17:20:00 -05:00
let visit_seq_expr = deserialize_seq(
cx,
2015-03-14 15:09:37 -05:00
builder,
builder.path().id(type_ident).id(variant_ident).build(),
fields,
2015-03-14 15:09:37 -05:00
);
Ok(quote_expr!(cx, {
2015-03-14 17:20:00 -05:00
$visitor_item
2015-03-14 15:09:37 -05:00
impl $visitor_generics ::serde::de::Visitor for $visitor_ty $where_clause {
type Value = $ty;
2015-03-14 15:09:37 -05:00
fn visit_seq<__V>(&mut self, mut visitor: __V) -> ::std::result::Result<$ty, __V::Error>
where __V: ::serde::de::SeqVisitor,
{
2015-03-14 15:09:37 -05:00
$visit_seq_expr
}
}
visitor.visit_tuple($fields, $visitor_expr)
}))
2015-03-14 15:09:37 -05:00
}
fn deserialize_struct_variant(
2015-03-14 15:09:37 -05:00
cx: &ExtCtxt,
builder: &aster::AstBuilder,
type_ident: ast::Ident,
variant_ident: ast::Ident,
generics: &ast::Generics,
ty: P<ast::Ty>,
fields: &[ast::StructField],
container_attrs: &attr::ContainerAttrs,
) -> Result<P<ast::Expr>, Error> {
let where_clause = &generics.where_clause;
2015-03-14 15:09:37 -05:00
let type_path = builder.path()
.id(type_ident)
.id(variant_ident)
.build();
2016-03-07 01:27:12 -06:00
let fields_with_attrs = try!(fields_with_attrs(cx, generics, &ty, fields, true));
let visit_seq_expr = try!(deserialize_struct_as_seq(
cx,
builder,
type_path.clone(),
2016-03-07 01:27:12 -06:00
&fields_with_attrs,
));
let (field_visitor, fields_stmt, field_expr) = try!(deserialize_struct_visitor(
2015-03-14 15:09:37 -05:00
cx,
builder,
type_path,
2016-03-07 01:27:12 -06:00
&fields_with_attrs,
container_attrs,
));
2015-03-14 15:09:37 -05:00
let (visitor_item, visitor_ty, visitor_expr, visitor_generics) = try!(deserialize_visitor(
builder,
generics,
vec![deserializer_ty_param(builder)],
vec![deserializer_ty_arg(builder)],
));
Ok(quote_expr!(cx, {
2015-03-14 23:47:23 -05:00
$field_visitor
2015-03-14 15:09:37 -05:00
2015-03-14 17:20:00 -05:00
$visitor_item
impl $visitor_generics ::serde::de::Visitor for $visitor_ty $where_clause {
type Value = $ty;
2015-03-14 15:09:37 -05:00
#[inline]
fn visit_seq<__V>(&mut self, mut visitor: __V) -> ::std::result::Result<$ty, __V::Error>
where __V: ::serde::de::SeqVisitor,
{
$visit_seq_expr
}
2015-07-21 23:40:39 -05:00
#[inline]
fn visit_map<__V>(&mut self, mut visitor: __V) -> ::std::result::Result<$ty, __V::Error>
where __V: ::serde::de::MapVisitor,
{
$field_expr
2015-03-14 15:09:37 -05:00
}
}
$fields_stmt
visitor.visit_struct(FIELDS, $visitor_expr)
}))
2015-03-14 15:09:37 -05:00
}
2015-03-15 17:31:31 -05:00
fn deserialize_field_visitor(
cx: &ExtCtxt,
builder: &aster::AstBuilder,
field_names: Vec<InternedString>,
container_attrs: &attr::ContainerAttrs,
is_variant: bool,
2015-03-15 17:31:31 -05:00
) -> Vec<P<ast::Item>> {
// Create the field names for the fields.
let field_idents: Vec<_> = (0 .. field_names.len())
.map(|i| builder.id(format!("__field{}", i)))
.collect();
2016-04-10 21:54:54 -05:00
let ignore_variant = if container_attrs.deny_unknown_fields() {
None
} else {
let skip_ident = builder.id("__ignore");
Some(builder.variant(skip_ident).unit())
};
2015-03-15 17:31:31 -05:00
let field_enum = builder.item()
.attr().allow(&["non_camel_case_types"])
.enum_("__Field")
.with_variants(
field_idents.iter().map(|field_ident| {
builder.variant(field_ident).unit()
2015-03-15 17:31:31 -05:00
})
)
.with_variants(ignore_variant.into_iter())
2015-03-15 17:31:31 -05:00
.build();
let index_field_arms: Vec<_> = field_idents.iter()
.enumerate()
.map(|(field_index, field_ident)| {
quote_arm!(cx, $field_index => { Ok(__Field::$field_ident) })
})
.collect();
let (index_error_msg, unknown_ident) = if is_variant {
(builder.expr().str("expected a variant"), builder.id("unknown_variant"))
} else {
(builder.expr().str("expected a field"), builder.id("unknown_field"))
};
let fallthrough_index_arm_expr = if !is_variant && !container_attrs.deny_unknown_fields() {
quote_expr!(cx, Ok(__Field::__ignore))
} else {
quote_expr!(cx, {
Err(::serde::de::Error::invalid_value($index_error_msg))
})
};
let index_body = quote_expr!(cx,
match value {
$index_field_arms
_ => $fallthrough_index_arm_expr
}
);
// Convert the field names into byte strings.
let str_field_names: Vec<_> = field_names.iter()
.map(|name| builder.expr().lit().str(&name))
.collect();
2015-03-15 17:31:31 -05:00
// Match arms to extract a field from a string
let str_field_arms: Vec<_> = field_idents.iter().zip(str_field_names.iter())
2016-02-08 10:03:46 -06:00
.map(|(field_ident, field_name)| {
quote_arm!(cx, $field_name => { Ok(__Field::$field_ident) })
2015-03-15 17:31:31 -05:00
})
.collect();
let fallthrough_str_arm_expr = if !is_variant && !container_attrs.deny_unknown_fields() {
quote_expr!(cx, Ok(__Field::__ignore))
} else {
quote_expr!(cx, Err(::serde::de::Error::$unknown_ident(value)))
};
let str_body = quote_expr!(cx,
match value {
$str_field_arms
_ => $fallthrough_str_arm_expr
}
);
// Convert the field names into byte strings.
let bytes_field_names: Vec<_> = field_names.iter()
.map(|name| {
let name: &str = name;
builder.expr().lit().byte_str(name)
})
.collect();
// Match arms to extract a field from a string
let bytes_field_arms: Vec<_> = field_idents.iter().zip(bytes_field_names.iter())
.map(|(field_ident, field_name)| {
quote_arm!(cx, $field_name => { Ok(__Field::$field_ident) })
})
.collect();
let fallthrough_bytes_arm_expr = if !is_variant && !container_attrs.deny_unknown_fields() {
quote_expr!(cx, Ok(__Field::__ignore))
} else {
quote_expr!(cx, {
let value = ::std::string::String::from_utf8_lossy(value);
Err(::serde::de::Error::$unknown_ident(&value))
})
};
let bytes_body = quote_expr!(cx,
match value {
$bytes_field_arms
_ => $fallthrough_bytes_arm_expr
}
);
2015-05-18 01:14:38 -05:00
let impl_item = quote_item!(cx,
impl ::serde::de::Deserialize for __Field {
#[inline]
fn deserialize<D>(deserializer: &mut D) -> ::std::result::Result<__Field, D::Error>
where D: ::serde::de::Deserializer,
{
use std::marker::PhantomData;
struct __FieldVisitor<D> {
phantom: PhantomData<D>
}
2015-03-15 17:31:31 -05:00
2015-05-18 01:14:38 -05:00
impl<__D> ::serde::de::Visitor for __FieldVisitor<__D>
where __D: ::serde::de::Deserializer
2015-03-15 17:31:31 -05:00
{
2015-05-18 01:14:38 -05:00
type Value = __Field;
fn visit_usize<E>(&mut self, value: usize) -> ::std::result::Result<__Field, E>
where E: ::serde::de::Error,
{
$index_body
}
2015-05-18 01:14:38 -05:00
fn visit_str<E>(&mut self, value: &str) -> ::std::result::Result<__Field, E>
where E: ::serde::de::Error,
{
$str_body
}
2015-03-15 17:31:31 -05:00
2015-05-18 01:14:38 -05:00
fn visit_bytes<E>(&mut self, value: &[u8]) -> ::std::result::Result<__Field, E>
where E: ::serde::de::Error,
{
$bytes_body
2015-03-15 17:31:31 -05:00
}
}
2015-05-18 01:14:38 -05:00
deserializer.deserialize_struct_field(__FieldVisitor::<D>{ phantom: PhantomData })
2015-03-15 17:31:31 -05:00
}
2015-05-18 01:14:38 -05:00
}
).unwrap();
vec![field_enum, impl_item]
2015-03-15 17:31:31 -05:00
}
fn deserialize_struct_visitor(
cx: &ExtCtxt,
builder: &aster::AstBuilder,
struct_path: ast::Path,
2016-03-07 01:27:12 -06:00
fields: &[(&ast::StructField, attr::FieldAttrs)],
container_attrs: &attr::ContainerAttrs,
) -> Result<(Vec<P<ast::Item>>, ast::Stmt, P<ast::Expr>), Error> {
let field_exprs = fields.iter()
2016-03-07 01:27:12 -06:00
.map(|&(_, ref attrs)| attrs.name().deserialize_name())
.collect();
2015-03-15 17:31:31 -05:00
let field_visitor = deserialize_field_visitor(
cx,
builder,
2016-03-07 01:27:12 -06:00
field_exprs,
container_attrs,
false,
2015-03-15 17:31:31 -05:00
);
let visit_map_expr = try!(deserialize_map(
2015-03-15 17:31:31 -05:00
cx,
builder,
struct_path,
fields,
container_attrs,
));
2015-03-15 17:31:31 -05:00
2016-02-12 23:06:47 -06:00
let fields_expr = builder.expr().ref_().slice()
.with_exprs(
fields.iter()
2016-03-07 01:27:12 -06:00
.map(|&(field, _)| {
2016-04-10 21:54:54 -05:00
match field.ident {
Some(name) => builder.expr().str(name),
None => {
cx.span_bug(field.span, "struct contains unnamed fields")
}
}
})
)
.build();
let fields_stmt = quote_stmt!(cx,
const FIELDS: &'static [&'static str] = $fields_expr;
).unwrap();
Ok((field_visitor, fields_stmt, visit_map_expr))
2015-03-15 17:31:31 -05:00
}
2015-03-15 20:38:52 -05:00
fn deserialize_map(
cx: &ExtCtxt,
builder: &aster::AstBuilder,
struct_path: ast::Path,
2016-03-07 01:27:12 -06:00
fields: &[(&ast::StructField, attr::FieldAttrs)],
container_attrs: &attr::ContainerAttrs,
) -> Result<P<ast::Expr>, Error> {
2015-03-15 20:38:52 -05:00
// Create the field names for the fields.
2016-03-07 01:27:12 -06:00
let fields_attrs_names = fields.iter()
.enumerate()
.map(|(i, &(ref field, ref attrs))|
(field, attrs, builder.id(format!("__field{}", i))))
.collect::<Vec<_>>();
2016-03-07 01:27:12 -06:00
// Declare each field that will be deserialized.
let let_values: Vec<ast::Stmt> = fields_attrs_names.iter()
.filter(|&&(_, ref attrs, _)| !attrs.skip_deserializing_field())
.map(|&(_, _, name)| quote_stmt!(cx, let mut $name = None;).unwrap())
2015-03-15 20:38:52 -05:00
.collect();
// Match arms to extract a value for a field.
2016-03-07 01:27:12 -06:00
let value_arms = fields_attrs_names.iter()
.filter(|&&(_, ref attrs, _)| !attrs.skip_deserializing_field())
.map(|&(_, ref attrs, name)| {
let expr = match attrs.deserialize_with() {
Some(expr) => expr.clone(),
None => quote_expr!(cx, visitor.visit_value()),
};
2015-03-15 20:38:52 -05:00
quote_arm!(cx,
2016-03-07 01:27:12 -06:00
__Field::$name => {
$name = Some(try!($expr));
}
)
})
.collect::<Vec<_>>();
// Match arms to ignore value for fields that have `skip_deserializing`.
// Ignored even if `deny_unknown_fields` is set.
let skipped_arms = fields_attrs_names.iter()
.filter(|&&(_, ref attrs, _)| attrs.skip_deserializing_field())
.map(|&(_, _, name)| {
quote_arm!(cx,
__Field::$name => {
try!(visitor.visit_value::<::serde::de::impls::IgnoredAny>());
2015-03-15 20:38:52 -05:00
}
)
})
.collect::<Vec<_>>();
2015-03-15 20:38:52 -05:00
2016-03-07 01:27:12 -06:00
// Visit ignored values to consume them
let ignored_arm = if !container_attrs.deny_unknown_fields() {
Some(quote_arm!(cx,
_ => { try!(visitor.visit_value::<::serde::de::impls::IgnoredAny>()); }
))
} else {
None
};
let extract_values = fields_attrs_names.iter()
.filter(|&&(_, ref attrs, _)| !attrs.skip_deserializing_field())
.map(|&(_, ref attrs, name)| {
let missing_expr = attrs.expr_is_missing();
2015-03-15 20:38:52 -05:00
Ok(quote_stmt!(cx,
2016-03-07 01:27:12 -06:00
let $name = match $name {
Some($name) => $name,
2015-05-14 16:28:44 -05:00
None => $missing_expr
2015-03-15 20:38:52 -05:00
};
).unwrap())
2015-03-15 20:38:52 -05:00
})
.collect::<Result<Vec<_>, _>>();
let extract_values = try!(extract_values);
2015-03-15 20:38:52 -05:00
let result = builder.expr().struct_path(struct_path)
.with_id_exprs(
2016-03-07 01:27:12 -06:00
fields_attrs_names.iter()
.map(|&(field, attrs, name)| {
2015-03-15 20:38:52 -05:00
(
2016-04-10 21:54:54 -05:00
match field.ident {
Some(name) => name.clone(),
None => {
cx.span_bug(field.span, "struct contains unnamed fields")
}
2015-03-15 20:38:52 -05:00
},
2016-03-07 01:27:12 -06:00
if attrs.skip_deserializing_field() {
builder.expr().default()
} else {
builder.expr().id(name)
}
2015-03-15 20:38:52 -05:00
)
})
)
.build();
Ok(quote_expr!(cx, {
2015-03-15 20:38:52 -05:00
$let_values
while let Some(key) = try!(visitor.visit_key()) {
match key {
$value_arms
2016-03-07 01:27:12 -06:00
$skipped_arms
$ignored_arm
2015-03-15 20:38:52 -05:00
}
}
$extract_values
try!(visitor.end());
2015-03-15 20:38:52 -05:00
Ok($result)
}))
2015-03-15 20:38:52 -05:00
}
2016-03-07 01:27:12 -06:00
fn fields_with_attrs<'a>(
cx: &ExtCtxt,
generics: &ast::Generics,
ty: &P<ast::Ty>,
fields: &'a [ast::StructField],
is_enum: bool
) -> Result<Vec<(&'a ast::StructField, attr::FieldAttrs)>, Error> {
fields.iter()
.map(|field| {
let attrs = try!(attr::FieldAttrs::from_field(cx, &ty, generics, field, is_enum));
Ok((field, attrs))
})
.collect()
}