#include "stdafx.h"
#include "SupportEnterCode.h"
#include <Project64-core/Settings/SettingType/SettingsType-Application.h>
#include "resource.h"

class CRequestCode :
    public CDialogImpl<CRequestCode>
{
public:
    BEGIN_MSG_MAP_EX(CRequestCode)
        MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
        MESSAGE_HANDLER(WM_CTLCOLORSTATIC, OnColorStatic)
        MESSAGE_HANDLER(WM_ERASEBKGND, OnEraseBackground)
        COMMAND_ID_HANDLER(IDOK, OnOkCmd)
        COMMAND_ID_HANDLER(IDCANCEL, OnCloseCmd)
    END_MSG_MAP()

    enum { IDD = IDD_Support_RequestCode };

    CRequestCode(CProjectSupport & Support);
    void ShowOldCodeMsg();

private:
    CRequestCode(void);
    CRequestCode(const CRequestCode&);
    CRequestCode& operator=(const CRequestCode&);

    LRESULT OnInitDialog(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/);
    LRESULT OnColorStatic(UINT /*uMsg*/, WPARAM wParam, LPARAM lParam, BOOL& bHandled);
    LRESULT OnEraseBackground(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& bHandled);
    LRESULT OnOkCmd(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/);
    LRESULT OnCloseCmd(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/);

    CProjectSupport & m_Support;
    bool m_ShowOldCodeMsg;
};

CSupportEnterCode::CSupportEnterCode(CProjectSupport & Support) :
    m_Support(Support)
{
}

LRESULT CSupportEnterCode::OnInitDialog(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/)
{
    SetWindowText(wGS(MSG_SUPPORT_ENTER_CODE_TITLE).c_str());
    CWindow hDescription = GetDlgItem(IDC_DESCRIPTION);
    CWindow MachineId = GetDlgItem(IDC_MACHINE_ID);
    CWindow OkBtn = GetDlgItem(IDOK);
    CWindow CancelBtn = GetDlgItem(IDCANCEL);

    std::wstring DescriptionText = wGS(MSG_SUPPORT_ENTER_CODE_DESC);
    hDescription.SetWindowText(DescriptionText.c_str());
    MachineId.SetWindowText(stdstr(m_Support.MachineID()).ToUTF16().c_str());
    OkBtn.SetWindowText(wGS(MSG_SUPPORT_OK).c_str());
    CancelBtn.SetWindowText(wGS(MSG_SUPPORT_CANCEL).c_str());

    m_RequestLink.SubclassWindow(GetDlgItem(IDC_REQUEST_LINK));
    m_RequestLink.SetHyperLinkExtendedStyle(HLINK_COMMANDBUTTON, HLINK_COMMANDBUTTON);

    CRect rcWin = { 0 };
    hDescription.GetClientRect(&rcWin);

    CDC hDC = hDescription.GetDC();
    HFONT hFont = hDescription.GetFont();
    if (hFont == nullptr)
    {
        hFont = (HFONT)::GetStockObject(SYSTEM_FONT);
    }
    hDC.SelectFont(hFont);
    if (hDC.DrawText(DescriptionText.c_str(), DescriptionText.length(), &rcWin, DT_LEFT | DT_CALCRECT | DT_WORDBREAK | DT_NOCLIP) > 0)
    {
        hDescription.SetWindowPos(NULL, 0, 0, rcWin.right, rcWin.bottom, SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOOWNERZORDER);
    }
    hDescription.GetWindowRect(&rcWin);
    ::MapWindowPoints(NULL, m_hWnd, (LPPOINT)&rcWin, 2);

    MachineId.SetWindowPos(NULL, rcWin.left, rcWin.bottom + 4, 0, 0, SWP_NOACTIVATE | SWP_NOSIZE | SWP_NOOWNERZORDER);
    MachineId.GetWindowRect(&rcWin);
    ::MapWindowPoints(NULL, m_hWnd, (LPPOINT)&rcWin, 2);

    CWindow Code = GetDlgItem(IDC_CODE);
    Code.SetWindowPos(NULL, rcWin.left, rcWin.bottom + 4, 0, 0, SWP_NOACTIVATE | SWP_NOSIZE | SWP_NOOWNERZORDER);
    Code.GetWindowRect(&rcWin);
    ::MapWindowPoints(NULL, m_hWnd, (LPPOINT)&rcWin, 2);

    CWindow RequestDescption = GetDlgItem(IDC_REQUEST_DESCPTION);
    RequestDescption.ShowWindow(SWP_HIDEWINDOW);
    RequestDescption.SetWindowPos(NULL, rcWin.left, rcWin.bottom + 10, 0, 0, SWP_NOACTIVATE | SWP_NOSIZE | SWP_NOOWNERZORDER);
    RequestDescption.GetWindowRect(&rcWin);
    ::MapWindowPoints(NULL, m_hWnd, (LPPOINT)&rcWin, 2);

    CWindow RequestLink = GetDlgItem(IDC_REQUEST_LINK);
    RequestLink.ShowWindow(SWP_HIDEWINDOW);
    RequestLink.SetWindowPos(NULL, rcWin.left, rcWin.bottom + 4, 0, 0, SWP_NOACTIVATE | SWP_NOSIZE | SWP_NOOWNERZORDER);
    RequestLink.GetWindowRect(&rcWin);
    ::MapWindowPoints(NULL, m_hWnd, (LPPOINT)&rcWin, 2);

    RECT CancelBtnWin = { 0 };
    CancelBtn.GetWindowRect(&CancelBtnWin);
    ::MapWindowPoints(NULL, m_hWnd, (LPPOINT)&CancelBtnWin, 2);
    CancelBtn.SetWindowPos(NULL, CancelBtnWin.left, rcWin.bottom + 40, 0, 0, SWP_NOACTIVATE | SWP_NOSIZE | SWP_NOOWNERZORDER);

    RECT OkBtnWin = { 0 };
    OkBtn.GetWindowRect(&OkBtnWin);
    ::MapWindowPoints(NULL, m_hWnd, (LPPOINT)&OkBtnWin, 2);
    OkBtn.SetWindowPos(NULL, OkBtnWin.left, rcWin.bottom + 40, 0, 0, SWP_NOACTIVATE | SWP_NOSIZE | SWP_NOOWNERZORDER);
    OkBtn.GetWindowRect(&OkBtnWin);
    ::MapWindowPoints(NULL, m_hWnd, (LPPOINT)&OkBtnWin, 2);

    GetWindowRect(&rcWin);
    SetRect(&rcWin, 0, 0, rcWin.Width(), OkBtnWin.bottom + 30);
    AdjustWindowRectEx(&rcWin, GetStyle(), GetMenu() != NULL, GetExStyle());
    int32_t Left = (GetSystemMetrics(SM_CXSCREEN) - rcWin.Width()) / 2;
    int32_t	Top = (GetSystemMetrics(SM_CYSCREEN) - rcWin.Height()) / 2;
    MoveWindow(Left, Top, rcWin.Width(), rcWin.Height(), TRUE);
    return TRUE;
}

