/*

* This file is part of the AzerothCore Project. See AUTHORS file for Copyright
information
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU Affero General Public License as published by the
* Free Software Foundation; either version 3 of the License, or (at your
* option) any later version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for
* more details.
*
* You should have received a copy of the GNU General Public License along
* with this program. If not, see <http://www.gnu.org/licenses/>.
*/

/* ScriptData
Name: disable_commandscript
%Complete: 100
Comment: All disable related commands
Category: commandscripts
EndScriptData */

#include "AchievementMgr.h"
#include "Chat.h"
#include "DisableMgr.h"
#include "Language.h"
#include "ObjectMgr.h"
#include "OutdoorPvP.h"
#include "Player.h"
#include "ScriptMgr.h"
#include "SpellMgr.h"

using namespace Acore::ChatCommands;

class disable_commandscript : public CommandScript

{
public:
disable_commandscript() : CommandScript("disable_commandscript") { }

ChatCommandTable GetCommands() const override

{
static ChatCommandTable removeDisableCommandTable =
{
{ "spell", HandleRemoveDisableSpellCommand,
SEC_ADMINISTRATOR, Console::Yes },
{ "quest", HandleRemoveDisableQuestCommand,
SEC_ADMINISTRATOR, Console::Yes },
{ "map", HandleRemoveDisableMapCommand,
SEC_ADMINISTRATOR, Console::Yes },
{ "battleground", HandleRemoveDisableBattlegroundCommand,
SEC_ADMINISTRATOR, Console::Yes },
{ "outdoorpvp", HandleRemoveDisableOutdoorPvPCommand,
SEC_ADMINISTRATOR, Console::Yes },
{ "vmap", HandleRemoveDisableVmapCommand,
SEC_ADMINISTRATOR, Console::Yes }
};
 static ChatCommandTable addDisableCommandTable =
{
{ "spell", HandleAddDisableSpellCommand,
SEC_ADMINISTRATOR, Console::Yes },
{ "quest", HandleAddDisableQuestCommand,
SEC_ADMINISTRATOR, Console::Yes },
{ "map", HandleAddDisableMapCommand,
SEC_ADMINISTRATOR, Console::Yes },
{ "battleground", HandleAddDisableBattlegroundCommand,
SEC_ADMINISTRATOR, Console::Yes },
{ "outdoorpvp", HandleAddDisableOutdoorPvPCommand,
SEC_ADMINISTRATOR, Console::Yes },
{ "vmap", HandleAddDisableVmapCommand,
SEC_ADMINISTRATOR, Console::Yes }
};
static ChatCommandTable disableCommandTable =
{
{ "add", addDisableCommandTable },
{ "remove", removeDisableCommandTable }
};
static ChatCommandTable commandTable =
{
{ "disable", disableCommandTable }
};
return commandTable;
}

static bool HandleAddDisables(ChatHandler* handler, uint32 entry, uint8 flags,

std::string disableComment, uint8 disableType)
{
std::string disableTypeStr = "";

switch (disableType)
{
case DISABLE_TYPE_SPELL:
{
if (!sSpellMgr->GetSpellInfo(entry))
{
handler->PSendSysMessage(LANG_COMMAND_NOSPELLFOUND);
handler->SetSentErrorMessage(true);
return false;
}
disableTypeStr = "spell";
break;
}
case DISABLE_TYPE_QUEST:
{
if (!sObjectMgr->GetQuestTemplate(entry))
{
handler->PSendSysMessage(LANG_COMMAND_QUEST_NOTFOUND,
entry);
handler->SetSentErrorMessage(true);
return false;
}
disableTypeStr = "quest";
break;
}
case DISABLE_TYPE_MAP:
{
 if (!sMapStore.LookupEntry(entry))
{
handler->PSendSysMessage(LANG_COMMAND_NOMAPFOUND);
handler->SetSentErrorMessage(true);
return false;
}
disableTypeStr = "map";
break;
}
case DISABLE_TYPE_BATTLEGROUND:
{
if (!sBattlemasterListStore.LookupEntry(entry))
{
handler-
>PSendSysMessage(LANG_COMMAND_NO_BATTLEGROUND_FOUND);
handler->SetSentErrorMessage(true);
return false;
}
disableTypeStr = "battleground";
break;
}
case DISABLE_TYPE_OUTDOORPVP:
{
if (entry > MAX_OUTDOORPVP_TYPES)
{
handler-
>PSendSysMessage(LANG_COMMAND_NO_OUTDOOR_PVP_FORUND);
handler->SetSentErrorMessage(true);
return false;
}
disableTypeStr = "outdoorpvp";
break;
}
case DISABLE_TYPE_VMAP:
{
if (!sMapStore.LookupEntry(entry))
{
handler->PSendSysMessage(LANG_COMMAND_NOMAPFOUND);
handler->SetSentErrorMessage(true);
return false;
}
disableTypeStr = "vmap";
break;
}
default:
break;
}

WorldDatabasePreparedStatement* stmt = nullptr;

stmt = WorldDatabase.GetPreparedStatement(WORLD_SEL_DISABLES);
stmt->setUInt32(0, entry);
stmt->setUInt8(1, disableType);
PreparedQueryResult result = WorldDatabase.Query(stmt);
if (result)
{
handler->PSendSysMessage("This %s (Id: %u) is already disabled.",
disableTypeStr.c_str(), entry);
handler->SetSentErrorMessage(true);
return false;
 }

stmt = WorldDatabase.GetPreparedStatement(WORLD_INS_DISABLES);
stmt->setUInt32(0, entry);
stmt->setUInt8(1, disableType);
stmt->setUInt16(2, flags);
stmt->setString(3, disableComment);
WorldDatabase.Execute(stmt);

handler->PSendSysMessage("Add Disabled %s (Id: %u) for reason %s",

disableTypeStr.c_str(), entry, disableComment.c_str());
return true;
}

static bool HandleAddDisableSpellCommand(ChatHandler* handler, uint32 entry,

uint8 flags, std::string disableComment)
{
return HandleAddDisables(handler, entry, flags, disableComment,
DISABLE_TYPE_SPELL);
}

static bool HandleAddDisableQuestCommand(ChatHandler* handler, uint32 entry,

uint8 flags, std::string disableComment)
{
return HandleAddDisables(handler, entry, flags, disableComment,
DISABLE_TYPE_QUEST);
}

static bool HandleAddDisableMapCommand(ChatHandler* handler, uint32 entry,

uint8 flags, std::string disableComment)
{
return HandleAddDisables(handler, entry, flags, disableComment,
DISABLE_TYPE_MAP);
}

static bool HandleAddDisableBattlegroundCommand(ChatHandler* handler, uint32

entry, uint8 flags, std::string disableComment)
{
return HandleAddDisables(handler, entry, flags, disableComment,
DISABLE_TYPE_BATTLEGROUND);
}

static bool HandleAddDisableAchievementCriteriaCommand(ChatHandler* handler,

uint32 entry, uint8 flags, std::string disableComment)
{
return HandleAddDisables(handler, entry, flags, disableComment,
DISABLE_TYPE_ACHIEVEMENT_CRITERIA);
}

static bool HandleAddDisableOutdoorPvPCommand(ChatHandler* handler, uint32

entry, uint8 flags, std::string disableComment)
{
HandleAddDisables(handler, entry, flags, disableComment,
DISABLE_TYPE_OUTDOORPVP);
return true;
}

static bool HandleAddDisableVmapCommand(ChatHandler* handler, uint32 entry,

uint8 flags, std::string disableComment)
 {
return HandleAddDisables(handler, entry, flags, disableComment,
DISABLE_TYPE_VMAP);
}

static bool HandleRemoveDisables(ChatHandler* handler, uint32 entry, uint8

disableType)
{
std::string disableTypeStr = "";

switch (disableType)
{
case DISABLE_TYPE_SPELL:
disableTypeStr = "spell";
break;
case DISABLE_TYPE_QUEST:
disableTypeStr = "quest";
break;
case DISABLE_TYPE_MAP:
disableTypeStr = "map";
break;
case DISABLE_TYPE_BATTLEGROUND:
disableTypeStr = "battleground";
break;
case DISABLE_TYPE_ACHIEVEMENT_CRITERIA:
disableTypeStr = "achievement criteria";
break;
case DISABLE_TYPE_OUTDOORPVP:
disableTypeStr = "outdoorpvp";
break;
case DISABLE_TYPE_VMAP:
disableTypeStr = "vmap";
break;
}

WorldDatabasePreparedStatement* stmt = nullptr;

stmt = WorldDatabase.GetPreparedStatement(WORLD_SEL_DISABLES);
stmt->setUInt32(0, entry);
stmt->setUInt8(1, disableType);
PreparedQueryResult result = WorldDatabase.Query(stmt);
if (!result)
{
handler->PSendSysMessage("This %s (Id: %u) is not disabled.",
disableTypeStr.c_str(), entry);
handler->SetSentErrorMessage(true);
return false;
}

stmt = WorldDatabase.GetPreparedStatement(WORLD_DEL_DISABLES);
stmt->setUInt32(0, entry);
stmt->setUInt8(1, disableType);
WorldDatabase.Execute(stmt);

handler->PSendSysMessage("Remove Disabled %s (Id: %u)",

disableTypeStr.c_str(), entry);
return true;
}

static bool HandleRemoveDisableSpellCommand(ChatHandler* handler, uint32 entry)

