ion-rs 1.0.0

use crate::element::builders::StructBuilder;
use crate::element::Element;
use crate::ion_data::{IonDataHash, IonDataOrd, IonEq};
use crate::symbol_ref::AsSymbolRef;
use crate::text::text_formatter::FmtValueFormatter;
use crate::Symbol;
use smallvec::SmallVec;
use std::cmp::Ordering;
use std::collections::{HashMap, VecDeque};
use std::fmt::{Display, Formatter};
use std::hash::Hasher;

// A convenient type alias for a vector capable of storing a single `usize` inline
// without heap allocation. This type should not be used in public interfaces directly.
type IndexVec = SmallVec<[usize; 1]>;

// This collection is broken out into its own type to allow instances of it to be shared with Arc/Rc.
#[derive(Debug, Clone)]
struct Fields {
    // Key/value pairs in the order they were inserted
    by_index: Vec<(Symbol, Element)>,
    // Maps symbols to a list of indexes where values may be found in `by_index` above
    by_name: HashMap<Symbol, IndexVec>,
}

impl Fields {
    /// Gets all of the indexes that contain a value associated with the given field name.
    fn get_indexes<A: AsSymbolRef>(&self, field_name: A) -> Option<&IndexVec> {
        field_name
            .as_symbol_ref()
            .text()
            .map(|text| {
                // If the symbol has defined text, look it up by &str
                self.by_name.get(text)
            })
            .unwrap_or_else(|| {
                // Otherwise, construct a (cheap, stack-allocated) Symbol with unknown text...
                let symbol = Symbol::unknown_text();
                // ...and use the unknown text symbol to look up matching field values
                self.by_name.get(&symbol)
            })
    }

    /// Iterates over the values found at the specified indexes.
    fn get_values_at_indexes<'a>(&'a self, indexes: &'a IndexVec) -> FieldValuesIterator<'a> {
        FieldValuesIterator {
            current: 0,
            indexes: Some(indexes),
            by_index: &self.by_index,
        }
    }

    /// Gets the last value in the Struct that is associated with the specified field name.
    ///
    /// Note that the Ion data model views a struct as a bag of (name, value) pairs and does not
    /// have a notion of field ordering. In most use cases, field names are distinct and the last
    /// appearance of a field in the struct's serialized form will have been the _only_ appearance.
    /// If a field name appears more than once, this method makes the arbitrary decision to return
    /// the value associated with the last appearance. If your application uses structs that repeat
    /// field names, you are encouraged to use [`get_all`](Self::get_all) instead.
    fn get_last<A: AsSymbolRef>(&self, field_name: A) -> Option<&Element> {
        self.get_indexes(field_name)
            .and_then(|indexes| indexes.last())
            .and_then(|index| self.by_index.get(*index))
            .map(|(_name, value)| value)
    }

    /// Iterates over all of the values associated with the given field name.
    fn get_all<A: AsSymbolRef>(&self, field_name: A) -> FieldValuesIterator<'_> {
        let indexes = self.get_indexes(field_name);
        FieldValuesIterator {
            current: 0,
            indexes,
            by_index: &self.by_index,
        }
    }

    /// Iterates over all of the (field name, field value) pairs in the struct.
    fn iter(&self) -> impl Iterator<Item = &(Symbol, Element)> {
        self.by_index.iter()
    }
}

/// Iterates over the (field name, field value) pairs in a Struct.
pub struct FieldIterator<'a> {
    values: Option<std::slice::Iter<'a, (Symbol, Element)>>,
}

impl<'a> FieldIterator<'a> {
    fn new(data: &'a [(Symbol, Element)]) -> Self {
        FieldIterator {
            values: Some(data.iter()),
        }
    }
}

impl<'a> Iterator for FieldIterator<'a> {
    type Item = (&'a Symbol, &'a Element);

    fn next(&mut self) -> Option<Self::Item> {
        self.values
            .as_mut()
            // Get the next &(name, value) and convert it to (&name, &value)
            .and_then(|iter| iter.next().map(|field| (&field.0, &field.1)))
    }
}

