sp-im 0.3.0

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

use serde::{
  de::{
    Deserialize,
    Deserializer,
    MapAccess,
    SeqAccess,
    Visitor,
  },
  ser::{
    Serialize,
    SerializeMap,
    SerializeSeq,
    Serializer,
  },
};
use core::{
  marker::PhantomData,
  ops::Deref,
};

use core::fmt;

use crate::{
  ordmap::OrdMap,
  ordset::OrdSet,
  vector::Vector,
};

struct SeqVisitor<'de, S, A>
where
  S: From<Vec<A>>,
  A: Deserialize<'de>, {
  phantom_s: PhantomData<S>,
  phantom_a: PhantomData<A>,
  phantom_lifetime: PhantomData<&'de ()>,
}

impl<'de, S, A> SeqVisitor<'de, S, A>
where
  S: From<Vec<A>>,
  A: Deserialize<'de>,
{
  pub(crate) fn new() -> SeqVisitor<'de, S, A> {
    SeqVisitor {
      phantom_s: PhantomData,
      phantom_a: PhantomData,
      phantom_lifetime: PhantomData,
    }
  }
}

impl<'de, S, A> Visitor<'de> for SeqVisitor<'de, S, A>
where
  S: From<Vec<A>>,
  A: Deserialize<'de>,
{
  type Value = S;

  fn expecting(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
    formatter.write_str("a sequence")
  }

  fn visit_seq<Access>(
    self,
    mut access: Access,
  ) -> Result<Self::Value, Access::Error>
  where
    Access: SeqAccess<'de>,
  {
    let mut v: Vec<A> = match access.size_hint() {
      None => Vec::new(),
      Some(l) => Vec::with_capacity(l),
    };
    while let Some(i) = access.next_element()? {
      v.push(i)
    }
    Ok(From::from(v))
  }
}

struct MapVisitor<'de, S, K, V>
where
  S: From<Vec<(K, V)>>,
  K: Deserialize<'de>,
  V: Deserialize<'de>, {
  phantom_s: PhantomData<S>,
  phantom_k: PhantomData<K>,
  phantom_v: PhantomData<V>,
  phantom_lifetime: PhantomData<&'de ()>,
}

impl<'de, S, K, V> MapVisitor<'de, S, K, V>
where
  S: From<Vec<(K, V)>>,
  K: Deserialize<'de>,
  V: Deserialize<'de>,
{
  pub(crate) fn new() -> MapVisitor<'de, S, K, V> {
    MapVisitor {
      phantom_s: PhantomData,
      phantom_k: PhantomData,
      phantom_v: PhantomData,
      phantom_lifetime: PhantomData,
    }
  }
}

impl<'de, S, K, V> Visitor<'de> for MapVisitor<'de, S, K, V>
where
  S: From<Vec<(K, V)>>,
  K: Deserialize<'de>,
  V: Deserialize<'de>,
{
  type Value = S;

  fn expecting(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
    formatter.write_str("a sequence")
  }

  fn visit_map<Access>(
    self,
    mut access: Access,
  ) -> Result<Self::Value, Access::Error>
  where
    Access: MapAccess<'de>,
  {
    let mut v: Vec<(K, V)> = match access.size_hint() {
      None => Vec::new(),
      Some(l) => Vec::with_capacity(l),
    };
    while let Some(i) = access.next_entry()? {
      v.push(i)
    }
    Ok(From::from(v))
  }
}

// Set

impl<'de, A: Deserialize<'de> + Ord + Clone> Deserialize<'de> for OrdSet<A> {
  fn deserialize<D>(des: D) -> Result<Self, D::Error>
  where D: Deserializer<'de> {
    des.deserialize_seq(SeqVisitor::new())
  }
}

impl<A: Ord + Clone + Serialize> Serialize for OrdSet<A> {
  fn serialize<S>(&self, ser: S) -> Result<S::Ok, S::Error>
  where S: Serializer {
    let mut s = ser.serialize_seq(Some(self.len()))?;
    for i in self.iter() {
      s.serialize_element(i.deref())?;
    }
    s.end()
  }
}

// Map

