#![cfg(test)]
#![allow(clippy::disallowed_names)]

use generic_mutability::*;

fn gen_index<M: Mutability>(gen_vec: GenRef<'_, M, Vec<i32>>, idx: usize) -> GenRef<'_, M, i32> {
    gen_mut! {M => {
        let ref_vec = from_gen!(gen_vec);
        into_gen!(&gen ref_vec[idx])
    }}
}
fn gen_get<M: Mutability>(
    gen_vec: GenRef<'_, M, Vec<i32>>,
    idx: usize,
) -> Option<GenRef<'_, M, i32>> {
    gen_mut! {M => {
        let ref_vec = from_gen!(gen_vec);
        switch_shared_mut![<[_]>::get, <[_]>::get_mut](ref_vec, idx).map(into_gen!())
    }}
}

#[test]
fn map_macro() {
    let mut vec = vec![1, 2, 3];
    let elem = GenRef::gen_into_mut(gen_index(GenRef::from(&mut vec), 1), Mutable::mutability());
    assert_eq!(*elem, 2);
}
#[test]
fn map_macro_with_get() {
    let mut vec = vec![1, 2, 3];
    let elem = gen_get(GenRef::from(&mut vec), 1);
    assert!(elem.is_some_and(|e| e == &2));
}

struct Foo {
    pub a: Bar,
}
struct Bar {
    pub b: i32,
}

impl Foo {
    fn gen_b<M: Mutability>(foo: GenRef<'_, M, Foo>) -> GenRef<'_, M, i32> {
        field!(&gen foo.a.b)
    }
}

#[test]
fn field_access_macro() {
    let foo = Foo { a: Bar { b: 1 } };

    let b = GenRef::into_shared(Foo::gen_b(GenRef::from(&foo)));

    assert_eq!(b, &1)
}