sea-orm-sync 2.0.0-rc.38

use crate::HasManyModel;
use core::ops::Index;
use std::hash::{Hash, Hasher};
use std::slice;

use super::super::EntityTrait;

#[derive(Debug, Default, Clone)]
pub enum HasMany<E: EntityTrait> {
    #[default]
    Unloaded,
    Loaded(Vec<E::ModelEx>),
}

impl<E> PartialEq for HasMany<E>
where
    E: EntityTrait,
    E::ModelEx: PartialEq,
{
    fn eq(&self, other: &Self) -> bool {
        match (self, other) {
            (HasMany::Unloaded, HasMany::Unloaded) => true,
            (HasMany::Loaded(a), HasMany::Loaded(b)) => a == b,
            _ => false,
        }
    }
}

impl<E> Eq for HasMany<E>
where
    E: EntityTrait,
    E::ModelEx: Eq,
{
}

impl<E: EntityTrait> HasMany<E> {
    /// Return true if variant is `Loaded`
    pub fn is_loaded(&self) -> bool {
        matches!(self, HasMany::Loaded(_))
    }

    /// Return true if variant is `Unloaded`
    pub fn is_unloaded(&self) -> bool {
        matches!(self, HasMany::Unloaded)
    }

    /// Return true if variant is `Unloaded` or underlying vector is empty
    pub fn is_empty(&self) -> bool {
        match self {
            HasMany::Unloaded => true,
            HasMany::Loaded(models) => models.is_empty(),
        }
    }

    /// Like `Vec::get`
    pub fn get(&self, index: usize) -> Option<&E::ModelEx> {
        match self {
            HasMany::Loaded(models) => models.get(index),
            HasMany::Unloaded => None,
        }
    }

    /// Length of underlying vector
    pub fn len(&self) -> usize {
        match self {
            HasMany::Loaded(models) => models.len(),
            HasMany::Unloaded => 0,
        }
    }
}

impl<E> HasMany<E>
where
    E: EntityTrait,
    E::ActiveModelEx: From<E::ModelEx>,
{
    pub fn into_active_model(self) -> HasManyModel<E> {
        match self {
            HasMany::Loaded(models) => {
                HasManyModel::Append(models.into_iter().map(Into::into).collect())
            }
            HasMany::Unloaded => HasManyModel::NotSet,
        }
    }
}

impl<E: EntityTrait> From<HasMany<E>> for Option<Vec<E::ModelEx>> {
    fn from(value: HasMany<E>) -> Self {
        match value {
            HasMany::Loaded(models) => Some(models),
            HasMany::Unloaded => None,
        }
    }
}

impl<E: EntityTrait> Index<usize> for HasMany<E> {
    type Output = E::ModelEx;

    fn index(&self, index: usize) -> &Self::Output {
        match self {
            HasMany::Unloaded => {
                panic!("index out of bounds: the HasMany is Unloaded (index: {index})")
            }
            HasMany::Loaded(items) => items.index(index),
        }
    }
}

impl<E: EntityTrait> IntoIterator for HasMany<E> {
    type Item = E::ModelEx;
    type IntoIter = std::vec::IntoIter<E::ModelEx>;

    fn into_iter(self) -> Self::IntoIter {
        match self {
            HasMany::Loaded(models) => models.into_iter(),
            HasMany::Unloaded => Vec::new().into_iter(),
        }
    }
}

impl<E: EntityTrait> From<Vec<E::ModelEx>> for HasMany<E> {
    fn from(value: Vec<E::ModelEx>) -> Self {
        HasMany::Loaded(value)
    }
}

impl<E, const N: usize> PartialEq<[<E as EntityTrait>::Model; N]> for HasMany<E>
where
    E: EntityTrait,
    E::ModelEx: PartialEq<<E as EntityTrait>::Model>,
    <E as EntityTrait>::Model: PartialEq<E::ModelEx>,
{
    fn eq(&self, other: &[<E as EntityTrait>::Model; N]) -> bool {
        match self {
            HasMany::Loaded(models) => models.as_slice() == other.as_slice(),
            HasMany::Unloaded => false,
        }
    }
}

