#include "GUIMain.h"

#include <Windows.h>

#define AS2CmdGenerate 1
#define AS2CmdQuit     2

GUIMain::GUIMain(HINSTANCE hInstance)
{
	this->hInstance = hInstance;
}

GUIMain::~GUIMain()
{

}

void GUIMain::Register()
{
	WNDCLASSEX wc;

	wc.cbSize        = sizeof(WNDCLASSEX);
	wc.cbClsExtra    = 0;
	wc.cbWndExtra    = 0;
	wc.hInstance     = hInstance;

	wc.lpszClassName = TEXT("AS2MainWindow");
	
	wc.lpfnWndProc   = WndProcA;

	wc.style         = CS_HREDRAW | CS_VREDRAW;
	wc.hbrBackground = (HBRUSH)GetStockObject(COLOR_BACKGROUND);
	
	wc.hCursor       = LoadCursor(0, IDC_ARROW);
	wc.hIcon         = LoadIcon(hInstance, IDI_APPLICATION);
	wc.hIconSm       = LoadIcon(hInstance, IDI_APPLICATION);

	wc.cbWndExtra    = sizeof(GUIMain *);

	wc.lpszMenuName  = 0;

	RegisterClassEx(&wc);
}

bool GUIMain::Create()
{
	hWnd = CreateWindowEx(0, TEXT("AS2MainWindow"),
						  TEXT("AS2"),
						  WS_BORDER | WS_CAPTION | WS_SYSMENU | WS_MINIMIZEBOX,
						  CW_USEDEFAULT, CW_USEDEFAULT,
						  824, 350,
						  0, 0,
						  hInstance, 0);

	return (bool)hWnd;
}

WPARAM GUIMain::Show(int nCmdShow)
{
	ShowWindow(hWnd, nCmdShow);

	UpdateWindow(hWnd);

	MSG msg;
	
	while (GetMessage(&msg, 0, 0, 0))
	{
		TranslateMessage(&msg);
		DispatchMessage(&msg);
	}

	return msg.wParam;
}

LRESULT CALLBACK GUIMain::wndProc(HWND hWnd, UINT Msg, WPARAM wParam, LPARAM lParam)
{
	switch(Msg)
	{
		case WM_COMMAND:
			handleCommand(LOWORD(wParam));
			return 0;
		
		case WM_CREATE:
			initialise(hWnd);
			return 0;
		case WM_DESTROY:
			PostQuitMessage(0);
			return 0;

		default:
			return DefWindowProc(hWnd, Msg, wParam, lParam);
	}

	return 0;
}

LRESULT CALLBACK GUIMain::WndProcA(HWND hWnd, UINT Msg, WPARAM wParam, LPARAM lParam)
{
	if (Msg == WM_NCCREATE)
	{
		SetWindowLong(hWnd, GWL_USERDATA, (long) ((LPCREATESTRUCT)lParam)->lpCreateParams);
		SetWindowLong(hWnd, GWL_WNDPROC,  (long) WndProcB);
	}

	GUIMain *g = (GUIMain *)GetWindowLong(hWnd, GWL_USERDATA);

	if (g)
		return g->wndProc(hWnd, Msg, wParam, lParam);
	else
		return DefWindowProc(hWnd, Msg, wParam, lParam);
}

LRESULT CALLBACK GUIMain::WndProcB(HWND hWnd, UINT Msg, WPARAM wParam, LPARAM lParam)
{
	return ((GUIMain *)GetWindowLong(hWnd, GWL_USERDATA))->wndProc(hWnd, Msg, wParam, lParam);
}

void GUIMain::initialise(HWND hWnd)
{
	this->hWnd = hWnd;

	cmdGenerate = CreateWindowEx(0, TEXT("BUTTON"), TEXT("&Generate..."),
							   WS_VISIBLE | WS_CHILD,
							   6, 6, 150, 25,        
							   hWnd, (HMENU)AS2CmdGenerate, 0, 0);

	cmdQuit     = CreateWindowEx(0, TEXT("BUTTON"), TEXT("&Quit"),
							   WS_VISIBLE | WS_CHILD,
							   6, 37, 150, 25,        
							   hWnd, (HMENU)AS2CmdQuit, 0, 0);
}

void GUIMain::handleCommand(int cmd)
{
	switch (cmd)
	{
		case AS2CmdQuit:
			PostQuitMessage(0);
			return;

		default:
			return;
	}
}