extension.cpp

// SPDX-License-Identifier: GPL-3.0-only
#include "extension.h"
#include "CDetour/detours.h"
#include <iplayerinfo.h>

#define GAMEDATA_FILE "defibfix"
#define HX_LOG 0

CDetour *hg_getPlayer = NULL;
CDetour *hg_defibStart = NULL;
CDetour *hg_defibEnd = NULL;
CDetour *hg_deadPlayer = NULL;

tCBaseEntity__SetAbsOrigin CBaseEntity__SetAbsOrigin;

IGameConfig *g_pGameConf = NULL;
IServerGameEnts *gameents = NULL;

DefibFix g_DefibFix;
SMEXT_LINK(&g_DefibFix);

int ig_defib;
int ig_noob;

/**
*	return -1           // Мертвый игрок не найден.
*	return (от 1 до 32) // Номер мертвого игрока.
**/
signed int HxGetDead(void)
{
	signed int iDead1 = -1;
	signed int iDead2 = -1;
	signed int iDead3 = -1;
	int i = 1;

	while (i <= 32)
	{
		IGamePlayer *player = playerhelpers->GetGamePlayer(i);
		if (player)
		{
			if (player->IsInGame())
			{
				IPlayerInfo *info = player->GetPlayerInfo();
				if (info)
				{
					if (info->GetTeamIndex() == 2)
					{
						if (info->IsObserver())
						{
#if HX_LOG
							g_pSM->LogMessage(myself, "Dead %d [%s]", i, info->GetName());
#endif
							if (info->IsFakeClient())
							{
								iDead3 = i;
							}
							else
							{
								iDead2 = i;
								if (ig_noob != i)
								{
									iDead1 = i;
								}
							}
						}

						if (info->IsDead())
						{
#if HX_LOG
							g_pSM->LogMessage(myself, "Dead %d [%s]", i, info->GetName());
#endif
							if (info->IsFakeClient())
							{
								iDead3 = i;
							}
							else
							{
								iDead2 = i;
								if (ig_noob != i)
								{
									iDead1 = i;
								}
							}
						}
					}
				}
			}
		}

		i += 1;
	}

	if (iDead1 > 0)
	{
#if HX_LOG
		g_pSM->LogMessage(myself, "Выбран %d", iDead1);
#endif
		return iDead1;
	}

	if (iDead2 > 0)
	{
#if HX_LOG
		g_pSM->LogMessage(myself, "Выбран %d", iDead2);
#endif
		return iDead2;
	}

#if HX_LOG
	g_pSM->LogMessage(myself, "Выбран %d", iDead3);
#endif
	return iDead3;
}

DETOUR_DECL_STATIC1(HxPlayerSpawn, void *, int, charaster)
{
	if (ig_defib)
	{
		ig_defib = 0;
		int iDead = HxGetDead();
		if (iDead > 0)
		{
			ig_noob = iDead;
#if HX_LOG
			g_pSM->LogMessage(myself, "Воскрешается %d", iDead);
#endif
			return gamehelpers->ReferenceToEntity(iDead);
		}

#if HX_LOG
		g_pSM->LogMessage(myself, "Ошибка");
#endif
		return 0;
	}

	return DETOUR_STATIC_CALL(HxPlayerSpawn)(charaster);
}

DETOUR_DECL_MEMBER4(HxDefibStart, void *, int, reserved, void*, player, void*, entity, int, reserved2)
{
	edict_t *edict = gameents->BaseEntityToEdict((CBaseEntity *) entity);

	if (edict)
	{
#if HX_LOG
		g_pSM->LogMessage(myself, "Начало дефибрилляции");
#endif
		if (HxGetDead() == -1)
		{
#if HX_LOG
			g_pSM->LogMessage(myself, "Труп удаляется");
#endif
			engine->RemoveEdict(edict);
		}
	}

	return DETOUR_MEMBER_CALL(HxDefibStart)(reserved, player, entity, reserved2);
}