impl<E, const N: usize> PartialEq<HasMany<E>> for [<E as EntityTrait>::Model; N]
where
    E: EntityTrait,
    E::ModelEx: PartialEq<<E as EntityTrait>::Model>,
    <E as EntityTrait>::Model: PartialEq<E::ModelEx>,
{
    fn eq(&self, other: &HasMany<E>) -> bool {
        other == self
    }
}

impl<E> PartialEq<[<E as EntityTrait>::Model]> for HasMany<E>
where
    E: EntityTrait,
    E::ModelEx: PartialEq<<E as EntityTrait>::Model>,
    <E as EntityTrait>::Model: PartialEq<E::ModelEx>,
{
    fn eq(&self, other: &[<E as EntityTrait>::Model]) -> bool {
        match self {
            HasMany::Loaded(models) => models.as_slice() == other,
            HasMany::Unloaded => false,
        }
    }
}

impl<E> PartialEq<HasMany<E>> for [<E as EntityTrait>::Model]
where
    E: EntityTrait,
    E::ModelEx: PartialEq<<E as EntityTrait>::Model>,
    <E as EntityTrait>::Model: PartialEq<E::ModelEx>,
{
    fn eq(&self, other: &HasMany<E>) -> bool {
        other == self
    }
}

#[derive(Debug, Clone)]
pub struct Iter<'a, E: EntityTrait> {
    inner: Option<slice::Iter<'a, E::ModelEx>>,
}

impl<E: EntityTrait> HasMany<E> {
    pub fn iter(&self) -> Iter<'_, E> {
        Iter {
            inner: match self {
                HasMany::Loaded(models) => Some(models.iter()),
                HasMany::Unloaded => None,
            },
        }
    }
}

impl<'a, E: EntityTrait> Iterator for Iter<'a, E> {
    type Item = &'a E::ModelEx;

    fn next(&mut self) -> Option<Self::Item> {
        self.inner.as_mut().and_then(|iter| iter.next())
    }

    fn size_hint(&self) -> (usize, Option<usize>) {
        self.inner
            .as_ref()
            .map(|iter| iter.size_hint())
            .unwrap_or((0, Some(0)))
    }

    fn count(self) -> usize
    where
        Self: Sized,
    {
        self.inner.map(|iter| iter.count()).unwrap_or(0)
    }

    fn last(self) -> Option<Self::Item>
    where
        Self: Sized,
    {
        self.inner.and_then(|iter| iter.last())
    }
}

impl<'a, E: EntityTrait> ExactSizeIterator for Iter<'a, E> {
    fn len(&self) -> usize {
        self.inner.as_ref().map(|iter| iter.len()).unwrap_or(0)
    }
}

impl<'a, E: EntityTrait> DoubleEndedIterator for Iter<'a, E> {
    fn next_back(&mut self) -> Option<Self::Item> {
        self.inner.as_mut().and_then(|iter| iter.next_back())
    }
}

impl<'a, E: EntityTrait> IntoIterator for &'a HasMany<E> {
    type Item = &'a E::ModelEx;
    type IntoIter = Iter<'a, E>;

    fn into_iter(self) -> Self::IntoIter {
        Iter {
            inner: match self {
                HasMany::Loaded(models) => Some(models.iter()),
                HasMany::Unloaded => None,
            },
        }
    }
}

impl<E> Hash for HasMany<E>
where
    E: EntityTrait,
    E::ModelEx: Hash,
{
    fn hash<H: Hasher>(&self, state: &mut H) {
        std::mem::discriminant(self).hash(state);
        match self {
            HasMany::Loaded(model) => model.hash(state),
            HasMany::Unloaded => {}
        }
    }
}

