#include <stdinc.hpp>
#include "game.hpp"

#include <utils/hook.hpp>
#include <utils/memory.hpp>

namespace game
{
	gamemode current = reinterpret_cast<const char*>(0x88A5DC) != "CoDWaW.exe"s
		? gamemode::multiplayer
		: gamemode::singleplayer;

	namespace environment
	{
		bool t4mp()
		{
			return current == gamemode::multiplayer;
		}

		bool t4sp()
		{
			return current == gamemode::singleplayer;
		}
	}

	std::map<std::string, team_t> team_map =
	{
		{ "free", TEAM_FREE },
		{ "axis", TEAM_AXIS },
		{ "allies", TEAM_ALLIES },
		{ "neutral", TEAM_NEUTRAL },
		{ "dead", TEAM_DEAD }
	};

	int Scr_GetInt(game::scriptInstance_t inst, unsigned int arg_index)
	{
		static const auto call_addr = SELECT(0x0, 0x699C50);
		int answer;

		__asm
		{
			mov ecx, arg_index;
			mov eax, inst;
			call call_addr;
			mov answer, eax;
		}

		return answer;
	}

	void Scr_AddInt(game::scriptInstance_t inst, int value)
	{
		static const auto call_addr = SELECT(0x0, 0x69A610);

		__asm
		{
			push value;
			mov eax, inst;
			call call_addr;
			add esp, 4;
		}
	}

	float Scr_GetFloat(game::scriptInstance_t inst, unsigned int arg_index)
	{
		static const auto call_addr = SELECT(0x0, 0x699E90);
		float answer;

		__asm
		{
			mov ecx, arg_index;
			mov eax, inst;
			call call_addr;
			movss answer, xmm0;
		}

		return answer;
	}

	void Scr_AddFloat(game::scriptInstance_t inst, float value)
	{
		static const auto call_addr = SELECT(0x0, 0x69A670);

		__asm
		{
			push value;
			mov eax, inst;
			call call_addr;
			add esp, 4;
		}
	}

	char* Scr_GetString(game::scriptInstance_t inst, unsigned int arg_index)
	{
		static const auto call_addr = SELECT(0x0, 0x69A0D0);
		char* answer;

		__asm
		{
			mov esi, inst;
			mov eax, arg_index;
			call call_addr;
			mov answer, eax;
		}

		return answer;
	}

	void Scr_AddString(game::scriptInstance_t inst, const char* string)
	{
		static const auto call_addr = SELECT(0x0, 0x69A7E0);

		__asm
		{
			push string;
			mov eax, inst;
			call call_addr;
			add esp, 4;
		}
	}

	const char* Scr_GetIString(game::scriptInstance_t inst, unsigned int arg_index)
	{
		static const auto call_addr = SELECT(0x0, 0x69A1A0); //Scr_GetConstIString

		int id;

		__asm
		{
			mov eax, arg_index;
			call call_addr;
			mov id, eax;
		}

		return SL_ConvertToString(inst, id);
	}

	void Scr_AddIString(game::scriptInstance_t inst, const char* string)
	{
		static const auto call_addr = SELECT(0x0, 0x69A860);

		__asm
		{
			mov esi, string;
			call call_addr;
		}
	}

	unsigned short Scr_GetConstString(game::scriptInstance_t inst, unsigned int arg_index)
	{
		static const auto call_addr = SELECT(0x0, 0x699F30);

		unsigned short answer;

		__asm
		{
			push arg_index;
			mov eax, inst;
			call call_addr;
			mov answer, cx;
			add esp, 4;
		}

		return answer;
	}

	void Scr_AddConstString(game::scriptInstance_t inst, unsigned short id)
	{
		static const auto call_addr = SELECT(0x0, 0x69A8D0);

		__asm
		{
			mov si, id;
			mov eax, inst;
			call call_addr;
		}
	}

