#![feature(type_alias_impl_trait, impl_trait_in_assoc_type)]

use std::time::Duration;

use nuit::{clone, Alignment, Animation, Bind, Button, Circle, ForEach, Frame, HStack, Rectangle, State, Text, VStack, Vec2, View, ViewExt, ZStack};

#[derive(Bind)]
struct AnimationsView<const COUNT: usize> {
    animations: [Animation; COUNT],
    flips: State<[bool; COUNT]>,
}

impl<const COUNT: usize> AnimationsView<COUNT> {
    pub fn new(animations: [Animation; COUNT]) -> Self {
        Self {
            animations,
            flips: State::new(animations.map(|_| false)),
        }
    }
}

impl<const COUNT: usize> View for AnimationsView<COUNT> {
    type Body = impl View;

    fn body(&self) -> Self::Body {
        let animations = self.animations;
        let flips = self.flips.clone();
        let width = 200.0;
        let radius = 10.0;
        let inner_width = width - 2.0 * radius;

        VStack::new((
            VStack::new(
                ForEach::with_index_id(animations, |i, animation| {
                    let factor = if flips.get()[i] { 1.0 } else { -1.0 };
                    HStack::new((
                        Text::new(format!("{}", animation))
                            .frame_with(Alignment::Trailing, Frame::with_width(100)),
                        ZStack::new((
                            Rectangle::new()
                                .frame(Frame::exact(inner_width, 2)),
                            Circle::new()
                                .frame(2.0 * radius)
                                .offset(Vec2::with_x(factor * inner_width / 2.0))
                        ))
                        .frame(Frame::with_width(width)),
                        Button::with_text("Flip", clone!(flips => move || {
                            let mut value = flips.get();
                            value[i] = !value[i];
                            flips.set_with(animation, value);
                        })),
                    ))
                })
            ),
        ))
    }
}

fn main() {
    let duration = Duration::from_secs(4);
    nuit::run_app(AnimationsView::new([
        Animation::linear(duration),
        Animation::ease_in(duration),
        Animation::ease_out(duration),
        Animation::ease_in_out(duration),
    ]));
}