#[cfg(feature = "with-json")]
impl<E> serde::Serialize for HasMany<E>
where
    E: EntityTrait,
    E::ModelEx: serde::Serialize,
{
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        match self {
            HasMany::Unloaded => None,
            HasMany::Loaded(models) => Some(models.as_slice()),
        }
        .serialize(serializer)
    }
}

#[cfg(feature = "with-json")]
impl<'de, E> serde::Deserialize<'de> for HasMany<E>
where
    E: EntityTrait,
    E::ModelEx: serde::Deserialize<'de>,
{
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        match <Option<Vec<E::ModelEx>>>::deserialize(deserializer)? {
            Some(models) => Ok(HasMany::Loaded(models)),
            None => Ok(HasMany::Unloaded),
        }
    }
}

#[cfg(test)]
mod test {
    use crate::compound::{HasMany, HasOne};
    use crate::tests_cfg::{cake, filling, fruit};

    #[test]
    fn test_serde_compound() {
        let cake = cake::ModelEx {
            id: 1,
            name: "A".into(),
            fruit: Default::default(),
            fillings: Default::default(),
        };

        assert_eq!(
            filling::Model {
                id: 2,
                name: "C".into(),
                vendor_id: None,
                ignored_attr: 3,
            }
            .into_ex(),
            filling::ModelEx {
                id: 2,
                name: "C".into(),
                vendor_id: None,
                ignored_attr: 3,
                ingredients: HasMany::Unloaded,
            }
        );

        assert_eq!(
            serde_json::to_string(&cake).unwrap(),
            r#"{"id":1,"name":"A","fruit":null,"fillings":null}"#
        );
        assert_eq!(
            serde_json::from_str::<cake::ModelEx>(&serde_json::to_string(&cake).unwrap()).unwrap(),
            cake
        );

        let cake = cake::ModelEx {
            id: 1,
            name: "A".into(),
            fruit: Default::default(),
            fillings: HasMany::Loaded(vec![]),
        };

        assert_eq!(
            serde_json::to_string(&cake).unwrap(),
            r#"{"id":1,"name":"A","fruit":null,"fillings":[]}"#
        );
        assert_eq!(
            serde_json::from_str::<cake::ModelEx>(&serde_json::to_string(&cake).unwrap()).unwrap(),
            cake
        );

        let mut cake = cake::ModelEx {
            id: 1,
            name: "A".into(),
            fruit: HasOne::Loaded(
                fruit::ModelEx {
                    id: 2,
                    name: "B".into(),
                    cake_id: None,
                }
                .into(),
            ),
            fillings: HasMany::Unloaded,
        };

        assert_eq!(
            serde_json::to_string(&cake).unwrap(),
            r#"{"id":1,"name":"A","fruit":{"id":2,"name":"B","cake_id":null},"fillings":null}"#
        );
        // fruit has skip_deserializing on id
        cake.fruit.as_mut().unwrap().id = 0;
        assert_eq!(
            serde_json::from_str::<cake::ModelEx>(&serde_json::to_string(&cake).unwrap()).unwrap(),
            cake
        );

        let cake = cake::ModelEx {
            id: 1,
            name: "A".into(),
            fruit: HasOne::Loaded(
                fruit::ModelEx {
                    id: 0,
                    name: "B".into(),
                    cake_id: None,
                }
                .into(),
            ),
            fillings: HasMany::Loaded(vec![
                filling::Model {
                    id: 2,
                    name: "C".into(),
                    vendor_id: None,
                    ignored_attr: 3,
                }
                .into_ex(),
            ]),
        };

        assert_eq!(
            serde_json::to_string(&cake).unwrap(),
            r#"{"id":1,"name":"A","fruit":{"id":0,"name":"B","cake_id":null},"fillings":[{"id":2,"name":"C","vendor_id":null,"ignored_attr":3,"ingredients":null}]}"#
        );
        assert_eq!(
            serde_json::from_str::<cake::ModelEx>(&serde_json::to_string(&cake).unwrap()).unwrap(),
            cake
        );
    }
}