#pragma once

#include <stdint.h>

#include <array>
#include <memory>
#include <phosg/JSON.hh>
#include <string>
#include <unordered_map>
#include <vector>

#include "ItemData.hh"
#include "ItemParameterTable.hh"

class ItemNameIndex {
public:
  struct ItemMetadata {
    uint32_t primary_identifier;
    std::string name;
  };

  ItemNameIndex(
      std::shared_ptr<const ItemParameterTable> pmt,
      std::shared_ptr<const ItemData::StackLimits> limits,
      const std::vector<std::string>& name_coll);

  inline size_t entry_count() const {
    return this->primary_identifier_index.size();
  }

  inline const std::unordered_map<uint32_t, std::shared_ptr<const ItemMetadata>>& all_by_primary_identifier() const {
    return this->primary_identifier_index;
  }
  inline const std::map<std::string, std::shared_ptr<const ItemMetadata>>& all_by_name() const {
    return this->name_index;
  }

  inline bool exists(const ItemData& item) const {
    return this->primary_identifier_index.count(item.primary_identifier());
  }

  enum Flag : uint8_t {
    INCLUDE_PSO_COLOR_ESCAPES = 0x01,
    NAME_ONLY = 0x02,
  };
  std::string describe_item(const ItemData& item, uint8_t flags = 0) const;
  ItemData parse_item_description(const std::string& description) const;

  void print_table(FILE* stream) const;

private:
  ItemData parse_item_description_phase(const std::string& description, bool skip_special) const;

  std::shared_ptr<const ItemParameterTable> item_parameter_table;
  std::shared_ptr<const ItemData::StackLimits> limits;

  std::unordered_map<uint32_t, std::shared_ptr<const ItemMetadata>> primary_identifier_index;
  std::map<std::string, std::shared_ptr<const ItemMetadata>> name_index;
  std::map<std::string, std::shared_ptr<const ItemMetadata>> es_name_index;
};