/****************************************************************************
*                                                                           *
* Project 64 - A Nintendo 64 emulator.                                      *
* http://www.pj64-emu.com/                                                  *
* Copyright (C) 2012 Project64. All rights reserved.                        *
*                                                                           *
* License:                                                                  *
* GNU/GPLv2 http://www.gnu.org/licenses/gpl-2.0.html                        *
*                                                                           *
****************************************************************************/
#include "stdafx.h"

#include "Settings Page.h"

COptionsShortCutsPage::COptionsShortCutsPage(HWND hParent, const RECT & rcDispay) :
m_EnableReset(false)
{
	if (!Create(hParent, rcDispay))
	{
		return;
	}

	SetDlgItemTextW(m_hWnd, IDC_S_CPU_STATE, GS(ACCEL_CPUSTATE_TITLE));
	SetDlgItemTextW(m_hWnd, IDC_MENU_ITEM_TEXT, GS(ACCEL_MENUITEM_TITLE));
	SetDlgItemTextW(m_hWnd, IDC_S_CURRENT_KEYS, GS(ACCEL_CURRENTKEYS_TITLE));
	SetDlgItemTextW(m_hWnd, IDC_S_SELECT_SHORT, GS(ACCEL_SELKEY_TITLE));
	SetDlgItemTextW(m_hWnd, IDC_S_CURRENT_ASSIGN, GS(ACCEL_ASSIGNEDTO_TITLE));
	SetDlgItemTextW(m_hWnd, IDC_ASSIGN, GS(ACCEL_ASSIGN_BTN));
	SetDlgItemTextW(m_hWnd, IDC_REMOVE, GS(ACCEL_REMOVE_BTN));
	SetDlgItemTextW(m_hWnd, IDC_KEY_PROMPT, GS(ACCEL_DETECTKEY));

	m_CreateNewShortCut.AttachToDlgItem(m_hWnd, IDC_S_SELECT_SHORT);
	m_CpuState.Attach(GetDlgItem(IDC_C_CPU_STATE));
	m_MenuItems.Attach(GetDlgItem(IDC_MENU_ITEMS));
	m_CurrentKeys.Attach(GetDlgItem(IDC_CURRENT_KEYS));
	m_VirtualKeyList.Attach(GetDlgItem(IDC_VIRTUALKEY));

	m_MenuItems.ModifyStyle(0, TVS_SHOWSELALWAYS);

	m_CpuState.SetItemData(m_CpuState.AddStringW(GS(ACCEL_CPUSTATE_1)), CMenuShortCutKey::GAME_NOT_RUNNING);
	m_CpuState.SetItemData(m_CpuState.AddStringW(GS(ACCEL_CPUSTATE_3)), CMenuShortCutKey::GAME_RUNNING_WINDOW);
	m_CpuState.SetItemData(m_CpuState.AddStringW(GS(ACCEL_CPUSTATE_4)), CMenuShortCutKey::GAME_RUNNING_FULLSCREEN);
	m_CpuState.SetCurSel(0);

	int VirtualKeyListSize;
	VIRTUAL_KEY * VirtualKeyList = CMenuShortCutKey::VirtualKeyList(VirtualKeyListSize);
	for (int count = 0; count < VirtualKeyListSize; count++)
	{
		m_VirtualKeyList.SetItemData(m_VirtualKeyList.AddString(VirtualKeyList[count].Name), VirtualKeyList[count].Key);
	}

	OnCpuStateChanged(LBN_SELCHANGE, IDC_C_CPU_STATE, GetDlgItem(IDC_C_CPU_STATE));
	CheckResetEnable();
}

void COptionsShortCutsPage::CheckResetEnable(void)
{
	MSC_MAP & ShortCuts = m_ShortCuts.GetShortCuts();
	for (MSC_MAP::iterator Item = ShortCuts.begin(); Item != ShortCuts.end(); Item++)
	{
		const SHORTCUT_KEY_LIST & ShortCutList = Item->second.GetAccelItems();
		for (SHORTCUT_KEY_LIST::const_iterator ShortCut_item = ShortCutList.begin(); ShortCut_item != ShortCutList.end(); ShortCut_item++)
		{
			if (!ShortCut_item->Inactive() && !ShortCut_item->UserAdded())
			{
				continue;
			}
			m_EnableReset = true;
			return;
		}
	}
	m_EnableReset = false;
}

