#include <iostream>
#include <API/ARK/Ark.h>

#pragma comment(lib, "ArkApi.lib")

DECLARE_HOOK(APrimalStructure_TakeDamage, float, APrimalStructure*, float, FDamageEvent*, AController*, AActor*, AShooterPlayerController*);

float Hook_APrimalStructure_TakeDamage(APrimalStructure* _this, float Damage, FDamageEvent* DamageEvent,
	AController* EventInstigator, AActor* DamageCauser, AShooterPlayerController* KillerShooterController)
{
	if (EventInstigator)
	{
		ACharacter* character = EventInstigator->CharacterField();

		if (character != nullptr && character->IsA(APrimalDinoCharacter::GetPrivateStaticClass()))
		{
			APrimalDinoCharacter* dino = static_cast<APrimalDinoCharacter*>(character);

			FString descr;
			dino->GetDinoDescriptiveName(&descr);




			if (descr.Contains("Dragon"))
			{
				return 0;
			}
			if (descr.Contains("Wyvern"))
			{
				return 0;
			}
			if (descr.Contains("Broodmother"))
			{
				return 0;
			}
			if (descr.Contains("Megapithecus"))
			{
				return 0;
			}
			if (descr.Contains("DodoRex"))
			{
				return 0;
			}
			if (descr.Contains("Odracir"))
			{
				return 0;
			}
			if (descr.Contains("Rawaiim"))
			{
				return 0;
			}
			if (descr.Contains("Nilloc"))
			{
				return 0;
			}
			if (descr.Contains("Smough"))
			{
				return 0;
			}
			if (descr.Contains("Karkinos"))
			{
				return 0;
			}
			if (descr.Contains("Gnashor"))
			{
				return 0;
			}
			if (descr.Contains("Bokito"))
			{
				return 0;
			}
			if (descr.Contains("Mormaw"))
			{
				return 0;
			}
			if (descr.Contains("Perdition"))
			{
				return 0;
			}
			if (descr.Contains("Raphus"))
			{
				return 0;
			}
			if (descr.Contains("Primordius"))
			{
				return 0;
			}
			if (descr.Contains("Demonic"))
			{
				return 0;
			}
			if (descr.Contains("Celestial"))
			{
				return 0;
			}
			if (descr.Contains("Chaos"))
			{
				return 0;
			}
			if (descr.Contains("Spirit"))
			{
				return 0;
			}
			if (descr.Contains("Badass"))
			{
				return 0;
			}
			if (descr.Contains("Champion"))
			{
				return 0;
			}
			if (descr.Contains("Alpha"))
			{
				return 0;
			}
			if (descr.Contains("Elite"))
			{
				return 0;
			}
			if (descr.Contains("Ancient"))
			{
				return 0;
			}
			if (descr.Contains("Zomdodo"))
			{
				return 0;
			}
			if (descr.Contains("Monkey"))
			{
				return 0;
			}
			if (descr.Contains("Apex"))
			{
				return 0;
			}
			if (descr.Contains("Toxic"))
			{
				return 0;
			}
			if (descr.Contains("Fabled"))
			{
				return 0;
			}
			if (descr.Contains("Primal"))
			{
				return 0;
			}
			if (descr.Contains("Elder"))
			{
				return 0;
			}
			if (descr.Contains("Buffon"))
			{
				return 0;
			}
			if (descr.Contains("Origins"))
			{
				return 0;
			}
			if (descr.Contains("Light"))
			{
				return 0;
			}
			if (descr.Contains("Ice"))
			{
				return 0;
			}
			if (descr.Contains("Fire"))
			{
				return 0;
			}
			if (descr.Contains("Electric"))
			{
				return 0;
			}
			if (descr.Contains("Captain"))
			{
				return 0;
			}
		}
	}
	float Hook_APrimalStructure_TakeDamage(APrimalStructure* _this, float Damage, FDamageEvent* DamageEvent, AController* EventInstigator, AActor* DamageCauser);
	{
		if (_this && EventInstigator && !EventInstigator->IsLocalController() && EventInstigator->IsA(AShooterPlayerController::StaticClass()))
		{
			AShooterPlayerController* AttackerShooterController = static_cast<AShooterPlayerController*>(EventInstigator);
			if (AttackerShooterController && AttackerShooterController->PlayerStateField() && AttackerShooterController->GetPlayerCharacter() && AttackerShooterController->GetPlayerCharacter()->CurrentWeaponField() && AttackerShooterController->GetPlayerCharacter()->CurrentWeaponField()->AssociatedPrimalItemField())
			{
				FString WeaponName;
				AttackerShooterController->GetPlayerCharacter()->CurrentWeaponField()->AssociatedPrimalItemField()->GetItemName(&WeaponName, false, true, nullptr);
				if (WeaponName.Contains(L"Flamethrower")) return 0;
			}
		}
		APrimalStructure_TakeDamage_original(_this, Damage, DamageEvent, EventInstigator, DamageCauser,KillerShooterController);
	}


void Load()
{
	Log::Get().Init("NoNunnaki");

	ArkApi::GetHooks().SetHook("APrimalStructure.TakeDamage", &Hook_APrimalStructure_TakeDamage,
		&APrimalStructure_TakeDamage_original);
}

void Unload()
{
	ArkApi::GetHooks().DisableHook("APrimalStructure.TakeDamage", &Hook_APrimalStructure_TakeDamage);
}

BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
{
	switch (ul_reason_for_call)
	{
	case DLL_PROCESS_ATTACH:
		Load();
		break;
	case DLL_PROCESS_DETACH:
		Unload();
		break;
	}
	return TRUE;
}