#include <string.h>
#include <stdbool.h>

#include "Oni.h"
#include "Oni_Character.h"

#include "Daodan.h"
#include "Daodan_Cheater.h"

oniCheatCode DDr_CheatTable[] = {
	{ "shapeshifter",   "Change Characters Enabled",       "Change Characters Disabled", cheat_shapeshifter   },
	{ "liveforever",    "Invincibility Enabled",           "Invincibility Disabled",     cheat_liveforever    },
	{ "touchofdeath",   "Omnipotence Enabled",             "Omnipotence Disabled",       cheat_touchofdeath   },
	{ "canttouchthis",  "Unstoppable Enabled",             "Unstoppable Disabled",       cheat_canttouchthis  },
	{ "fatloot",        "Fat Loot Received",               NULL,                         cheat_fatloot        },
	{ "glassworld",     "Glass Furniture Enabled",         "Glass Furniture Disabled",   cheat_glassworld     },
	{ "winlevel",       "Instantly Win Level",             NULL,                         cheat_winlevel       },
	{ "loselevel",      "Instantly Lose Level",            NULL,                         cheat_loselevel      },
	{ "bighead",        "Big Head Enabled",                "Big Head Disabled",          cheat_bighead        },
	{ "minime",         "Mini Mode Enabled",               "Mini Mode Disabled",         cheat_minime         },
	{ "superammo",      "Super Ammo Mode Enabled",         "Super Ammo Mode Disabled",   cheat_superammo      },
	{ "thedayismine",   "Developer Access Enabled",        "Developer Access Disabled",  cheat_thedayismine   },
	{ "reservoirdogs",  "Last Man Standing Enabled",       "Last Man Standing Disabled", cheat_reservoirdogs  },
	{ "roughjustice",   "Gatling Guns Enabled",            "Gatling Guns Disabled",      cheat_roughjustice   },
	{ "chenille",       "Daodan Power Enabled",            "Daodan Power Disabled",      cheat_chenille       },
	{ "behemoth",       "Godzilla Mode Enabled",           "Godzilla Mode Disabled",     cheat_behemoth       },
	{ "elderrune",      "Regeneration Enabled",            "Regeneration Disabled",      cheat_elderrune      },
	{ "moonshadow",     "Phase Cloak Enabled",             "Phase Cloak Disabled",       cheat_moonshadow     },
	{ "munitionfrenzy", "Weapons Locker Created",          NULL,                         cheat_munitionfrenzy },
	{ "fistsoflegend",  "Fists Of Legend Enabled",         "Fists Of Legend Disabled",   cheat_fistsoflegend  },
	{ "killmequick",    "Ultra Mode Enabled",              "Ultra Mode Disabled",        cheat_killmequick    },
	{ "carousel",       "Slow Motion Enabled",             "Slow Motion Disabled",       cheat_carousel       },
	{ "bigbadboss",     "Boss Shield Enabled",             "Boss Shield Disabled",       cheat_bigbadboss     },
	{ "bulletproof",    "Force Field Enabled",             "Force Field Disabled",       cheat_bulletproof    },
	{ "kangaroo",       "Kangaroo Jump Enabled",           "Kangaroo Jump Disabled",     cheat_kangaroo       },
	{ "marypoppins",    "Jet Pack Mode Enabled",           "Jet Pack Mode Disabled",     cheat_marypoppins    },
	{ "buddha",         "Unkillable Enabled",              "Unkillable Disabled",        cheat_buddha         },
	{ "shinobi",        "Ninja Mode Enabled (good luck!)", "Ninja Mode Disabled",        cheat_shinobi        },
	{ "x",              "Developer Access Enabled",        "Developer Access Disabled",  cheat_x              },
	{ "testcheat",      "Testing...",                      "",                           cheat_testcheat      },
	{0}
};

// Just copied all these defines from the old daodan, they were originaly from SFeLi's code.

#define GSA_camera        (0x00000080)
#define GSA_player        (0x000000AC)
#define GSA_carousel      (0x00000104) /* char */
#define GSA_slowmotimer   (0x00000108)
#define GSA_splashscreen  (0x00000118) /* char */

