#include "Stats.hpp"

#include <iostream>

#include <components/esm/attr.hpp>
#include <components/esm/loadskil.hpp>
#include <components/misc/stringops.hpp>
#include <components/openmw-mp/Log.hpp>
#include <components/openmw-mp/NetworkMessages.hpp>

#include <apps/openmw-mp/Networking.hpp>
#include <apps/openmw-mp/Script/ScriptFunctions.hpp>

using namespace std;
using namespace ESM;

int StatsFunctions::GetAttributeCount() noexcept
{
    return Attribute::Length;
}

int StatsFunctions::GetSkillCount() noexcept
{
    return Skill::Length;
}

int StatsFunctions::GetAttributeId(const char *name) noexcept
{
    for (int x = 0; x < Attribute::Length; x++)
    {
        if (Misc::StringUtils::ciEqual(name, Attribute::sAttributeNames[x]))
        {
            return x;
        }
    }

    return -1;
}

int StatsFunctions::GetSkillId(const char *name) noexcept
{
    for (int x = 0; x < Skill::Length; x++)
    {
        if (Misc::StringUtils::ciEqual(name, Skill::sSkillNames[x]))
        {
            return x;
        }
    }

    return -1;
}

const char *StatsFunctions::GetAttributeName(unsigned short attributeId) noexcept
{
    if (attributeId >= Attribute::Length)
        return "invalid";

    return Attribute::sAttributeNames[attributeId].c_str();
}

const char *StatsFunctions::GetSkillName(unsigned short skillId) noexcept
{
    if (skillId >= Skill::Length)
        return "invalid";

    return Skill::sSkillNames[skillId].c_str();
}

const char *StatsFunctions::GetName(unsigned short pid) noexcept
{

    Player *player;
    GET_PLAYER(pid, player, 0);

    return player->npc.mName.c_str();
}

const char *StatsFunctions::GetRace(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    return player->npc.mRace.c_str();
}

const char *StatsFunctions::GetHead(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    return player->npc.mHead.c_str();
}

const char *StatsFunctions::GetHairstyle(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    return player->npc.mHair.c_str();
}

int StatsFunctions::GetIsMale(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, false);

    return player->npc.isMale();
}

const char *StatsFunctions::GetBirthsign(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    return player->birthsign.c_str();
}

int StatsFunctions::GetLevel(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    return player->creatureStats.mLevel;
}

int StatsFunctions::GetLevelProgress(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    return player->npcStats.mLevelProgress;
}

double StatsFunctions::GetHealthBase(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0.0f);

    return player->creatureStats.mDynamic[0].mBase;
}

double StatsFunctions::GetHealthCurrent(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0.0f);

    return player->creatureStats.mDynamic[0].mCurrent;
}

double StatsFunctions::GetMagickaBase(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0.0f);

    return player->creatureStats.mDynamic[1].mBase;
}

double StatsFunctions::GetMagickaCurrent(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0.0f);

    return player->creatureStats.mDynamic[1].mCurrent;
}

double StatsFunctions::GetFatigueBase(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0.0f);

    return player->creatureStats.mDynamic[2].mBase;
}

double StatsFunctions::GetFatigueCurrent(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0.0f);

    return player->creatureStats.mDynamic[2].mCurrent;
}

int StatsFunctions::GetAttributeBase(unsigned short pid, unsigned short attributeId) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    if (attributeId >= Attribute::Length)
        return 0;

    return player->creatureStats.mAttributes[attributeId].mBase;
}

int StatsFunctions::GetAttributeModifier(unsigned short pid, unsigned short attributeId) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    if (attributeId >= Attribute::Length)
        return 0;

    return player->creatureStats.mAttributes[attributeId].mMod;
}

int StatsFunctions::GetSkillBase(unsigned short pid, unsigned short skillId) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    if (skillId >= Skill::Length)
        return 0;

    return player->npcStats.mSkills[skillId].mBase;
}