void COptionsShortCutsPage::OnCpuStateChanged(UINT /*Code*/, int /*id*/, HWND /*ctl*/)
{
	ACCESS_MODE AccessLevel = (ACCESS_MODE)m_CpuState.GetItemData(m_CpuState.GetCurSel());

	MSC_MAP & ShortCuts = m_ShortCuts.GetShortCuts();
	m_MenuItems.DeleteAllItems();

	for (MSC_MAP::iterator Item = ShortCuts.begin(); Item != ShortCuts.end(); Item++)
	{
		ACCESS_MODE ItemMode = Item->second.AccessMode();
		if ((ItemMode & AccessLevel) != AccessLevel)
		{
			continue;
		}
		//find Parent
		HTREEITEM hParent = m_MenuItems.GetChildItem(TVI_ROOT);
		while (hParent)
		{
			if (m_MenuItems.GetItemData(hParent) == (DWORD_PTR)Item->second.Section())
			{
				break;
			}
			hParent = m_MenuItems.GetNextSiblingItem(hParent);
		}

		if (hParent == NULL)
		{
			hParent = m_MenuItems.InsertItemW(TVIF_TEXT | TVIF_PARAM, GS(Item->second.Section()), 0, 0, 0, 0, Item->second.Section(), TVI_ROOT, TVI_LAST);
		}

		wstring str = GS(Item->second.Title());
		std::wstring::size_type pos = str.find(L"&");
		while (pos != std::wstring::npos)
		{
			str.replace(pos, 1, L"");
			pos = str.find(L"&", pos);
		}
		pos = str.find(L"...");
		while (pos != std::wstring::npos)
		{
			str.replace(pos, 3, L"");
			pos = str.find(L"...", pos);
		}

		HTREEITEM hItem = m_MenuItems.InsertItemW(TVIF_TEXT | TVIF_PARAM, str.c_str(), 0, 0, 0, 0, (DWORD_PTR)&Item->second, hParent, TVI_LAST);

		const SHORTCUT_KEY_LIST & ShortCutList = Item->second.GetAccelItems();
		for (SHORTCUT_KEY_LIST::const_iterator ShortCut_item = ShortCutList.begin(); ShortCut_item != ShortCutList.end(); ShortCut_item++)
		{
			if (!ShortCut_item->Inactive() && !ShortCut_item->UserAdded())
			{
				continue;
			}
			m_MenuItems.SetItemState(hItem, TVIS_BOLD, TVIS_BOLD);
			m_MenuItems.SetItemState(hParent, TVIS_BOLD, TVIS_BOLD);
			break;
		}
	}
}

void COptionsShortCutsPage::OnRemoveClicked(UINT /*Code*/, int /*id*/, HWND /*ctl*/)
{
	HTREEITEM hSelectedItem = m_MenuItems.GetSelectedItem();
	if (hSelectedItem == NULL)
	{
		g_Notify->DisplayError(GS(MSG_NO_SEL_SHORTCUT));
		return;
	}
	HTREEITEM hParent = m_MenuItems.GetParentItem(hSelectedItem);
	if (hParent == NULL)
	{
		g_Notify->DisplayError(GS(MSG_NO_SEL_SHORTCUT));
		return;
	}

	CShortCutItem * ShortCut = (CShortCutItem *)m_MenuItems.GetItemData(hSelectedItem);

	//Make sure an item is selected
	int index = m_CurrentKeys.GetCurSel();
	if (index < 0)
	{
		g_Notify->DisplayError(GS(MSG_NO_SEL_SHORTCUT));
		return;
	}
	ShortCut->RemoveItem((CMenuShortCutKey *)m_CurrentKeys.GetItemData(index));
	m_MenuItems.SetItemState(hSelectedItem, TVIS_BOLD, TVIS_BOLD);
	m_MenuItems.SetItemState(hParent, TVIS_BOLD, TVIS_BOLD);
	m_EnableReset = true;

	RefreshShortCutOptions(hSelectedItem);
	SendMessage(GetParent(), PSM_CHANGED, (WPARAM)m_hWnd, 0);
}

void COptionsShortCutsPage::OnDetectKeyClicked(UINT /*Code*/, int /*id*/, HWND /*ctl*/)
{
	CloseHandle(CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)stInputGetKeys, this, 0, NULL));
}

