registry.rs

use better_any::{Tid, TidAble};
use derive_more::{Deref, DerefMut};
use mahf::{CustomState, StateRegistry};

#[derive(Debug, Deref, DerefMut, Tid)]
pub struct A(usize);
impl CustomState<'_> for A {}

#[derive(Debug, Deref, DerefMut, Tid)]
pub struct B<'a>(&'a mut A);
impl<'a> CustomState<'a> for B<'a> {}

fn main() {
    let mut b_source = A(10);

    let mut registry: StateRegistry = StateRegistry::new();

    registry.insert(A(0));
    registry.insert(B(&mut b_source));

    let a = registry.borrow::<A>();
    let _a2 = registry.borrow::<A>();
    let mut b = registry.borrow_mut::<B>();

    assert!(registry.try_borrow_mut::<A>().is_err());

    println!("{a:?}");
    println!("{b:?}");

    b.0 .0 += 1;

    assert!(registry.try_borrow_mut::<B>().is_err());

    println!("{b:?}");

    drop(a);
    drop(_a2);
    drop(b);

    let (a, b) = registry.get_multiple_mut::<(A, B)>();
    a.0 += 1;
    b.0 .0 += 1;

    println!("{a:?}");
    println!("{b:?}");

    let a = registry.entry::<A>().or_insert(A(0));
    println!("{a:?}");
    drop(a);

    assert!(registry.try_get_multiple_mut::<(A, A)>().is_err());
}