/*
 * Copyright (C) 2008-2018 TrinityCore <https://www.trinitycore.org/>
 *
 * 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 "NPCPackets.h"
#include "Optional.h"

void WorldPackets::NPC::Hello::Read()
{
    _worldPacket >> Unit;
}

WorldPacket const* WorldPackets::NPC::TrainerList::Write()
{
    _worldPacket << TrainerGUID;
    _worldPacket << uint32(TrainerType);
    _worldPacket << uint32(TrainerID);

    _worldPacket << uint32(Spells.size());
    for (TrainerListSpell const& spell : Spells)
    {
        _worldPacket << uint32(spell.SpellID);
        _worldPacket << uint8(spell.Usable);
        _worldPacket << uint32(spell.MoneyCost);
        _worldPacket << uint8(spell.ReqLevel);
        _worldPacket << uint32(spell.ReqSkillLine);
        _worldPacket << uint32(spell.ReqSkillRank);
        _worldPacket.append(spell.ReqAbility.data(), spell.ReqAbility.size());
        _worldPacket << uint32(spell.ProfessionDialog);
        _worldPacket << uint32(spell.ProfessionButton);
    }

    _worldPacket << Greeting;

    return &_worldPacket;
}

void WorldPackets::NPC::TrainerBuySpell::Read()
{
    _worldPacket >> TrainerGUID;
    _worldPacket >> TrainerID;
    _worldPacket >> SpellID;
}

WorldPacket const* WorldPackets::NPC::TrainerBuyFailed::Write()
{
    _worldPacket << TrainerGUID;
    _worldPacket << int32(SpellID);
    _worldPacket << int32(TrainerFailedReason);

    return &_worldPacket;
}

WorldPacket const* WorldPackets::NPC::TrainerBuySucceeded::Write()
{
    _worldPacket << TrainerGUID;
    _worldPacket << int32(SpellID);

    return &_worldPacket;
}


ByteBuffer& operator<<(ByteBuffer& data, WorldPackets::NPC::VendorItem const& item)
{
    data << int32(item.MuID);
    data << int32(item.Durability);

    if (item.ExtendedCostID)
        data << uint32(*item.ExtendedCostID);

    data << uint32(item.ItemID);
    data << uint32(item.Type);
    data << uint32(item.Price);
    data << uint32(item.ItemDisplayInfoID);

    if (item.PlayerConditionFailed.is_initialized())
        data << int32(*item.PlayerConditionFailed);

    data << int32(item.Quantity);
    data << uint32(item.StackCount);

    return data;
}

WorldPacket const* WorldPackets::NPC::VendorInventory::Write()
{
    _worldPacket.WriteBit(Vendor[1]);
    _worldPacket.WriteBit(Vendor[0]);
    _worldPacket.WriteBits(Items.size(), 21);
    _worldPacket.WriteBit(Vendor[3]);
    _worldPacket.WriteBit(Vendor[6]);
    _worldPacket.WriteBit(Vendor[5]);
    _worldPacket.WriteBit(Vendor[2]);
    _worldPacket.WriteBit(Vendor[7]);

    for (VendorItem const& item : Items)
    {
        _worldPacket.WriteBit(!item.ExtendedCostID.is_initialized());
        _worldPacket.WriteBit(!item.PlayerConditionFailed.is_initialized());
    }

    _worldPacket.WriteBit(Vendor[4]);
    _worldPacket.FlushBits();

    for (VendorItem const& item : Items)
        _worldPacket << item;

    _worldPacket.WriteByteSeq(Vendor[5]);
    _worldPacket.WriteByteSeq(Vendor[4]);
    _worldPacket.WriteByteSeq(Vendor[1]);
    _worldPacket.WriteByteSeq(Vendor[0]);
    _worldPacket.WriteByteSeq(Vendor[6]);

    _worldPacket << uint8(Reason);

    _worldPacket.WriteByteSeq(Vendor[2]);
    _worldPacket.WriteByteSeq(Vendor[3]);
    _worldPacket.WriteByteSeq(Vendor[7]);

    return &_worldPacket;
}