openmw-tes3mp/apps/openmw-mp/Script/Functions/Spells.cpp

#include "Spells.hpp"
#include <apps/openmw-mp/Script/ScriptFunctions.hpp>
#include <components/openmw-mp/NetworkMessages.hpp>
#include <apps/openmw-mp/Networking.hpp>
#include <components/misc/stringops.hpp>

using namespace mwmp;

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

    return player->packetSpells.count;
}

void SpellFunctions::AddSpell(unsigned short pid, const char* spellId) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    ESM::Spell spell;
    spell.mId = spellId;

    player->packetSpellsBuffer.spells.push_back(spell);
    player->packetSpellsBuffer.action = PacketSpells::ADD;
}

void SpellFunctions::RemoveSpell(unsigned short pid, const char* spellId) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    ESM::Spell spell;
    spell.mId = spellId;

    player->packetSpellsBuffer.spells.push_back(spell);
    player->packetSpellsBuffer.action = PacketSpells::REMOVE;
}

void SpellFunctions::ClearSpellbook(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    player->packetSpellsBuffer.spells.clear();
    player->packetSpellsBuffer.action = PacketSpells::SET;
}

const char *SpellFunctions::GetSpellId(unsigned short pid, unsigned int i) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, "");

    if (i >= player->packetSpells.count)
        return "invalid";

    return player->packetSpells.spells.at(i).mId.c_str();
}

void SpellFunctions::SendSpells(unsigned short pid) noexcept
{
    Player *player;
    GET_PLAYER(pid, player, );

    std::swap(player->packetSpells, player->packetSpellsBuffer);
    mwmp::Networking::get().getPlayerController()->GetPacket(ID_GAME_SPELLBOOK)->Send(player, false);
    player->packetSpells = std::move(player->packetSpellsBuffer);
    player->packetSpellsBuffer.spells.clear();
}