jay-compositor 1.15.0

use crate::utils::cell_ext::CellExt;
use crate::utils::clonecell::CloneCell;
use crate::utils::markers::JayClone;
use std::cell::Cell;
use std::rc::Rc;

pub trait ObjWithId {
    type Id: Copy;

    fn id(&self) -> Self::Id;
}

pub trait ObjWithIdOptExt {
    type Id: Copy;

    fn id_or_default(&self) -> Self::Id;
}

impl<T> ObjWithId for Option<T>
where
    T: ObjWithId,
{
    type Id = Option<T::Id>;

    fn id(&self) -> Self::Id {
        self.as_ref().map(ObjWithId::id)
    }
}

impl<T> ObjWithIdOptExt for Option<T>
where
    T: ObjWithId,
    T::Id: Default,
{
    type Id = T::Id;

    fn id_or_default(&self) -> Self::Id {
        self.id().unwrap_or_default()
    }
}

impl<T> ObjWithId for Rc<T>
where
    T: ObjWithId,
{
    type Id = T::Id;

    fn id(&self) -> Self::Id {
        <T as ObjWithId>::id(self)
    }
}

impl<T> ObjWithId for &T
where
    T: ObjWithId,
{
    type Id = T::Id;

    fn id(&self) -> Self::Id {
        <T as ObjWithId>::id(self)
    }
}

pub struct ObjAndId<T>
where
    T: ObjWithId,
{
    obj: CloneCell<T>,
    id: Cell<T::Id>,
}

impl<T> Default for ObjAndId<T>
where
    T: ObjWithId + Default,
{
    fn default() -> Self {
        Self::new(T::default())
    }
}

impl<T> ObjAndId<T>
where
    T: ObjWithId,
{
    pub fn new(obj: T) -> Self {
        Self {
            id: Cell::new(obj.id()),
            obj: CloneCell::new(obj),
        }
    }

    #[inline]
    pub fn get(&self) -> T
    where
        T: JayClone,
    {
        self.obj.get()
    }

    #[inline]
    pub fn id(&self) -> T::Id {
        self.id.get()
    }

    pub fn set(&self, obj: T) -> T {
        self.id.set(obj.id());
        self.obj.set(obj)
    }
}

impl<T> ObjAndId<Option<T>>
where
    T: ObjWithId,
{
    #[inline]
    pub fn is_some(&self) -> bool {
        self.id.is_some()
    }

    #[inline]
    pub fn is_none(&self) -> bool {
        self.id.is_none()
    }
}