int StatsFunctions::GetSkillModifier(unsigned short pid, unsigned short skillId) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    if (skillId >= Skill::Length)
        return 0;

    return player->npcStats.mSkills[skillId].mMod;
}

double StatsFunctions::GetSkillProgress(unsigned short pid, unsigned short skillId) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0.0f);

    if (skillId >= Skill::Length)
        return 0;

    return player->npcStats.mSkills[skillId].mProgress;
}

int StatsFunctions::GetSkillIncrease(unsigned short pid, unsigned int attributeId) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    if (attributeId > Attribute::Length)
        return 0;

    return player->npcStats.mSkillIncrease[attributeId];
}

int StatsFunctions::GetBounty(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    return player->npcStats.mBounty;
}

const char *StatsFunctions::GetDeathReason(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, 0);

    return player->deathReason.c_str();
}

void StatsFunctions::SetName(unsigned short pid, const char *name) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    if (player->npc.mName == name)
        return;

    player->npc.mName = name;
}

void StatsFunctions::SetRace(unsigned short pid, const char *race) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    if (player->npc.mRace == race)
        return;

    LOG_MESSAGE_SIMPLE(Log::LOG_VERBOSE, "Setting race for %s: %s -> %s", player->npc.mName.c_str(),
                       player->npc.mRace.c_str(), race);

    player->npc.mRace = race;
}

void StatsFunctions::SetHead(unsigned short pid, const char *head) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    if (player->npc.mHead == head)
        return;

    player->npc.mHead = head;
}

void StatsFunctions::SetHairstyle(unsigned short pid, const char *hairstyle) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    if (player->npc.mHair == hairstyle)
        return;

    player->npc.mHair = hairstyle;
}

void StatsFunctions::SetIsMale(unsigned short pid, int state) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    player->npc.setIsMale(state == true);
}

void StatsFunctions::SetBirthsign(unsigned short pid, const char *sign) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    if (player->birthsign == sign)
        return;

    player->birthsign = sign;
}

void StatsFunctions::SetResetStats(unsigned short pid, bool resetStats) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    player->resetStats = resetStats;
}

void StatsFunctions::SetLevel(unsigned short pid, int value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    player->creatureStats.mLevel = value;
}

void StatsFunctions::SetLevelProgress(unsigned short pid, int value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    player->npcStats.mLevelProgress = value;
}

void StatsFunctions::SetHealthBase(unsigned short pid, double value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    player->creatureStats.mDynamic[0].mBase = value;

    if (!Utils::vectorContains(&player->statsDynamicIndexChanges, 0))
        player->skillIndexChanges.push_back(0);
}

void StatsFunctions::SetHealthCurrent(unsigned short pid, double value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    player->creatureStats.mDynamic[0].mCurrent = value;

    if (!Utils::vectorContains(&player->statsDynamicIndexChanges, 0))
        player->skillIndexChanges.push_back(0);
}

void StatsFunctions::SetMagickaBase(unsigned short pid, double value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    player->creatureStats.mDynamic[1].mBase = value;

    if (!Utils::vectorContains(&player->statsDynamicIndexChanges, 1))
        player->skillIndexChanges.push_back(1);
}

void StatsFunctions::SetMagickaCurrent(unsigned short pid, double value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    player->creatureStats.mDynamic[1].mCurrent = value;

    if (!Utils::vectorContains(&player->statsDynamicIndexChanges, 1))
        player->skillIndexChanges.push_back(1);
}

void StatsFunctions::SetFatigueBase(unsigned short pid, double value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    player->creatureStats.mDynamic[2].mBase = value;

    if (!Utils::vectorContains(&player->statsDynamicIndexChanges, 2))
        player->skillIndexChanges.push_back(2);
}

void StatsFunctions::SetFatigueCurrent(unsigned short pid, double value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    player->creatureStats.mDynamic[2].mCurrent = value;

    if (!Utils::vectorContains(&player->statsDynamicIndexChanges, 2))
        player->skillIndexChanges.push_back(2);
}

