#pragma once

#include <unordered_map>
#include "DataSpec/DNACommon/DNACommon.hpp"
#include "DataSpec/DNACommon/STRG.hpp"
#include "DNAMP1.hpp"

namespace DataSpec::DNAMP1 {

struct STRG : ISTRG {
  AT_DECL_EXPLICIT_DNA_YAMLV
  void _read(athena::io::IStreamReader& reader);
  std::vector<std::pair<FourCC, std::vector<std::u16string>>> langs;
  std::unordered_map<FourCC, std::vector<std::u16string>*> langMap;

  int32_t lookupIdx(std::string_view name) const override { return -1; }

  size_t count() const override {
    size_t retval = 0;
    for (const auto& item : langs) {
      size_t sz = item.second.size();
      if (sz > retval)
        retval = sz;
    }
    return retval;
  }
  std::string getUTF8(const FourCC& lang, size_t idx) const override {
    auto search = langMap.find(lang);
    if (search != langMap.end())
      return hecl::Char16ToUTF8(search->second->at(idx));
    return std::string();
  }
  std::u16string getUTF16(const FourCC& lang, size_t idx) const override {
    auto search = langMap.find(lang);
    if (search != langMap.end())
      return search->second->at(idx);
    return std::u16string();
  }
  hecl::SystemString getSystemString(const FourCC& lang, size_t idx) const override {
    auto search = langMap.find(lang);
    if (search != langMap.end())
#if HECL_UCS2
      return hecl::Char16ToWide(search->second->at(idx));
#else
      return hecl::Char16ToUTF8(search->second->at(idx));
#endif
    return hecl::SystemString();
  }

  static bool Extract(PAKEntryReadStream& rs, const hecl::ProjectPath& outPath) {
    STRG strg;
    strg.read(rs);
    athena::io::TransactionalFileWriter writer(outPath.getAbsolutePath());
    athena::io::ToYAMLStream(strg, writer);
    return true;
  }

  static bool Cook(const hecl::ProjectPath& inPath, const hecl::ProjectPath& outPath) {
    STRG strg;
    athena::io::FileReader reader(inPath.getAbsolutePath());
    athena::io::FromYAMLStream(strg, reader);
    athena::io::TransactionalFileWriter ws(outPath.getAbsolutePath());
    strg.write(ws);
    return true;
  }

  static bool Cook(const STRG& strg, const hecl::ProjectPath& outPath) {
    athena::io::TransactionalFileWriter ws(outPath.getAbsolutePath());
    strg.write(ws);
    return true;
  }

  void gatherDependencies(std::vector<hecl::ProjectPath>& pathsOut) const override;
};

} // namespace DataSpec::DNAMP1
