use crate::{
    render::{sprite::SpriteDraw, AtlasLayout, Renderer},
    tilemap::Tilemap,
    ui::UiState,
};
use hurrycurry_protocol::{
    glam::{IVec2, Vec2},
    TileIndex,
};
use rand::{random, seq::IndexedRandom, thread_rng};
use sdl2::{
    keyboard::{KeyboardState, Keycode},
    mouse::MouseState,
};

#[derive(Debug)]
pub struct Menu {
    map: Tilemap,
    fade_in: f32,
    ui_state: UiState,
    background: Vec2,
}

impl Menu {
    pub fn new(layout: &AtlasLayout) -> Self {
        let mut map = Tilemap::default();
        map.init(
            &[
                "floor",
                "tomato-crate",
                "raw-steak-crate",
                "table",
                "chair",
                "counter",
                "sink",
                "stove",
            ]
            .map(String::from),
            layout,
        );
        static BUCKETS: &[&[usize]] = &[&[], &[0, 0, 0, 0, 1, 2], &[3, 4, 5], &[6, 7]];

        for x in -10..11 {
            for y in -10..11 {
                let p = Vec2::new(x as f32, y as f32);
                let w = (-p.length() * 0.15).exp();
                let k = ((random::<f32>() * w) * BUCKETS.len() as f32) as usize;
                if let Some(ti) = BUCKETS[k.min(BUCKETS.len())].choose(&mut thread_rng()) {
                    map.set(IVec2::new(x, y), Some(TileIndex(*ti)), [None; 4])
                }
            }
        }

        Self {
            map,
            fade_in: 0.,
            ui_state: UiState::default(),
            background: Vec2::ZERO,
        }
    }
    pub fn tick(
        &mut self,
        dt: f32,
        keyboard: &KeyboardState,
        mouse: &MouseState,
        _layout: &AtlasLayout,
    ) {
        self.fade_in = (self.fade_in + dt).min(1.);
        self.background += Vec2::new(2., 3.) * dt;
        self.ui_state.update(keyboard, mouse, dt);
    }
    pub fn keyboard_event(&mut self, keycode: Keycode, down: bool) {
        self.ui_state.keyboard_event(keycode, down);
    }
    pub fn draw(&mut self, ctx: &mut Renderer) {
        ctx.set_world_view(
            ctx.size / ctx.get_world_scale() * Vec2::new(0.8, 0.2),
            ctx.size.max_element() / 32. / 15.,
        );

        for x in -1..=2 {
            for y in -1..=2 {
                ctx.draw_ui(SpriteDraw::underlay(
                    ctx.misc_textures.clouds,
                    Vec2::new(x as f32, y as f32) * 256. + self.background,
                    Vec2::ONE * 256.,
                    None,
                ));
            }
        }
        ctx.draw_ui(SpriteDraw::underlay(
            ctx.misc_textures.solid,
            Vec2::ZERO,
            ctx.ui_size,
            Some([0, 0, 0, 50]),
        ));

        self.map.draw(ctx);

        self.ui_state.draw(ctx, |ui| {
            if ui.button(80., "Join") {
                eprintln!("join button")
            }
            if ui.button(80., "Settings") {
                eprintln!("settings button")
            }
            if ui.button(80., "Quit") {
                eprintln!("quit button")
            }
            ui.fill();
        });

        ctx.draw_ui(SpriteDraw::overlay(
            ctx.misc_textures.solid,
            Vec2::ZERO,
            ctx.ui_size,
            Some([0, 0, 0, 255 - (self.fade_in * 255.) as u8]),
        ));
    }
}