#include "loadinfo.hpp"

namespace ESM
{

void DialInfo::load(ESMReader &esm)
{
    id = esm.getHNString("INAM");
    prev = esm.getHNString("PNAM");
    next = esm.getHNString("NNAM");

    // Not present if deleted
    if (esm.isNextSub("DATA"))
        esm.getHT(data, 12);

    // What follows is somewhat spaghetti-ish, but it's worth if for
    // an extra speedup. INFO is by far the most common record type.

    // subName is a reference to the original, so it changes whenever
    // a new sub name is read. esm.isEmptyOrGetName() will get the
    // next name for us, or return true if there are no more records.
    esm.getSubName();
    const NAME &subName = esm.retSubName();

    if (subName.val == REC_ONAM)
    {
        actor = esm.getHString();
        if (esm.isEmptyOrGetName())
            return;
    }
    if (subName.val == REC_RNAM)
    {
        race = esm.getHString();
        if (esm.isEmptyOrGetName())
            return;
    }
    if (subName.val == REC_CNAM)
    {
        clas = esm.getHString();
        if (esm.isEmptyOrGetName())
            return;
    }

    factionLess = false;
    if (subName.val == REC_FNAM)
    {
        npcFaction = esm.getHString();
        if (npcFaction == "FFFF")
            factionLess = true;
        if (esm.isEmptyOrGetName())
            return;
    }
    if (subName.val == REC_ANAM)
    {
        cell = esm.getHString();
        if (esm.isEmptyOrGetName())
            return;
    }
    if (subName.val == REC_DNAM)
    {
        pcFaction = esm.getHString();
        if (esm.isEmptyOrGetName())
            return;
    }
    if (subName.val == REC_SNAM)
    {
        sound = esm.getHString();
        if (esm.isEmptyOrGetName())
            return;
    }
    if (subName.val == REC_NAME)
    {
        response = esm.getHString();
        if (esm.isEmptyOrGetName())
            return;
    }

    while (subName.val == REC_SCVR)
    {
        SelectStruct ss;

        ss.selectRule = esm.getHString();
        esm.isEmptyOrGetName();

        if (subName.val == REC_INTV)
        {
            ss.type = VT_Int;
            esm.getHT(ss.i);
        }
        else if (subName.val == REC_FLTV)
        {
            ss.type = VT_Float;
            esm.getHT(ss.f);
        }
        else
            esm.fail(
                    "INFO.SCVR must precede INTV or FLTV, not "
                            + subName.toString());

        selects.push_back(ss);

        if (esm.isEmptyOrGetName())
            return;
    }

    if (subName.val == REC_BNAM)
    {
        resultScript = esm.getHString();
        if (esm.isEmptyOrGetName())
            return;
    }

    questStatus = QS_None;

    if (subName.val == REC_QSTN)
        questStatus = QS_Name;
    else if (subName.val == REC_QSTF)
        questStatus = QS_Finished;
    else if (subName.val == REC_QSTR)
        questStatus = QS_Restart;
    else if (subName.val == REC_DELE)
        questStatus = QS_Deleted;
    else
        esm.fail(
                "Don't know what to do with " + subName.toString()
                        + " in INFO " + id);

    if (questStatus != QS_None)
        // Skip rest of record
        esm.skipRecord();
}

void DialInfo::save(ESMWriter &esm)
{
    esm.writeHNString("INAM", id);
    esm.writeHNString("PNAM", prev);
    esm.writeHNString("NNAM", next);
    esm.writeHNT("DATA", data, 12);
    esm.writeHNOString("ONAM", actor);
    esm.writeHNOString("RNAM", race);
    esm.writeHNOString("CNAM", clas);
    esm.writeHNOString("FNAM", npcFaction);
    esm.writeHNOString("ANAM", cell);
    esm.writeHNOString("DNAM", pcFaction);
    esm.writeHNOString("SNAM", sound);
    esm.writeHNOString("NAME", response);

    for (std::vector<SelectStruct>::iterator it = selects.begin(); it != selects.end(); ++it)
    {
        esm.writeHNString("SCVR", it->selectRule);
        switch(it->type)
        {
        case VT_Int: esm.writeHNT("INTV", it->i); break;
        case VT_Float: esm.writeHNT("FLTV", it->f); break;
        default: break;
        }
    }

    esm.writeHNOString("BNAM", resultScript);
    
    switch(questStatus)
    {
    case QS_Name: esm.writeHNT("QSTN",'\1'); break;
    case QS_Finished: esm.writeHNT("QSTF", '\1'); break;
    case QS_Restart: esm.writeHNT("QSTR", '\1'); break;
    case QS_Deleted: esm.writeHNT("DELE", '\1'); break;
    default: break;
    }
}

}