#include "PickDllGui.h"

BOOL PickDllGui::OnInitDialog(CWindow wndFocus, LPARAM lInitParam)
{
	ListDLLSelect.Attach(GetDlgItem(IDC_LIST_DLLSELECT));

	addColumnsToModuleList(ListDLLSelect);
	displayModuleList(ListDLLSelect);

	CenterWindow();

	GetClientRect(&MinSize);

	return TRUE;
}

void PickDllGui::OnGetMinMaxInfo(MINMAXINFO* lpMMI)
{
	lpMMI->ptMinTrackSize.x = MinSize.right;
	lpMMI->ptMinTrackSize.y = MinSize.bottom;
}

void PickDllGui::OnSizing(UINT fwSide, RECT* pRect)
{
	int toResize[] = {IDC_LIST_DLLSELECT};
	int toMove[] = {IDC_BTN_PICKDLL_OK, IDC_BTN_PICKDLL_CANCEL};

	// Get size difference
	RECT rectOld;
	GetWindowRect(&rectOld);
	long deltaX = (pRect->right  - pRect->left) - (rectOld.right  - rectOld.left);
	long deltaY = (pRect->bottom - pRect->top)  - (rectOld.bottom - rectOld.top);

	HDWP hdwp = BeginDeferWindowPos(_countof(toResize) + _countof(toMove));

	for(int i = 0; i < _countof(toResize); i++)
	{
		RECT rectControl;
		CWindow control(GetDlgItem(toResize[i]));

		control.GetWindowRect(&rectControl); // Why doesn't GetClientRect work?
		// calculate new width and height
		int cx = rectControl.right - rectControl.left + deltaX;
		int cy = rectControl.bottom - rectControl.top + deltaY;

		control.DeferWindowPos(hdwp, NULL, 0, 0, cx, cy, SWP_NOMOVE | SWP_NOOWNERZORDER);
	}

	for(int i = 0; i < _countof(toMove); i++)
	{
		RECT rectControl;
		CWindow control(GetDlgItem(toMove[i]));
		control.GetClientRect(&rectControl);
		control.MapWindowPoints(m_hWnd, &rectControl);
		control.DeferWindowPos(hdwp, NULL, rectControl.left + deltaX, rectControl.top + deltaY, 0, 0, SWP_NOSIZE | SWP_NOOWNERZORDER);
	}

	EndDeferWindowPos(hdwp);
}

void PickDllGui::OnOK(UINT uNotifyCode, int nID, CWindow wndCtl)
{
	int index = ListDLLSelect.GetSelectionMark();
	if (index != -1)
	{
		selectedModule = &moduleList[index];
		EndDialog(1);
	}
}

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

void PickDllGui::addColumnsToModuleList(CListViewCtrl& list)
{
	list.SetExtendedListViewStyle(LVS_EX_FULLROWSELECT, LVS_EX_FULLROWSELECT);

	list.InsertColumn(COL_PATH, L"Path", LVCFMT_LEFT, 210);
	list.InsertColumn(COL_NAME, L"Name", LVCFMT_CENTER, 130);
	list.InsertColumn(COL_IMAGEBASE, L"ImageBase", LVCFMT_CENTER, 70);
	list.InsertColumn(COL_IMAGESIZE, L"ImageSize", LVCFMT_CENTER, 70);
}

void PickDllGui::displayModuleList(CListViewCtrl& list)
{
	WCHAR temp[20];

	list.DeleteAllItems();

	std::vector<ModuleInfo>::const_iterator iter;
	int count = 0;

	for( iter = moduleList.begin(); iter != moduleList.end(); iter++ , count++)
	{
		list.InsertItem(count, iter->fullPath);

		list.SetItemText(count, COL_NAME, iter->getFilename());

		swprintf_s(temp,_countof(temp),L"%08X",iter->modBaseAddr);
		list.SetItemText(count, COL_IMAGEBASE, temp);

		swprintf_s(temp,_countof(temp),L"%08X",iter->modBaseSize);
		list.SetItemText(count, COL_IMAGESIZE, temp);
	}

	//list.SetColumnWidth(COL_PATH, LVSCW_AUTOSIZE);
	list.SetColumnWidth(COL_NAME, LVSCW_AUTOSIZE);
	//list.SetColumnWidth(COL_IMAGEBASE, LVSCW_AUTOSIZE);

	list.SetColumnWidth(COL_IMAGESIZE, LVSCW_AUTOSIZE_USEHEADER);

	//m_hotkeysListView.SetColumnWidth(1, LVSCW_AUTOSIZE_USEHEADER);
}