use std::error;
use std::fmt::{self, Display, Formatter};
use std::vec::IntoIter;
use serde::{self, Serialize, Deserialize};
use serde::bytes::Bytes;
use serde::de::{self, DeserializeSeed, SeqVisitor, Unexpected, Visitor};
use serde::de::value::ValueDeserializer;
use serde::ser::{self, SerializeSeq, SerializeTuple, SerializeTupleStruct, SerializeMap, SerializeStruct};
use {Integer, IntPriv, Utf8String, Value};
impl Serialize for Value {
fn serialize<S>(&self, s: S) -> Result<S::Ok, S::Error>
where S: ser::Serializer
{
match *self {
Value::Nil => s.serialize_unit(),
Value::Boolean(v) => s.serialize_bool(v),
Value::Integer(Integer { n }) => {
match n {
IntPriv::PosInt(n) => s.serialize_u64(n),
IntPriv::NegInt(n) => s.serialize_i64(n),
}
}
Value::F32(v) => s.serialize_f32(v),
Value::F64(v) => s.serialize_f64(v),
Value::String(ref v) => {
match v.s {
Ok(ref v) => s.serialize_str(v),
Err(ref v) => Bytes::from(&v.0[..]).serialize(s),
}
}
Value::Binary(ref v) => Bytes::from(v).serialize(s),
Value::Array(ref array) => {
let mut state = s.serialize_seq(Some(array.len()))?;
for item in array {
state.serialize_element(item)?;
}
state.end()
}
Value::Map(ref map) => {
let mut state = s.serialize_map(Some(map.len()))?;
for &(ref key, ref val) in map {
state.serialize_entry(key, val)?;
}
state.end()
}
Value::Ext(ty, ref buf) => {
let mut state = s.serialize_seq(Some(2))?;
state.serialize_element(&ty)?;
state.serialize_element(buf)?;
state.end()
}
}
}
}
impl Deserialize for Value {
#[inline]
fn deserialize<D>(de: D) -> Result<Self, D::Error>
where D: de::Deserializer
{
struct ValueVisitor;
impl serde::de::Visitor for ValueVisitor {
type Value = Value;
fn expecting(&self, fmt: &mut Formatter) -> Result<(), fmt::Error> {
"any valid MessagePack value".fmt(fmt)
}
#[inline]
fn visit_some<D>(self, de: D) -> Result<Value, D::Error>
where D: de::Deserializer
{
Deserialize::deserialize(de)
}
#[inline]
fn visit_none<E>(self) -> Result<Value, E> {
Ok(Value::Nil)
}
#[inline]
fn visit_unit<E>(self) -> Result<Value, E> {
Ok(Value::Nil)
}
#[inline]
fn visit_bool<E>(self, value: bool) -> Result<Value, E> {
Ok(Value::Boolean(value))
}
#[inline]
fn visit_u64<E>(self, value: u64) -> Result<Value, E> {
Ok(Value::from(value))
}
#[inline]
fn visit_i64<E>(self, value: i64) -> Result<Value, E> {
Ok(Value::from(value))
}
#[inline]
fn visit_f32<E>(self, value: f32) -> Result<Value, E> {
Ok(Value::F32(value))
}
#[inline]
fn visit_f64<E>(self, value: f64) -> Result<Value, E> {
Ok(Value::F64(value))
}
#[inline]
fn visit_string<E>(self, value: String) -> Result<Value, E> {
Ok(Value::String(Utf8String::from(value)))
}
#[inline]
fn visit_str<E>(self, value: &str) -> Result<Value, E>
where E: de::Error
{
self.visit_string(String::from(value))
}
#[inline]
fn visit_seq<V>(self, visitor: V) -> Result<Value, V::Error>
where V: serde::de::SeqVisitor
{
let values: Vec<Value> = try!(serde::de::impls::VecVisitor::new()
.visit_seq(visitor));
let values = values.into_iter().collect();
Ok(Value::Array(values))
}
#[inline]
fn visit_bytes<E>(self, v: &[u8]) -> Result<Self::Value, E>
where E: serde::de::Error
{
Ok(Value::Binary(v.to_owned()))
}
#[inline]
fn visit_map<V>(self, mut visitor: V) -> Result<Value, V::Error>
where V: serde::de::MapVisitor
{
let mut pairs = vec![];
loop {
let key: Option<Value> = try!(visitor.visit_key());
if let Some(key) = key {
let value: Value = try!(visitor.visit_value());
pairs.push((key, value));
} else {
break;
}
}
Ok(Value::Map(pairs))
}
}
de.deserialize(ValueVisitor)
}
}
#[derive(Debug)]
pub enum Error {
Syntax(String),
}
impl Display for Error {
fn fmt(&self, fmt: &mut Formatter) -> Result<(), fmt::Error> {
match *self {
Error::Syntax(ref err) => write!(fmt, "{}: {}", error::Error::description(self), err)
}
}
}
impl error::Error for Error {
fn description(&self) -> &str {
"error while decoding value"
}
fn cause(&self) -> Option<&error::Error> {
match *self {
Error::Syntax(..) => None,
}
}
}
impl de::Error for Error {
fn custom<T: Display>(msg: T) -> Self {
Error::Syntax(format!("{}", msg))
}
}
impl ser::Error for Error {
fn custom<T: Display>(msg: T) -> Self {
Error::Syntax(format!("{}", msg))
}
}
pub struct Deserializer;
impl de::Deserializer for Value {
type Error = Error;
#[inline]
fn deserialize<V>(self, visitor: V) -> Result<V::Value, Self::Error>
where V: Visitor
{
match self {
Value::Nil => visitor.visit_unit(),
Value::Boolean(v) => visitor.visit_bool(v),
Value::Integer(Integer { n }) => {
match n {
IntPriv::PosInt(v) => visitor.visit_u64(v),
IntPriv::NegInt(v) => visitor.visit_i64(v)
}
}
Value::F32(v) => visitor.visit_f32(v),
Value::F64(v) => visitor.visit_f64(v),
Value::String(v) => {
match v.s {
Ok(v) => visitor.visit_string(v),
Err(v) => visitor.visit_byte_buf(v.0),
}
}
Value::Binary(v) => visitor.visit_byte_buf(v),
Value::Array(v) => {
let len = v.len();
let mut de = SeqDeserializer::new(v);
let seq = visitor.visit_seq(&mut de)?;
if de.iter.len() == 0 {
Ok(seq)
} else {
Err(de::Error::invalid_length(len, &"fewer elements in array"))
}
}
Value::Map(v) => {
let len = v.len();
let mut de = MapDeserializer::new(v);
let map = visitor.visit_map(&mut de)?;
if de.iter.len() == 0 {
Ok(map)
} else {
Err(de::Error::invalid_length(len, &"fewer elements in map"))
}
}
Value::Ext(..) => {
unimplemented!();
}
}
}
#[inline]
fn deserialize_option<V>(self, visitor: V) -> Result<V::Value, Self::Error>
where V: Visitor
{
if let Value::Nil = self {
visitor.visit_none()
} else {
visitor.visit_some(self)
}
}
#[inline]
fn deserialize_enum<V>(self, _name: &str, _variants: &'static [&'static str], visitor: V) -> Result<V::Value, Self::Error>
where V: de::Visitor
{
match self {
Value::Array(v) => {
let mut iter = v.into_iter();
if !(iter.len() == 1 || iter.len() == 2) {
return Err(de::Error::invalid_value(Unexpected::Seq, &"array with one or two elements"));
}
let id = match iter.next() {
Some(id) => from_value(id)?,
None => {
return Err(de::Error::invalid_value(Unexpected::Seq, &"array with one or two elements"));
}
};
visitor.visit_enum(EnumDeserializer {
id: id,
value: iter.next(),
})
}
other => {
Err(de::Error::invalid_type(other.unexpected(), &"array, map or int"))
}
}
}
#[inline]
fn deserialize_newtype_struct<V>(self, _name: &'static str, visitor: V) -> Result<V::Value, Self::Error>
where V: de::Visitor
{
match self {
Value::Array(v) => {
let iter = v.into_iter();
if iter.len() != 1 {
return Err(de::Error::invalid_value(Unexpected::Seq, &"array with one element"));
}
visitor.visit_seq(SeqDeserializer {
iter: iter,
})
}
other => Err(de::Error::invalid_type(other.unexpected(), &"array")),
}
}
forward_to_deserialize! {
bool u8 u16 u32 u64 i8 i16 i32 i64 f32 f64 char str string unit seq
seq_fixed_size bytes byte_buf map unit_struct tuple_struct struct
struct_field tuple ignored_any
}
}
struct SeqDeserializer {
iter: IntoIter<Value>,
}
impl SeqDeserializer {
fn new(vec: Vec<Value>) -> Self {
SeqDeserializer {
iter: vec.into_iter(),
}
}
}
impl SeqVisitor for SeqDeserializer {
type Error = Error;
fn visit_seed<T>(&mut self, seed: T) -> Result<Option<T::Value>, Self::Error>
where T: de::DeserializeSeed
{
match self.iter.next() {
Some(val) => seed.deserialize(val).map(Some),
None => Ok(None),
}
}
fn size_hint(&self) -> (usize, Option<usize>) {
self.iter.size_hint()
}
}
impl de::Deserializer for SeqDeserializer {
type Error = Error;
#[inline]
fn deserialize<V>(mut self, visitor: V) -> Result<V::Value, Self::Error>
where V: de::Visitor
{
let len = self.iter.len();
if len == 0 {
visitor.visit_unit()
} else {
let ret = visitor.visit_seq(&mut self)?;
let remaining = self.iter.len();
if remaining == 0 {
Ok(ret)
} else {
Err(de::Error::invalid_length(len, &"fewer elements in array"))
}
}
}
forward_to_deserialize! {
bool u8 u16 u32 u64 i8 i16 i32 i64 f32 f64 char str string unit option
seq seq_fixed_size bytes byte_buf map unit_struct newtype_struct
tuple_struct struct struct_field tuple enum ignored_any
}
}
struct MapDeserializer {
val: Option<Value>,
iter: IntoIter<(Value, Value)>,
}
impl MapDeserializer {
fn new(map: Vec<(Value, Value)>) -> Self {
MapDeserializer {
val: None,
iter: map.into_iter(),
}
}
}
impl de::MapVisitor for MapDeserializer {
type Error = Error;
fn visit_key_seed<T>(&mut self, seed: T) -> Result<Option<T::Value>, Self::Error>
where T: DeserializeSeed
{
match self.iter.next() {
Some((key, val)) => {
self.val = Some(val);
seed.deserialize(key).map(Some)
}
None => Ok(None),
}
}
fn visit_value_seed<T>(&mut self, seed: T) -> Result<T::Value, Self::Error>
where T: DeserializeSeed
{
match self.val.take() {
Some(val) => seed.deserialize(val),
None => Err(de::Error::custom("value is missing")),
}
}
fn size_hint(&self) -> (usize, Option<usize>) {
self.iter.size_hint()
}
}
impl de::Deserializer for MapDeserializer {
type Error = Error;
#[inline]
fn deserialize<V>(self, visitor: V) -> Result<V::Value, Self::Error>
where V: de::Visitor
{
visitor.visit_map(self)
}
forward_to_deserialize! {
bool u8 u16 u32 u64 i8 i16 i32 i64 f32 f64 char str string unit option
seq seq_fixed_size bytes byte_buf map unit_struct newtype_struct
tuple_struct struct struct_field tuple enum ignored_any
}
}
struct EnumDeserializer {
id: u32,
value: Option<Value>,
}
impl de::EnumVisitor for EnumDeserializer {
type Error = Error;
type Variant = VariantDeserializer;
fn visit_variant_seed<V>(self, seed: V) -> Result<(V::Value, Self::Variant), Self::Error>
where V: de::DeserializeSeed
{
let variant = self.id.into_deserializer();
let visitor = VariantDeserializer { value: self.value };
seed.deserialize(variant).map(|v| (v, visitor))
}
}
struct VariantDeserializer {
value: Option<Value>,
}
impl de::VariantVisitor for VariantDeserializer {
type Error = Error;
fn visit_unit(self) -> Result<(), Error> {
match self.value {
Some(value) => de::Deserialize::deserialize(value),
None => Ok(()),
}
}
fn visit_newtype_seed<T>(self, seed: T) -> Result<T::Value, Error>
where T: de::DeserializeSeed,
{
match self.value {
Some(Value::Array(v)) => {
let mut iter = v.into_iter();
let val = match iter.next() {
Some(val) => seed.deserialize(val),
None => return Err(de::Error::invalid_value(Unexpected::Seq, &"array with one element")),
};
if iter.next().is_some() {
Err(de::Error::invalid_value(Unexpected::Seq, &"array with one element"))
} else {
val
}
}
Some(value) => seed.deserialize(value),
None => Err(de::Error::invalid_type(Unexpected::UnitVariant, &"newtype variant")),
}
}
fn visit_tuple<V>(self, _len: usize, visitor: V) -> Result<V::Value, Error>
where V: de::Visitor,
{
match self.value {
Some(Value::Array(v)) => {
de::Deserializer::deserialize(SeqDeserializer::new(v), visitor)
}
Some(other) => Err(de::Error::invalid_type(other.unexpected(), &"tuple variant")),
None => Err(de::Error::invalid_type(Unexpected::UnitVariant, &"tuple variant"))
}
}
fn visit_struct<V>(self, _fields: &'static [&'static str], visitor: V) -> Result<V::Value, Error>
where V: de::Visitor,
{
match self.value {
Some(Value::Array(v)) => {
de::Deserializer::deserialize(SeqDeserializer::new(v), visitor)
}
Some(Value::Map(v)) => {
de::Deserializer::deserialize(MapDeserializer::new(v), visitor)
}
Some(other) => Err(de::Error::invalid_type(other.unexpected(), &"struct variant")),
_ => Err(de::Error::invalid_type(Unexpected::UnitVariant, &"struct variant"))
}
}
}
trait ValueExt {
fn unexpected(&self) -> Unexpected;
}
impl ValueExt for Value {
fn unexpected(&self) -> Unexpected {
match *self {
Value::Nil => Unexpected::Unit,
Value::Boolean(v) => Unexpected::Bool(v),
Value::Integer(Integer { n }) => {
match n {
IntPriv::PosInt(v) => Unexpected::Unsigned(v),
IntPriv::NegInt(v) => Unexpected::Signed(v),
}
}
Value::F32(v) => Unexpected::Float(v as f64),
Value::F64(v) => Unexpected::Float(v),
Value::String(ref v) => {
match v.s {
Ok(ref v) => Unexpected::Str(v),
Err(ref v) => Unexpected::Bytes(&v.0[..]),
}
}
Value::Binary(ref v) => Unexpected::Bytes(v),
Value::Array(..) => Unexpected::Seq,
Value::Map(..) => Unexpected::Map,
Value::Ext(..) => Unexpected::Seq,
}
}
}
pub fn from_value<T>(val: Value) -> Result<T, Error>
where T: Deserialize
{
Deserialize::deserialize(val)
}
struct Serializer;
pub fn to_value<T: Serialize>(value: T) -> Result<Value, Error> {
value.serialize(Serializer)
}
impl ser::Serializer for Serializer {
type Ok = Value;
type Error = Error;
type SerializeSeq = SerializeVec;
type SerializeTuple = SerializeVec;
type SerializeTupleStruct = SerializeVec;
type SerializeTupleVariant = SerializeTupleVariant;
type SerializeMap = DefaultSerializeMap;
type SerializeStruct = SerializeVec;
type SerializeStructVariant = SerializeStructVariant;
#[inline]
fn serialize_bool(self, val: bool) -> Result<Self::Ok, Self::Error> {
Ok(Value::Boolean(val))
}
#[inline]
fn serialize_i8(self, val: i8) -> Result<Self::Ok, Self::Error> {
self.serialize_i64(val as i64)
}
#[inline]
fn serialize_i16(self, val: i16) -> Result<Self::Ok, Self::Error> {
self.serialize_i64(val as i64)
}
#[inline]
fn serialize_i32(self, val: i32) -> Result<Self::Ok, Self::Error> {
self.serialize_i64(val as i64)
}
#[inline]
fn serialize_i64(self, val: i64) -> Result<Self::Ok, Self::Error> {
Ok(Value::from(val))
}
#[inline]
fn serialize_u8(self, val: u8) -> Result<Self::Ok, Self::Error> {
self.serialize_u64(val as u64)
}
#[inline]
fn serialize_u16(self, val: u16) -> Result<Self::Ok, Self::Error> {
self.serialize_u64(val as u64)
}
#[inline]
fn serialize_u32(self, val: u32) -> Result<Self::Ok, Self::Error> {
self.serialize_u64(val as u64)
}
#[inline]
fn serialize_u64(self, val: u64) -> Result<Self::Ok, Self::Error> {
Ok(Value::from(val))
}
#[inline]
fn serialize_f32(self, val: f32) -> Result<Self::Ok, Self::Error> {
Ok(Value::F32(val))
}
#[inline]
fn serialize_f64(self, val: f64) -> Result<Self::Ok, Self::Error> {
Ok(Value::F64(val))
}
#[inline]
fn serialize_char(self, val: char) -> Result<Self::Ok, Self::Error> {
let mut buf = String::new();
buf.push(val);
self.serialize_str(&buf)
}
#[inline]
fn serialize_str(self, val: &str) -> Result<Self::Ok, Self::Error> {
Ok(Value::String(val.into()))
}
#[inline]
fn serialize_bytes(self, val: &[u8]) -> Result<Self::Ok, Self::Error> {
Ok(Value::Binary(val.into()))
}
#[inline]
fn serialize_unit(self) -> Result<Self::Ok, Self::Error> {
Ok(Value::Nil)
}
#[inline]
fn serialize_unit_struct(self, _name: &'static str) -> Result<Self::Ok, Self::Error> {
Ok(Value::Array(Vec::new()))
}
#[inline]
fn serialize_unit_variant(self, _name: &'static str, idx: usize, _variant: &'static str) -> Result<Self::Ok, Self::Error> {
let vec = vec![
Value::from(idx),
Value::Array(Vec::new())
];
Ok(Value::Array(vec))
}
#[inline]
fn serialize_newtype_struct<T: ?Sized>(self, _name: &'static str, value: &T) -> Result<Self::Ok, Self::Error>
where T: Serialize
{
Ok(Value::Array(vec![to_value(value)?]))
}
fn serialize_newtype_variant<T: ?Sized>(self, _name: &'static str, idx: usize, _variant: &'static str, value: &T) -> Result<Self::Ok, Self::Error>
where T: Serialize
{
let vec = vec![
Value::from(idx),
Value::Array(vec![to_value(value)?]),
];
Ok(Value::Array(vec))
}
#[inline]
fn serialize_none(self) -> Result<Self::Ok, Self::Error> {
self.serialize_unit()
}
#[inline]
fn serialize_some<T: ?Sized>(self, value: &T) -> Result<Self::Ok, Self::Error>
where T: Serialize
{
value.serialize(self)
}
fn serialize_seq(self, len: Option<usize>) -> Result<Self::SerializeSeq, Self::Error> {
let se = SerializeVec {
vec: Vec::with_capacity(len.unwrap_or(0))
};
Ok(se)
}
fn serialize_seq_fixed_size(self, len: usize) -> Result<Self::SerializeSeq, Error> {
self.serialize_seq(Some(len))
}
fn serialize_tuple(self, len: usize) -> Result<Self::SerializeTuple, Error> {
self.serialize_seq(Some(len))
}
fn serialize_tuple_struct(self, _name: &'static str, len: usize) -> Result<Self::SerializeTupleStruct, Error> {
self.serialize_seq(Some(len))
}
fn serialize_tuple_variant(self, _name: &'static str, idx: usize, _variant: &'static str, len: usize) -> Result<Self::SerializeTupleVariant, Error> {
let se = SerializeTupleVariant {
idx: idx,
vec: Vec::with_capacity(len),
};
Ok(se)
}
fn serialize_map(self, len: Option<usize>) -> Result<Self::SerializeMap, Error> {
let se = DefaultSerializeMap {
map: Vec::with_capacity(len.unwrap_or(0)),
next_key: None,
};
Ok(se)
}
fn serialize_struct(self, name: &'static str, len: usize) -> Result<Self::SerializeStruct, Error> {
self.serialize_tuple_struct(name, len)
}
fn serialize_struct_variant(self, _name: &'static str, idx: usize, _variant: &'static str, len: usize) -> Result<Self::SerializeStructVariant, Error> {
let se = SerializeStructVariant {
idx: idx,
vec: Vec::with_capacity(len),
};
Ok(se)
}
}
#[doc(hidden)]
pub struct SerializeVec {
vec: Vec<Value>,
}
#[doc(hidden)]
pub struct SerializeTupleVariant {
idx: usize,
vec: Vec<Value>,
}
#[doc(hidden)]
pub struct DefaultSerializeMap {
map: Vec<(Value, Value)>,
next_key: Option<Value>,
}
#[doc(hidden)]
pub struct SerializeStructVariant {
idx: usize,
vec: Vec<Value>,
}
impl SerializeSeq for SerializeVec {
type Ok = Value;
type Error = Error;
fn serialize_element<T: ?Sized>(&mut self, value: &T) -> Result<(), Error>
where T: Serialize
{
self.vec.push(to_value(&value)?);
Ok(())
}
fn end(self) -> Result<Value, Error> {
Ok(Value::Array(self.vec))
}
}
impl SerializeTuple for SerializeVec {
type Ok = Value;
type Error = Error;
fn serialize_element<T: ?Sized>(&mut self, value: &T) -> Result<(), Error>
where T: Serialize
{
ser::SerializeSeq::serialize_element(self, value)
}
fn end(self) -> Result<Value, Error> {
ser::SerializeSeq::end(self)
}
}
impl SerializeTupleStruct for SerializeVec {
type Ok = Value;
type Error = Error;
fn serialize_field<T: ?Sized>(&mut self, value: &T) -> Result<(), Error>
where T: Serialize
{
ser::SerializeSeq::serialize_element(self, value)
}
fn end(self) -> Result<Value, Error> {
ser::SerializeSeq::end(self)
}
}
impl ser::SerializeTupleVariant for SerializeTupleVariant {
type Ok = Value;
type Error = Error;
fn serialize_field<T: ?Sized>(&mut self, value: &T) -> Result<(), Error>
where T: Serialize
{
self.vec.push(to_value(&value)?);
Ok(())
}
fn end(self) -> Result<Value, Error> {
Ok(Value::Array(vec![Value::from(self.idx), Value::Array(self.vec)]))
}
}
impl ser::SerializeMap for DefaultSerializeMap {
type Ok = Value;
type Error = Error;
fn serialize_key<T: ?Sized>(&mut self, key: &T) -> Result<(), Error>
where T: Serialize
{
self.next_key = Some(to_value(key)?);
Ok(())
}
fn serialize_value<T: ?Sized>(&mut self, value: &T) -> Result<(), Error>
where T: ser::Serialize
{
let key = self.next_key.take()
.expect("`serialize_value` called before `serialize_key`");
self.map.push((key, to_value(&value)?));
Ok(())
}
fn end(self) -> Result<Value, Error> {
Ok(Value::Map(self.map))
}
}
impl SerializeStruct for SerializeVec {
type Ok = Value;
type Error = Error;
fn serialize_field<T: ?Sized>(&mut self, _key: &'static str, value: &T) -> Result<(), Error>
where T: Serialize
{
ser::SerializeSeq::serialize_element(self, value)
}
fn end(self) -> Result<Value, Error> {
ser::SerializeSeq::end(self)
}
}
impl ser::SerializeStructVariant for SerializeStructVariant {
type Ok = Value;
type Error = Error;
fn serialize_field<T: ?Sized>(&mut self, _key: &'static str, value: &T) -> Result<(), Error>
where T: Serialize
{
self.vec.push(to_value(&value)?);
Ok(())
}
fn end(self) -> Result<Value, Error> {
Ok(Value::Array(vec![Value::from(self.idx), Value::Array(self.vec)]))
}
}