use std::sync::{Arc, Mutex};
use std::any::Any;
use druid_shell::{Application, WinHandler, WindowHandle, WinCtx, KeyEvent, WindowBuilder, RunLoop, HotKey, KeyCode};
use druid_shell::piet::{
    Piet, PietFont, Color, FontBuilder, RenderContext, Text, TextLayoutBuilder
};
use druid_shell::kurbo::Rect;
use neovim_lib::NeovimApi;

use crate::editor::{DrawCommand, Editor, Colors};

#[derive(new)]
struct WindowState {
    editor: Arc<Mutex<Editor>>,

    #[new(default)]
    pub size: (f64, f64),
    #[new(default)]
    pub handle: WindowHandle,
    #[new(default)]
    pub font: Option<PietFont>
}

const FONT_NAME: &str = "Delugia Nerd Font";
const FONT_SIZE: f64 = 14.0;
const FONT_WIDTH: f64 = 8.2;
const FONT_HEIGHT: f64 = 16.4;

fn process_draw_commands(draw_commands: &Vec<DrawCommand>, default_colors: &Colors, piet: &mut Piet, font: &PietFont) {
    for command in draw_commands {
        let x = command.col_start as f64 * FONT_WIDTH;
        let y = command.row as f64 * FONT_HEIGHT + FONT_HEIGHT;
        let top = y - FONT_HEIGHT * 0.8;
        let top_left = (x, top);
        let width = command.text.chars().count() as f64 * FONT_WIDTH;
        let height = FONT_HEIGHT;
        let bottom_right = (x + width, top + height);
        let region = Rect::from_points(top_left, bottom_right);
        piet.fill(region, &command.style.colors.background.clone().or(default_colors.background.clone()).unwrap());
            
        let piet_text = piet.text();
        let text_layout = piet_text.new_text_layout(&font, &command.text).build().unwrap();
        piet.draw_text(&text_layout, (x, y), &command.style.colors.foreground.clone().or(default_colors.foreground.clone()).unwrap());
    }
}

impl WinHandler for WindowState {
    fn connect(&mut self, handle: &WindowHandle) {
        self.handle = handle.clone();
    }

    fn paint(&mut self, piet: &mut Piet, _ctx: &mut dyn WinCtx) -> bool {
        let text = piet.text();
        if self.font.is_none() {
            self.font = Some(text.new_font_by_name(FONT_NAME, FONT_SIZE).build().unwrap());
        }
        let font = self.font.as_ref().unwrap();

        let (draw_commands, default_colors, cursor_pos) = {
            let editor = self.editor.lock().unwrap();
            (editor.build_draw_commands().clone(), editor.default_colors.clone(), editor.cursor_pos.clone())
        };

        piet.clear(default_colors.background.clone().unwrap());
        process_draw_commands(&draw_commands, &default_colors, piet, font);

        let (cursor_grid_x, cursor_grid_y) = cursor_pos;
        let cursor_x = cursor_grid_x as f64 * FONT_WIDTH;
        let cursor_width = FONT_WIDTH / 8.0;
        let cursor_y = cursor_grid_y as f64 * FONT_HEIGHT + FONT_HEIGHT * 0.2;
        let cursor_height = FONT_HEIGHT;
        let cursor = Rect::from_points((cursor_x, cursor_y), (cursor_x + cursor_width, cursor_y + cursor_height));
        piet.fill(cursor, &Color::rgb8(0xff, 0xff, 0xff));
        true
    }

    fn key_down(&mut self, key_event: KeyEvent, _ctx: &mut dyn WinCtx) -> bool {
        let mut editor = self.editor.lock().unwrap();
        match key_event {
            k_e if k_e.key_code.is_printable() => {
                let incoming_text = k_e.unmod_text().unwrap_or("");
                editor.nvim.input(incoming_text).expect("Input call failed...");
            },
            k_e if (HotKey::new(None, KeyCode::Escape)).matches(k_e) => {
                editor.nvim.input("<Esc>").expect("Input call failed...");
            },
            k_e if (HotKey::new(None, KeyCode::Backspace)).matches(k_e) => {
                editor.nvim.input("<BS>").expect("Input call failed...");
            }
            _ => ()
        };
        true
    }

    fn size(&mut self, width: u32, height: u32, _ctx: &mut dyn WinCtx) {
        let dpi = self.handle.get_dpi();
        let dpi_scale = dpi as f64 / 96.0;
        let width_f = (width as f64) / dpi_scale;
        let height_f = (height as f64) / dpi_scale;
        self.size = (width_f, height_f);

        let mut editor = self.editor.lock().unwrap();
        editor.resize((width_f / FONT_WIDTH) as u16, (height_f / FONT_HEIGHT) as u16);
    }

    fn as_any(&mut self) -> &mut dyn Any {
        self
    }
}

pub fn ui_loop(editor: Arc<Mutex<Editor>>) {
    let window_state = WindowState::new(editor);

    Application::init();
    let mut run_loop = RunLoop::new();
    let mut builder = WindowBuilder::new();
    builder.set_handler(Box::new(window_state));
    builder.set_title("Nvim editor");
    let window = builder.build().unwrap();
    window.show();
    run_loop.run();
}