#include <windows.h>
#include <CharFunc.h>
#include <iostream>
#include <fstream>

#pragma data_seg("SharedBlock")
	HINSTANCE	hInst = NULL;
	HHOOK		keyHook = NULL;
	FILE*		file1 = NULL;
	char*		filepath = NULL;
	int	        hookState = -1;
#pragma data_seg()

extern "C" __declspec (dllexport) int _stdcall GetHookState()
{
	return hookState;
}

extern "C" __declspec (dllexport) int _stdcall AppendToLog(char* append)
{
	fopen_s(&file1, filepath, "a");
	fwrite(append, 1, strlen(append), file1);
	fclose(file1);
	return 0;
}

extern "C" __declspec (dllexport) LRESULT CALLBACK __stdcall KeyboardProc(int code, WPARAM wParam, LPARAM lParam)
{
	if ( (code < 0) || (wParam == VK_CONTROL) || (wParam >= VK_F1 && wParam <= VK_F11))
		return CallNextHookEx(keyHook, code, wParam, lParam);

	if (code == HC_ACTION)
	{
		if( (GetKeyState(VK_SHIFT) & 0x8000) != 0)
			shiftState = true;
		else
			shiftState = false;

		if ((HIWORD(lParam) & KF_UP) == 0)
		{
			MessageBox(NULL, ConvertCharToWide(filepath), TEXT("FilePath"), MB_OK);
			fopen_s(&file1, filepath, "a");

                        count++;
                        //..output int to file
                       
			fclose(file1);
		}//end key up
	}//end if (code == HC_ACTION)
	return  CallNextHookEx(keyHook, code, wParam, lParam);	
}

extern "C" int __declspec (dllexport) __stdcall SetKeyboardHook(char* fpath)
{
	//set the keyboard hook
	keyHook = SetWindowsHookEx(WH_KEYBOARD, KeyboardProc, hInst, 0); 

	if (keyHook == NULL)
	{
		//MessageBox(NULL, TEXT("Could Not Set Hook"), TEXT("Error #003"), NULL);
		hookState = -1;
		return -1;
	}
	else
	{
		filepath = new char[strlen(fpath) + 1];
		strcpy_s(filepath, strlen(fpath) + 2, fpath);
		//MessageBox(NULL,TEXT("Hooked"),TEXT("Hookoed"),MB_OK);
		hookState = 0;
		return 0;
	}
}

extern "C" int __declspec (dllexport) __stdcall UnsetKeyboardHook()
{
	//unhook
	UnhookWindowsHookEx(keyHook);

	if (filepath != NULL)
		delete [] filepath;

	hookState = -1;
	//MessageBox(NULL,TEXT("Unhooked"),TEXT("Unhooked"),MB_OK);
	return 0;
}

extern "C" BOOL __declspec (dllexport) __stdcall DllMain (HINSTANCE hInstance, DWORD reason, LPVOID reserved)
{
	switch (reason)
	{
		case DLL_PROCESS_ATTACH:
		{
			//save instance
			hInst = hInstance;
			MessageBox(NULL,TEXT("Attached"),TEXT("Attached"),MB_OK);
			break;
		}
		case DLL_PROCESS_DETACH:
		{
			MessageBox(NULL,TEXT("Detached"),TEXT("Detached"),MB_OK);
			break;
		}
    }
    return TRUE;
}