/// Iterates over the (field name, field value) pairs in a Struct.
pub struct OwnedFieldIterator {
    fields: VecDeque<(Symbol, Element)>,
}

impl OwnedFieldIterator {
    fn new(data: Vec<(Symbol, Element)>) -> Self {
        OwnedFieldIterator {
            fields: data.into(),
        }
    }
}

impl Iterator for OwnedFieldIterator {
    type Item = (Symbol, Element);

    fn next(&mut self) -> Option<Self::Item> {
        self.fields.pop_front()
    }

    fn size_hint(&self) -> (usize, Option<usize>) {
        let len = self.fields.len();
        (len, Some(len))
    }
}

/// Iterates over the values associated with a given field name in a Struct.
pub(crate) struct FieldValuesIterator<'a> {
    current: usize,
    indexes: Option<&'a IndexVec>,
    by_index: &'a Vec<(Symbol, Element)>,
}

impl<'a> Iterator for FieldValuesIterator<'a> {
    type Item = &'a Element;

    fn next(&mut self) -> Option<Self::Item> {
        self.indexes
            .and_then(|i| i.get(self.current))
            .and_then(|i| {
                self.current += 1;
                self.by_index.get(*i)
            })
            .map(|(_name, value)| value)
    }
}

/// An in-memory representation of an Ion Struct
/// ```
/// use ion_rs::{Element, ion_struct};
/// # use ion_rs::IonResult;
/// # fn main() -> IonResult<()> {
/// let struct_ = ion_struct! {
///   "foo": 1,
///   "bar": true,
///   "baz": "hello"
/// };
/// assert_eq!(struct_.len(), 3);
/// assert_eq!(struct_.get("baz"), Some(&Element::string("hello")));
/// # Ok(())
/// # }
/// ```
#[derive(Debug, Clone)]
pub struct Struct {
    fields: Fields,
}

impl Display for Struct {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        let mut ivf = FmtValueFormatter { output: f };
        ivf.format_struct(self).map_err(|_| std::fmt::Error)?;
        Ok(())
    }
}

impl Struct {
    pub fn builder() -> StructBuilder {
        StructBuilder::new()
    }

    pub fn clone_builder(&self) -> StructBuilder {
        StructBuilder::with_initial_fields(&self.fields.by_index)
    }

    /// Returns an iterator over the field name/value pairs in this Struct.
    #[allow(clippy::map_identity)]
    // ^-- This is a temporary workaround for a bug in Clippy that should be fixed in the next release.
    // See: https://github.com/rust-lang/rust-clippy/issues/9280
    pub fn fields(&self) -> impl Iterator<Item = (&Symbol, &Element)> {
        self.fields
            .iter()
            // Here we convert from &(name, value) to (&name, &value).
            // The former makes a stronger assertion about how the data is being stored. We don't
            // want that to be a mandatory part of the public API.
            .map(|(name, element)| (name, element))
    }

    fn fields_eq(&self, other: &Self, eq: impl Fn(&Element, &Element) -> bool) -> bool {
        if self.fields.by_index.len() != self.fields.by_name.len() {
            return self.fields_with_repeated_names_eq(other, eq);
        }

        for (field_name, field_value_indexes) in &self.fields.by_name {
            let other_value_indexes = match other.fields.get_indexes(field_name) {
                Some(indexes) => indexes,
                // The other struct doesn't have a field with this name so they're not equal.
                None => return false,
            };

            debug_assert_eq!(field_value_indexes.len(), 1);
            if other_value_indexes.len() != 1 {
                // The other struct has fields with the same name, but a different number of them.
                return false;
            }

            // Direct indexing is safe here because we've already guaranteed that len() == 1
            let self_value = &self.fields.by_index[field_value_indexes[0]].1;
            let other_value = &other.fields.by_index[other_value_indexes[0]].1;
            if !eq(self_value, other_value) {
                return false;
            }
        }

        true
    }

