#[macro_use]
extern crate lazy_static;

extern crate direct2d;
extern crate winapi;
extern crate wio;
extern crate math2d;

use math2d::*;
use direct2d::brush::SolidColorBrush;
use direct2d::enums::{FigureBegin, FigureEnd, FillMode};
use direct2d::geometry::PathGeometry;
use direct2d::layer::Layer;
use direct2d::render_target::HwndRenderTarget;
use direct2d::factory::Factory;
use std::{mem, ptr};

use winapi::ctypes::c_int;
use winapi::shared::minwindef::*;
use winapi::shared::windef::HWND;
use winapi::um::libloaderapi::*;
use winapi::um::winuser::*;
use wio::wide::ToWide;

lazy_static! {
    pub static ref BACKGROUND: Color = Color::from_u32(0x2A14CC, 1.0);
    pub static ref HIGHLIGHT: Color = Color::from_u32(0x483D99, 1.0);
    pub static ref ACCENT: Color = Color::from_u32(0x006AFF, 1.0);
    pub static ref FOREGROUND: Color = Color::from_u32(0xFFA940, 1.0);
    pub static ref FADED: Color = Color::from_u32(0xCC5E14, 1.0);
}

fn paint_window(window: &mut Window) {
    let rt = window.target.as_mut().unwrap();

    let accent_brush = SolidColorBrush::create(&rt)
        .with_color(*ACCENT)
        .build()
        .unwrap();
    let foreground_brush = SolidColorBrush::create(&rt)
        .with_color(*FOREGROUND)
        .build()
        .unwrap();
    let diamond_brush = SolidColorBrush::create(&rt)
        .with_color(*HIGHLIGHT)
        .build()
        .unwrap();

    rt.begin_draw();
    rt.clear(*BACKGROUND);

    rt.fill_rectangle([50.0, 50.0, 750.0, 430.0], &accent_brush);
    rt.fill_rectangle([150.0, 150.0, 650.0, 330.0], &foreground_brush);

    let path = build_path(&rt.factory());

    let layer = Layer::create(rt, None).unwrap();
    rt.push_layer(&layer)
        .with_mask(&path)
        .with_mask_transform(
            Matrix3x2f::scaling([300.0, 300.0], (0.5, 0.5))
                * Matrix3x2f::translation([400.0, 240.0]),
        )
        .push();

    rt.fill_rectangle([0.0, 0.0, 800.0, 480.0], &diamond_brush);

    rt.pop_layer();

    rt.end_draw().unwrap();
}

struct Window {
    hwnd: HWND,
    factory: Factory,
    target: Option<HwndRenderTarget>,
}

unsafe extern "system" fn wnd_proc(hwnd: HWND, msg: UINT, wp: WPARAM, lp: LPARAM) -> LRESULT {
    match msg {
        WM_CREATE => {
            // Set the window pointer into the creation parameters
            let ptr = lp as *const CREATESTRUCTW;
            if ptr.is_null() {
                std::process::abort();
            }
            let cs = &*ptr;

            let ptr = cs.lpCreateParams as *mut Window;
            SetWindowLongPtrW(hwnd, GWLP_USERDATA, ptr as _);
            if ptr.is_null() {
                std::process::abort();
            }
            let window = &mut *ptr;

            // Create the direct2d stuff
            let target = HwndRenderTarget::create(&window.factory)
                .with_hwnd(hwnd)
                .with_pixel_size(800, 480)
                .build()
                .unwrap();

            window.target = Some(target);

            0
        }
        WM_PAINT => {
            let ptr = GetWindowLongPtrW(hwnd, GWLP_USERDATA) as *mut Window;
            if !ptr.is_null() {
                let window = &mut *ptr;
                paint_window(window);
            }
            DefWindowProcW(hwnd, msg, wp, lp)
        }
        WM_TIMER => {
            DestroyWindow(hwnd);
            0
        }
        _ => DefWindowProcW(hwnd, msg, wp, lp),
    }
}

unsafe fn real_window_test() {
    let factory = Factory::new().unwrap();

    let mut window = Window {
        hwnd: ptr::null_mut(),
        factory: factory,
        target: None,
    };

    let hinst: HINSTANCE = GetModuleHandleW(ptr::null());
    let class_name = "Test D2D1 Window Class".to_wide_null();
    let window_name = "Test D2D1 Window".to_wide_null();

    RegisterClassW(&WNDCLASSW {
        style: CS_DBLCLKS | CS_HREDRAW | CS_VREDRAW,
        lpfnWndProc: Some(wnd_proc),
        cbWndExtra: mem::size_of::<&mut Window>() as c_int,
        hInstance: hinst,
        lpszClassName: class_name.as_ptr(),

        ..mem::zeroed()
    });

    let hwnd = CreateWindowExW(
        0, // dwExStyle
        class_name.as_ptr(),
        window_name.as_ptr(),
        WS_OVERLAPPED,
        CW_USEDEFAULT, // x
        CW_USEDEFAULT, // y
        800,           // width
        480,           // height
        ptr::null_mut(),
        ptr::null_mut(),
        hinst,
        &mut window as *mut _ as LPVOID,
    );

    assert!(hwnd != ptr::null_mut());
    window.hwnd = hwnd;

    ShowWindow(hwnd, SW_SHOW);
    SetTimer(hwnd, 0, 1_000, None);

    let mut msg: MSG = mem::uninitialized();
    loop {
        match GetMessageW(&mut msg, hwnd, 0, 0) {
            -1 => break,
            _ => {
                TranslateMessage(&msg);
                DispatchMessageW(&msg);
            }
        }
    }
}

#[test]
fn window_test() {
    unsafe { real_window_test() };
}

fn build_path(factory: &Factory) -> PathGeometry {
    let path = PathGeometry::create(factory)
        .unwrap()
        .fill_mode(FillMode::Winding)
        // Square with a triangle base
        .with_figure(
            Point2f::new(0.0, 0.0),
            FigureBegin::Filled,
            FigureEnd::Closed,
            |figure| {
                figure
                    .add_line(Point2f::new(1.0, 0.0))
                    .add_line(Point2f::new(1.0, 1.0))
                    .add_line(Point2f::new(0.5, 1.5))
                    .add_line(Point2f::new(0.0, 1.0))
                    .add_line(Point2f::new(0.0, 0.0))
            },
        )
        // Add a triangle hat
        .with_figure(
            Point2f::new(0.0, 0.0),
            FigureBegin::Filled,
            FigureEnd::Closed,
            |figure| {
                figure
                    .add_line(Point2f::new(0.5, -0.5))
                    .add_line(Point2f::new(1.0, 0.0))
            },
        )
        // Cut a hole in the middle
        .fill_mode(FillMode::Alternate)
        .with_figure(
            Point2f::new(0.25, 0.25),
            FigureBegin::Filled,
            FigureEnd::Closed,
            |figure| {
                figure
                    .add_line(Point2f::new(0.75, 0.25))
                    .add_line(Point2f::new(0.75, 0.75))
                    .add_line(Point2f::new(0.25, 0.75))
            },
        )
        .finish()
        .unwrap();

    path
}