serde_utils 0.6.0

use std::collections::BTreeMap;
use std::hash::{Hash, Hasher};
use std::mem::transmute;
use std::fmt;
use std::cmp::Ordering;

use serde_bytes::ByteBuf;
use serde::{Serialize, Serializer, Deserialize, Deserializer};
use serde::de::{Visitor, SeqAccess, MapAccess, Error};

/// A generic object that can hold any value deserialized via Serde.
///
/// The important aspect of this generic object enum is that it can consume all possible values and
/// does not expect any specific types. This makes it possible to deserialize data first and
/// interpret it later.
///
/// **Warning**: Deserializing and even serializing unknown content allows attackers to control the
///              recursion depth of the process and potentially crash it (although in a safe way).
///
/// Note: The implementations of `PartialEq`, `PartialOrd`, and `Ord` traits treat `NAN` floats as
///       equal.
///       The implementations of `Hash`, `PartialEq`, `PartialOrd`, and `Ord` traits treat
///       non-negative `Signed` objects like `Unsigned` objects with the same value.
#[allow(dead_code)]
#[derive(Debug, Clone)]
pub enum Obj {
    /// `None` / `null` / `nil`
    Null,

    /// Boolean value
    Bool(bool),

    /// Signed integer
    Signed(i64),

    /// Unsigned integer
    Unsigned(u64),

    /// Floating-point value
    Float(f64),

    /// Utf-8 string
    Str(String),

    /// Byte sequence
    Bin(ByteBuf),

    /// List / tuple / sequence
    List(Vec<Obj>),

    /// Mapping / object
    Map(BTreeMap<Obj, Obj>)
}

impl Obj {
    fn type_num(&self) -> u8 {
        match *self {
            Obj::Null => 0,
            Obj::Bool(_) => 1,
            Obj::Signed(_) => 2,
            Obj::Unsigned(_) => 3,
            Obj::Float(_) => 4,
            Obj::Str(_) => 5,
            Obj::Bin(_) => 6,
            Obj::List(_) => 7,
            Obj::Map(_) => 8,
        }
    }
}

impl Default for Obj {
    #[inline]
    fn default() -> Obj {
        Obj::Null
    }
}

impl PartialEq for Obj {
    fn eq(&self, other: &Self) -> bool {
        if let Obj::Signed(val) = *self {
            if val >= 0 {
                return &Obj::Unsigned(val as u64) == other;
            }
        }
        if let Obj::Signed(val) = *other {
            if val >= 0 {
                return self == &Obj::Unsigned(val as u64);
            }
        }
        match *self {
            Obj::Null => if let Obj::Null = *other { true } else { false },
            Obj::Bool(val) => if let Obj::Bool(oval) = *other { val == oval } else { false },
            Obj::Unsigned(val) => if let Obj::Unsigned(oval) = *other { val == oval } else { false },
            Obj::Signed(val) => if let Obj::Signed(oval) = *other { val == oval } else { false },
            Obj::Float(val) => if let Obj::Float(oval) = *other {
                if val.is_nan() && oval.is_nan() { true } else { val == oval }
            } else { false },
            Obj::Str(ref val) => if let Obj::Str(ref oval) = *other { val == oval } else { false },
            Obj::Bin(ref val) => if let Obj::Bin(ref oval) = *other { val == oval } else { false },
            Obj::List(ref val) => if let Obj::List(ref oval) = *other { val == oval } else { false },
            Obj::Map(ref val) => if let Obj::Map(ref oval) = *other { val == oval } else { false },
        }
    }
}

impl Eq for Obj {}

impl PartialOrd for Obj {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}

impl Ord for Obj {
    fn cmp(&self, other: &Self) -> Ordering {
        if let Obj::Signed(val) = *self {
            if val >= 0 {
                return Obj::Unsigned(val as u64).cmp(other);
            }
        }
        if let Obj::Signed(val) = *other {
            if val >= 0 {
                return self.cmp(&Obj::Unsigned(val as u64));
            }
        }
        let stype = self.type_num();
        let otype = other.type_num();
        if stype != otype {
            return stype.cmp(&otype);
        }
        match *self {
            Obj::Null => Ordering::Equal,
            Obj::Bool(val) => if let Obj::Bool(ref oval) = *other {
                val.cmp(oval)
            } else {
                unreachable!()
            },
            Obj::Unsigned(val) => if let Obj::Unsigned(ref oval) = *other {
                val.cmp(oval)
            } else {
                unreachable!()
            },
            Obj::Signed(val) => if let Obj::Signed(ref oval) = *other {
                val.cmp(oval)
            } else {
                unreachable!()
            },
            Obj::Float(val) => if let Obj::Float(oval) = *other {
                if !val.is_nan() && !oval.is_nan() {
                    val.partial_cmp(&oval).unwrap()
                } else if val.is_nan() {
                    Ordering::Greater
                } else {
                    Ordering::Less
                }
            } else {
                unreachable!()
            },
            Obj::Str(ref val) => if let Obj::Str(ref oval) = *other {
                val.cmp(oval)
            } else {
                unreachable!()
            },
            Obj::Bin(ref val) => if let Obj::Bin(ref oval) = *other {
                val.cmp(oval)
            } else {
                unreachable!()
            },
            Obj::List(ref val) => if let Obj::List(ref oval) = *other {
                val.cmp(oval)
            } else {
                unreachable!()
            },
            Obj::Map(ref val) => if let Obj::Map(ref oval) = *other {
                val.cmp(oval)
            } else {
                unreachable!()
            },
        }
    }
}