void StatsFunctions::SetAttributeBase(unsigned short pid, unsigned short attributeId, int value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    if (attributeId >= Attribute::Length)
        return;

    player->creatureStats.mAttributes[attributeId].mBase = value;

    if (!Utils::vectorContains(&player->attributeIndexChanges, attributeId))
        player->attributeIndexChanges.push_back(attributeId);
}

void StatsFunctions::ClearAttributeModifier(unsigned short pid, unsigned short attributeId) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    if (attributeId >= Attribute::Length)
        return;

    player->creatureStats.mAttributes[attributeId].mMod = 0;

    if (!Utils::vectorContains(&player->attributeIndexChanges, attributeId))
        player->attributeIndexChanges.push_back(attributeId);
}

void StatsFunctions::SetSkillBase(unsigned short pid, unsigned short skillId, int value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    if (skillId >= Skill::Length)
        return;

    player->npcStats.mSkills[skillId].mBase = value;

    if (!Utils::vectorContains(&player->skillIndexChanges, skillId))
        player->skillIndexChanges.push_back(skillId);
}

void StatsFunctions::ClearSkillModifier(unsigned short pid, unsigned short skillId) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    if (skillId >= Skill::Length)
        return;

    player->npcStats.mSkills[skillId].mMod = 0;

    if (!Utils::vectorContains(&player->skillIndexChanges, skillId))
        player->skillIndexChanges.push_back(skillId);
}

void StatsFunctions::SetSkillProgress(unsigned short pid, unsigned short skillId, double value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    if (skillId >= Skill::Length)
        return;

    player->npcStats.mSkills[skillId].mProgress = value;

    if (!Utils::vectorContains(&player->skillIndexChanges, skillId))
        player->skillIndexChanges.push_back(skillId);
}

void StatsFunctions::SetSkillIncrease(unsigned short pid, unsigned int attributeId, int value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    if (attributeId > Attribute::Length)
        return;

    player->npcStats.mSkillIncrease[attributeId] = value;

    if (!Utils::vectorContains(&player->attributeIndexChanges, attributeId))
        player->attributeIndexChanges.push_back(attributeId);
}

void StatsFunctions::SetBounty(unsigned short pid, int value) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    player->npcStats.mBounty = value;
}

void StatsFunctions::SetCharGenStage(unsigned short pid, int currentStage, int endStage) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    player->charGenState.currentStage = currentStage;
    player->charGenState.endStage = endStage;
    player->charGenState.isFinished = false;

    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_CHARGEN)->setPlayer(player);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_CHARGEN)->Send(false);
}

void StatsFunctions::SendBaseInfo(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_BASEINFO)->setPlayer(player);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_BASEINFO)->Send(false);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_BASEINFO)->Send(true);
}

void StatsFunctions::SendStatsDynamic(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_STATS_DYNAMIC)->setPlayer(player);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_STATS_DYNAMIC)->Send(false);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_STATS_DYNAMIC)->Send(true);

    player->statsDynamicIndexChanges.clear();
}

void StatsFunctions::SendAttributes(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_ATTRIBUTE)->setPlayer(player);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_ATTRIBUTE)->Send(false);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_ATTRIBUTE)->Send(true);

    player->attributeIndexChanges.clear();
}

void StatsFunctions::SendSkills(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player,);

    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_SKILL)->setPlayer(player);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_SKILL)->Send(false);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_SKILL)->Send(true);

    player->skillIndexChanges.clear();
}

void StatsFunctions::SendLevel(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_LEVEL)->setPlayer(player);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_LEVEL)->Send(false);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_LEVEL)->Send(true);
}

void StatsFunctions::SendBounty(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_BOUNTY)->setPlayer(player);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_BOUNTY)->Send(false);
    mwmp::Networking::get().getPlayerPacketController()->GetPacket(ID_PLAYER_BOUNTY)->Send(true);
}