#define CHR_flags         (0x00000004)
#define CHR_flags2        (0x00000008)
#define CHR_oncc          (0x0000000C)
#define CHR_team          (0x00000012) /* short */
#define CHR_name          (0x00000014) /* char[32] */
#define CHR_scalemodel    (0x00000034)
#define CHR_weapon1       (0x00000194)
#define CHR_weapon2       (0x00000198)
#define CHR_weapon3       (0x0000019C)
#define CHR_shield_curr   (0x000001B6) /* short */
#define CHR_shield        (0x000001B8) /* short */
#define CHR_phasecloak    (0x000001BA) /* short */
#define CHR_stats_kills   (0x00001670)
#define CHR_stats_damage  (0x00001674)

#define ONCC_jet_accel    (0x00000010) /* float */
#define ONCC_jet_timer    (0x00000016) /* short */
#define ONCC_height1      (0x00000018) /* float */
#define ONCC_height2      (0x0000001C) /* float */
#define ONCC_bodysize_min (0x00000C58) /* float */
#define ONCC_bodysize_max (0x00000C8C) /* float */

#define kangaroo_h     (60)
#define kangaroo_jp    (0.06)
#define marypoppins_jp (0.14)

uint16_t cheat_oldshield = 0;
int32_t cheat_oldhealth = 0;
int32_t cheat_oldmaxhealth = 0;
float cheat_oldjet_accel = 0.03f;
uint16_t cheat_oldjet_timer = 20;
float cheat_oldheight1 = 45;
float cheat_oldheight2 = 135;
bool inc_fallingframes = true;

