use hex_color::HexColor;
use tauri::{ Runtime, Window};

use std::mem::transmute;
use std::{ptr, ffi::c_void, mem::size_of};
use tauri::plugin::{Builder, TauriPlugin};

use windows::Win32::UI::Controls::{WTA_NONCLIENT, WTNCA_NODRAWICON, WTNCA_NOSYSMENU, WTNCA_NOMIRRORHELP};

use windows::Win32::UI::Controls::SetWindowThemeAttribute;
use windows::Win32::UI::Controls::WTNCA_NODRAWCAPTION;
use windows::Win32::Graphics::Dwm::DWMWA_CAPTION_COLOR;
use windows::Win32::Foundation::COLORREF;
use windows::Win32::Foundation::BOOL;
use windows::Win32::Graphics::Dwm::DwmSetWindowAttribute;
use windows::Win32::Foundation::HWND;
use windows::Win32::Graphics::Dwm::{DWMWA_USE_IMMERSIVE_DARK_MODE};

pub fn init<R: Runtime>() -> TauriPlugin<R> {
    Builder::new("windows_window")
        .on_window_ready(|window| {
            #[cfg(target_os = "windows")]
            setup_win_window(window);
            return;
        })
        .build()
}

fn hex_color_to_colorref(color: HexColor) -> COLORREF {
    // TODO: Remove this unsafe, This operation doesn't need to be unsafe!
    unsafe {
        COLORREF(transmute::<[u8; 4], u32>([color.r, color.g, color.b, 0]))
    }
}

struct WinThemeAttribute {
    flag: u32,
    mask: u32
}

#[cfg(target_os = "windows")]
fn update_bg_color(hwnd: &HWND, bg_color: HexColor) {

    let use_dark_mode = BOOL::from(true);

    let final_color = hex_color_to_colorref(bg_color);

    unsafe {
        DwmSetWindowAttribute(
            HWND(hwnd.0),
            DWMWA_USE_IMMERSIVE_DARK_MODE,
            ptr::addr_of!(use_dark_mode) as *const c_void,
            size_of::<BOOL>().try_into().unwrap()
        ).unwrap();

        DwmSetWindowAttribute(
            HWND(hwnd.0),
            DWMWA_CAPTION_COLOR,
            ptr::addr_of!(final_color) as *const c_void,
            size_of::<COLORREF>().try_into().unwrap()
        ).unwrap();

        let flags = WTNCA_NODRAWCAPTION | WTNCA_NODRAWICON;
        let mask  = WTNCA_NODRAWCAPTION | WTNCA_NODRAWICON | WTNCA_NOSYSMENU | WTNCA_NOMIRRORHELP;
        let options = WinThemeAttribute { flag: flags, mask };

        SetWindowThemeAttribute(
            HWND(hwnd.0),
            WTA_NONCLIENT,
            ptr::addr_of!(options) as *const c_void,
            size_of::<WinThemeAttribute>().try_into().unwrap()
        ).unwrap();
    }
}

#[cfg(target_os = "windows")]
pub fn setup_win_window<R: Runtime>(window: Window<R>) {
    let win_handle = window.hwnd().unwrap();
    let win_clone = win_handle.clone();

    window.listen_global("yaak_bg_changed", move |ev| {
        let payload = serde_json::from_str::<&str>(ev.payload().unwrap())
            .unwrap()
            .trim();

        let color = HexColor::parse_rgb(payload).unwrap();
        update_bg_color(&HWND(win_clone.0), color);
    });
}