DETOUR_DECL_MEMBER2(HxDefibEnd, void *, void*, player, void*, entity)
{
	edict_t *edict = gameents->BaseEntityToEdict((CBaseEntity *) entity);
	if (edict)
	{
#if HX_LOG
		g_pSM->LogMessage(myself, "Окончание дефибрилляции");
#endif
		ig_defib = 1;
	}

	return DETOUR_MEMBER_CALL(HxDefibEnd)(player, entity);
}

DETOUR_DECL_STATIC1(HxDeathModel, CBaseEntity *, CBasePlayer*, bplayer)
{
	edict_t *pEdict = gameents->BaseEntityToEdict((CBaseEntity *) bplayer);
	int client = gamehelpers->IndexOfEdict(pEdict);

	CBaseEntity *result = DETOUR_STATIC_CALL(HxDeathModel)(bplayer);
	if (client > 0)
	{
		Vector PlayerOrigin = pEdict->GetCollideable()->GetCollisionOrigin();
		CBaseEntity__SetAbsOrigin(result, &PlayerOrigin);
	}

	return result;
}

void HxDestroyAll(void)
{
	if (hg_getPlayer)
	{
		hg_getPlayer->Destroy();
		hg_getPlayer = NULL;
	}
	if (hg_defibStart)
	{
		hg_defibStart->Destroy();
		hg_defibStart = NULL;
	}
	if (hg_defibEnd)
	{
		hg_defibEnd->Destroy();
		hg_defibEnd = NULL;
	}
	if (hg_deadPlayer)
	{
		hg_deadPlayer->Destroy();
		hg_deadPlayer = NULL;
	}
}

bool HxStart(void)
{
	CDetourManager::Init(g_pSM->GetScriptingEngine(), g_pGameConf);
	g_pGameConf->GetMemSig("CBaseEntity::SetAbsOrigin", (void **) &CBaseEntity__SetAbsOrigin);

	hg_getPlayer = DETOUR_CREATE_STATIC(HxPlayerSpawn, "GetPlayerByCharacter");
	hg_defibStart = DETOUR_CREATE_MEMBER(HxDefibStart, "DefibrillatorOnStartAction");
	hg_defibEnd = DETOUR_CREATE_MEMBER(HxDefibEnd, "DefibrillatorOnActionComplete");
	hg_deadPlayer = DETOUR_CREATE_STATIC(HxDeathModel, "CSurvivorDeathModel::Create");

	if (CBaseEntity__SetAbsOrigin && hg_getPlayer && hg_defibStart && hg_defibEnd && hg_deadPlayer)
	{
		hg_getPlayer->EnableDetour();
		hg_defibStart->EnableDetour();
		hg_defibEnd->EnableDetour();
		hg_deadPlayer->EnableDetour();
		return true;
	}

	HxDestroyAll();
	return false;
}

bool DefibFix::SDK_OnMetamodLoad(ISmmAPI *ismm, char *error, size_t maxlength, bool late)
{
	size_t maxlen = maxlength;
	GET_V_IFACE_CURRENT(GetEngineFactory, engine, IVEngineServer, INTERFACEVERSION_VENGINESERVER);
	GET_V_IFACE_ANY(GetServerFactory, gameents, IServerGameEnts, INTERFACEVERSION_SERVERGAMEENTS);

	return true;
}

bool DefibFix::SDK_OnLoad(char *error, size_t maxlength, bool late)
{
	ig_defib = 0;
	ig_noob = 0;

	char conf_error[255];
	if (!gameconfs->LoadGameConfigFile(GAMEDATA_FILE, &g_pGameConf, conf_error, sizeof(conf_error)))
	{
		snprintf(error, maxlength, "Проблема с defibfix.txt: %s", conf_error);
		return false;
	}

	if (!HxStart())
	{
		snprintf(error, maxlength, "Проблема с Offset");
		return false;
	}

	return true;
}

void DefibFix::SDK_OnUnload()
{
	HxDestroyAll();
	gameconfs->CloseGameConfigFile(g_pGameConf);
}