void COptionsShortCutsPage::OnAssignClicked(UINT /*Code*/, int /*id*/, HWND /*ctl*/)
{
	//Get the virtual key info
	int index = m_VirtualKeyList.GetCurSel();
	if (index < 0)
	{
		g_Notify->DisplayError(GS(MSG_NO_SHORTCUT_SEL));
		return;
	}

	WORD key = (WORD)SendDlgItemMessage(IDC_VIRTUALKEY, CB_GETITEMDATA, index, 0);
	bool bCtrl = (SendDlgItemMessage(IDC_CTL, BM_GETCHECK, 0, 0) == BST_CHECKED);
	bool bAlt = (SendDlgItemMessage(IDC_ALT, BM_GETCHECK, 0, 0) == BST_CHECKED);
	bool bShift = (SendDlgItemMessage(IDC_SHIFT, BM_GETCHECK, 0, 0) == BST_CHECKED);

	ACCESS_MODE AccessLevel = (ACCESS_MODE)m_CpuState.GetItemData(m_CpuState.GetCurSel());

	HTREEITEM hSelectedItem = m_MenuItems.GetSelectedItem();
	if (hSelectedItem == NULL)
	{
		g_Notify->DisplayError(GS(MSG_NO_MENUITEM_SEL));
		return;
	}
	HTREEITEM hParent = m_MenuItems.GetParentItem(hSelectedItem);
	if (hParent == NULL)
	{
		g_Notify->DisplayError(GS(MSG_NO_MENUITEM_SEL));
		return;
	}

	CShortCutItem * ShortCut = (CShortCutItem *)m_MenuItems.GetItemData(hSelectedItem);

	LanguageStringID strid = m_ShortCuts.GetMenuItemName(key, bCtrl, bAlt, bShift, AccessLevel);
	if (strid != EMPTY_STRING)
	{
		g_Notify->DisplayError(GS(MSG_MENUITEM_ASSIGNED));
		return;
	}
	ShortCut->AddShortCut(key, bCtrl, bAlt, bShift, AccessLevel, true, false);
	m_MenuItems.SetItemState(hSelectedItem, TVIS_BOLD, TVIS_BOLD);
	m_MenuItems.SetItemState(hParent, TVIS_BOLD, TVIS_BOLD);
	m_EnableReset = true;

	RefreshShortCutOptions(hSelectedItem);
	SendMessage(GetParent(), PSM_CHANGED, (WPARAM)m_hWnd, 0);
}

void COptionsShortCutsPage::OnShortCutChanged(UINT /*Code*/, int /*id*/, HWND /*ctl*/)
{
	//Get the virtual key info
	int index = m_VirtualKeyList.GetCurSel();
	if (index < 0) { return; }
	WORD key = (WORD)m_VirtualKeyList.GetItemData(index);
	bool bCtrl = (SendDlgItemMessage(IDC_CTL, BM_GETCHECK, 0, 0) == BST_CHECKED);
	bool bAlt = (SendDlgItemMessage(IDC_ALT, BM_GETCHECK, 0, 0) == BST_CHECKED);
	bool bShift = (SendDlgItemMessage(IDC_SHIFT, BM_GETCHECK, 0, 0) == BST_CHECKED);

	ACCESS_MODE AccessLevel = (ACCESS_MODE)m_CpuState.GetItemData(m_CpuState.GetCurSel());

	stdstr str;
	str.FromUTF16(GS(m_ShortCuts.GetMenuItemName(key, bCtrl, bAlt, bShift, AccessLevel)));
	if (str.length() > 0)
	{
		str.resize(std::remove(str.begin(), str.end(), '&') - str.begin());
	}
	else
	{
		str = "None";
	}
	SetDlgItemText(IDC_ASSIGNED_MENU_ITEM, str.c_str());
}

LRESULT COptionsShortCutsPage::OnMenuItemChanged(LPNMHDR lpnmh)
{
	RefreshShortCutOptions(((LPNMTREEVIEW)lpnmh)->itemNew.hItem);
	return true;
}

void COptionsShortCutsPage::RefreshShortCutOptions(HTREEITEM hItem)
{
	HTREEITEM hParent = m_MenuItems.GetParentItem(hItem);
	if (hParent == NULL)
	{
		return;
	}

	ACCESS_MODE AccessLevel = (ACCESS_MODE)m_CpuState.GetItemData(m_CpuState.GetCurSel());
	CShortCutItem * ShortCut = (CShortCutItem *)m_MenuItems.GetItemData(hItem);

	m_CurrentKeys.ResetContent();

	const SHORTCUT_KEY_LIST & ShortCutList = ShortCut->GetAccelItems();
	for (SHORTCUT_KEY_LIST::const_iterator ShortCut_item = ShortCutList.begin(); ShortCut_item != ShortCutList.end(); ShortCut_item++)
	{
		if (ShortCut_item->Inactive())
		{
			continue;
		}

		ACCESS_MODE ItemMode = ShortCut_item->AccessMode();
		if ((ItemMode & AccessLevel) != AccessLevel)
		{
			continue;
		}
		stdstr Name = ShortCut_item->Name();
		m_CurrentKeys.SetItemData(m_CurrentKeys.AddString(Name.c_str()), (DWORD_PTR)&*ShortCut_item);
	}
}