uint8_t ONICALL DDrCheater(uint32_t cheat)
{
	switch (cheat)
	{
		case cheat_bigbadboss:
		{
			char* player = *((char**)(ONgGameState + GSA_player));
			if (*(unsigned int*)(player + CHR_flags) & chr_bossshield)
			{
				*(unsigned int*)(player + CHR_flags) = *(unsigned int*)(player + CHR_flags) & ~chr_bossshield;
				return 0;
			}
			else
			{
				*(unsigned int*)(player + CHR_flags) = *(unsigned int*)(player + CHR_flags) | chr_bossshield;
				return 1;
			}
		}
		case cheat_bulletproof:
		{
			char* player = *((char**)(ONgGameState + GSA_player));
			if (*(unsigned int*)(player + CHR_flags) & chr_weaponimmune)
			{
				*(unsigned int*)(player + CHR_flags) = *(unsigned int*)(player + CHR_flags) & ~chr_weaponimmune;
				*(unsigned short*)(player + CHR_shield) = cheat_oldshield;
				return 0;
			}
			else
			{
				*(unsigned int*)(player + CHR_flags) = *(unsigned int*)(player + CHR_flags) | chr_weaponimmune;
				cheat_oldshield = *(unsigned short*)(player + CHR_shield);
				*(unsigned short*)(player + CHR_shield) = 100;
				return 1;
			}
		}
		case cheat_kangaroo:
		{
			char* player = *((char**)(ONgGameState + GSA_player));
			char* oncc = *(char**)(player + CHR_oncc);
			if (!inc_fallingframes)
				inc_fallingframes = true;
			if (*(unsigned int*)(oncc + ONCC_jet_timer) == kangaroo_h)
			{
				*(float*)(oncc + ONCC_jet_accel) = cheat_oldjet_accel;
				*(unsigned short*)(oncc + ONCC_jet_timer) = cheat_oldjet_timer;
				*(float*)(oncc + ONCC_height1) = cheat_oldheight1;
				*(float*)(oncc + ONCC_height2) = cheat_oldheight2;
				return 0;
			}
			else if (*(unsigned int*)(oncc + ONCC_jet_timer) == 0xFFFF)
			{
				*(float*)(oncc + ONCC_jet_accel) = kangaroo_jp;
				*(unsigned short*)(oncc + ONCC_jet_timer) = kangaroo_h;
				*(float*)(oncc + ONCC_height1) = 0x7f800000;
				*(float*)(oncc + ONCC_height2) = 0x7f800000;
				return 1;
			}
			else
			{
				cheat_oldjet_accel = *(float*)(oncc + ONCC_jet_accel);
				cheat_oldjet_timer = *(unsigned short*)(oncc + ONCC_jet_timer);
				cheat_oldheight1 = *(float*)(oncc + ONCC_height1);
				cheat_oldheight2 = *(float*)(oncc + ONCC_height2);
				*(float*)(oncc + ONCC_jet_accel) = kangaroo_jp;
				*(unsigned short*)(oncc + ONCC_jet_timer) = kangaroo_h;
				*(float*)(oncc + ONCC_height1) = 0x7f800000;
				*(float*)(oncc + ONCC_height2) = 0x7f800000;
				return 1;
			}
		}
		case cheat_marypoppins:
		{
			char* player = *((char**)(ONgGameState + GSA_player));
			char* oncc = *(char**)(player + CHR_oncc);
			if (!inc_fallingframes)
			{
				inc_fallingframes = true;
				if (*(unsigned int*)(oncc + ONCC_jet_timer) == 0xFFFF)
				{
					*(float*)(oncc + ONCC_jet_accel) = cheat_oldjet_accel;
					*(unsigned short*)(oncc + ONCC_jet_timer) = cheat_oldjet_timer;
					*(float*)(oncc + ONCC_height1) = cheat_oldheight1;
					*(float*)(oncc + ONCC_height2) = cheat_oldheight2;
				}
				return 0;
			}
			else if (*(unsigned int*)(oncc + ONCC_jet_timer) == kangaroo_h)
			{
				*(float*)(oncc + ONCC_jet_accel) = marypoppins_jp;
				*(unsigned short*)(oncc + ONCC_jet_timer) = 0xFFFF;
				*(float*)(oncc + ONCC_height1) = 0x7f800000;
				*(float*)(oncc + ONCC_height2) = 0x7f800000;
				inc_fallingframes = false;
				return 1;
			}
			else
			{
				cheat_oldjet_accel = *(float*)(oncc + ONCC_jet_accel);
				cheat_oldjet_timer = *(unsigned short*)(oncc + ONCC_jet_timer);
				cheat_oldheight1 = *(float*)(oncc + ONCC_height1);
				cheat_oldheight2 = *(float*)(oncc + ONCC_height2);
				*(float*)(oncc + ONCC_jet_accel) = marypoppins_jp;
				*(unsigned short*)(oncc + ONCC_jet_timer) = 0xFFFF;
				*(float*)(oncc + ONCC_height1) = 0x7f800000;
				*(float*)(oncc + ONCC_height2) = 0x7f800000;
				inc_fallingframes = false;
				return 1;
			}
		}
		case cheat_buddha:
		{
			char* player = *((char**)(ONgGameState + GSA_player));
			if (*(unsigned int*)(player + CHR_flags) & chr_unkillable)
			{
				*(unsigned int*)(player + CHR_flags) = *(unsigned int*)(player + CHR_flags) & ~chr_unkillable;
				return 0;
			}
			else
			{
				*(unsigned int*)(player + CHR_flags) = *(unsigned int*)(player + CHR_flags) | chr_unkillable;
				return 1;
			}
		}
		case cheat_shinobi:
		{
			Character* player = (Character*)*((void**)(ONgGameState + GSA_player));
			if (player->MaxHealth == 1)
			{
				player->Health = cheat_oldhealth;
				player->MaxHealth = cheat_oldmaxhealth;
				player->Flags = player->Flags & ~(chr_bossshield | chr_weaponimmune);
				return 0;
			}
			else
			{
				cheat_oldhealth = player->Health;
				cheat_oldmaxhealth = player->MaxHealth;
				player->Health = 1;
				player->MaxHealth = 1;
				player->Flags = player->Flags | chr_bossshield | chr_weaponimmune;
				return 1;
			}
			
		}
		case cheat_testcheat:
		{
			char* player = *((char**)(ONgGameState + GSA_player));
			*(unsigned int*)(player + CHR_flags) = *(unsigned int*)(player + CHR_flags) | chr_noncombatant;
			return 1;
		}
		case cheat_x:
			return ONrCheater(cheat_thedayismine);
		default:
			return ONrCheater(cheat);
	}
}

void ONICALL DDrCheater_LevelLoad()
{
	inc_fallingframes = true;
}

__stdcall void FallingFrames(void* Ebp)
{
	if (inc_fallingframes)
		*((unsigned int*)(Ebp + 0xf6)) = *((unsigned int*)(Ebp + 0xf6)) + 1;
}