LRESULT CSupportEnterCode::OnColorStatic(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& /*bHandled*/)
{
    CDCHandle hdcStatic = (HDC)wParam;
    hdcStatic.SetTextColor(RGB(0, 0, 0));
    hdcStatic.SetBkMode(TRANSPARENT);
    return (LONG)(LRESULT)((HBRUSH)GetStockObject(NULL_BRUSH));
}

LRESULT CSupportEnterCode::OnEraseBackground(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& /*bHandled*/)
{
    static HPEN Outline = CreatePen(PS_SOLID, 1, 0x00FFFFFF);
    static HBRUSH Fill = CreateSolidBrush(0x00FFFFFF);

    CDCHandle hdc = (HDC)wParam;
    hdc.SelectPen(Outline);
    hdc.SelectBrush(Fill);

    RECT rect;
    GetClientRect(&rect);
    hdc.Rectangle(&rect);
    return TRUE;
}

LRESULT CSupportEnterCode::OnCloseCmd(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/)
{
    EndDialog(wID);
    return TRUE;
}

LRESULT CSupportEnterCode::OnOkCmd(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/)
{
    wchar_t code[50];
    if (!GetDlgItemText(IDC_CODE,code,sizeof(code) /sizeof(code[0])))
    {
        MessageBox(wGS(MSG_SUPPORT_ENTER_SUPPORT_CODE).c_str(), wGS(MSG_SUPPORT_PROJECT64).c_str(), MB_OK);
        return false;
    }
    GetDlgItem(IDOK).EnableWindow(false);
    GetDlgItem(IDCANCEL).EnableWindow(false);

    bool ValidCode = false;
    if (_wcsicmp(code,L"thank you from project64") == 0)
    {
        SetDlgItemText(IDC_CODE, L"");
        CRequestCode RequestWindow(m_Support);
        RequestWindow.ShowOldCodeMsg();
        RequestWindow.DoModal(m_hWnd);
        GetDlgItem(IDOK).EnableWindow(TRUE);
        GetDlgItem(IDCANCEL).EnableWindow(TRUE);
        return TRUE;
    }
    else if (m_Support.ValidateCode(stdstr().FromUTF16(code).c_str()))
    {
        ValidCode = true;
    }
    if (ValidCode)
    {
        MessageBox(wGS(MSG_SUPPORT_COMPLETE).c_str(), wGS(MSG_SUPPORT_PROJECT64).c_str(), MB_OK);
        EndDialog(wID);
    }
    else
    {
        MessageBox(wGS(MSG_SUPPORT_INCORRECT_CODE).c_str(), wGS(MSG_SUPPORT_PROJECT64).c_str(), MB_OK);
        GetDlgItem(IDOK).EnableWindow(TRUE);
        GetDlgItem(IDCANCEL).EnableWindow(TRUE);
    }
    return TRUE;
}