	void Scr_GetVector(game::scriptInstance_t inst, unsigned int arg_index, float* value)
	{
		static const auto call_addr = SELECT(0x0, 0x69A220);

		__asm
		{
			push arg_index;
			mov ecx, value;
			mov eax, inst;
			call call_addr;
			add esp, 4;
		}
	}

	void Scr_AddVector(game::scriptInstance_t inst, float* value)
	{
		static const auto call_addr = SELECT(0x0, 0x69A940);

		__asm
		{
			push value;
			mov eax, inst;
			call call_addr;
			add esp, 4;
		}
	}

	void Scr_AddUndefined(game::scriptInstance_t inst)
	{
		static const auto call_addr = SELECT(0x0, 0x69A720);

		__asm
		{
			mov eax, inst;
			call call_addr;
		}
	}

	gentity_s* Scr_GetEntity(unsigned int arg_index)
	{
		static const auto call_addr = SELECT(0x0, 0x546E30);
		gentity_s* answer;

		__asm
		{
			mov eax, arg_index;
			call call_addr;
			mov answer, eax;
		}

		return answer;
	}

	void Scr_AddEntity(game::scriptInstance_t inst, gentity_s* ent)
	{
		static const auto call_addr = SELECT(0x0, 0x546D90);

		__asm
		{
			mov edi, inst;
			mov eax, ent;
			call call_addr;
		}
	}

	unsigned int Scr_GetEntityId(scriptInstance_t inst, int entNum, classNum_e classnum, unsigned int clientnum)
	{
		static const auto call_addr = SELECT(0x0, 0x692520);

		unsigned int answer;

		__asm
		{
			push clientnum;
			push classnum;
			push inst;
			mov eax, entNum;
			call call_addr;
			add esp, 0xC;
			mov answer, eax;
		}

		return answer;
	}

	void Scr_AddEntityNum(scriptInstance_t inst, unsigned int entid)
	{
		static const auto call_addr = SELECT(0x0, 0x69A770);

		__asm
		{
			mov esi, entid;
			mov eax, inst;
			call call_addr;
		}
	}

	//Only supports getting the first argument as a path node
	pathnode_t* Scr_GetPathnode(scriptInstance_t inst)
	{
		static const auto call_addr = SELECT(0x0, 0x559E20);
		pathnode_t* answer;

		__asm
		{
			mov eax, inst;
			call call_addr;
			mov answer, eax;
		}

		return answer;
	}

	void Scr_AddPathnode(scriptInstance_t inst, pathnode_t* node)
	{
		unsigned int entnum = node - (*gameWorldCurrent)->path.nodes;
		auto entid = Scr_GetEntityId(inst, entnum, CLASS_NUM_PATHNODE, 0);
		Scr_AddEntityNum(inst, entid);
	}

	void Scr_MakeArray(scriptInstance_t inst)
	{
		static const auto call_addr = SELECT(0x0, 0x69A9D0);

		__asm
		{
			mov eax, inst;
			call call_addr;
		}
	}

	void Scr_AddArrayStringIndexed(scriptInstance_t inst, unsigned short id)
	{
		static const auto call_addr = SELECT(0x0, 0x69AAF0);

		__asm
		{
			mov edi, inst;
			movzx ecx, id;
			call call_addr;
		}
	}

	unsigned int Scr_GetNumParam(scriptInstance_t inst)
	{
		return scrVmPub[inst].outparamcount;
	}

	VariableType Scr_GetType(scriptInstance_t inst, unsigned int index)
	{
		static const auto call_addr = SELECT(0x0, 0x69A4E0);
		VariableType answer;

		__asm
		{
			mov eax, inst;
			mov ecx, index;
			call call_addr;
			mov answer, eax;
		}

		return answer;
	}

	void Scr_Error(const char* err, scriptInstance_t inst, bool is_terminal)
	{
		static const auto call_addr = SELECT(0x0, 0x69AB70);

		__asm
		{
			push is_terminal;
			mov edi, inst;
			mov ecx, err;
			call call_addr;
			add esp, 4;
		}
	}

