xsd-parser 1.5.2

use std::ops::DerefMut;

use crate::models::{
    meta::{AttributeMeta, ElementMeta, EnumerationMetaVariant},
    PropertyIdent,
};

/// Helper trait that implements additional methods for vectors.
pub trait VecHelper {
    /// Item stored in the vector.
    type Item;

    /// Try to find the object with the passed `ident` in the vector. If it was
    /// not found `None` is returned.
    fn find(&mut self, ident: PropertyIdent) -> Option<&mut Self::Item>;

    /// Try to find the object with the passed `ident` in the vector. If it was
    /// not found, a new one is created by invoking `f`.
    ///
    /// Returns a mutable reference either to the found or the newly created object.
    fn find_or_insert<F>(&mut self, ident: PropertyIdent, f: F) -> &mut Self::Item
    where
        F: FnOnce(PropertyIdent) -> Self::Item;
}

impl<X, T> VecHelper for X
where
    X: DerefMut<Target = Vec<T>>,
    T: WithIdent,
{
    type Item = T;

    fn find(&mut self, ident: PropertyIdent) -> Option<&mut Self::Item> {
        self.iter_mut().find(|x| x.ident().eq(&ident))
    }

    fn find_or_insert<F>(&mut self, ident: PropertyIdent, f: F) -> &mut Self::Item
    where
        F: FnOnce(PropertyIdent) -> Self::Item,
    {
        let vec = &mut **self;

        if let Some(i) = vec.iter().position(|x| x.ident().eq(&ident)) {
            &mut vec[i]
        } else {
            vec.push(f(ident));

            vec.last_mut().unwrap()
        }
    }
}

/// Helper trait that adds name information to the implementing object.
pub trait WithIdent {
    /// Returns the name of the object.
    fn ident(&self) -> &PropertyIdent;
}

impl WithIdent for EnumerationMetaVariant {
    fn ident(&self) -> &PropertyIdent {
        &self.ident
    }
}

impl WithIdent for ElementMeta {
    fn ident(&self) -> &PropertyIdent {
        &self.ident
    }
}

impl WithIdent for AttributeMeta {
    fn ident(&self) -> &PropertyIdent {
        &self.ident
    }
}