#include "loadbook.hpp"

#include "esmreader.hpp"
#include "esmwriter.hpp"

#include <components/esm3/loadskil.hpp>
#include <components/misc/concepts.hpp>

namespace ESM
{
    namespace
    {
        struct EsmBKDTstruct
        {
            float mWeight;
            int32_t mValue, mIsScroll, mSkillId, mEnchant;
        };

        void toBinary(const Book::BKDTstruct& src, EsmBKDTstruct& dst)
        {
            dst.mWeight = src.mWeight;
            dst.mValue = src.mValue;
            dst.mIsScroll = src.mIsScroll;
            dst.mSkillId = ESM::Skill::refIdToIndex(src.mSkillId);
            dst.mEnchant = src.mEnchant;
        }

        void fromBinary(const EsmBKDTstruct& src, Book::BKDTstruct& dst)
        {
            dst.mWeight = src.mWeight;
            dst.mValue = src.mValue;
            dst.mIsScroll = src.mIsScroll;
            dst.mSkillId = ESM::Skill::indexToRefId(src.mSkillId);
            dst.mEnchant = src.mEnchant;
        }
    }

    template <Misc::SameAsWithoutCvref<EsmBKDTstruct> T>
    void decompose(T&& v, const auto& f)
    {
        f(v.mWeight, v.mValue, v.mIsScroll, v.mSkillId, v.mEnchant);
    }

    template <Misc::SameAsWithoutCvref<Book::BKDTstruct> T>
    void decompose(T&& v, const auto& f)
    {
        f(v.mWeight, v.mValue, v.mIsScroll, v.mEnchant);
    }

    void Book::load(ESMReader& esm, bool& isDeleted)
    {
        isDeleted = false;
        mRecordFlags = esm.getRecordFlags();

        bool hasName = false;
        bool hasData = false;
        while (esm.hasMoreSubs())
        {
            esm.getSubName();
            switch (esm.retSubName().toInt())
            {
                case SREC_NAME:
                    mId = esm.getRefId();
                    hasName = true;
                    break;
                case fourCC("MODL"):
                    mModel = esm.getHString();
                    break;
                case fourCC("FNAM"):
                    mName = esm.getHString();
                    break;
                case fourCC("BKDT"):
                {
                    if (esm.getFormatVersion() <= MaxFixedStatsFormatVersion)
                    {
                        EsmBKDTstruct data;
                        esm.getSubComposite(data);
                        fromBinary(data, mData);
                    }
                    else
                    {
                        esm.getSubComposite(mData);
                        mData.mSkillId = esm.getHNORefId("SKIL");
                    }
                    hasData = true;
                    break;
                }
                case fourCC("SCRI"):
                    mScript = esm.getRefId();
                    break;
                case fourCC("ITEX"):
                    mIcon = esm.getHString();
                    break;
                case fourCC("ENAM"):
                    mEnchant = esm.getRefId();
                    break;
                case fourCC("TEXT"):
                    mText = esm.getHString();
                    break;
                case SREC_DELE:
                    esm.skipHSub();
                    isDeleted = true;
                    break;
                default:
                    esm.fail("Unknown subrecord");
                    break;
            }
        }

        if (!hasName)
            esm.fail("Missing NAME subrecord");
        if (!hasData && !isDeleted)
            esm.fail("Missing BKDT subrecord");
    }
    void Book::save(ESMWriter& esm, bool isDeleted) const
    {
        esm.writeHNCRefId("NAME", mId);

        if (isDeleted)
        {
            esm.writeHNString("DELE", "", 3);
            return;
        }

        esm.writeHNCString("MODL", mModel.getOriginal());
        esm.writeHNOCString("FNAM", mName);
        if (esm.getFormatVersion() <= MaxFixedStatsFormatVersion)
        {
            EsmBKDTstruct data;
            toBinary(mData, data);
            esm.writeNamedComposite("BKDT", data);
        }
        else
        {
            esm.writeNamedComposite("BKDT", mData);
            esm.writeHNORefId("SKIL", mData.mSkillId);
        }
        esm.writeHNOCRefId("SCRI", mScript);
        esm.writeHNOCString("ITEX", mIcon.getOriginal());
        esm.writeHNOString("TEXT", mText);
        esm.writeHNOCRefId("ENAM", mEnchant);
    }

    void Book::blank()
    {
        mRecordFlags = 0;
        mData.mWeight = 0;
        mData.mValue = 0;
        mData.mIsScroll = 0;
        mData.mSkillId = {};
        mData.mEnchant = 0;
        mName.clear();
        mModel.clear();
        mIcon.clear();
        mScript = ESM::RefId();
        mEnchant = ESM::RefId();
        mText.clear();
    }
}