use std::sync::Arc;

use crate::lenses::{Lens, LensOver, LensView};

type Getter<T, U> = dyn Fn(T) -> U;
type Setter<T, U> = dyn Fn(T, U) -> T;

#[derive(Clone)]
pub struct FuncLens<T, U>(pub(crate) Arc<Getter<T, U>>, pub(crate) Arc<Setter<T, U>>);

impl<T, U> LensView<T> for FuncLens<T, U> {
    type Field = U;

    fn view(&self, thing: T) -> Self::Field {
        (self.0)(thing)
    }
}
impl<T: Clone, U> LensOver<T> for FuncLens<T, U> {
    fn over<F>(&self, thing: T, f: F) -> T
    where
        F: FnOnce(Self::Field) -> Self::Field,
    {
        let v = f((self.0)(thing.clone()));
        (self.1)(thing, v)
    }

    fn set(&self, thing: T, v: Self::Field) -> T {
        (self.1)(thing, v)
    }
}

/// Makes a lens that implements `LensView<T>` and `LensOver<T>` with the provided functions
pub fn lens_from_boxed<T, U>(
    getter: Arc<Getter<T, U>>,
    setter: Arc<Setter<T, U>>,
) -> Lens<FuncLens<T, U>> {
    Lens(FuncLens(getter, setter))
}
/// Makes a lens that implements `LensView<T>` and `LensOver<T>` with the provided functions
pub fn lens<T, U>(
    getter: impl Fn(T) -> U + 'static,
    setter: impl Fn(T, U) -> T + 'static,
) -> Lens<FuncLens<T, U>> {
    Lens(FuncLens(Arc::new(getter), Arc::new(setter)))
}