    /// Bitvector multiset matching fallback for structs with repeated field names.
    /// O(p * m^2) where p is the number of unique field names and m is the max repeat count.
    fn fields_with_repeated_names_eq(
        &self,
        other: &Self,
        eq: impl Fn(&Element, &Element) -> bool,
    ) -> bool {
        for (field_name, field_value_indexes) in &self.fields.by_name {
            let other_value_indexes = match other.fields.get_indexes(field_name) {
                Some(indexes) => indexes,
                None => return false,
            };

            if field_value_indexes.len() != other_value_indexes.len() {
                return false;
            }

            let mut matched = vec![false; other_value_indexes.len()];
            for field_value in self.fields.get_values_at_indexes(field_value_indexes) {
                let found = other
                    .fields
                    .get_values_at_indexes(other_value_indexes)
                    .enumerate()
                    .find(|(i, other_value)| !matched[*i] && eq(field_value, other_value));
                match found {
                    Some((i, _)) => matched[i] = true,
                    None => return false,
                }
            }
        }

        true
    }

    /// Returns the number of fields in this Struct.
    pub fn len(&self) -> usize {
        self.fields.by_index.len()
    }

    /// Returns `true` if this struct has zero fields.
    pub fn is_empty(&self) -> bool {
        self.len() == 0
    }

    pub fn iter(&self) -> FieldIterator<'_> {
        FieldIterator::new(&self.fields.by_index)
    }

    /// Returns the value associated with the specified field name.
    ///
    /// If more than one field in this struct has that name, this method will return the value of
    /// the _last_ field with that name. To access other fields with the same name, see
    /// [`get_all`](Self::get_all).
    pub fn get<A: AsSymbolRef>(&self, field_name: A) -> Option<&Element> {
        self.fields.get_last(field_name)
    }

    /// Returns an iterator over all of the values associated with the specified field name.
    pub fn get_all<A: AsSymbolRef>(&self, field_name: A) -> impl Iterator<Item = &Element> {
        self.fields.get_all(field_name)
    }
}

// Allows `for (name, value) in &my_struct {...}` syntax
impl<'a> IntoIterator for &'a Struct {
    type Item = (&'a Symbol, &'a Element);
    type IntoIter = FieldIterator<'a>;

    fn into_iter(self) -> Self::IntoIter {
        self.iter()
    }
}

// Allows `for (name, value) in my_struct {...}` syntax
impl IntoIterator for Struct {
    type Item = (Symbol, Element);
    type IntoIter = OwnedFieldIterator;

    fn into_iter(self) -> Self::IntoIter {
        OwnedFieldIterator::new(self.fields.by_index)
    }
}

impl<K, V> FromIterator<(K, V)> for Struct
where
    K: Into<Symbol>,
    V: Into<Element>,
{
    /// Returns an owned struct from the given iterator of field names/values.
    fn from_iter<I: IntoIterator<Item = (K, V)>>(iter: I) -> Self {
        let mut by_index: Vec<(Symbol, Element)> = Vec::new();
        let mut by_name: HashMap<Symbol, IndexVec> = HashMap::new();
        for (field_name, field_value) in iter {
            let field_name = field_name.into();
            let field_value = field_value.into();

            by_name
                .entry(field_name.clone())
                .or_default()
                .push(by_index.len());
            by_index.push((field_name, field_value));
        }

        let fields = Fields { by_index, by_name };
        Self { fields }
    }
}

impl PartialEq for Struct {
    fn eq(&self, other: &Self) -> bool {
        self.len() == other.len() && self.fields_eq(other, |a, b| a == b)
    }
}

impl IonEq for Struct {
    fn ion_eq(&self, other: &Self) -> bool {
        self.len() == other.len() && self.fields_eq(other, |a, b| a.ion_eq(b))
    }
}

