use crate::{
    render::{sprite::SpriteDraw, AtlasLayout, Renderer},
    tilemap::Tilemap,
};
use hurrycurry_protocol::{
    glam::{IVec2, Vec2},
    TileIndex,
};
use rand::{random, seq::IndexedRandom, thread_rng};

pub struct MenuBackground {
    background: Vec2,
    map: Tilemap,
}

impl MenuBackground {
    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,
            background: Vec2::ZERO,
        }
    }

    pub fn tick(&mut self, dt: f32) {
        self.background += Vec2::new(2., 3.) * dt;
    }
    pub fn draw(&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);
    }
}