CRequestCode::CRequestCode(CProjectSupport & Support) :
    m_Support(Support),
    m_ShowOldCodeMsg(false)
{
}

void CRequestCode::ShowOldCodeMsg()
{
    m_ShowOldCodeMsg = true;
}

LRESULT CSupportEnterCode::OnRequestCode(WORD /*wNotifyCode*/, WORD /*wID*/, HWND /*hWndCtl*/, BOOL& /*bHandled*/)
{
    CRequestCode RequestWindow(m_Support);
    RequestWindow.DoModal(m_hWnd);
    return 0;
}

LRESULT CRequestCode::OnInitDialog(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/)
{
    if (m_ShowOldCodeMsg)
    {
        SetDlgItemText(IDC_DESCRIPTION, L"We have changed the code to be unique to a machine, please enter the email you used to support Project64 with.");
    }
    return TRUE;
}

LRESULT CRequestCode::OnColorStatic(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& /*bHandled*/)
{
    CDCHandle hdcStatic = (HDC)wParam;
    hdcStatic.SetTextColor(RGB(0, 0, 0));
    hdcStatic.SetBkMode(TRANSPARENT);
    return (LONG)(LRESULT)((HBRUSH)GetStockObject(NULL_BRUSH));
}

LRESULT CRequestCode::OnEraseBackground(UINT /*uMsg*/, WPARAM wParam, LPARAM /*lParam*/, BOOL& /*bHandled*/)
{
    static HPEN Outline = CreatePen(PS_SOLID, 1, 0x00FFFFFF);
    static HBRUSH Fill = CreateSolidBrush(0x00FFFFFF);

    CDCHandle hdc = (HDC)wParam;
    hdc.SelectPen(Outline);
    hdc.SelectBrush(Fill);

    RECT rect;
    GetClientRect(&rect);
    hdc.Rectangle(&rect);
    return TRUE;
}

LRESULT CRequestCode::OnOkCmd(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/)
{
    CWindow EmailWnd(GetDlgItem(IDC_EMAIL));
    int EmailLen = EmailWnd.GetWindowTextLength();
    std::wstring Email;
    Email.resize(EmailLen + 1);
    EmailWnd.GetWindowText((wchar_t *)Email.c_str(), Email.length());
    GetDlgItem(IDOK).EnableWindow(false);
    GetDlgItem(IDCANCEL).EnableWindow(false);
    if (m_Support.RequestCode(stdstr().FromUTF16(Email.c_str()).c_str()))
    {
        MessageBox(wGS(MSG_SUPPORT_REQUESTCODE_SUCCESS).c_str(), wGS(MSG_SUPPORT_REQUESTCODE_TITLE).c_str(), MB_OK);
        EndDialog(wID);
    }
    else
    {
        MessageBox(wGS(MSG_SUPPORT_REQUESTCODE_FAIL).c_str(), wGS(MSG_SUPPORT_REQUESTCODE_TITLE).c_str(), MB_OK);
        GetDlgItem(IDOK).EnableWindow(true);
        GetDlgItem(IDCANCEL).EnableWindow(true);
    }
    return TRUE;
}

LRESULT CRequestCode::OnCloseCmd(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/)
{
    EndDialog(wID);
    return TRUE;
}