diff options
Diffstat (limited to 'vendor/serde-value/src/ser.rs')
-rw-r--r-- | vendor/serde-value/src/ser.rs | 438 |
1 files changed, 438 insertions, 0 deletions
diff --git a/vendor/serde-value/src/ser.rs b/vendor/serde-value/src/ser.rs new file mode 100644 index 0000000..654fa5d --- /dev/null +++ b/vendor/serde-value/src/ser.rs @@ -0,0 +1,438 @@ +use serde::ser; +use std::collections::BTreeMap; +use std::error::Error; +use std::fmt; + +use crate::Value; + +#[derive(Debug)] +pub enum SerializerError { + Custom(String), +} + +impl fmt::Display for SerializerError { + fn fmt(&self, fmt: &mut fmt::Formatter) -> fmt::Result { + match *self { + SerializerError::Custom(ref s) => fmt.write_str(s), + } + } +} + +impl Error for SerializerError { + fn description(&self) -> &str { + "Value serializer error" + } +} + +impl ser::Error for SerializerError { + fn custom<T: fmt::Display>(msg: T) -> SerializerError { + SerializerError::Custom(msg.to_string()) + } +} + +impl ser::Serialize for Value { + fn serialize<S: ser::Serializer>(&self, s: S) -> Result<S::Ok, S::Error> { + match *self { + Value::Bool(v) => s.serialize_bool(v), + Value::U8(v) => s.serialize_u8(v), + Value::U16(v) => s.serialize_u16(v), + Value::U32(v) => s.serialize_u32(v), + Value::U64(v) => s.serialize_u64(v), + Value::I8(v) => s.serialize_i8(v), + Value::I16(v) => s.serialize_i16(v), + Value::I32(v) => s.serialize_i32(v), + Value::I64(v) => s.serialize_i64(v), + Value::F32(v) => s.serialize_f32(v), + Value::F64(v) => s.serialize_f64(v), + Value::Char(v) => s.serialize_char(v), + Value::String(ref v) => s.serialize_str(v), + Value::Unit => s.serialize_unit(), + Value::Option(None) => s.serialize_none(), + Value::Option(Some(ref v)) => s.serialize_some(v), + Value::Newtype(ref v) => s.serialize_newtype_struct("", v), + Value::Seq(ref v) => v.serialize(s), + Value::Map(ref v) => v.serialize(s), + Value::Bytes(ref v) => s.serialize_bytes(v), + } + } +} + +pub fn to_value<T: ser::Serialize>(value: T) -> Result<Value, SerializerError> { + value.serialize(Serializer) +} + +struct Serializer; + +impl ser::Serializer for Serializer { + type Ok = Value; + type Error = SerializerError; + type SerializeSeq = SerializeSeq; + type SerializeTuple = SerializeTuple; + type SerializeTupleStruct = SerializeTupleStruct; + type SerializeTupleVariant = SerializeTupleVariant; + type SerializeMap = SerializeMap; + type SerializeStruct = SerializeStruct; + type SerializeStructVariant = SerializeStructVariant; + + fn serialize_bool(self, v: bool) -> Result<Self::Ok, Self::Error> { + Ok(Value::Bool(v)) + } + + fn serialize_i8(self, v: i8) -> Result<Self::Ok, Self::Error> { + Ok(Value::I8(v)) + } + + fn serialize_i16(self, v: i16) -> Result<Self::Ok, Self::Error> { + Ok(Value::I16(v)) + } + + fn serialize_i32(self, v: i32) -> Result<Self::Ok, Self::Error> { + Ok(Value::I32(v)) + } + + fn serialize_i64(self, v: i64) -> Result<Self::Ok, Self::Error> { + Ok(Value::I64(v)) + } + + fn serialize_u8(self, v: u8) -> Result<Self::Ok, Self::Error> { + Ok(Value::U8(v)) + } + + fn serialize_u16(self, v: u16) -> Result<Self::Ok, Self::Error> { + Ok(Value::U16(v)) + } + + fn serialize_u32(self, v: u32) -> Result<Self::Ok, Self::Error> { + Ok(Value::U32(v)) + } + + fn serialize_u64(self, v: u64) -> Result<Self::Ok, Self::Error> { + Ok(Value::U64(v)) + } + + fn serialize_f32(self, v: f32) -> Result<Self::Ok, Self::Error> { + Ok(Value::F32(v)) + } + + fn serialize_f64(self, v: f64) -> Result<Self::Ok, Self::Error> { + Ok(Value::F64(v)) + } + + fn serialize_char(self, v: char) -> Result<Self::Ok, Self::Error> { + Ok(Value::Char(v)) + } + + fn serialize_str(self, v: &str) -> Result<Self::Ok, Self::Error> { + Ok(Value::String(v.to_string())) + } + + fn serialize_bytes(self, v: &[u8]) -> Result<Self::Ok, Self::Error> { + Ok(Value::Bytes(v.to_vec())) + } + + fn serialize_none(self) -> Result<Self::Ok, Self::Error> { + Ok(Value::Option(None)) + } + + fn serialize_some<T: ?Sized>( + self, + value: &T + ) -> Result<Self::Ok, Self::Error> + where + T: ser::Serialize + { + value.serialize(Serializer).map(|v| Value::Option(Some(Box::new(v)))) + } + + fn serialize_unit(self) -> Result<Self::Ok, Self::Error> { + Ok(Value::Unit) + } + + fn serialize_unit_struct( + self, + _name: &'static str + ) -> Result<Self::Ok, Self::Error> { + Ok(Value::Unit) + } + + fn serialize_unit_variant( + self, + _name: &'static str, + _variant_index: u32, + variant: &'static str, + ) -> Result<Self::Ok, Self::Error> { + Ok(Value::String(variant.to_string())) + } + + fn serialize_newtype_struct<T: ?Sized>( + self, + _name: &'static str, + value: &T + ) -> Result<Self::Ok, Self::Error> + where + T: ser::Serialize + { + value.serialize(Serializer).map(|v| Value::Newtype(Box::new(v))) + } + + fn serialize_newtype_variant<T: ?Sized>( + self, + _name: &'static str, + _variant_index: u32, + variant: &'static str, + value: &T + ) -> Result<Self::Ok, Self::Error> + where + T: ser::Serialize + { + value.serialize(Serializer).map(|v| { + let mut map = BTreeMap::new(); + map.insert(Value::String(variant.to_string()), v); + Value::Map(map) + }) + } + + fn serialize_seq( + self, + _len: Option<usize> + ) -> Result<Self::SerializeSeq, Self::Error> { + Ok(SerializeSeq(vec![])) + } + + fn serialize_tuple( + self, + _len: usize + ) -> Result<Self::SerializeTuple, Self::Error> { + Ok(SerializeTuple(vec![])) + } + + fn serialize_tuple_struct( + self, + _name: &'static str, + _len: usize + ) -> Result<Self::SerializeTupleStruct, Self::Error> { + Ok(SerializeTupleStruct(vec![])) + } + + fn serialize_tuple_variant( + self, + _name: &'static str, + _variant_index: u32, + variant: &'static str, + len: usize + ) -> Result<Self::SerializeTupleVariant, Self::Error> { + Ok(SerializeTupleVariant( + Value::String(variant.to_string()), + Vec::with_capacity(len), + )) + } + + fn serialize_map( + self, + _len: Option<usize> + ) -> Result<Self::SerializeMap, Self::Error> { + Ok(SerializeMap { map: BTreeMap::new(), key: None }) + } + + fn serialize_struct( + self, + _name: &'static str, + _len: usize + ) -> Result<Self::SerializeStruct, Self::Error> { + Ok(SerializeStruct(BTreeMap::new())) + } + + fn serialize_struct_variant( + self, + _name: &'static str, + _variant_index: u32, + variant: &'static str, + _len: usize + ) -> Result<Self::SerializeStructVariant, Self::Error> { + Ok(SerializeStructVariant( + Value::String(variant.to_string()), + BTreeMap::new(), + )) + } +} + +struct SerializeSeq(Vec<Value>); + +impl ser::SerializeSeq for SerializeSeq { + type Ok = Value; + type Error = SerializerError; + + fn serialize_element<T: ?Sized>( + &mut self, + value: &T + ) -> Result<(), Self::Error> + where + T: ser::Serialize + { + let value = value.serialize(Serializer)?; + self.0.push(value); + Ok(()) + } + + fn end(self) -> Result<Self::Ok, Self::Error> { + Ok(Value::Seq(self.0)) + } +} + +struct SerializeTuple(Vec<Value>); + +impl ser::SerializeTuple for SerializeTuple { + type Ok = Value; + type Error = SerializerError; + + fn serialize_element<T: ?Sized>( + &mut self, + value: &T + ) -> Result<(), Self::Error> + where + T: ser::Serialize + { + let value = value.serialize(Serializer)?; + self.0.push(value); + Ok(()) + } + + fn end(self) -> Result<Self::Ok, Self::Error> { + Ok(Value::Seq(self.0)) + } +} + +struct SerializeTupleStruct(Vec<Value>); + +impl ser::SerializeTupleStruct for SerializeTupleStruct { + type Ok = Value; + type Error = SerializerError; + + fn serialize_field<T: ?Sized>( + &mut self, + value: &T + ) -> Result<(), Self::Error> + where + T: ser::Serialize + { + let value = value.serialize(Serializer)?; + self.0.push(value); + Ok(()) + } + + fn end(self) -> Result<Self::Ok, Self::Error> { + Ok(Value::Seq(self.0)) + } +} + +struct SerializeTupleVariant(Value, Vec<Value>); + +impl ser::SerializeTupleVariant for SerializeTupleVariant { + type Ok = Value; + type Error = SerializerError; + + fn serialize_field<T: ?Sized>( + &mut self, + value: &T + ) -> Result<(), Self::Error> + where + T: ser::Serialize + { + let value = value.serialize(Serializer)?; + self.1.push(value); + Ok(()) + } + + fn end(self) -> Result<Self::Ok, Self::Error> { + let mut map = BTreeMap::new(); + map.insert(self.0, Value::Seq(self.1)); + Ok(Value::Map(map)) + } +} + +struct SerializeMap { + map: BTreeMap<Value, Value>, + key: Option<Value>, +} + +impl ser::SerializeMap for SerializeMap { + type Ok = Value; + type Error = SerializerError; + + fn serialize_key<T: ?Sized>(&mut self, key: &T) -> Result<(), Self::Error> + where + T: ser::Serialize + { + let key = key.serialize(Serializer)?; + self.key = Some(key); + Ok(()) + } + + fn serialize_value<T: ?Sized>( + &mut self, + value: &T + ) -> Result<(), Self::Error> + where + T: ser::Serialize + { + let value = value.serialize(Serializer)?; + self.map.insert(self.key.take().unwrap(), value); + Ok(()) + } + + fn end(self) -> Result<Self::Ok, Self::Error> { + Ok(Value::Map(self.map)) + } +} + +struct SerializeStruct(BTreeMap<Value, Value>); + +impl ser::SerializeStruct for SerializeStruct { + type Ok = Value; + type Error = SerializerError; + + fn serialize_field<T: ?Sized>( + &mut self, + key: &'static str, + value: &T + ) -> Result<(), Self::Error> + where + T: ser::Serialize + { + let key = Value::String(key.to_string()); + let value = value.serialize(Serializer)?; + self.0.insert(key, value); + Ok(()) + } + + fn end(self) -> Result<Self::Ok, Self::Error> { + Ok(Value::Map(self.0)) + } +} + +struct SerializeStructVariant(Value, BTreeMap<Value, Value>); + +impl ser::SerializeStructVariant for SerializeStructVariant { + type Ok = Value; + type Error = SerializerError; + + fn serialize_field<T: ?Sized>( + &mut self, + key: &'static str, + value: &T + ) -> Result<(), Self::Error> + where + T: ser::Serialize + { + let key = Value::String(key.to_string()); + let value = value.serialize(Serializer)?; + self.1.insert(key, value); + Ok(()) + } + + fn end(self) -> Result<Self::Ok, Self::Error> { + let mut map = BTreeMap::new(); + map.insert(self.0, Value::Map(self.1)); + Ok(Value::Map(map)) + } +} |