cfpve.cpp

/*
 * This file is part of the TrinityCore 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 General Public License as published by the
 * Free Software Foundation; either version 2 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 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/>.
 */


#include "ScriptMgr.h"
#include "Player.h"
#include "Group.h"
#include "ObjectAccessor.h"

enum MiscCrossFactionPVE
{
    ZONE_ICECROWN_CITADEL = 4812,
    ICC_MAP_ID = 631,
    ZONE_TRIAL_OF_THE_CHAMPION = 4723,
    TOCHAMPION_MAP_ID = 650,
    ZONE_TRIAL_OF_THE_CRUSADER = 4722,
    TOCRUSADER_MAP_ID = 649,
    ZONE_PIT_OF_SARON = 4813,
    POS_MAP_ID = 658,
    ZONE_HALLS_OF_REFLECTION = 4820,
    HOR_MAP_ID = 668,
    ZONE_FORGE_OF_SOULS = 4809,
    FOS_MAP_ID = 632,
    ZONE_HALLS_OF_STONE = 4264,
    HOS_MAP_ID = 599,
    ZONE_THE_NEXUS = 4265,
    TN_MAP_ID = 576,
    ZONE_WARSONG_GULCH = 3277,
    WSG_MAP_ID = 489,
    ZONE_ARATHI_BASIN = 3358,
    AB_MAP_ID = 529
};

class CfPlayerScript : public PlayerScript
{
public:
    CfPlayerScript() : PlayerScript("CfPlayerScript") {}

    // Called when a player enters the world (logs in or teleports)
    void OnLogin(Player* player, bool firstLogin) override
    {
        HandleFactionChange(player, player->GetMapId());
    }

    // Called when a player changes zones
    void OnUpdateZone(Player* player, uint32 newZone, uint32 /*newArea*/) override
    {
        HandleFactionChange(player, newZone);
    }

private:
    // Store the original faction in a map
    std::unordered_map<uint64, uint32> originalFactionMap;

    void HandleFactionChange(Player* player, uint32 zoneOrMapId)
    {
        static const std::set<uint32> zoneSet = {
            ICC_MAP_ID, TOCHAMPION_MAP_ID, TOCRUSADER_MAP_ID, POS_MAP_ID,
            HOR_MAP_ID, FOS_MAP_ID, HOS_MAP_ID, TN_MAP_ID, WSG_MAP_ID, AB_MAP_ID
        };

        if (zoneSet.count(zoneOrMapId))
        {
            // Change faction to match the group leader
            if (Group* group = player->GetGroup())
            {
                if (Player* leader = ObjectAccessor::FindPlayer(group->GetLeaderGUID()))
                {
                    if (originalFactionMap.find(player->GetGUID()) == originalFactionMap.end())
                    {
                        // Store the original faction
                        originalFactionMap[player->GetGUID()] = player->GetFaction();
                    }
                    player->SetFaction(leader->GetFaction());
                }
            }
        }
        else
        {
            // Restore player's original faction
            auto it = originalFactionMap.find(player->GetGUID());
            if (it != originalFactionMap.end())
            {
                player->SetFaction(it->second);
                originalFactionMap.erase(it); // Clean up the map after restoring
            }
        }
    }
};

void AddSC_cfpve()
{
    new CfPlayerScript();
}