#pragma once

#include "DataSpec/DNACommon/PAK.hpp"
#include "DataSpec/DNACommon/MLVL.hpp"
#include "DNAMP2.hpp"

namespace DataSpec::DNAMP2 {

struct MLVL : BigDNA {
  AT_DECL_DNA_YAML
  Value<atUint32> magic;
  Value<atUint32> version;
  UniqueID32 worldNameId;
  UniqueID32 darkWorldNameId;
  Value<atUint32> unk;
  UniqueID32 saveWorldId;
  UniqueID32 worldSkyboxId;

  Value<atUint32> areaCount;
  struct Area : BigDNA {
    AT_DECL_DNA_YAML
    UniqueID32 areaNameId;
    Value<atVec4f> transformMtx[3];
    Value<atVec3f> aabb[2];
    UniqueID32 areaMREAId;
    Value<atUint32> areaId;

    Value<atUint32> attachedAreaCount;
    Vector<atUint16, AT_DNA_COUNT(attachedAreaCount)> attachedAreas;
    Value<atUint32> padding;

    Value<atUint32> depCount;
    struct Dependency : BigDNA {
      AT_DECL_DNA_YAML
      UniqueID32 id;
      DNAFourCC type;
    };
    Vector<Dependency, AT_DNA_COUNT(depCount)> deps;

    Value<atUint32> depLayerCount;
    Vector<atUint32, AT_DNA_COUNT(depLayerCount)> depLayers;

    Value<atUint32> dockCount;
    struct Dock : BigDNA {
      AT_DECL_DNA_YAML
      Value<atUint32> endpointCount;
      struct Endpoint : BigDNA {
        AT_DECL_DNA_YAML
        Value<atUint32> areaIdx;
        Value<atUint32> dockIdx;
      };
      Vector<Endpoint, AT_DNA_COUNT(endpointCount)> endpoints;

      Value<atUint32> planeVertCount;
      Vector<atVec3f, AT_DNA_COUNT(planeVertCount)> planeVerts;
    };
    Vector<Dock, AT_DNA_COUNT(dockCount)> docks;

    Value<atUint32> relCount;
    Vector<String<-1>, AT_DNA_COUNT(relCount)> relFilenames;
    Value<atUint32> relOffsetCount;
    Vector<atUint32, AT_DNA_COUNT(relOffsetCount)> relOffsets;

    String<-1> internalAreaName;
  };
  Vector<Area, AT_DNA_COUNT(areaCount)> areas;

  UniqueID32 worldMap;
  Value<atUint8> unknown2;
  Value<atUint32> unknown3;

  Value<atUint32> layerFlagCount;
  struct LayerFlags : BigDNA {
    AT_DECL_DNA_YAML
    Value<atUint32> layerCount;
    Value<atUint64> flags;
  };
  Vector<LayerFlags, AT_DNA_COUNT(layerFlagCount)> layerFlags;

  Value<atUint32> layerNameCount;
  Vector<String<-1>, AT_DNA_COUNT(layerNameCount)> layerNames;

  Value<atUint32> layerNameOffsetCount;
  Vector<atUint32, AT_DNA_COUNT(layerNameOffsetCount)> layerNameOffsets;

  static bool Extract(const SpecBase& dataSpec, PAKEntryReadStream& rs, const hecl::ProjectPath& outPath,
                      PAKRouter<PAKBridge>& pakRouter, const DNAMP2::PAK::Entry& entry, bool force,
                      hecl::blender::Token& btok, std::function<void(const char*)> fileChanged) {
    MLVL mlvl;
    mlvl.read(rs);
    athena::io::FileWriter writer(outPath.getWithExtension(".yaml", true).getAbsolutePath());
    athena::io::ToYAMLStream(mlvl, writer);
    hecl::blender::Connection& conn = btok.getBlenderConnection();
    return DNAMLVL::ReadMLVLToBlender(conn, mlvl, outPath, pakRouter, entry, force, fileChanged);
  }
};

} // namespace DataSpec::DNAMP2