impl Hash for Obj {
    fn hash<H>(&self, state: &mut H) where H: Hasher {
        if let Obj::Signed(val) = *self {
            if val >= 0 {
                return Obj::Unsigned(val as u64).hash(state);
            }
        }
        state.write_u8(self.type_num());
        match *self {
            Obj::Null => (),
            Obj::Bool(val) => state.write_u8(if val { 1 } else { 0 }),
            Obj::Unsigned(val) => state.write_u64(val),
            Obj::Signed(val) => state.write_i64(val),
            Obj::Float(val) => state.write_u64(unsafe { transmute(val) }),
            Obj::Str(ref val) => val.hash(state),
            Obj::Bin(ref val) => val.hash(state),
            Obj::List(ref val) => val.hash(state),
            Obj::Map(ref val) => val.hash(state),
        }
    }
}

impl Serialize for Obj {
    #[inline]
    fn serialize<S: Serializer>(&self, ser: S) -> Result<S::Ok, S::Error> {
        match *self {
            Obj::Null => ser.serialize_none(),
            Obj::Bool(val) => ser.serialize_bool(val),
            Obj::Unsigned(val) => ser.serialize_u64(val),
            Obj::Signed(val) => ser.serialize_i64(val),
            Obj::Float(val) => ser.serialize_f64(val),
            Obj::Str(ref val) => ser.serialize_str(val),
            Obj::Bin(ref val) => ser.serialize_bytes(val),
            Obj::List(ref val) => val.serialize(ser),
            Obj::Map(ref val) => val.serialize(ser)
        }
    }
}

struct GenericVisitor;

impl<'a> Visitor<'a> for GenericVisitor {
    type Value = Obj;

    #[inline]
    fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
        write!(formatter, "any value")
    }

    #[inline]
    fn visit_none<E: Error>(self) -> Result<Self::Value, E> {
        Ok(Obj::Null)
    }

    #[inline]
    fn visit_bool<E: Error>(self, val: bool) -> Result<Self::Value, E> {
        Ok(Obj::Bool(val))
    }

    #[inline]
    fn visit_u64<E: Error>(self, val: u64) -> Result<Self::Value, E> {
        Ok(Obj::Unsigned(val))
    }

    #[inline]
    fn visit_i64<E: Error>(self, val: i64) -> Result<Self::Value, E> {
        Ok(Obj::Signed(val))
    }

    #[inline]
    fn visit_f64<E: Error>(self, val: f64) -> Result<Self::Value, E> {
        Ok(Obj::Float(val))
    }

    #[inline]
    fn visit_str<E: Error>(self, val: &str) -> Result<Self::Value, E> {
        Ok(Obj::Str(val.to_owned()))
    }

    #[inline]
    fn visit_string<E: Error>(self, val: String) -> Result<Self::Value, E> {
        Ok(Obj::Str(val))
    }

    #[inline]
    fn visit_bytes<E: Error>(self, val: &[u8]) -> Result<Self::Value, E> {
        let mut bin = Vec::with_capacity(val.len());
        bin.extend(val.iter().cloned());
        Ok(Obj::Bin(ByteBuf::from(bin)))
    }

    #[inline]
    fn visit_byte_buf<E: Error>(self, val: Vec<u8>) -> Result<Self::Value, E> {
        Ok(Obj::Bin(ByteBuf::from(val)))
    }

    #[inline]
    fn visit_unit<E: Error>(self) -> Result<Self::Value, E> {
        Ok(Obj::Null)
    }

    #[inline]
    fn visit_seq<V: SeqAccess<'a>>(self, mut visitor: V) -> Result<Self::Value, V::Error> {
        let mut list = if let Some(len) = visitor.size_hint() {
            Vec::with_capacity(len)
        } else {
            Vec::new()
        };
        while let Some(value) = try!(visitor.next_element()) {
            list.push(value);
        }
        Ok(Obj::List(list))
    }

    #[inline]
    fn visit_map<V: MapAccess<'a>>(self, mut visitor: V) -> Result<Self::Value, V::Error> {
        let mut map = BTreeMap::new();
        while let Some((key, value)) = try!(visitor.next_entry()) {
            map.insert(key, value);
        }
        Ok(Obj::Map(map))
    }
}

impl<'a> Deserialize<'a> for Obj {
    #[inline]
    fn deserialize<D: Deserializer<'a>>(de: D) -> Result<Self, D::Error> {
        de.deserialize_any(GenericVisitor)
    }
}

impl fmt::Display for Obj {
    fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
        match *self {
            Obj::Null => write!(f, "null"),
            Obj::Bool(val) => write!(f, "{}", val),
            Obj::Unsigned(val) => write!(f, "{}", val),
            Obj::Signed(val) => write!(f, "{}", val),
            Obj::Float(val) => write!(f, "{}", val),
            Obj::Str(ref val) => write!(f, "{}", val),
            Obj::Bin(ref val) => write!(f, "{:?}", val),
            Obj::List(ref val) => write!(f, "{:?}", val),
            Obj::Map(ref val) => write!(f, "{:?}", val),
        }
    }
}