#include "PickApiGui.h"

#include <atlconv.h> // string conversion

PickApiGui::PickApiGui(const std::vector<ModuleInfo> &moduleList) : moduleList(moduleList)
{
	selectedApi = 0;
}

BOOL PickApiGui::OnInitDialog(CWindow wndFocus, LPARAM lInitParam)
{
	DoDataExchange(); // attach controls
	DlgResize_Init(true, true);
	
	fillDllComboBox(ComboDllSelect);

	CenterWindow();
	return TRUE;
}

void PickApiGui::OnOK(UINT uNotifyCode, int nID, CWindow wndCtl)
{
	actionApiSelected();
}

void PickApiGui::OnCancel(UINT uNotifyCode, int nID, CWindow wndCtl)
{
	EndDialog(0);
}

void PickApiGui::OnDllListSelected(UINT uNotifyCode, int nID, CWindow wndCtl)
{
	int indexDll = ComboDllSelect.GetCurSel();
	if (indexDll != CB_ERR)
	{
		fillApiListBox(ListApiSelect, moduleList[indexDll].apiList);
		EditApiFilter.SetWindowText(L"");
	}
}

void PickApiGui::OnApiListDoubleClick(UINT uNotifyCode, int nID, CWindow wndCtl)
{
	actionApiSelected();
}

void PickApiGui::OnApiFilterUpdated(UINT uNotifyCode, int nID, CWindow wndCtl)
{
	int indexDll = ComboDllSelect.GetCurSel();
	if (indexDll == CB_ERR)
		return;

	std::vector<ApiInfo *> newApis;
	WCHAR filter[MAX_PATH];

	int lenFilter = EditApiFilter.GetWindowText(filter, _countof(filter));
	if(lenFilter > 0)
	{
		const std::vector<ApiInfo *> &apis = moduleList[indexDll].apiList;

		for (size_t i = 0; i < apis.size(); i++)
		{
			ApiInfo* api = apis[i];
			if(api->name[0] != '\0')
			{
				CA2WEX<MAX_PATH> wStr(api->name);
				if(!_wcsnicmp(wStr, filter, lenFilter))
				{
					newApis.push_back(api);
				}
			}
			else
			{
				WCHAR buf[6];
				swprintf_s(buf, L"#%04X", api->ordinal);
				if(!_wcsnicmp(buf, filter, lenFilter))
				{
					newApis.push_back(api);
				}
			}
		}
	}
	else
	{
		newApis = moduleList[indexDll].apiList;
	}

	fillApiListBox(ListApiSelect, newApis);
}

void PickApiGui::actionApiSelected()
{
	int indexDll = ComboDllSelect.GetCurSel();
	int indexApi;
	if(ListApiSelect.GetCount() == 1)
	{
		indexApi = 0;
	}
	else
	{
		indexApi = ListApiSelect.GetCurSel();
	}
	if (indexDll != CB_ERR && indexApi != LB_ERR)
	{
		selectedApi = (ApiInfo *)ListApiSelect.GetItemData(indexApi);
		EndDialog(1);
	}
}

void PickApiGui::fillDllComboBox(CComboBox& combo)
{
	combo.ResetContent();

	for (size_t i = 0; i < moduleList.size(); i++)
	{
		combo.AddString(moduleList[i].fullPath);
	}
}

void PickApiGui::fillApiListBox(CListBox& list, const std::vector<ApiInfo *> &apis)
{
	list.ResetContent();

	for (size_t i = 0; i < apis.size(); i++)
	{
		const ApiInfo* api = apis[i];
		int item;
		if(api->name[0] != '\0')
		{
			CA2WEX<MAX_PATH> wStr(api->name);
			item = list.AddString(wStr);
		}
		else
		{
			WCHAR buf[6];
			swprintf_s(buf, L"#%04X", api->ordinal);
			item = list.AddString(buf);
		}
		list.SetItemData(item, (DWORD_PTR)api);
	}
}