#include <CodeAnalysis\Warnings.h>

#pragma warning(push)
#pragma warning(disable : ALL_CODE_ANALYSIS_WARNINGS)

#include <iostream>
#include <ppl.h>
#include <agents.h>
#include <Windows.h>

LRESULT __stdcall WndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
{
	return DefWindowProc(hwnd, message, wParam, lParam);
}

class MessageAgent : public Concurrency::agent
{
	HWND hWindow;

public:
	MessageAgent(HWND hWindow)
		: hWindow(hWindow)
	{
		
	}

	void run()
	{
		SIZE_T windowThread = GetWindowThreadProcessId(hWindow, nullptr);
		SIZE_T currThread = GetCurrentThreadId();

		std::cout << "Current Thread: " << currThread << "\n";
		std::cout << "Window Thread: " << windowThread << "\n";

		if (AttachThreadInput(currThread, windowThread, TRUE) == FALSE)
			std::cout << "Attach Thread 1 fail\n";

		bool doLoop = true;
		MSG msg;

		while (doLoop)
		{
			while(PeekMessage(&msg, nullptr, 0, 0, PM_REMOVE) != FALSE)
			{
				if (msg.message == WM_QUIT)
				{
					doLoop = false;
				}

				TranslateMessage(&msg);

				DispatchMessage(&msg);
			}
		}

		done();
	}
};

int main()
{
	WNDCLASSEX windowClassEx;

	windowClassEx.cbSize = sizeof(WNDCLASSEX);
	windowClassEx.cbClsExtra = 0;
	windowClassEx.cbWndExtra = 0;
	windowClassEx.style = CS_CLASSDC;
	windowClassEx.hInstance = GetModuleHandle(nullptr);
	windowClassEx.lpfnWndProc = WndProc;
	windowClassEx.lpszMenuName = 0;
	windowClassEx.lpszClassName = L"Window Class Test";
	windowClassEx.hCursor = LoadCursor(nullptr, IDC_ARROW);
	windowClassEx.hIcon = LoadIcon(windowClassEx.hInstance, IDI_APPLICATION);
	windowClassEx.hIconSm = LoadIcon(windowClassEx.hInstance, IDI_APPLICATION);
	windowClassEx.hbrBackground = static_cast<HBRUSH>(GetStockObject(WHITE_BRUSH));

	if (!RegisterClassEx(&windowClassEx))
	{
		std::cout << "Fail to register window class\n";
	}

	SIZE_T windowStyleFlags = WS_SYSMENU | WS_MINIMIZEBOX | WS_MAXIMIZEBOX | WS_OVERLAPPED;

	HWND windowHandle = CreateWindowEx(WS_EX_CLIENTEDGE, L"Window Class Test", L"Parallel Window Test",
		windowStyleFlags , 0, 0, 1024, 768, 0, 0, GetModuleHandle(nullptr), nullptr);

	if (windowHandle == nullptr)
	{
		std::cout << "Fail to create window\n";
	}

	ShowWindow(windowHandle, SW_SHOWNORMAL);
	UpdateWindow(windowHandle);
	SetFocus(windowHandle);

	std::cout << "Current Thread: " << GetCurrentThreadId() << "\n";

	MessageAgent msgAgent(windowHandle);
	msgAgent.start();
	Concurrency::agent::wait(&msgAgent);

	UnregisterClass(L"Window Class Test", GetModuleHandle(nullptr));

	return 0;
}

#pragma warning(pop)