use winit::event::{Event, WindowEvent};
use winit::event_loop::{ControlFlow, EventLoop};
use winit::window::WindowBuilder;
use softbuffer::GraphicsContext;

const BUFFER_WIDTH: usize = 256;
const BUFFER_HEIGHT: usize = 128;

fn main() {
    let event_loop = EventLoop::new();
    let window = WindowBuilder::new().build(&event_loop).unwrap();
    let mut graphics_context = unsafe { GraphicsContext::new(window) };

    event_loop.run(move |event, _, control_flow| {
        *control_flow = ControlFlow::Wait;

        match event {
            Event::RedrawRequested(window_id) if window_id == graphics_context.window().id() => {
                let buffer = (0..((BUFFER_WIDTH*BUFFER_HEIGHT) as usize)).map(|index|{
                    let y = index / (BUFFER_WIDTH as usize);
                    let x = index % (BUFFER_WIDTH as usize);
                    let red = x % 255;
                    let green = y % 255;
                    let blue = (x*y) % 255;

                    let color = blue | (green << 8) | (red << 16);

                    color as u32
                }).collect::<Vec<_>>();

                graphics_context.set_buffer(&buffer, BUFFER_WIDTH as u16, BUFFER_HEIGHT as u16);
            }
            Event::WindowEvent {
                event: WindowEvent::CloseRequested,
                window_id
            } if window_id == graphics_context.window().id() => {
                *control_flow = ControlFlow::Exit;
            },
            _ => {}
        }
    });
}