#[cfg(not(target_family = "wasm"))]
use rayon::prelude::*;
use std::f64::consts::PI;
use std::num::NonZeroU32;
use web_time::Instant;
use winit::event::{KeyEvent, WindowEvent};
use winit::event_loop::{ControlFlow, EventLoop};
use winit::keyboard::{Key, NamedKey};

#[path = "utils/winit_app.rs"]
mod winit_app;

fn main() {
    let event_loop = EventLoop::new().unwrap();
    let start = Instant::now();

    let context = softbuffer::Context::new(event_loop.owned_display_handle()).unwrap();

    let app = winit_app::WinitAppBuilder::with_init(
        |event_loop| {
            let window = winit_app::make_window(event_loop, |w| w);

            let old_size = (0, 0);
            let frames = pre_render_frames(0, 0);

            (window, old_size, frames)
        },
        move |_elwft, (window, _old_size, _frames)| {
            softbuffer::Surface::new(&context, window.clone()).unwrap()
        },
    )
    .with_event_handler(move |state, surface, window_id, event, elwt| {
        let (window, old_size, frames) = state;

        elwt.set_control_flow(ControlFlow::Poll);

        if window_id != window.id() {
            return;
        }

        match event {
            WindowEvent::Resized(size) => {
                let Some(surface) = surface else {
                    eprintln!("Resized fired before Resumed or after Suspended");
                    return;
                };

                if let (Some(width), Some(height)) =
                    (NonZeroU32::new(size.width), NonZeroU32::new(size.height))
                {
                    surface.resize(width, height).unwrap();
                }
            }
            WindowEvent::RedrawRequested => {
                let Some(surface) = surface else {
                    eprintln!("RedrawRequested fired before Resumed or after Suspended");
                    return;
                };

                let elapsed = start.elapsed().as_secs_f64() % 1.0;

                let mut buffer = surface.buffer_mut().unwrap();

                let size = (buffer.width().get(), buffer.height().get());
                if size != *old_size {
                    *old_size = size;
                    *frames = pre_render_frames(size.0, size.1);
                }

                let frame = &frames[((elapsed * 60.0).round() as usize).clamp(0, 59)];

                buffer.copy_from_slice(frame);
                buffer.present().unwrap();
            }
            WindowEvent::CloseRequested
            | WindowEvent::KeyboardInput {
                event:
                    KeyEvent {
                        logical_key: Key::Named(NamedKey::Escape),
                        ..
                    },
                ..
            } => {
                elwt.exit();
            }
            _ => {}
        }
    })
    .with_about_to_wait_handler(|state, _, _| {
        let (window, _, _) = state;
        window.request_redraw();
    });

    winit_app::run_app(event_loop, app);
}

fn pre_render_frames(width: u32, height: u32) -> Vec<Vec<u32>> {
    let render = |frame_id| {
        let elapsed = ((frame_id as f64) / (60.0)) * 2.0 * PI;

        let coords = (0..height).flat_map(|x| (0..width).map(move |y| (x, y)));
        coords
            .map(|(x, y)| {
                let y = (y as f64) / (height as f64);
                let x = (x as f64) / (width as f64);
                let red =
                    ((((y + elapsed).sin() * 0.5 + 0.5) * 255.0).round() as u32).clamp(0, 255);
                let green =
                    ((((x + elapsed).sin() * 0.5 + 0.5) * 255.0).round() as u32).clamp(0, 255);
                let blue =
                    ((((y - elapsed).cos() * 0.5 + 0.5) * 255.0).round() as u32).clamp(0, 255);

                blue | (green << 8) | (red << 16)
            })
            .collect::<Vec<_>>()
    };

    #[cfg(target_family = "wasm")]
    return (0..60).map(render).collect();

    #[cfg(not(target_family = "wasm"))]
    (0..60).into_par_iter().map(render).collect()
}