impl IonDataOrd for Struct {
    fn ion_cmp(&self, other: &Self) -> Ordering {
        let mut these_fields = self.fields.by_index.iter().collect::<Vec<_>>();
        let mut those_fields = other.fields.by_index.iter().collect::<Vec<_>>();
        these_fields.sort_by(ion_cmp_field);
        those_fields.sort_by(ion_cmp_field);

        let mut i0 = these_fields.iter();
        let mut i1 = those_fields.iter();
        loop {
            match [i0.next(), i1.next()] {
                [None, Some(_)] => return Ordering::Less,
                [None, None] => return Ordering::Equal,
                [Some(_), None] => return Ordering::Greater,
                [Some(a), Some(b)] => {
                    let ord = ion_cmp_field(a, b);
                    if ord != Ordering::Equal {
                        return ord;
                    }
                }
            }
        }
    }
}

fn ion_cmp_field(this: &&(Symbol, Element), that: &&(Symbol, Element)) -> Ordering {
    let ord = this.0.ion_cmp(&that.0);
    if !ord.is_eq() {
        return ord;
    }
    IonDataOrd::ion_cmp(&this.1, &that.1)
}

impl IonDataHash for Struct {
    fn ion_data_hash<H: Hasher>(&self, state: &mut H) {
        let mut these_fields = self.fields.by_index.iter().collect::<Vec<_>>();
        these_fields.sort_by(ion_cmp_field);
        for (name, value) in these_fields {
            name.ion_data_hash(state);
            value.ion_data_hash(state);
        }
    }
}

#[cfg(test)]
mod tests {
    use crate::element::Element;
    use crate::ion_data::IonEq;
    use crate::ion_struct;

    #[test]
    fn duplicate_field_values_not_equal() {
        let s1 = ion_struct! { "a": 1, "a": 1, "a": 2 };
        let s2 = ion_struct! { "a": 1, "a": 2, "a": 2 };
        assert_ne!(s1, s2);
        assert!(!s1.ion_eq(&s2));
    }

    #[test]
    fn same_multiset_different_order_is_equal() {
        let s1 = ion_struct! { "a": 1, "a": 2, "b": 3 };
        let s2 = ion_struct! { "b": 3, "a": 2, "a": 1 };
        assert_eq!(s1, s2);
        assert!(s1.ion_eq(&s2));
    }

    #[test]
    fn nan_partial_eq_vs_ion_eq() {
        let s1 = ion_struct! { "x": f64::NAN };
        let s2 = ion_struct! { "x": f64::NAN };
        // PartialEq uses standard f64 semantics: NaN != NaN
        assert_ne!(s1, s2);
        // IonEq treats NaN as equivalent
        assert!(s1.ion_eq(&s2));
    }

    #[test]
    fn signed_zero_partial_eq_vs_ion_eq() {
        let s1 = ion_struct! { "x": 0.0f64 };
        let s2 = ion_struct! { "x": -0.0f64 };
        // PartialEq uses standard f64 semantics: 0.0 == -0.0
        assert_eq!(s1, s2);
        // IonEq distinguishes signed zeros
        assert!(!s1.ion_eq(&s2));
    }

    #[test]
    fn for_field_in_struct() {
        // Simple example to exercise Struct's implementation of IntoIterator
        let s = ion_struct! { "foo": 1, "bar": 2, "baz": 3};
        let _fields = s.clone().iter().collect::<Vec<_>>(); // exercises `size_hint`
        let mut baz_value = None;
        for (name, value) in &s {
            if *name == "baz" {
                baz_value = Some(value);
            }
        }
        assert_eq!(baz_value, Some(&Element::int(3)));
    }

    #[test]
    fn for_field_in_owned_struct() {
        // Simple example to exercise Struct's implementation of IntoIterator
        let s = ion_struct! { "foo": 1, "bar": 2, "baz": 3};
        let _fields = s.clone().into_iter().collect::<Vec<_>>(); // exercises `size_hint`
        let mut baz_value = None;
        for (name, value) in s {
            if name == "baz" {
                baz_value = Some(value);
            }
        }
        assert_eq!(baz_value, Some(Element::int(3)));
    }
}