impl<'de, K: Deserialize<'de> + Ord + Clone, V: Deserialize<'de> + Clone>
  Deserialize<'de> for OrdMap<K, V>
{
  fn deserialize<D>(des: D) -> Result<Self, D::Error>
  where D: Deserializer<'de> {
    des.deserialize_map(MapVisitor::<'de, OrdMap<K, V>, K, V>::new())
  }
}

impl<K: Serialize + Ord + Clone, V: Serialize + Clone> Serialize
  for OrdMap<K, V>
{
  fn serialize<S>(&self, ser: S) -> Result<S::Ok, S::Error>
  where S: Serializer {
    let mut s = ser.serialize_map(Some(self.len()))?;
    for (k, v) in self.iter() {
      s.serialize_entry(k.deref(), v.deref())?;
    }
    s.end()
  }
}

// HashMap

// impl<'de, K, V, S> Deserialize<'de> for HashMap<K, V, S>
// where
//   K: Deserialize<'de> + Hash + Eq + Clone,
//   V: Deserialize<'de> + Clone,
//   S: BuildHasher + Default,
// {
//   fn deserialize<D>(des: D) -> Result<Self, D::Error>
//   where D: Deserializer<'de> {
//     des.deserialize_map(MapVisitor::<'de, HashMap<K, V, S>, K, V>::new())
//   }
// }

// impl<K, V, S> Serialize for HashMap<K, V, S>
// where
//   K: Serialize + Hash + Eq + Clone,
//   V: Serialize + Clone,
//   S: BuildHasher + Default,
// {
//   fn serialize<Ser>(&self, ser: Ser) -> Result<Ser::Ok, Ser::Error>
//   where Ser: Serializer {
//     let mut s = ser.serialize_map(Some(self.len()))?;
//     for (k, v) in self.iter() {
//       s.serialize_entry(k.deref(), v.deref())?;
//     }
//     s.end()
//   }
// }

// HashSet

// impl<'de, A: Deserialize<'de> + Hash + Eq + Clone, S: BuildHasher + Default>
//   Deserialize<'de> for HashSet<A, S>
// {
//   fn deserialize<D>(des: D) -> Result<Self, D::Error>
//   where D: Deserializer<'de> {
//     des.deserialize_seq(SeqVisitor::new())
//   }
// }

// impl<A: Serialize + Hash + Eq + Clone, S: BuildHasher + Default> Serialize
//   for HashSet<A, S>
// {
//   fn serialize<Ser>(&self, ser: Ser) -> Result<Ser::Ok, Ser::Error>
//   where Ser: Serializer {
//     let mut s = ser.serialize_seq(Some(self.len()))?;
//     for i in self.iter() {
//       s.serialize_element(i.deref())?;
//     }
//     s.end()
//   }
// }

// Vector

impl<'de, A: Clone + Deserialize<'de>> Deserialize<'de> for Vector<A> {
  fn deserialize<D>(des: D) -> Result<Self, D::Error>
  where D: Deserializer<'de> {
    des.deserialize_seq(SeqVisitor::<'de, Vector<A>, A>::new())
  }
}

impl<A: Clone + Serialize> Serialize for Vector<A> {
  fn serialize<S>(&self, ser: S) -> Result<S::Ok, S::Error>
  where S: Serializer {
    let mut s = ser.serialize_seq(Some(self.len()))?;
    for i in self.iter() {
      s.serialize_element(i.deref())?;
    }
    s.end()
  }
}

// Tests

#[cfg(test)]
mod test {
  use super::*;
  use serde_json::{
    from_str,
    to_string,
  };

  quickcheck! {
    fn ser_ordset(v: OrdSet<i32>) -> bool {
      v == from_str::<OrdSet<i32>>(&to_string(&v).unwrap()).unwrap()
    }

    fn ser_ordmap(v: OrdMap<i32, i32>) -> bool {
      v == from_str::<OrdMap<i32, i32>>(&to_string(&v).unwrap()).unwrap()
    }


    fn ser_vector(v: Vector<i32>) -> bool {
      v == from_str::<Vector<i32>>(&to_string(&v).unwrap()).unwrap()
    }
  }
}