BOOL CALLBACK KeyPromptDlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM /*lParam*/)
{
	switch (uMsg)
	{
	case WM_INITDIALOG:
		break;
	case WM_COMMAND:
		switch (LOWORD(wParam))
		{
		case IDCANCEL:
			SetForegroundWindow(GetParent(hDlg));
			DestroyWindow(hDlg);
			break;
		}
		break;
	default:
		return FALSE;
	}
	return TRUE;
}

void COptionsShortCutsPage::InputGetKeys(void)
{
	HWND hKeyDlg = CreateDialogParam(GetModuleHandle(NULL), MAKEINTRESOURCE(IDD_Key_Prompt), m_hWnd, (DLGPROC)KeyPromptDlgProc, (LPARAM)::GetDlgItem(m_hWnd, IDC_VIRTUALKEY));
	::EnableWindow(GetParent(), false);
	MSG msg;

	for (bool fDone = false; !fDone; MsgWaitForMultipleObjects(0, NULL, false, 45, QS_ALLINPUT)) {
		while (PeekMessage(&msg, 0, 0, 0, PM_REMOVE)) {
			if (msg.message == WM_QUIT) {
				fDone = true;
				::PostMessage(NULL, WM_QUIT, 0, 0);
				break;
			}
			if (msg.message == WM_KEYDOWN || msg.message == WM_SYSKEYDOWN) {
				int nVirtKey = (int)msg.wParam;
				if (nVirtKey == VK_SHIFT) { continue; }
				if (nVirtKey == VK_CONTROL) { continue; }
				if (nVirtKey == VK_MENU) { continue; }
				SendDlgItemMessage(IDC_VIRTUALKEY, CB_SETCURSEL, (WPARAM)-1, 0);
				for (int count = 0; count < SendDlgItemMessage(IDC_VIRTUALKEY, CB_GETCOUNT, 0, 0); count++) {
					int Data = (int)SendDlgItemMessage(IDC_VIRTUALKEY, CB_GETITEMDATA, count, 0);
					if (Data != nVirtKey) { continue; }
					SendDlgItemMessage(IDC_VIRTUALKEY, CB_SETCURSEL, count, 0);
					SendDlgItemMessage(IDC_CTL, BM_SETCHECK, (GetKeyState(VK_CONTROL) & 0x80) != 0 ? BST_CHECKED : BST_UNCHECKED, 0);
					SendDlgItemMessage(IDC_ALT, BM_SETCHECK, (GetKeyState(VK_MENU) & 0x80) != 0 ? BST_CHECKED : BST_UNCHECKED, 0);
					SendDlgItemMessage(IDC_SHIFT, BM_SETCHECK, (GetKeyState(VK_SHIFT) & 0x80) != 0 ? BST_CHECKED : BST_UNCHECKED, 0);
					SendMessage(WM_COMMAND, MAKELPARAM(IDC_VIRTUALKEY, LBN_SELCHANGE), (LPARAM)::GetDlgItem(m_hWnd, IDC_VIRTUALKEY));
					SetForegroundWindow(GetParent());
					::DestroyWindow(hKeyDlg);
				}
				continue;
			}
			if (!::IsDialogMessage(hKeyDlg, &msg)) {
				TranslateMessage(&msg);
				DispatchMessage(&msg);
			}
		}

		if (!::IsWindow(hKeyDlg)) { fDone = true; }
	}
	::SetFocus(GetParent());
	::EnableWindow(GetParent(), true);
}

void COptionsShortCutsPage::HidePage()
{
	ShowWindow(SW_HIDE);
}

void COptionsShortCutsPage::ShowPage()
{
	ShowWindow(SW_SHOW);
}

void COptionsShortCutsPage::ApplySettings(bool /*UpdateScreen*/)
{
	m_ShortCuts.Save();
	g_Settings->SaveBool(Info_ShortCutsChanged, true);
}

bool COptionsShortCutsPage::EnableReset(void)
{
	return m_EnableReset;
}

void COptionsShortCutsPage::ResetPage()
{
	m_EnableReset = false;
	m_ShortCuts.Load(true);
	OnCpuStateChanged(LBN_SELCHANGE, IDC_C_CPU_STATE, GetDlgItem(IDC_C_CPU_STATE));
	SendMessage(GetParent(), PSM_CHANGED, (WPARAM)m_hWnd, 0);
	m_CurrentKeys.ResetContent();
	CSettingsPageImpl<COptionsShortCutsPage>::ResetPage();
}