 {
return HandleRemoveDisables(handler, entry, DISABLE_TYPE_SPELL);
}

static bool HandleRemoveDisableQuestCommand(ChatHandler* handler, uint32 entry)

{
return HandleRemoveDisables(handler, entry, DISABLE_TYPE_QUEST);
}

static bool HandleRemoveDisableMapCommand(ChatHandler* handler, uint32 entry)

{
return HandleRemoveDisables(handler, entry, DISABLE_TYPE_MAP);
}

static bool HandleRemoveDisableBattlegroundCommand(ChatHandler* handler, uint32

entry)
{
return HandleRemoveDisables(handler, entry, DISABLE_TYPE_BATTLEGROUND);
}

static bool HandleRemoveDisableAchievementCriteriaCommand(ChatHandler* handler,

uint32 entry)
{
return HandleRemoveDisables(handler, entry,
DISABLE_TYPE_ACHIEVEMENT_CRITERIA);
}

static bool HandleRemoveDisableOutdoorPvPCommand(ChatHandler* handler, uint32

entry)
{
return HandleRemoveDisables(handler, entry, DISABLE_TYPE_OUTDOORPVP);
}

static bool HandleRemoveDisableVmapCommand(ChatHandler* handler, uint32 entry)

{
return HandleRemoveDisables(handler, entry, DISABLE_TYPE_VMAP);
}
};

void AddSC_disable_commandscript()
{
new disable_commandscript();
}