#include "main.h"
#include "../keys.h"

int main(char argv[][], int argc)
{
	HWND hM = NULL;
	do {
		hM = FindWindowA(NULL, "Minesweeper");
	} while (hM == NULL);

	HANDLE outfile = NULL;
	outfile = CreateFile("minemap.txt", GENERIC_WRITE, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
	char newline = '\n';
	
	RECT rM, rM2;
	GetClientRect(hM, &rM);
	GetWindowRect(hM, &rM2);
	POINT pM; pM.x = 0; pM.y = 0;
	ClientToScreen(hM, &pM);
	rM.left = pM.x; rM.top = pM.y;
	MoveMouse(rM.left, rM.top);

	PressKey(hM, KEY_X);
	PressKey(hM, KEY_Y);
	PressKey(hM, KEY_Z);
	PressKey(hM, KEY_Z);
	PressKey(hM, KEY_Y);
	PressKey(hM, VK_SHIFT);

	//Init shit
	int startX = 20, startY = 63;
	int incX = 16, incY = 16;
	int rows = Xatoi(argv[1]);
	int columns = Xatoi(argv[2]);
	int rowIdx = 0, columnIdx = 0;
	int currentIsMine = FALSE;
	int sleepTime = Xatoi(argv[3]);
	int doRealSolve = Xatoi(argv[4]);
	//End Init

	Sleep(1000);

	SetFocus(hM);

	for (; columnIdx < columns; ++columnIdx)
	{
		for (; rowIdx < rows; ++rowIdx)
		{
			if (GetAsyncKeyState(VK_END)) { break; }
			MoveMouseInWindow(hM, startX + (rowIdx*incX), startY + (columnIdx*incY));
			ClickMouse(3, 10); //*** Pixel doesnt change without this ***
			currentIsMine = IsMine();
			WriteFile(outfile, &currentIsMine, sizeof(currentIsMine), NULL, NULL);
			if (doRealSolve && !currentIsMine) { ClickMouse(1, 5); }
			Sleep(sleepTime);
		}
		if (GetAsyncKeyState(VK_END)) { break; }
		WriteFile(outfile, &newline, sizeof(newline), NULL, NULL);
		rowIdx = 0;
	}

	CloseHandle(outfile);
	return 0;
}

int IsMine()
{
	HDC dc = GetDC(0);
	COLORREF pixel = GetPixel(dc, 0, 0);
	BYTE r = GetRValue(pixel);
	BYTE g = GetGValue(pixel);
	BYTE b = GetBValue(pixel);
	if (r==255 && g == 255 && b == 255)
	{
		return FALSE;
	}
	return TRUE;
}

void ClickMouse(int btn, int sleep)
{
	switch(btn)
	{
		case 1:
			mouse_event(MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0);
			Sleep(sleep);
			mouse_event(MOUSEEVENTF_LEFTUP, 0, 0, 0, 0);
		case 2:
			mouse_event(MOUSEEVENTF_RIGHTDOWN, 0, 0, 0, 0);
			Sleep(sleep);
			mouse_event(MOUSEEVENTF_RIGHTUP, 0, 0, 0, 0);
		case 3:
			mouse_event(MOUSEEVENTF_MIDDLEDOWN, 0, 0, 0, 0);
			Sleep(sleep);
			mouse_event(MOUSEEVENTF_MIDDLEUP, 0, 0, 0, 0);
	}
}

void MoveMouseInWindow(HWND window, int x, int y)
{
	POINT pM; pM.x = x; pM.y = y;
	ClientToScreen(window, &pM);
	SetCursorPos((int)pM.x, (int)pM.y);
}

void MoveMouse(int x, int y)
{
	POINT moveTo;
	moveTo.x = x;
	moveTo.y = y;
	SetCursorPos((int)moveTo.x, (int)moveTo.y);
}

void PressKey(HWND window, BYTE key)
{
	PostMessage(window, WM_KEYDOWN, key, 1);
}

int Xatoi(char const *str)
{
	int result = 0;
	while( char c = *str++ )
		result = result*10 + (c-'0');
	return result;
}

void Xitoa(int n, char s[])
{
	int i, sign;

	if ((sign = n) < 0)  /* record sign */
	n = -n;          /* make n positive */
	i = 0;
	do {       /* generate digits in reverse order */
		s[i++] = n % 10 + '0';   /* get next digit */
	} while ((n /= 10) > 0);     /* delete it */
	if (sign < 0)
		s[i++] = '-';
	s[i] = '\0';
	Xreverse(s);
}
void Xreverse(char s[])
{
	int i, j;
	char c;

	for (i = 0, j = strlen(s)-1; i<j; i++, j--) {
		c = s[i];
		s[i] = s[j];
		s[j] = c;
	}
}