diff options
author | Sam Scott <sam.scott89@gmail.com> | 2017-03-08 18:14:48 -0500 |
---|---|---|
committer | Sam Scott <sam.scott89@gmail.com> | 2017-03-09 11:44:42 -0500 |
commit | 92ba37c0679cb79b6ef686866e5b4c0cc7575878 (patch) | |
tree | 55e887e7f70fb00a525715edf1a7cfbd09e6bcea /src | |
parent | c74ae78b8469a68460f4d77b74b14dcaa75df64e (diff) |
Refactoring aproach.
Diffstat (limited to 'src')
-rw-r--r-- | src/de.rs | 424 | ||||
-rw-r--r-- | src/lib.rs | 5 | ||||
-rw-r--r-- | src/ser/mod.rs | 45 | ||||
-rw-r--r-- | src/ser/pair.rs | 263 |
4 files changed, 320 insertions, 417 deletions
@@ -2,9 +2,8 @@ use serde::de; -use std::collections::{ - HashMap, -}; +use fnv::FnvHashMap; + use std::borrow::Cow; #[doc(inline)] @@ -13,40 +12,61 @@ use serde::de::value::MapDeserializer; use std::io::Read; // use url::form_urlencoded::Parse as UrlEncodedParse; use url::form_urlencoded::parse; +use url::percent_encoding; -/// Deserializes a `application/x-wwww-url-encoded` value from a `&[u8]`. +/// Deserializes a query-string from a `&[u8]`. /// /// ``` -/// let meal = vec![ -/// ("bread".to_owned(), "baguette".to_owned()), -/// ("cheese".to_owned(), "comté".to_owned()), -/// ("fat".to_owned(), "butter".to_owned()), -/// ("meat".to_owned(), "ham".to_owned()), -/// ]; -/// -/// let mut res = serde_urlencoded::from_bytes::<Vec<(String, String)>>( -/// b"bread=baguette&cheese=comt%C3%A9&meat=ham&fat=butter").unwrap(); -/// res.sort(); -/// assert_eq!(res, meal); +/// # #[macro_use] +/// # extern crate serde_derive; +/// # extern crate serde_qs; +/// #[derive(Debug, Deserialize, PartialEq, Serialize)] +/// struct Query { +/// name: String, +/// age: u8, +/// occupation: String, +/// } +/// +/// # fn main(){ +/// let q = Query { +/// name: "Alice".to_owned(), +/// age: 24, +/// occupation: "Student".to_owned(), +/// }; +/// +/// assert_eq!( +/// serde_qs::from_bytes::<Query>("name=Alice&age=24&occupation=Student".as_bytes()), +/// Ok(q)); +/// # } /// ``` pub fn from_bytes<T: de::Deserialize>(input: &[u8]) -> Result<T, Error> { T::deserialize(Deserializer::new(input)) } -/// Deserializes a `application/x-wwww-url-encoded` value from a `&str`. +/// Deserializes a query-string from a `&str`. /// /// ``` -/// let meal = vec![ -/// ("bread".to_owned(), "baguette".to_owned()), -/// ("cheese".to_owned(), "comté".to_owned()), -/// ("fat".to_owned(), "butter".to_owned()), -/// ("meat".to_owned(), "ham".to_owned()), -/// ]; +/// # #[macro_use] +/// # extern crate serde_derive; +/// # extern crate serde_qs; +/// #[derive(Debug, Deserialize, PartialEq, Serialize)] +/// struct Query { +/// name: String, +/// age: u8, +/// occupation: String, +/// } /// -/// let mut res = serde_urlencoded::from_str::<Vec<(String, String)>>( -/// "bread=baguette&cheese=comt%C3%A9&meat=ham&fat=butter").unwrap(); -/// res.sort(); -/// assert_eq!(res, meal); +/// # fn main(){ +/// let q = Query { +/// name: "Alice".to_owned(), +/// age: 24, +/// occupation: "Student".to_owned(), +/// }; +/// +/// assert_eq!( +/// serde_qs::from_str::<Query>("name=Alice&age=24&occupation=Student"), +/// Ok(q)); +/// # } /// ``` pub fn from_str<T: de::Deserialize>(input: &str) -> Result<T, Error> { from_bytes(input.as_bytes()) @@ -65,20 +85,11 @@ pub fn from_reader<T, R>(mut reader: R) -> Result<T, Error> from_bytes(&buf) } -/// A deserializer for the `application/x-www-form-urlencoded` format. -/// -/// * Supported top-level outputs are structs, maps and sequences of pairs, -/// with or without a given length. +/// A deserializer for the query-string format. /// -/// * Main `deserialize` methods defers to `deserialize_map`. -/// -/// * Everything else but `deserialize_seq` and `deserialize_seq_fixed_size` -/// defers to `deserialize`. +/// Supported top-level outputs are structs and maps. pub struct Deserializer<'a> { - // value: &'a [u8], - // map: HashMap<Cow<'a, str>, Level<'a>>, - // parser: Option<UrlEncodedParse<'a>>, - iter: iter::Peekable<iter::Fuse<IntoIter<Cow<'a, str>, Level<'a>>>>, + iter: iter::Peekable<iter::Fuse<IntoIter<String, Level<'a>>>>, } @@ -88,135 +99,271 @@ use std::collections::hash_map::{Entry, IntoIter}; #[derive(Debug)] enum Level<'a> { - Nested(HashMap<Cow<'a, str>, Level<'a>>), - Sequence(Vec<Cow<'a, str>>), + Nested(FnvHashMap<String, Level<'a>>), + Sequence(Vec<Level<'a>>), Flat(Cow<'a, str>), Invalid(&'static str), } -impl<'a> Deserializer<'a> { +macro_rules! tu { + ($x:expr) => ( + match $x { + Some(x) => x, + // None => return Err(de::Error::custom("query string ended before expected")) + None => panic!("None found here"), + } + ) +} - // Call this with a map, with key k, and rest should the rest of the key. - // I.e. a[b][c]=v would be called as parse(map, "a", "b][c]", v) - fn parse(map: &mut HashMap<Cow<'a, str>, Level<'a>>, k: Cow<'a, str>, rest: Cow<'a, str>, v: Cow<'a, str>) { - if rest.is_empty() { - match map.entry(k) { - Entry::Occupied(mut o) => { - o.insert(Level::Invalid("Multiple values for one key")); +use std::str; +use std::iter::Iterator; + +struct Parser<I: Iterator<Item=u8>> { + inner: I, + acc: Vec<u8>, + peeked: Option<u8>, + array_limit: u8, +} + +impl<I: Iterator<Item=u8>> Iterator for Parser<I> +{ + type Item = u8; + fn next(&mut self) -> Option<Self::Item> { + self.inner.next() + } +} + +impl<I: Iterator<Item=u8>> Parser<I> { + fn new(iter: I) -> Self { + Parser { + inner: iter, + acc: Vec::new(), + peeked: None, + array_limit: 20, + } + } + + fn peek(&mut self) -> Option<<Self as Iterator>::Item> { + if !self.acc.is_empty() { + self.peeked + } else { + if let Some(x) = self.inner.next() { + self.acc.push(x); + self.peeked = Some(x); + Some(x) + } else { + None + } + } + } + + fn parse_string_key(&mut self, end_on: u8, consume: bool) -> Result<String, Error> { + loop { + match tu!(self.next()) { + x if x == end_on => { + let res = String::from_utf8(self.acc.split_off(0)); + self.acc.clear(); + + // Add this character back to the buffer for peek. + if !consume { + self.acc.push(x); + self.peeked = Some(x); + } + // println!("Key parsed as: {:?}", res); + return res.map_err(|_| de::Error::custom("blah")) }, - Entry::Vacant(vm) => { - vm.insert(Level::Flat(v)); + x @ b'=' => { + let res = String::from_utf8(self.acc.split_off(0)); + self.acc.clear(); + + // Add this character back to the buffer for peek. + self.acc.push(x); + self.peeked = Some(x); + // println!("Key parsed as: {:?}", res); + return res.map_err(|_| de::Error::custom("blah")) } - } - return; - } else { - // rest is not empty - // "b][c]" =? "b", "[c]" - let (next_key, next_rest) = split(rest, ']'); - if next_key.is_empty() { - // key is of the form a[] - // We assume this is at the bottom layer of nesting, otherwise we have - // ambiguity: a[][b]=1, a[][b]=2, a[][c]=3, a[][c] = 4 - // ==> [{b:1, c:3}, {b:2, c:4}] or - // ==> [{b:1, c:4}, {b:2, c:3}] ? Ordering not clear. - if next_rest != "]" { - map.insert(k, Level::Invalid("unindexed nested structs is unsupported")); - return; + x @ b']' | x @ b'[' => { + return Err(de::Error::custom(format!("unexpected character {} in query string, waiting for: {}.", x as char, end_on as char))); } + x @ 0x20 ... 0x7e => { + self.acc.push(x); + }, + _ => { + return Err(de::Error::custom("unexpected character in query string.")); + } + } + } - match map.entry(k) { - Entry::Vacant(vm) => { - let vec: Vec<Cow<'a, str>> = Vec::new(); - vm.insert(Level::Sequence(vec)); - }, - Entry::Occupied(o) => { - match o.into_mut() { - &mut Level::Sequence(ref mut inner) => { inner.push(v); }, - x => { *x = Level::Invalid("multiple types for one key"); } + } + + fn parse_int_key(&mut self, end_on: u8) -> Result<Option<u8>, Error> { + Ok(None) + } + + fn parse_map_value(&mut self, key: String, node: &mut Level) -> Result<(), Error> { + match tu!(self.peek()) { + b'=' => { + self.acc.clear(); + let vec: Vec<u8> = self.take_while(|b| *b != b'&').collect(); + self.acc.extend(vec); + let value = String::from_utf8(self.acc.split_off(0)); + // println!("Value parsed as: {:?}", value); + let value = value.map_err(|_e| de::Error::custom("blah"))?; + + // Reached the end of the key string + if let Level::Nested(ref mut map) = *node { + match map.entry(key) { + Entry::Occupied(mut o) => { + o.insert(Level::Invalid("Multiple values for one key")); + }, + Entry::Vacant(vm) => { + vm.insert(Level::Flat(value.into())); } } - }; - return; - } else { - // assert_eq!(&rest.as_ref()[0..1], "["); - // println!("{:?}", next_rest); - let (e, next_rest) = split(next_rest, '['); - assert_eq!(e, ""); - match map.entry(k).or_insert(Level::Nested(HashMap::new())) { - &mut Level::Nested(ref mut m) => Deserializer::parse(m, next_key, next_rest, v), - x => { *x = Level::Invalid(""); return; } - + } else { + panic!(""); + } + Ok(()) + }, + _ => { + if let Level::Nested(ref mut map) = *node { + self.parse( + map.entry(key).or_insert(Level::Nested(FnvHashMap::default())) + )?; + Ok(()) + } else { + Ok(()) } - return; } } } - /// Returns a new `Deserializer`. - pub fn new(input: &'a [u8]) -> Self { - let mut map = HashMap::<Cow<str>, Level<'a>>::new(); - let parser = parse(input).into_iter(); - - for (k, v) in parser { - let (ldepth, rdepth) = k.chars().fold((0, 0), |(acc0, acc1), x| { - match x { - '[' => (acc0+1, acc1), - ']' => (acc0, acc1+1), - _ => (acc0, acc1) + fn parse_seq_value(&mut self, node: &mut Level) -> Result<(), Error> { + match tu!(self.peek()) { + b'=' => { + self.acc.clear(); + // let value = str::from_utf8(input.take_while(|b| *b != &b'&').collect()); + // self.acc.extend_from_slice(&self.take_while(|b| *b != &b'&').collect()); + let end = self.position(|b| b == b'&'); + let value = match end { + Some(idx) => { + for b in self.inner.by_ref().take(idx) { + self.acc.push(b); + } + String::from_utf8(self.acc.split_off(0)).map(|s| s.into()) + // Ok("") + }, + None => Ok("".into()), + }; + let value = value.map_err(|e| de::Error::custom(e.to_string()))?; + // Reached the end of the key string + if let Level::Sequence(ref mut seq) = *node { + seq.push(Level::Flat(value)); + } else { + panic!(""); } - }); - debug_assert!(ldepth == rdepth); + Ok(()) + }, + _ => { + Err(de::Error::custom("non-indexed sequence of structs not supported")) + } + } + } - // Split keystring into the `root` key and the `rest`. - // a[b][c]/// => "a", "b][c]..." - let (root, rest) = split(k, '['); - Deserializer::parse(&mut map, root, rest, v); } + // Call this with a map, with key k, and rest should the rest of the key. + // I.e. a[b][c]=v would be called as parse(map, "a", "b][c]", v) + fn parse(&mut self, node: &mut Level) -> Result<bool, Error> { + // First character determines parsing type + + // loop { + match self.peek() { + Some(x) => match x { + b'a' ... b'z' | b'A' ... b'Z' => { + let key = self.parse_string_key(b'[', false).unwrap(); + self.parse_map_value(key.into(), node)?; + Ok(true) + }, + b'0' ... b'9' => { + let key = self.parse_int_key(b'[').unwrap(); + if let Some(key) = key { + self.parse_map_value(key.to_string().into(), node)?; + Ok(true) + } else { + self.parse_seq_value(node)?; + Ok(true) + } - // println!("{:?}", map); + }, + b'[' => { + self.acc.clear(); + // let _ = self.next(); + match tu!(self.peek()) { + b'a' ... b'z' | b'A' ... b'Z' => { + let key = self.parse_string_key(b']', true).unwrap(); + // key.into() + self.parse_map_value(key.into(), node)?; + Ok(true) + + }, + b'0' ... b'9' => { + let key = self.parse_int_key(b']').unwrap(); + if let Some(key) = key { + self.parse_map_value(key.to_string().into(), node)?; + Ok(true) + } else { + self.parse_seq_value(node)?; + Ok(true) + } + }, + _ => { + panic!(""); + } + } + }, + _ => { + panic!(""); + } + }, + None => return Ok(false) + // } - Deserializer { - iter: map.into_iter().fuse().peekable(), + // println!("Map: {:?}", node); } } - fn with_map(map: HashMap<Cow<'a, str>, Level<'a>>) -> Self { +} + +impl<'a> Deserializer<'a> { + fn with_map(map: FnvHashMap<String, Level<'a>>) -> Self { Deserializer { iter: map.into_iter().fuse().peekable(), } } -} -fn split<'a>(input: Cow<'a, str>, split: char) -> (Cow<'a, str>, Cow<'a, str>) { - match input { - Cow::Borrowed(v) => { - let mut split2 = v.splitn(2, split); - let s1 = split2.next().unwrap(); - let s2 = split2.next().unwrap_or(""); - (Cow::Borrowed(s1), Cow::Borrowed(s2)) - }, - Cow::Owned(v) => { - // let v = v.into_bytes(); - let mut split_idx = v.len(); - for (idx, c) in v.chars().enumerate() { - if c == split { - split_idx = idx; - break; - } - } - // b][c] split = ], idx = 1 - if split_idx < v.len() { - let mut v = v.into_bytes(); - let v2 = v.split_off(split_idx+1); - v.pop(); - unsafe { - return (Cow::Owned(String::from_utf8_unchecked(v)), - Cow::Owned(String::from_utf8_unchecked(v2))) - } - } else { - return (Cow::Owned(v), Cow::Owned("".to_string())) + + + /// Returns a new `Deserializer`. + pub fn new(input: &'a [u8]) -> Self { + let map = FnvHashMap::<String, Level<'a>>::default(); + let mut root = Level::Nested(map); + + let decoded = percent_encoding::percent_decode(&input); + let mut parser = Parser::new(decoded); + while let Ok(x) = parser.parse(&mut root) { + if !x { + break } - // (Cow::Owned(v),Cow::Borrowed("")) + } + // self.input = Some(decoded.as_bytes()); + // println!("{:?}", root); + let iter = match root { + Level::Nested(map) => map.into_iter().fuse().peekable(), + _ => panic!(""), + }; + Deserializer { + // input: Some(decoded.as_bytes().into()), + iter: iter, } } } @@ -251,7 +398,6 @@ impl<'a, 'b> de::Deserializer for Deserializer<'a> { where V: de::Visitor { visitor.visit_seq(MapDeserializer::new(self.iter)) - } forward_to_deserialize! { bool @@ -357,6 +503,8 @@ impl<'a> de::Deserializer for LevelDeserializer<'a> { // visitor.visit_seq(self) if let Level::Sequence(x) = self.0 { SeqDeserializer::new(x.into_iter()).deserialize(visitor) + } else if let Level::Nested(map) = self.0 { + SeqDeserializer::new(map.into_iter().map(|(_k, v)| v)).deserialize(visitor) } else { Err(de::Error::custom("value does not appear to be a sequence")) } @@ -1,13 +1,16 @@ -//! `x-www-form-urlencoded` meets Serde +//! Serde support for querystring-style strings #![warn(unused_extern_crates)] +extern crate fnv; extern crate itoa; extern crate dtoa; #[macro_use] extern crate serde; extern crate url; +#[macro_use] +extern crate serde_derive; pub mod de; diff --git a/src/ser/mod.rs b/src/ser/mod.rs index f36be4f..4b8f484 100644 --- a/src/ser/mod.rs +++ b/src/ser/mod.rs @@ -1,7 +1,6 @@ //! Serialization support for the `application/x-www-form-urlencoded` format. mod key; -mod pair; mod part; mod value; @@ -13,19 +12,34 @@ use std::str; use url::form_urlencoded::Serializer as UrlEncodedSerializer; use url::form_urlencoded::Target as UrlEncodedTarget; + + + /// Serializes a value into a `application/x-wwww-url-encoded` `String` buffer. /// /// ``` -/// let meal = &[ -/// ("bread", "baguette"), -/// ("cheese", "comté"), -/// ("meat", "ham"), -/// ("fat", "butter"), -/// ]; +/// # #[macro_use] +/// # extern crate serde_derive; +/// # extern crate serde_qs; +/// #[derive(Deserialize, Serialize)] +/// struct Query { +/// name: String, +/// age: u8, +/// occupation: String, +/// } +/// +/// # fn main(){ +/// let q = Query { +/// name: "Alice".to_owned(), +/// age: 24, +/// occupation: "Student".to_owned(), +/// }; +/// /// /// assert_eq!( -/// serde_urlencoded::to_string(meal), -/// Ok("bread=baguette&cheese=comt%C3%A9&meat=ham&fat=butter".to_owned())); +/// serde_qs::to_string(&q), +/// Ok("name=Alice&age=24&occupation=Student".to_owned())); +/// # } /// ``` pub fn to_string<T: ser::Serialize>(input: &T) -> Result<String, Error> { let mut urlencoder = UrlEncodedSerializer::new("".to_owned()); @@ -93,8 +107,8 @@ impl ser::Error for Error { } /// Sequence serializer. -pub struct SeqSerializer<'output, Target: 'output + UrlEncodedTarget> { - urlencoder: &'output mut UrlEncodedSerializer<Target>, +pub struct SeqSerializer<'output, T: 'output + UrlEncodedTarget> { + inner: ser::Impossible<&'output mut UrlEncodedSerializer<T>, Error>, } /// Tuple serializer. @@ -277,14 +291,14 @@ impl<'output, Target> ser::Serializer for Serializer<'output, Target> fn serialize_seq(self, _len: Option<usize>) -> Result<Self::SerializeSeq, Error> { - Ok(SeqSerializer { urlencoder: self.urlencoder }) + Err(Error::top_level()) } /// Serializes a sequence, given length is ignored. fn serialize_seq_fixed_size(self, _len: usize) -> Result<Self::SerializeSeq, Error> { - Ok(SeqSerializer { urlencoder: self.urlencoder }) + Err(Error::top_level()) } /// Returns an error. @@ -352,14 +366,15 @@ impl<'output, Target> ser::SerializeSeq for SeqSerializer<'output, Target> fn serialize_element<T: ?Sized + ser::Serialize>(&mut self, value: &T) -> Result<(), Error> { - value.serialize(pair::PairSerializer::new(self.urlencoder)) + self.inner.serialize_element(value) } fn end(self) -> Result<Self::Ok, Error> { - Ok(self.urlencoder) + self.inner.end() } } + impl<'output, Target> ser::SerializeTuple for TupleSerializer<'output, Target> where Target: 'output + UrlEncodedTarget, { diff --git a/src/ser/pair.rs b/src/ser/pair.rs deleted file mode 100644 index 6815647..0000000 --- a/src/ser/pair.rs +++ /dev/null @@ -1,263 +0,0 @@ -use ser::Error; -use ser::key::KeySink; -use ser::part::PartSerializer; -use ser::value::ValueSink; -use serde::ser; -use std::borrow::Cow; -use std::mem; -use url::form_urlencoded::Serializer as UrlEncodedSerializer; -use url::form_urlencoded::Target as UrlEncodedTarget; - -pub struct PairSerializer<'target, Target: 'target + UrlEncodedTarget> { - urlencoder: &'target mut UrlEncodedSerializer<Target>, - state: PairState, -} - -impl<'target, Target> PairSerializer<'target, Target> - where Target: 'target + UrlEncodedTarget, -{ - pub fn new(urlencoder: &'target mut UrlEncodedSerializer<Target>) -> Self { - PairSerializer { - urlencoder: urlencoder, - state: PairState::WaitingForKey, - } - } -} - -impl<'target, Target> ser::Serializer for PairSerializer<'target, Target> - where Target: 'target + UrlEncodedTarget, -{ - type Ok = (); - type Error = Error; - type SerializeSeq = ser::Impossible<(), Error>; - type SerializeTuple = Self; - type SerializeTupleStruct = ser::Impossible<(), Error>; - type SerializeTupleVariant = ser::Impossible<(), Error>; - type SerializeMap = ser::Impossible<(), Error>; - type SerializeStruct = ser::Impossible<(), Error>; - type SerializeStructVariant = ser::Impossible<(), Error>; - - fn serialize_bool(self, _v: bool) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_i8(self, _v: i8) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_i16(self, _v: i16) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_i32(self, _v: i32) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_i64(self, _v: i64) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_u8(self, _v: u8) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_u16(self, _v: u16) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_u32(self, _v: u32) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_u64(self, _v: u64) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_f32(self, _v: f32) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_f64(self, _v: f64) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_char(self, _v: char) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_str(self, _value: &str) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_bytes(self, _value: &[u8]) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_unit(self) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_unit_struct(self, _name: &'static str) -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_unit_variant(self, - _name: &'static str, - _variant_index: usize, - _variant: &'static str) - -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_newtype_struct<T: ?Sized + ser::Serialize> - (self, - _name: &'static str, - value: &T) - -> Result<(), Error> { - value.serialize(self) - } - - fn serialize_newtype_variant<T: ?Sized + ser::Serialize> - (self, - _name: &'static str, - _variant_index: usize, - _variant: &'static str, - _value: &T) - -> Result<(), Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_none(self) -> Result<(), Error> { - Ok(()) - } - - fn serialize_some<T: ?Sized + ser::Serialize>(self, - value: &T) - -> Result<(), Error> { - value.serialize(self) - } - - fn serialize_seq(self, - _len: Option<usize>) - -> Result<Self::SerializeSeq, Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_seq_fixed_size(self, - _len: usize) - -> Result<Self::SerializeSeq, Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_tuple(self, len: usize) -> Result<Self, Error> { - if len == 2 { - Ok(self) - } else { - Err(Error::unsupported_pair()) - } - } - - fn serialize_tuple_struct(self, - _name: &'static str, - _len: usize) - -> Result<Self::SerializeTupleStruct, Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_tuple_variant - (self, - _name: &'static str, - _variant_index: usize, - _variant: &'static str, - _len: usize) - -> Result<Self::SerializeTupleVariant, Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_map(self, - _len: Option<usize>) - -> Result<Self::SerializeMap, Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_struct(self, - _name: &'static str, - _len: usize) - -> Result<Self::SerializeStruct, Error> { - Err(Error::unsupported_pair()) - } - - fn serialize_struct_variant - (self, - _name: &'static str, - _variant_index: usize, - _variant: &'static str, - _len: usize) - -> Result<Self::SerializeStructVariant, Error> { - Err(Error::unsupported_pair()) - } -} - -impl<'target, Target> ser::SerializeTuple for PairSerializer<'target, Target> - where Target: 'target + UrlEncodedTarget, -{ - type Ok = (); - type Error = Error; - - fn serialize_element<T: ?Sized + ser::Serialize>(&mut self, - value: &T) - -> Result<(), Error> { - match mem::replace(&mut self.state, PairState::Done) { - PairState::WaitingForKey => { - let key_sink = KeySink::new(|key| Ok(key.into())); - let key_serializer = PartSerializer::new(key_sink); - self.state = PairState::WaitingForValue { - key: value.serialize(key_serializer)?, - }; - Ok(()) - }, - PairState::WaitingForValue { key } => { - let result = { - let value_sink = ValueSink::new(self.urlencoder, &key); - let value_serializer = PartSerializer::new(value_sink); - value.serialize(value_serializer) - }; - if result.is_ok() { - self.state = PairState::Done; - } else { - self.state = PairState::WaitingForValue { key: key }; - } - result - }, - PairState::Done => Err(Error::done()), - } - } - - fn end(self) -> Result<(), Error> { - if let PairState::Done = self.state { - Ok(()) - } else { - Err(Error::not_done()) - } - } -} - -enum PairState { - WaitingForKey, - WaitingForValue { key: Cow<'static, str> }, - Done, -} - -impl Error { - fn done() -> Self { - Error::Custom("this pair has already been serialized".into()) - } - - fn not_done() -> Self { - Error::Custom("this pair has not yet been serialized".into()) - } - - fn unsupported_pair() -> Self { - Error::Custom("unsupported pair".into()) - } -} |