/*  ZeroPAD - author: zerofrog(@gmail.com)
 *  Copyright (C) 2006-2007
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include <string.h>
#include <windows.h>
#include <windowsx.h>
#include <commctrl.h>

#include "resource.h"
#include "../zeropad.h"

#include <pthread.h>
#include <string>

using namespace std;

HINSTANCE hInst = NULL;
static pthread_spinlock_t s_mutexStatus;
static u32 s_keyPress[2], s_keyRelease[2];
extern u16 status[2];

extern string s_strIniPath;
LRESULT WINAPI PADwndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam);
WNDPROC GSwndProc = NULL;
HWND GShwnd = NULL;

extern keyEvent event;


void SaveConfig()
{
	char *szTemp;
	char szIniFile[256], szValue[256], szProf[256];
	int i, j;

	GetModuleFileName(GetModuleHandle((LPCSTR)hInst), szIniFile, 256);
	szTemp = strrchr(szIniFile, '\\');

	if (!szTemp) return;
	strcpy(szTemp, "\\inis\\zeropad.ini");

	for (j = 0; j < 2 * PADSUBKEYS; j++)
	{
		for (i = 0; i < PADKEYS; i++)
		{
			sprintf(szProf, "%d_%d", j, i);
			sprintf(szValue, "%d", conf.keys[j][i]);
			WritePrivateProfileString("Interface", szProf, szValue, szIniFile);
		}
	}

	sprintf(szValue, "%u", conf.log);
	WritePrivateProfileString("Interface", "Logging", szValue, szIniFile);
}

void LoadConfig()
{
	FILE *fp;
	char *szTemp;
	char szIniFile[256], szValue[256], szProf[256];
	int i, j;

	memset(&conf, 0, sizeof(conf));
#ifdef _WIN32
	conf.keys[0][0] = 'W';			// L2
	conf.keys[0][1] = 'O';			// R2
	conf.keys[0][2] = 'A';			// L1
	conf.keys[0][3] = ';';			// R1
	conf.keys[0][4] = 'I';			// TRIANGLE
	conf.keys[0][5] = 'L';			// CIRCLE
	conf.keys[0][6] = 'K';			// CROSS
	conf.keys[0][7] = 'J';			// SQUARE
	conf.keys[0][8] = 'V';	// SELECT
	conf.keys[0][11] = 'N';	// START
	conf.keys[0][12] = 'E';	// UP
	conf.keys[0][13] = 'F';	// RIGHT
	conf.keys[0][14] = 'D';	// DOWN
	conf.keys[0][15] = 'S';	// LEFT
#endif
	conf.log = 0;

	GetModuleFileName(GetModuleHandle((LPCSTR)hInst), szIniFile, 256);
	szTemp = strrchr(szIniFile, '\\');

	if (!szTemp) return ;
	strcpy(szTemp, "\\inis\\zeropad.ini");
	fp = fopen("inis\\zeropad.ini", "rt");//check if usbnull.ini really exists
	if (!fp)
	{
		CreateDirectory("inis", NULL);
		SaveConfig();//save and return
		return ;
	}
	fclose(fp);

	for (j = 0; j < 2 * PADSUBKEYS; j++)
	{
		for (i = 0; i < PADKEYS; i++)
		{
			sprintf(szProf, "%d_%d", j, i);
			GetPrivateProfileString("Interface", szProf, NULL, szValue, 20, szIniFile);
			conf.keys[j][i] = strtoul(szValue, NULL, 10);
		}
	}

	GetPrivateProfileString("Interface", "Logging", NULL, szValue, 20, szIniFile);
	conf.log = strtoul(szValue, NULL, 10);
}

void SysMessage(char *fmt, ...)
{
	va_list list;
	char tmp[512];

	va_start(list, fmt);
	vsprintf(tmp, fmt, list);
	va_end(list);
	MessageBox(0, tmp, "PADwinKeyb Msg", 0);
}

s32  _PADopen(void *pDsp)
{
	memset(&event, 0, sizeof(event));
	LoadConfig();
	pthread_spin_init(&s_mutexStatus, PTHREAD_PROCESS_PRIVATE);
	s_keyPress[0] = s_keyPress[1] = 0;
	s_keyRelease[0] = s_keyRelease[1] = 0;

	if (GShwnd != NULL && GSwndProc != NULL)
	{
		// revert
		SetWindowLongPtr(GShwnd, GWLP_WNDPROC, (LPARAM)(WNDPROC)(GSwndProc));
	}

	GShwnd = (HWND) * (long*)pDsp;
	GSwndProc = (WNDPROC)GetWindowLongPtr(GShwnd, GWLP_WNDPROC);
	GSwndProc = ((WNDPROC)SetWindowLongPtr(GShwnd, GWLP_WNDPROC, (LPARAM)(WNDPROC)(PADwndProc)));

	return 0;
}

void _PADclose()
{
	if (GShwnd != NULL && GSwndProc != NULL)
	{
		SetWindowLongPtr(GShwnd, GWLP_WNDPROC, (LPARAM)(WNDPROC)(GSwndProc));
		GSwndProc = NULL;
		GShwnd = NULL;
	}
	pthread_spin_destroy(&s_mutexStatus);
}

void _PADupdate(int pad)
{
	pthread_spin_lock(&s_mutexStatus);
	status[pad] |= s_keyRelease[pad];
	status[pad] &= ~s_keyPress[pad];
	s_keyRelease[pad] = 0;
	s_keyPress[pad] = 0;
	pthread_spin_unlock(&s_mutexStatus);
}

void CALLBACK PADupdate(int pad)
{
}

LRESULT WINAPI PADwndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
	int i, pad, keyPress[2] = {0}, keyRelease[2] = {0};
	static bool lbutton = false, rbutton = false;

	switch (msg)
	{
		case WM_KEYDOWN:
			if (lParam & 0x40000000)
				return TRUE;

			i = FindKey(wParam, pad);
			if (i != -1) {
				keyPress[pad] |= (1 << i);
				keyRelease[pad] &= ~(1 << i);
			}
			event.evt = KEYPRESS;
			event.key = wParam;
			break;

		case WM_KEYUP:
			i = FindKey(wParam, pad);
			if (i != -1) {
				keyPress[pad] &= ~(1 << i);
				keyRelease[pad] |= (1 << i);
			}
			event.evt = KEYRELEASE;
			event.key = wParam;
			break;

		case WM_LBUTTONDOWN:
			lbutton = true;
			break;

		case WM_LBUTTONUP:
			g_lanalog[0].x = 0x80;
			g_lanalog[0].y = 0x80;
			g_lanalog[1].x = 0x80;
			g_lanalog[1].y = 0x80;
			lbutton = false;
			break;

		case WM_RBUTTONDOWN:
			rbutton = true;
			break;

		case WM_RBUTTONUP:
			g_ranalog[0].x = 0x80;
			g_ranalog[0].y = 0x80;
			g_ranalog[1].x = 0x80;
			g_ranalog[1].y = 0x80;
			rbutton = false;
			break;

		case WM_MOUSEMOVE:
			if (lbutton)
			{
				g_lanalog[0].x = LOWORD(lParam) & 254;
				g_lanalog[0].y = HIWORD(lParam) & 254;
				g_lanalog[1].x = LOWORD(lParam) & 254;
				g_lanalog[1].y = HIWORD(lParam) & 254;
			}
			if (rbutton)
			{
				g_ranalog[0].x = LOWORD(lParam) & 254;
				g_ranalog[0].y = HIWORD(lParam) & 254;
				g_ranalog[1].x = LOWORD(lParam) & 254;
				g_ranalog[1].y = HIWORD(lParam) & 254;
			}
			break;

		case WM_DESTROY:
		case WM_QUIT:
			event.evt = KEYPRESS;
			event.key = VK_ESCAPE;
			return GSwndProc(hWnd, msg, wParam, lParam);

		default:
			return GSwndProc(hWnd, msg, wParam, lParam);
	}

	pthread_spin_lock(&s_mutexStatus);
	for (pad = 0; pad < 2; ++pad)
	{
		s_keyPress[pad] |= keyPress[pad];
		s_keyPress[pad] &= ~keyRelease[pad];
		s_keyRelease[pad] |= keyRelease[pad];
		s_keyRelease[pad] &= ~keyPress[pad];
	}
	pthread_spin_unlock(&s_mutexStatus);

	return TRUE;
}

string GetKeyLabel(const int pad, const int index)
{
	const int key = conf.keys[pad][index];
	char buff[16] = "NONE)";
	
	if (key < 0x100)
	{
		if (key == 0)
			strcpy(buff, "NONE");
		else
		{
			if (key >= 0x60 && key <= 0x69)
				sprintf(buff, "NumPad %c", '0' + key - 0x60);
			else 
				sprintf(buff, "%c", key);
		}
	}
	else if (key >= 0x1000 && key < 0x2000)
	{
		sprintf(buff, "J%d_%d", (key & 0xfff) / 0x100, (key & 0xff) + 1);
	}
	else if (key >= 0x2000 && key < 0x3000)
	{
		static const char name[][4] = { "MIN", "MAX" };
		const int axis = (key & 0xff);
		
		sprintf(buff, "J%d_AXIS%d_%s", (key & 0xfff) / 0x100, axis / 2, name[axis % 2]);
		if (index >= 17 && index <= 20) buff[strlen(buff) -4] = '\0';
	}
	else if (key >= 0x3000 && key < 0x4000)
	{
		static const char name[][7] = { "FOWARD", "RIGHT", "BACK", "LEFT" };
		const int pov = (key & 0xff);
		sprintf(buff, "J%d_POV%d_%s", (key & 0xfff) / 0x100, pov / 4, name[pov % 4]);
	}

	return buff;
}

BOOL CALLBACK ConfigureDlgProc(HWND hW, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	HWND hWC;
	TCITEM tcI;
	int i, key, numkeys;
	u8* pkeyboard;
	static int disabled = 0;
	static int padn = 0;

	switch (uMsg)
	{
		case WM_INITDIALOG:
			LoadConfig();
			padn = 0;
			if (conf.log) CheckDlgButton(hW, IDC_LOG, TRUE);

			for (i = 0; i < PADKEYS; i++)
			{
				hWC = GetDlgItem(hW, IDC_L2 + i * 2);
				Button_SetText(hWC, GetKeyLabel(padn, i).c_str());
			}

			hWC = GetDlgItem(hW, IDC_TABC);

			tcI.mask = TCIF_TEXT;
			tcI.pszText = "PAD 1";

			TabCtrl_InsertItem(hWC, 0, &tcI);

			tcI.mask = TCIF_TEXT;
			tcI.pszText = "PAD 2";

			TabCtrl_InsertItem(hWC, 1, &tcI);
			return TRUE;

		case WM_TIMER:
			if (disabled)
			{
				key = 0;
				//pkeyboard = SDL_GetKeyState(&numkeys);
				for (int i = 0; i < numkeys; ++i)
				{
					if (pkeyboard[i])
					{
						key = i;
						break;
					}
				}

				if (key == 0)
				{
					// check joystick
				}

				if (key != 0)
				{
					KillTimer(hW, 0x80);
					hWC = GetDlgItem(hW, disabled);
					conf.keys[padn][disabled-IDC_L2] = key;
					Button_SetText(hWC, GetKeyLabel(padn, disabled - IDC_L2).c_str());
					EnableWindow(hWC, TRUE);
					disabled = 0;
					return TRUE;
				}
			}
			return TRUE;

		case WM_COMMAND:
			for (i = IDC_L2; i <= IDC_LEFT; i += 2)
			{
				if (LOWORD(wParam) == i)
				{
					if (disabled)//change selection
						EnableWindow(GetDlgItem(hW, disabled), TRUE);

					EnableWindow(GetDlgItem(hW, disabled = wParam), FALSE);

					SetTimer(hW, 0x80, 250, NULL);

					return TRUE;
				}
			}

			switch (LOWORD(wParam))
			{
				case IDCANCEL:
					KillTimer(hW, 0x80);
					EndDialog(hW, TRUE);
					return TRUE;
				
				case IDOK:
					KillTimer(hW, 0x80);
				
					if (IsDlgButtonChecked(hW, IDC_LOG))
						conf.log = 1;
					else 
						conf.log = 0;
					
					SaveConfig();
					EndDialog(hW, FALSE);
					return TRUE;
			}
			break;

		case WM_NOTIFY:
			switch (wParam)
			{
				case IDC_TABC:
					hWC = GetDlgItem(hW, IDC_TABC);
					padn = TabCtrl_GetCurSel(hWC);

					for (i = 0; i < PADKEYS; i++)
					{
						hWC = GetDlgItem(hW, IDC_EL3 + i);
						Button_SetText(hWC, GetKeyLabel(padn, i).c_str());
					}

					return TRUE;
			}
			return FALSE;
	}
	return FALSE;
}

BOOL CALLBACK AboutDlgProc(HWND hW, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	switch (uMsg)
	{
		case WM_INITDIALOG:
			return TRUE;

		case WM_COMMAND:
			switch (LOWORD(wParam))
			{
				case IDOK:
					EndDialog(hW, FALSE);
					return TRUE;
			}
	}
	return FALSE;
}

void CALLBACK PADconfigure()
{
	DialogBox(hInst,
	          MAKEINTRESOURCE(IDD_DIALOG1),
	          GetActiveWindow(),
	          (DLGPROC)ConfigureDlgProc);
}

void CALLBACK PADabout()
{
	SysMessage("Author: zerofrog\nThanks to SSSPSXPad, TwinPAD, and PADwin plugins");
}

s32 CALLBACK PADtest()
{
	return 0;
}

BOOL APIENTRY DllMain(HANDLE hModule,                  // DLL INIT
                      DWORD  dwReason,
                      LPVOID lpReserved)
{
	hInst = (HINSTANCE)hModule;
	return TRUE;                                          // very quick :)
}