#include "../bot_command.h"

void bot_command_cast(Client* c, const Seperator* sep)
{
	if (helper_is_help_or_usage(sep->arg[1])) {
		std::vector<std::string> description =
		{
			"Commands bots to force cast a specific spell type, ignoring all settings (holds, delays, thresholds, etc)"
		};

		std::vector<std::string> notes =
		{
			"- This will interrupt any spell currently being cast by bots told to use the command.",
			"- Bots will still check to see if they have the spell in their spell list, whether the target is immune, spell is allowed and all other sanity checks for spells"
		};

		std::vector<std::string> example_format =
		{
			fmt::format(
				"{} [Type Shortname] [actionable, default: spawned]"
				, sep->arg[0]
			),
			fmt::format(
				"{} [Type ID] [actionable, default: spawned]"
				, sep->arg[0]
			)
		};
		std::vector<std::string> examples_one =
		{
			"To tell everyone to Nuke the target:",
			fmt::format(
				"{} {} spawned",
				sep->arg[0],
				c->GetSpellTypeShortNameByID(BotSpellTypes::Nuke)
			),
			fmt::format(
				"{} {} spawned",
				sep->arg[0],
				BotSpellTypes::Nuke
			)
		};
		std::vector<std::string> examples_two =
		{
			"To tell all Enchanters to slow the target:",
			fmt::format(
				"{} {} byclass {}",
				sep->arg[0],
				Class::Enchanter,
				c->GetSpellTypeShortNameByID(BotSpellTypes::Slow)
			),
			fmt::format(
				"{} {} byclass {}",
				sep->arg[0],
				Class::Enchanter,
				BotSpellTypes::Slow
			)
		};
		std::vector<std::string> examples_three =
		{
			"To tell Clrbot to resurrect the targeted corpse:",
			fmt::format(
				"{} {} byname Clrbot",
				sep->arg[0],
				c->GetSpellTypeShortNameByID(BotSpellTypes::Resurrect)
			),
			fmt::format(
				"{} {} byname Clrbot",
				sep->arg[0],
				BotSpellTypes::Resurrect
			)
		};

		std::vector<std::string> actionables =
		{
			"target, byname, ownergroup, ownerraid, targetgroup, namesgroup, healrotationtargets mmr, byclass, byrace, spawned"
		};

		std::vector<std::string> options = { };
		std::vector<std::string> options_one = { };
		std::vector<std::string> options_two = { };
		std::vector<std::string> options_three = { }; 

		std::string popup_text = c->SendCommandHelpWindow(
			c,
			description,
			notes,
			example_format,
			examples_one, examples_two, examples_three,
			actionables,
			options,
			options_one, options_two, options_three
		);

		popup_text = DialogueWindow::Table(popup_text);

		c->SendPopupToClient(sep->arg[0], popup_text.c_str());
		c->SendSpellTypePrompts(true);

		c->Message(
			Chat::Yellow,
			fmt::format(
				"Use help after any command type for more subtypes to use, for example: {}.",
				Saylink::Silent("^cast invisibility help")
			).c_str()
		);

		if (RuleB(Bots, SendClassRaceOnHelp)) {
			c->Message(
				Chat::Yellow,
				fmt::format(
					"Use {} for information about race/class IDs.",
					Saylink::Silent("^classracelist")
				).c_str()
			);
		}

		return;
	}

	std::string arg1 = sep->arg[1];
	std::string arg2 = sep->arg[2];

	//Commanded type help prompts
	if (!arg2.compare("help")) {
		c->Message(Chat::Yellow, "You can also use [single], [group], [ae]. Ex: ^cast movementspeed group.", sep->arg[0]);
	}

	if (!arg1.compare("invisibility") && !arg2.compare("help")) {
		c->Message(
			Chat::Yellow,
			fmt::format(
				"Available options for {} are: {}, {}, {}, {}.",
				sep->arg[0],
				Saylink::Silent("^cast invisibility see", "see"),
				Saylink::Silent("^cast invisibility invis", "invis"),
				Saylink::Silent("^cast invisibility undead", "undead"),
				Saylink::Silent("^cast invisibility animals", "animals")
			).c_str()
		);

		return;
	}

	if (!arg1.compare("size") && !arg2.compare("help")) {
		c->Message(
			Chat::Yellow,
			fmt::format(
				"Available options for {} are: {}, {}.",
				sep->arg[0],
				Saylink::Silent("^cast size grow", "grow"),
				Saylink::Silent("^cast size shrink", "shrink")
			).c_str()
		);

		return;
	}

	if (!arg1.compare("movementspeed") && !arg2.compare("help")) {
		c->Message(
			Chat::Yellow,
			fmt::format(
				"Available options for {} are: {}, {}.",
				sep->arg[0],
				Saylink::Silent("^cast movementspeed selo"), "selo"
			).c_str()
		);

		return;
	}

	if (!arg2.compare("help")) {
		c->Message(Chat::Yellow, "There are no additional options for {}.", sep->arg[0]);
		return;
	}

	int ab_arg = 2;
	uint16 spellType = 0;

	// String/Int type checks
	if (sep->IsNumber(1)) {
		spellType = atoi(sep->arg[1]);

		if (spellType < BotSpellTypes::START || (spellType > BotSpellTypes::END && spellType < BotSpellTypes::COMMANDED_START) || spellType > BotSpellTypes::COMMANDED_END) {
			c->Message(
				Chat::Yellow,
				fmt::format(
					"You must choose a valid spell type. Use {} for information regarding this command.",
					Saylink::Silent(
						fmt::format("{} help", sep->arg[0])
					)
				).c_str()
			);

			return;
		}
	}
	else {
		if (c->GetSpellTypeIDByShortName(arg1) != UINT16_MAX) {
			spellType = c->GetSpellTypeIDByShortName(arg1);
		}
		else {
			c->Message(
				Chat::Yellow,
				fmt::format(
					"Incorrect argument, use {} for information regarding this command.",
					Saylink::Silent(
						fmt::format("{} help", sep->arg[0])
					)
				).c_str()
			);

			return;
		}
	}

	switch (spellType) { //Allowed command checks
		case BotSpellTypes::Charm:
			if (!RuleB(Bots, AllowCommandedCharm)) {
				c->Message(Chat::Yellow, "This commanded type is currently disabled.");
				return;
			}

			break;
		case BotSpellTypes::AEMez:
		case BotSpellTypes::Mez:
			if (!RuleB(Bots, AllowCommandedMez)) {
				c->Message(Chat::Yellow, "This commanded type is currently disabled.");
				return;
			}

			break;
		case BotSpellTypes::Resurrect:
			if (!RuleB(Bots, AllowCommandedResurrect)) {
				c->Message(Chat::Yellow, "This commanded type is currently disabled.");
				return;
			}

			break;
		case BotSpellTypes::SummonCorpse:
			if (!RuleB(Bots, AllowCommandedSummonCorpse)) {
				c->Message(Chat::Yellow, "This commanded type is currently disabled.");
				return;
			}

			break;
		default:
			break;
	}

	std::string argString = sep->arg[ab_arg];
	uint16 subType = UINT16_MAX;
	uint16 subTargetType = UINT16_MAX;
	
	if (!argString.compare("shrink")) {
		subType = CommandedSubTypes::Shrink;
		++ab_arg;
	}
	else if (!argString.compare("grow")) {
		subType = CommandedSubTypes::Grow;
		++ab_arg;
	}
	else if (!argString.compare("see")) {
		subType = CommandedSubTypes::SeeInvis;
		++ab_arg;
	}
	else if (!argString.compare("invis")) {
		subType = CommandedSubTypes::Invis;
		++ab_arg;
	}
	else if (!argString.compare("undead")) {
		subType = CommandedSubTypes::InvisUndead;
		++ab_arg;
	}
	else if (!argString.compare("animals")) {
		subType = CommandedSubTypes::InvisAnimals;
		++ab_arg;
	}
	else if (!argString.compare("selo")) {
		subType = CommandedSubTypes::Selo;
		++ab_arg;
	}

	argString = sep->arg[ab_arg];

	if (!argString.compare("single")) {
		subTargetType = CommandedSubTypes::SingleTarget;
		++ab_arg;
	}
	else if (!argString.compare("group")) {
		subTargetType = CommandedSubTypes::GroupTarget;
		++ab_arg;
	}
	else if (!argString.compare("ae")) {
		subTargetType = CommandedSubTypes::AETarget;
		++ab_arg;
	}

	if (
		spellType == BotSpellTypes::PetBuffs ||
		spellType == BotSpellTypes::PetCompleteHeals ||
		spellType == BotSpellTypes::PetFastHeals ||
		spellType == BotSpellTypes::PetHoTHeals ||
		spellType == BotSpellTypes::PetRegularHeals ||
		spellType == BotSpellTypes::PetVeryFastHeals
	) {
		c->Message(Chat::Yellow, "Pet type heals and buffs are not supported, use the regular spell type.");
		return;
	}

	Mob* tar = c->GetTarget();
	//LogTestDebug("{}: 'Attempting {} [{}-{}] on {}'", __LINE__, c->GetSpellTypeNameByID(spellType), (subType != UINT16_MAX ? c->GetSubTypeNameByID(subType) : "Standard"), (subTargetType != UINT16_MAX ? c->GetSubTypeNameByID(subTargetType) : "Standard"), (tar ? tar->GetCleanName() : "NOBODY")); //deleteme

	if (!tar) {
		if (spellType != BotSpellTypes::Escape && spellType != BotSpellTypes::Pet) {
			c->Message(Chat::Yellow, "You need a target for that.");
			return;
		}
	}

	switch (spellType) { //Target Checks
		case BotSpellTypes::Resurrect:
			if (!tar->IsCorpse() || !tar->CastToCorpse()->IsPlayerCorpse()) {
				c->Message(Chat::Yellow, "[%s] is not a player's corpse.", tar->GetCleanName());

				return;
			}

			break;
		case BotSpellTypes::Identify:
		case BotSpellTypes::SendHome:
		case BotSpellTypes::BindAffinity:
		case BotSpellTypes::SummonCorpse:
			if (!tar->IsClient() || !c->IsInGroupOrRaid(tar)) {
				c->Message(Chat::Yellow, "[%s] is an invalid target. Only players in your group or raid are eligible targets.", tar->GetCleanName());

				return;
			}

			break;
		default:
			if (
				(IsBotSpellTypeDetrimental(spellType) && !c->IsAttackAllowed(tar)) ||
				(
					spellType == BotSpellTypes::Charm && 
					(
						tar->IsClient() || 
						tar->IsCorpse() ||
						tar->GetOwner()
					)
				)
			) {
				c->Message(Chat::Yellow, "You cannot attack [%s].", tar->GetCleanName());

				return;
			}

			if (IsBotSpellTypeBeneficial(spellType)) {
				if (
					(!tar->IsOfClientBot() && !(tar->IsPet() && tar->GetOwner() && tar->GetOwner()->IsOfClientBot())) ||
					((tar->IsOfClientBot() && !c->IsInGroupOrRaid(tar)) || (tar->GetOwner() && tar->GetOwner()->IsOfClientBot() && !c->IsInGroupOrRaid(tar->GetOwner())))
				) {
					c->Message(Chat::Yellow, "[%s] is an invalid target. Only players in your group or raid are eligible targets.", tar->GetCleanName());

					return;
				}
			}

			break;
	}

	const int ab_mask = ActionableBots::ABM_Type1;
	std::string actionableArg = sep->arg[ab_arg];

	if (actionableArg.empty()) {
		actionableArg = "spawned";
	}

	std::string class_race_arg = sep->arg[ab_arg];
	bool class_race_check = false;

	if (!class_race_arg.compare("byclass") || !class_race_arg.compare("byrace")) {
		class_race_check = true;
	}

	std::list<Bot*> sbl;

	if (ActionableBots::PopulateSBL(c, actionableArg, sbl, ab_mask, !class_race_check ? sep->arg[ab_arg + 1] : nullptr, class_race_check ? atoi(sep->arg[ab_arg + 1]) : 0) == ActionableBots::ABT_None) {
		return;
	}

	sbl.remove(nullptr);

	BotSpell botSpell;
	botSpell.SpellId = 0;
	botSpell.SpellIndex = 0;
	botSpell.ManaCost = 0;
	bool isSuccess = false;
	uint16 successCount = 0;
	Bot* firstFound = nullptr;

	for (auto bot_iter : sbl) {
		if (!bot_iter->IsInGroupOrRaid(c)) {
			continue;
		}

		if (bot_iter->GetBotStance() == Stance::Passive || bot_iter->GetHoldFlag() || bot_iter->GetAppearance() == eaDead || bot_iter->IsFeared() || bot_iter->IsSilenced() || bot_iter->IsAmnesiad() || bot_iter->GetHP() < 0) {
			continue;
		}

		Mob* newTar = tar;
		//LogTestDebug("{}: {} says, 'Attempting {} [{}-{}] on {}'", __LINE__, bot_iter->GetCleanName(), c->GetSpellTypeNameByID(spellType), (subType != UINT16_MAX ? c->GetSubTypeNameByID(subType) : "Standard"), (subTargetType != UINT16_MAX ? c->GetSubTypeNameByID(subTargetType) : "Standard"), (newTar ? newTar->GetCleanName() : "NOBODY")); //deleteme
		if (!SpellTypeRequiresTarget(spellType, bot_iter->GetClass())) {
			newTar = bot_iter;
		}

		if (!newTar) {
			continue;
		}

		if (
			IsBotSpellTypeBeneficial(spellType) &&
			!RuleB(Bots, CrossRaidBuffingAndHealing) &&
			!bot_iter->IsInGroupOrRaid(newTar, true)
		) {
			continue;
		}

		if (IsBotSpellTypeDetrimental(spellType, bot_iter->GetClass()) && !bot_iter->IsAttackAllowed(newTar)) {
			bot_iter->BotGroupSay(
				bot_iter,
				fmt::format(
					"I cannot attack [{}].",
					newTar->GetCleanName()
				).c_str()
			);

			continue;
		}

		LogTestDebug("{}: {} says, 'Attempting {} [{}-{}] on {}'", __LINE__, bot_iter->GetCleanName(), c->GetSpellTypeNameByID(spellType), (subType != UINT16_MAX ? c->GetSubTypeNameByID(subType) : "Standard"), (subTargetType != UINT16_MAX ? c->GetSubTypeNameByID(subTargetType) : "Standard"), (newTar ? newTar->GetCleanName() : "NOBODY")); //deleteme

		bot_iter->SetCommandedSpell(true);

		if (bot_iter->AICastSpell(newTar, 100, spellType, subTargetType, subType)) {
			if (!firstFound) {
				firstFound = bot_iter;
			}

			isSuccess = true;
			++successCount;
		}

		bot_iter->SetCommandedSpell(false);

		continue;
	}

	if (!isSuccess) {
		c->Message(
			Chat::Yellow,
			fmt::format(
				"No bots are capable of casting [{}] on {}.",
				c->GetSpellTypeNameByID(spellType),
				tar ? tar->GetCleanName() : "your target"
			).c_str()
		);
	}
	else {
		c->Message( Chat::Yellow,
			fmt::format(
				"{} {} [{}]{}",
				((successCount == 1 && firstFound) ? firstFound->GetCleanName() : (fmt::format("{}", successCount).c_str())),
				((successCount == 1 && firstFound) ? "casted" : "of your bots casted"),
				c->GetSpellTypeNameByID(spellType),
				tar ? (fmt::format(" on {}.", tar->GetCleanName()).c_str()) : "."
			).c_str()
		);
	}
}