#include "window.h"
#include "controls.h"
#include <shellapi.h>
HICON GetExeIcon(const std::wstring &filePath)
{
    SHFILEINFO fileInfo;
    HICON hIcon = NULL;
    if (SHGetFileInfo(filePath.c_str(), 0, &fileInfo, sizeof(fileInfo), SHGFI_ICON | SHGFI_LARGEICON))
    {
        hIcon = fileInfo.hIcon;
    }
    return hIcon;
}
void mainwindow::visfont()
{
    if (hfont == 0)
        hfont = parent->hfont;
    if (hfont)
    {
        for (auto ctr : controls)
        {
            SendMessage(ctr->winId, WM_SETFONT, (LPARAM)hfont, TRUE);
        }
    }
}
void mainwindow::setfont(const Font &font)
{
    hfont = font.hfont();
    SendMessage(winId, WM_SETFONT, (WPARAM)hfont, TRUE);
    visfont();
    for (auto child : childrens)
    {
        child->setfont(font);
    }
}
std::wstring basewindow::text()
{
    int textLength = GetWindowTextLength(winId);
    std::vector<wchar_t> buffer(textLength + 1);
    GetWindowText(winId, buffer.data(), buffer.size());
    return buffer.data();
}
void basewindow::settext(const std::wstring &text)
{
    SetWindowText(winId, text.c_str());
}

void basewindow::setgeo(int x, int y, int w, int h)
{
    MoveWindow(winId, x, y, w, h, TRUE);
    on_size(w, h);
}
RECT basewindow::getgeo()
{
    RECT rect;
    GetWindowRect(winId, &rect);
    return rect;
}

LRESULT mainwindow::wndproc(UINT message, WPARAM wParam, LPARAM lParam)
{
    switch (message)
    {
    case WM_SHOWWINDOW:
    {
        on_show();
        visfont();
        break;
    }
    case WM_SIZE:
    {
        int width = LOWORD(lParam);
        int height = HIWORD(lParam);
        on_size(width, height);
        break;
    }
    case WM_NOTIFY:
    {
        NMHDR *pnmhdr = (NMHDR *)lParam;
        for (auto ctl : controls)
        {
            if (pnmhdr->hwndFrom == ctl->winId)
            {
                ctl->dispatch_2(wParam, lParam);
                break;
            }
        }
    }
    case WM_COMMAND:
    {
        if (lParam == 0)
        {
            for (auto ctl : controls)
            {
                if (lastcontexthwnd == ctl->winId)
                {
                    if (ctl->menu)
                        ctl->menu.value().dispatch(wParam);
                    break;
                }
            }
        }
        else
            for (auto ctl : controls)
            {
                if ((HWND)lParam == ctl->winId)
                {
                    ctl->dispatch(wParam);
                    break;
                }
            }
        break;
    }
    case WM_CONTEXTMENU:
    {
        bool succ = false;
        lastcontexthwnd = 0;
        for (auto ctl : controls)
        {
            if ((HWND)wParam == ctl->winId)
            {
                auto hm = ctl->on_menu();
                ctl->menu = hm;
                if (hm)
                {
                    int xPos = LOWORD(lParam);
                    int yPos = HIWORD(lParam);
                    TrackPopupMenu(hm.value().load(), TPM_LEFTALIGN | TPM_TOPALIGN | TPM_RIGHTBUTTON,
                                   xPos, yPos, 0, winId, NULL);
                    lastcontexthwnd = ctl->winId;
                    succ = true;
                }
                break;
            }
        }
        if (succ == false)
            return DefWindowProc(winId, message, wParam, lParam);
        break;
    }
    case WM_CLOSE:
    {
        on_close();
        if (parent == 0)
            PostQuitMessage(0);
        else
            ShowWindow(winId, SW_HIDE);
        break;
    }
    default:
        return DefWindowProc(winId, message, wParam, lParam);
    }

    return 0;
}
std::pair<int, int> mainwindow::calculateXY(int w, int h)
{
    int cx, cy;
    if (parent == 0)
    {
        int screenWidth = GetSystemMetrics(SM_CXSCREEN);
        int screenHeight = GetSystemMetrics(SM_CYSCREEN);
        cx = screenWidth / 2;
        cy = screenHeight / 2;
    }
    else
    {
        auto rect = parent->getgeo();
        cx = (rect.left + rect.right) / 2;
        cy = (rect.top + rect.bottom) / 2;
    }
    return {cx - w / 2, cy - h / 2};
}
void mainwindow::setcentral(int w, int h)
{
    auto [x, y] = calculateXY(w, h);
    setgeo(x, y, w, h);
}
void mainwindow::setlayout(control *_l)
{
    layout = _l;
}
mainwindow::mainwindow(mainwindow *_parent)
{
    layout = 0;
    const wchar_t CLASS_NAME[] = L"LunaHostWindow";

    WNDCLASS wc = {};
    wc.lpfnWndProc = [](HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
    {
        mainwindow *_window = reinterpret_cast<mainwindow *>(GetWindowLongPtrW(hWnd, GWLP_USERDATA));
        if ((!_window) || (_window->winId != hWnd))
            return DefWindowProc(hWnd, message, wParam, lParam);
        return _window->wndproc(message, wParam, lParam);
    };
    wc.hInstance = GetModuleHandle(0);
    wc.lpszClassName = CLASS_NAME;
    wc.hbrBackground = (HBRUSH)(COLOR_WINDOW);
    wc.hIcon = GetExeIcon(getModuleFilename().value()); // LoadIconW(GetModuleHandle(0),L"IDI_ICON1");

    static auto _ = RegisterClass(&wc);
    HWND hWnd = CreateWindowEx(
        WS_EX_CLIENTEDGE, CLASS_NAME, CLASS_NAME, WS_OVERLAPPEDWINDOW,
        CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
        _parent ? _parent->winId : NULL, NULL, GetModuleHandle(0), this);
    winId = hWnd;
    parent = _parent;
    if (parent)
        parent->childrens.push_back(this);
    SetWindowLongPtrW(hWnd, GWLP_USERDATA, (LONG_PTR)this);
}
void mainwindow::show()
{
    ShowWindow(winId, SW_SHOW);
    SetForegroundWindow(winId);
}
void mainwindow::close()
{
    ShowWindow(winId, SW_HIDE);
}
void mainwindow::run()
{
    MSG msg = {};
    while (GetMessage(&msg, NULL, 0, 0))
    {
        TranslateMessage(&msg);
        DispatchMessage(&msg);
    }
}

void mainwindow::on_close() {}
void mainwindow::on_show() {}
void mainwindow::on_size(int w, int h)
{
    if (layout)
    {
        layout->setgeo(0, 0, w, h);
    }
}
void basewindow::on_size(int w, int h) {}