	const char* SL_ConvertToString(scriptInstance_t inst, int id)
	{
		static const auto call_addr = SELECT(0x0, 0x68D950);
		const char* answer;

		__asm
		{
			mov ecx, inst;
			mov eax, id;
			call call_addr;
			mov answer, eax;
		}

		return answer;
	}

	const char* Cmd_Argv(int index)
	{
		static const auto call_addr = SELECT(0x0, 0x435CE0);
		const char* answer;

		__asm
		{
			mov eax, index;
			call call_addr;
			mov answer, eax;
		}

		return answer;
	}

	unsigned int Cmd_Argc()
	{
		return cmd_args->argc[cmd_args->nesting];
	}

	void Cmd_AddCommand(const char* name, void(__cdecl* function)())
	{
		auto newCmd = utils::memory::allocate<game::cmd_function_s>();
		auto name_dup = utils::memory::duplicate_string(name);

		*newCmd = {};
		newCmd->next = *game::cmd_functions;
		newCmd->function = function;
		newCmd->name = name_dup;

		*game::cmd_functions = newCmd;
	}

	dvar_s* Dvar_RegisterInt(const char* name, int value, int min, int max, DvarFlags flags, const char* desc)
	{
		DvarLimits limits;
		DvarValue dvar_value;
		limits.integer.min = min;
		limits.integer.max = max;
		dvar_value.integer = value;
		return Dvar_RegisterVariant(name, game::DVAR_TYPE_INT, flags, dvar_value, limits, desc);
	}

	dvar_s* Dvar_RegisterString(const char* name, const char* value, DvarFlags flags, const char* desc)
	{
		DvarLimits limits = {};
		DvarValue dvar_value;
		dvar_value.string = value;
		return Dvar_RegisterVariant(name, game::DVAR_TYPE_STRING, flags, dvar_value, limits, desc);
	}

	int Path_FindPath(path_t* pPath, team_t eTeam, float* vStartPos, float* vGoalPos, int bAllowNegotiationLinks)
	{
		static const auto call_addr = SELECT(0x0, 0x4CF280);

		int answer;

		__asm
		{
			push bAllowNegotiationLinks;
			push vGoalPos;
			push vStartPos;
			mov edx, eTeam;
			mov ecx, pPath;
			call call_addr;
			mov answer, eax;
		}

		return answer;
	}

	pathnode_t* Path_NearestNodeNotCrossPlanes(int typeFlags, int maxNodes, float* vOrigin, pathsort_t* nodes, float fMaxDist, float a6, float a7, int iPlaneCount, int* returnCount, nearestNodeHeightCheck heightCheck)
	{
		static const auto call_addr = SELECT(0x0, 0x55C210);

		pathnode_t* answer;

		__asm
		{
			push heightCheck;
			push returnCount;
			push iPlaneCount;
			push a7;
			push a6;
			push fMaxDist;
			push nodes;
			push vOrigin;
			mov ecx, maxNodes;
			mov edx, typeFlags;
			call call_addr;
			add esp, 0x20;
			mov answer, eax;
		}

		return answer;
	}

	int Path_FindPathFromTo(float* startPos, pathnode_t* pNodeTo, path_t* pPath, team_t eTeam, pathnode_t* pNodeFrom, float* vGoalPos, int bAllowNegotiationLinks, int bIgnoreBadplaces)
	{
		static const auto call_addr = SELECT(0x0, 0x4CF3F0);

		int answer;

		__asm
		{
			push bIgnoreBadplaces;
			push bAllowNegotiationLinks;
			push vGoalPos;
			push pNodeFrom;
			push eTeam;
			push pPath;
			mov edx, pNodeTo;
			mov eax, startPos;
			call call_addr;
			mov answer, eax;
		}

		return answer;
	}

	namespace plutonium
	{
	}
}