#ifndef __COMMON_FSM2_HPP__
#define __COMMON_FSM2_HPP__

#include "PAK.hpp"
#include "DNACommon.hpp"
#include "athena/FileWriter.hpp"

namespace DataSpec::DNAFSM2
{
struct IFSM : BigDNAVYaml
{
    Delete _d;
};

template <class IDType>
struct AT_SPECIALIZE_PARMS(DataSpec::UniqueID32, DataSpec::UniqueID64) FSM2 : BigDNA
{
    struct Header : BigDNA
    {
        AT_DECL_DNA_YAML
        DNAFourCC magic = FOURCC('FSM2');
        Value<atUint32> version;
    } header;

    struct CommonStruct : BigDNA
    {
        AT_DECL_DNA_YAML
        String<-1> name;
        Value<atUint32> unknown;
    };

    struct FSMV1 : IFSM
    {
        AT_DECL_DNA_YAML
        AT_DECL_DNAV
        Value<atUint32> stateCount;
        Value<atUint32> unknown1Count;
        Value<atUint32> unknown2Count;
        Value<atUint32> unknown3Count;
        struct State : BigDNA
        {
            AT_DECL_DNA_YAML
            String<-1> name;
            Value<atUint32> unknownCount;
            Vector<CommonStruct, AT_DNA_COUNT(unknownCount)> unknown;
        };

        struct Unknown1 : BigDNA
        {
            AT_DECL_DNA_YAML
            String<-1> name;
            Value<float> unknown1;
            Value<atUint32> unknown2Count;
            Vector<CommonStruct, AT_DNA_COUNT(unknown2Count)> unknown2;
            Value<atUint8> unknown3;
        };

        struct Unknown2 : BigDNA
        {
            AT_DECL_DNA_YAML
            String<-1> name;
            Value<atUint32> unknownCount;
            Vector<CommonStruct, AT_DNA_COUNT(unknownCount)> unknown;
        };

        struct Unknown3 : BigDNA
        {
            AT_DECL_DNA_YAML
            String<-1> name;
            Value<atUint32> unknownCount;
            Vector<CommonStruct, AT_DNA_COUNT(unknownCount)> unknown;
            Value<IDType> fsmId;
        };

        Vector<State, AT_DNA_COUNT(stateCount)> states;
        Vector<Unknown1, AT_DNA_COUNT(unknown1Count)> unknown1;
        Vector<Unknown2, AT_DNA_COUNT(unknown2Count)> unknown2;
        Vector<Unknown3, AT_DNA_COUNT(unknown3Count)> unknown3;
    };

    struct FSMV2 : IFSM
    {
        AT_DECL_DNA_YAML
        AT_DECL_DNAV
        Value<atUint32> stateCount;
        Value<atUint32> unknown1Count;
        Value<atUint32> unknown2Count;
        Value<atUint32> unknown3Count;
        struct State : BigDNA
        {
            AT_DECL_DNA_YAML
            String<-1> name;
            Value<atUint32> unknown1;
            Value<atUint32> unknown2;
            Value<atUint32> unknown3;
            Value<atUint32> unknown4;
            Value<atUint32> unknown5Count;
            Vector<CommonStruct, AT_DNA_COUNT(unknown5Count)> unknown5;
        };

        struct Unknown1 : BigDNA
        {
            AT_DECL_DNA_YAML
            String<-1> name;
            Value<atUint32> unknown1;
            Value<atUint32> unknown2;
            Value<atUint32> unknown3;
            Value<atUint32> unknown4;
            Value<float> unknown5;
            Value<atUint32> unknown6Count;
            Vector<CommonStruct, AT_DNA_COUNT(unknown6Count)> unknown6;
            Value<atUint8> unknown7;
        };

        struct Unknown2 : BigDNA
        {
            AT_DECL_DNA_YAML
            String<-1> name;
            Value<atUint32> unknown1;
            Value<atUint32> unknown2;
            Value<atUint32> unknown3;
            Value<atUint32> unknown4;
            Value<atUint32> unknown5Count;
            Vector<CommonStruct, AT_DNA_COUNT(unknown5Count)> unknown5;
        };

        struct Unknown3 : BigDNA
        {
            AT_DECL_DNA_YAML
            String<-1> name;
            Value<atUint32> unknown1;
            Value<atUint32> unknown2;
            Value<atUint32> unknown3;
            Value<atUint32> unknown4;
            Value<atUint32> unknown5Count;
            Vector<CommonStruct, AT_DNA_COUNT(unknown5Count)> unknown5;
            Value<IDType> fsmId;
        };

        Vector<State, AT_DNA_COUNT(stateCount)> states;
        Vector<Unknown1, AT_DNA_COUNT(unknown1Count)> unknown1;
        Vector<Unknown2, AT_DNA_COUNT(unknown2Count)> unknown2;
        Vector<Unknown3, AT_DNA_COUNT(unknown3Count)> unknown3;
    };

    std::unique_ptr<IFSM> detail;
    AT_DECL_EXPLICIT_DNA_YAML
};

template <class IDType>
bool ExtractFSM2(PAKEntryReadStream& rs, const hecl::ProjectPath& outPath);
template <class IDType>
bool WriteFSM2(const FSM2<IDType>& fsm2, const hecl::ProjectPath& outPath);

}
#endif // __COMMON_FSM2_HPP__