#include "ScriptPCH.h"
#define TYPE_NON_COMBAT_PET 12
class remover : public WorldScript
{
public:
	remover() : WorldScript("remover") {}
	void OnStartup()
	{
		SpellInfoMap spellInfo;
		spellInfo.resize(sSpellStore.GetNumRows(), NULL);
		for(int i = 0; i < sSpellStore.GetNumRows(); ++i)
		{
			if (SpellEntry const* spellEntry = sSpellStore.LookupEntry(i))
				spellInfo[i] = new SpellInfo(spellEntry);
		}

		sLog->outError(LOG_FILTER_GENERAL, "loaded %u spells, checking for all mounting spells...", spellInfo.size());

		std::vector<uint32> mountedSpells;

		for(SpellInfoMap::iterator itr = spellInfo.begin(); itr != spellInfo.end(); ++itr)
		{
			for(int i = 0; i < MAX_SPELL_EFFECTS; ++i)
				if(*itr && (*itr)->Effects[i].ApplyAuraName == SPELL_AURA_MOUNTED)
					mountedSpells.push_back((*itr)->Id);
		}

		sLog->outError(LOG_FILTER_GENERAL, "Found %u mounted spells, removing these spells from players...", mountedSpells.size());

		uint32 counter = 0;

		for(std::vector<uint32>::iterator itr = mountedSpells.begin(); itr != mountedSpells.end(); ++itr)
		{
			QueryResult result = CharacterDatabase.PQuery("SELECT * FROM `character_spell` WHERE `spell` = %u", *itr);
			if(!result)
				continue;
			CharacterDatabase.PExecute("DELETE FROM `character_spell` WHERE `spell` = %u", *itr);
			++counter;
		}

		sLog->outError(LOG_FILTER_GENERAL, "Removed %u spell(s) from characters", counter);

		counter = 0;

		sLog->outError(LOG_FILTER_GENERAL, "Checking for companions...");

		std::vector<uint32> companionSpells;

		for(SpellInfoMap::iterator itr = spellInfo.begin(); itr != spellInfo.end(); ++itr)
		{
			if(SpellEntry const* spellEntry = sSpellStore.LookupEntry((*itr)->Id))
			{
				for(int i = 0; i < 3; ++i)
				{
					if(const CreatureTemplate* creatureTemplate = sObjectMgr->GetCreatureTemplate(spellEntry->EffectMiscValue[i]))
						if(creatureTemplate->type == TYPE_NON_COMBAT_PET)
							companionSpells.push_back((*itr)->Id);
				}
			}
		}

		sLog->outError(LOG_FILTER_GENERAL, "Found %u companion spells, removing them from every character..", companionSpells.size());

		for(std::vector<uint32>::iterator itr = companionSpells.begin(); itr != companionSpells.end(); ++itr)
		{
			QueryResult result = CharacterDatabase.PQuery("SELECT * FROM `character_spell` WHERE `spell` = %u", *itr);
			if(!result)
				continue;
			CharacterDatabase.PExecute("DELETE FROM `character_spell` WHERE `spell` = %u", *itr);
			++counter;
		}

		sLog->outError(LOG_FILTER_GENERAL, "Removed %u spell(s) from characters", counter);
	}
};

void AddSC_remover()
{
	new remover;
}