#include <Windows.h>
#include <stdio.h>
#include "MemoryModule.h"
#include "WFClasses.h"

bool IsNum(char v) { return (v <= '9' && v >= '0'); }

bool CheckName(const char* pszName)
{
	if (pszName == 0)
		return false;

	size_t len = strlen(pszName);

	if (len < 4)
		return false;

	if (len == 5)
	{
		if (IsNum(pszName[0]) && IsNum(pszName[2]) && IsNum(pszName[4]))
			return false;
	}

	if (pszName[0] >= 'A' && pszName[0] <= 'z')
		return false;

	if (pszName[0] >= 33 && pszName[0] <= 44)
		return false;

	if (pszName[0] == '@')
		return false;

	return true;
}

DWORD WINAPI Thread(LPVOID)
{
	Beep(1000, 90);
	while (true)
	{
		Sleep(1);

		if (GetAsyncKeyState(VK_NUMPAD9))
			TerminateProcess(GetCurrentProcess(), 0);

		SSystemGlobalEnvironment* pEnv = *(SSystemGlobalEnvironment**)0x141C1D150;

		if (!pEnv || !pEnv->pD3DEngine || !pEnv->pGame || !pEnv->pEntitySystem)
			continue;


		// SCVars* pVars = pEnv->pGame->pCVars();
		//*pVars->i_pelletsDisp() = 20000.f;
		//*pVars->g_crosshairSpreadTuning_Enable() = 0;
		//*pVars->g_parametric_spread_attack_enabled() = 0;
		//*pVars->g_parametric_recoil_random_enabled() = 0;

		static bool enablewh = false;
		if (GetAsyncKeyState(VK_NUMPAD7))
		{
			while (GetAsyncKeyState(VK_NUMPAD7)) { Sleep(1); }
			if (enablewh)
				Beep(700, 90);
			else
				Beep(1000, 90);
			enablewh = !enablewh;
		}

		if (enablewh)
		{
			IEntityIt* pIt = pEnv->pEntitySystem->GetEntityIterator();
			IEntity* pEnt = pIt->Next();

			if (pEnt)
			{
				do
				{
					IGameFramework* pFramework = *(IGameFramework**)0x141F269A8;
					if (!pFramework)
					{
						Beep(1000, 90);
						break;
					}

					IActor* pActor = pFramework->GetIActorSystem()->GetActor(pEnt->GetID());
					if (pActor)
					{
						IEntityRenderProxy* pProx = (IEntityRenderProxy*)pEnt->GetProxy(EEntityProxy::ENTITY_PROXY_RENDER);
						if (pProx)
						{
							pProx->SetHUDSilhouettesParams(255, 255, 0, 0);



						}
					}



					/*
							//*(int*)((DWORD64)pEnt + (4 * 4)) = 0x11111111;
							//*(int*)((DWORD64)pEnt + (5 * 4)) = 0x11111111;


							pProx->SetRndFlags(0x00000000F0000000);

							*/

					/*
					const char* pszName = pEnt->GetName();
					if (CheckName(pszName))
					{
						IEntityRenderProxy* pProx = (IEntityRenderProxy*)pEnt->GetProxy(EEntityProxy::ENTITY_PROXY_RENDER);
						if (pProx)
						{
							pProx->SetHUDSilhouettesParams(255, 255, 0, 0);
							//pProx->SetRndFlags(0x10);
						}
					}
					*/



				} while ((pEnt = pIt->Next()));
			}

			pEnv->pD3DEngine->SetPostEffectParam("Flashbang_Time", 0, true);
		}

		if (GetAsyncKeyState(VK_NUMPAD6))
		{
			while (GetAsyncKeyState(VK_NUMPAD6)) { Sleep(1); }

			HANDLE hFile = CreateFileA("D:\\wfl.txt", GENERIC_WRITE, 0, 0, CREATE_ALWAYS, 0, 0);
			IEntityIt* pIt = pEnv->pEntitySystem->GetEntityIterator();
			IEntity* pEnt = pIt->Next();

			static char result[80000];
			ZeroMemory(result, sizeof(result));

			do
			{

				const char* pszName = pEnt->GetName();
				if (CheckName(pszName))
				{
					IEntityRenderProxy* pProx = (IEntityRenderProxy*)pEnt->GetProxy(EEntityProxy::ENTITY_PROXY_RENDER);
					if (pProx)
					{
						char buffer[1000];
						sprintf(buffer, "Name: %s\n", pEnt->GetName());
						strcat(result, buffer);

						for (int i = 0; i < 35; i++)
						{
							sprintf(buffer, "idx: %d, value: 0x%p\n", i, *(DWORD64*)((DWORD64)pProx + (i * 8)));
							strcat(result, buffer);
						}

						strcat(result, "\n");
					}
				}



			} while ((pEnt = pIt->Next()));

			WriteFile(hFile, result, strlen(result), 0, 0);
			CloseHandle(hFile);

		}

	}
}
//((void (__fastcall *)(ID3DEngine *, char *, __int64, _QWORD))
BOOL CALLBACK DllMain(HMODULE hModule, DWORD dwReason, LPVOID lpReserved)
{
	static HMEMORYMODULE hMM = NULL;
	static HANDLE hThread = NULL;

	switch (dwReason)
	{
	case DLL_PROCESS_ATTACH:

		//extern unsigned char* GetGClay64Raw();
		//hMM = MemoryLoadLibrary(GetGClay64Raw());

		hThread = CreateThread(0, 0, Thread, 0, 0, 0);

		break;
	case DLL_PROCESS_DETACH:
		TerminateThread(hThread, 0);
		//MemoryFreeLibrary(hMM);
		break;
	}

	return TRUE;
}