#include "Chat.h"
#include "MpLogger.h"
#include "Map.h"
#include "MpDataStore.h"
#include "MythicPlus.h"
#include "Player.h"
#include "ScriptMgr.h"


class MythicPlus_AllMapScript : public AllMapScript
{
public:
    MythicPlus_AllMapScript() : AllMapScript("MythicPlus_AllMapScript")
    {
    }

    void OnCreateMap(Map* map) { }

    /**
     * When a player enters the map check it needs to set up the instance data
     */
    void OnPlayerEnterAll(Map* map, Player* player)
    {
        if (!sMythicPlus->IsMapEligible(map)) {
            return;
        }

        if(!sMythicPlus->IsDifficultySet(player)) {
            return;
        }

        Group* group = player->GetGroup();
        if (group) {
            MpLogger::debug("Player {} entered map {} in groupLeader {}", player->GetName(), map->GetMapName(), group->GetLeaderName());
        } else {
            return;
        }

        // if there is not any group data for this group then just bail
        const MpGroupData* groupData = sMpDataStore->GetGroupData(group->GetGUID());
        if (!groupData) {
            return;
        }

        // Check if we already have mythic instance data set for this map and group
        MpInstanceData* existingData = sMpDataStore->GetInstanceData(map->GetId(), map->GetInstanceId());
        if (existingData) {
            if(player->GetName() == group->GetLeaderName()) {
                MpLogger::debug("Instance data already set for Map: {} InstanceId: {} for GroupLeader: {} ",
                    map->GetMapName(),
                    map->GetInstanceId(),
                    group->GetLeaderName()
                );
            }
            return;
        }

        MpInstanceData instanceData;
        switch(groupData->difficulty) {
            case MP_DIFFICULTY_MYTHIC:
                instanceData.boss = sMythicPlus->mythicBossModifiers;
                instanceData.creature = sMythicPlus->mythicDungeonModifiers;
                instanceData.itemRewards = sMythicPlus->EnableItemRewards;
                instanceData.deathLimits = sMythicPlus->mythicDeathAllowance;
                instanceData.itemOffset = sMythicPlus->mythicItemOffset;
                break;
            case MP_DIFFICULTY_LEGENDARY:
                instanceData.boss = sMythicPlus->legendaryBossModifiers;
                instanceData.creature = sMythicPlus->legendaryDungeonModifiers;
                instanceData.itemRewards = sMythicPlus->EnableItemRewards;
                instanceData.deathLimits = sMythicPlus->legendaryDeathAllowance;
                instanceData.itemOffset = sMythicPlus->legendaryItemOffset;
                break;
            case MP_DIFFICULTY_ASCENDANT:
                instanceData.boss = sMythicPlus->ascendantBossModifiers;
                instanceData.creature = sMythicPlus->ascendantDungeonModifiers;
                instanceData.itemRewards = sMythicPlus->EnableItemRewards;
                instanceData.deathLimits = sMythicPlus->ascendantDeathAllowance;
                instanceData.itemOffset = sMythicPlus->ascendantItemOffset;
                break;
            default:
                MpLogger::debug("No difficulty set for group {}", group->GetGUID().GetCounter());
                return;
        }

        instanceData.difficulty = groupData->difficulty;

        // Attempt to cast map to InstanceMap, making sure it is not null
        instanceData.instance = dynamic_cast<InstanceMap*>(sMapMgr->FindMap(map->GetId(), map->GetInstanceId()));
        if (!instanceData.instance)
        {
            MpLogger::error("Failed to find InstanceMap for map ID {} and instance ID {}.", map->GetId(), map->GetInstanceId());
            return;
        }

        MpLogger::debug("Setting up instance data for group {} for map {} instance {} data {}",
            group->GetGUID().GetCounter(),
            map->GetMapName(),
            map->GetInstanceId(),
            instanceData.ToString()
        );
        sMpDataStore->AddInstanceData(map->GetId(), map->GetInstanceId(), instanceData);

        // Once we have instance data set we can scale the remaining characters in our instance
        sMythicPlus->ScaleRemaining(player, &instanceData);
    }

    // When an instance is destroyed remove the instance data from the data store
    virtual void OnDestroyInstance(MapInstanced* /*mapInstanced*/, Map* map)
    {
        if (!sMythicPlus->IsMapEligible(map)) {
            return;
        }
        sMpDataStore->RemoveInstanceData(map->GetId(), map->GetInstanceId());
    }
};

void Add_MP_AllMapScripts()
{
    MpLogger::debug("Add_MP_AllMapScripts()");
    new MythicPlus_AllMapScript();
}