use std::path::Path;

use simple_logger::SimpleLogger;
use winit::event::ElementState;
use winit::event::VirtualKeyCode::Key1;
use winit::window::Icon;

use LemoGUI::device::container::Container;
use LemoGUI::device::display_window::*;
use LemoGUI::device::event_context::ELContext;
use LemoGUI::device::wgpu_context::WGContext;
use LemoGUI::graphic::base::color::*;
use LemoGUI::graphic::base::shape::{Point, Rectangle};
use LemoGUI::graphic::render_middle::pipeline_state::PipelineState;
use LemoGUI::graphic::render_middle::render_function::RenderUtil;
use LemoGUI::graphic::style::*;
use LemoGUI::widget::button::Button;
use LemoGUI::widget::component::ComponentModel;
use LemoGUI::widget::frame::Frame;
use LemoGUI::widget::listener::Listener;

#[derive(Debug, Copy, Clone, PartialOrd, PartialEq)]
enum Ms {
    Add,
    Sub,
}

fn main() {
    SimpleLogger::new().with_level(log::LevelFilter::Info).init().unwrap();
    log::info!("build window");
    let path = concat!(env!("CARGO_MANIFEST_DIR"), "/res/icon.png");

    let icon = load_icon(Path::new(path));
    let mut builder = winit::window::WindowBuilder::new();
    builder = builder.with_title("Counter")
        .with_inner_size(winit::dpi::LogicalSize::new(428.0, 433.0))
        .with_window_icon(Some(icon));

    start(builder, &frame)
}

fn frame(wgcontext: WGContext) -> Frame<Ms>
{
    // 自定义设置
    let rect = Rectangle::new(100.0, 100.0, 170, 40);
    let style = Style::default()
        .no_border()
        .hover_color(RGBA(0.0, 0.75, 1.0, 0.5))
        .back_color(RGBA(1.0, 0.5, 0.5, 1.0))
        .font_color(RGBA(0.0, 0.0, 0.0, 1.0), 45.)
        .round();
    let b1
        = Button::new_with_style(rect, style, "button1").action(Ms::Add);
    let b2 = Button::new(
        Point { x: 100.0, y: 300.0 },
        "sub button减")
        .action(Ms::Sub);

    let counter = Counter {
        value: 0,
        b1,
        b2,
    };

    let mut frame = Frame::new(wgcontext);
    frame.add_comp(counter);

    frame
}

struct Counter<M: Copy + PartialEq> {
    value: i32,
    b1: Button<M>,
    b2: Button<M>,
}

impl<M: Copy + PartialEq> Listener<M> for Counter<M> {
    fn action_listener(&mut self, action_state: ElementState, el_context: &ELContext<'_, M>) -> bool {
        self.b1.action_listener(action_state, el_context) ||
            self.b2.action_listener(action_state, el_context)
    }
    fn message_listener(&mut self, broadcast: &M) -> bool {
        if self.b1.match_message(broadcast) {
            self.value += 1;
        }
        if self.b2.match_message(broadcast) {
            self.value -= 1;
        }
        true
    }
}

impl<M: Copy + PartialEq> ComponentModel<M> for Counter<M> {
    fn draw(&self, wgcontext: &WGContext, render_utils: &mut RenderUtil, glob_pipeline: &PipelineState) {
        let v = Button::<M>::new(Point::new(120., 20.), self.value.to_string());
        v.draw(wgcontext, render_utils, glob_pipeline);
        self.b1.draw(wgcontext, render_utils, glob_pipeline);
        self.b2.draw(wgcontext, render_utils, glob_pipeline);
    }
}

fn load_icon(path: &Path) -> Icon {
    let (icon_rgba, icon_width, icon_height) = {
        let image = image::open(path)
            .expect("Failed to open icon path")
            .into_rgba8();
        let (width, height) = image.dimensions();
        let rgba = image.into_raw();
        (rgba, width, height)
    };
    Icon::from_rgba(icon_rgba, icon_width, icon_height).expect("Failed to open icon")
}