#ifndef TOOLS_DATAPACKER_CONTAINER_H_
#define TOOLS_DATAPACKER_CONTAINER_H_

#include <cstdint>
#include <map>
#include <optional>
#include <string>

#include "proto/data.pb.h"

namespace com::fourisland::lingo2_archipelago {

class Container {
 public:
  uint64_t FindOrAddMap(std::string map_name);

  uint64_t FindOrAddRoom(std::optional<std::string> map_name,
                         std::string room_name,
                         std::optional<std::string> map_fallback);

  uint64_t FindOrAddPainting(std::optional<std::string> map_name,
                             std::optional<std::string> room_name,
                             std::string painting_name,
                             std::optional<std::string> map_fallback,
                             std::optional<std::string> room_fallback);

  uint64_t FindOrAddPort(std::optional<std::string> map_name,
                         std::optional<std::string> room_name,
                         std::string port_name,
                         std::optional<std::string> map_fallback,
                         std::optional<std::string> room_fallback);

  uint64_t FindOrAddPanel(std::optional<std::string> map_name,
                          std::optional<std::string> room_name,
                          std::string panel_name,
                          std::optional<std::string> map_fallback,
                          std::optional<std::string> room_fallback);

  uint64_t FindOrAddLetter(std::string key, bool level2);

  uint64_t FindLetterByName(std::string letter_name);

  uint64_t FindOrAddEnding(std::string ending_name);

  uint64_t FindOrAddMastery(std::optional<std::string> map_name,
                            std::optional<std::string> room_name,
                            std::string mastery_name,
                            std::optional<std::string> map_fallback,
                            std::optional<std::string> room_fallback);

  uint64_t FindOrAddKeyholder(std::optional<std::string> map_name,
                              std::optional<std::string> room_name,
                              std::string keyholder_name,
                              std::optional<std::string> map_fallback,
                              std::optional<std::string> room_fallback);

  uint64_t FindOrAddDoor(std::optional<std::string> map_name,
                         std::string door_name,
                         std::optional<std::string> map_fallback);

  void AddConnection(const Connection& connection);

  uint64_t FindOrAddProgressive(std::string prog_name);

  uint64_t FindOrAddDoorGroup(std::string group_name);

  AllObjects& all_objects() { return all_objects_; }

 private:
  AllObjects all_objects_;

  std::map<std::string, uint64_t> map_id_by_name_;
  std::map<std::string, std::map<std::string, uint64_t>>
      room_id_by_map_room_names_;
  std::map<std::string, std::map<std::string, std::map<std::string, uint64_t>>>
      painting_id_by_map_room_painting_names_;
  std::map<std::string, std::map<std::string, std::map<std::string, uint64_t>>>
      port_id_by_map_room_port_names_;
  std::map<std::string, std::map<std::string, std::map<std::string, uint64_t>>>
      panel_id_by_map_room_panel_names_;
  std::map<std::string, uint64_t> letter_id_by_name_;
  std::map<std::string, std::map<std::string, std::map<std::string, uint64_t>>>
      mastery_id_by_map_room_mastery_names_;
  std::map<std::string, std::map<std::string, std::map<std::string, uint64_t>>>
      keyholder_id_by_map_room_keyholder_names_;
  std::map<std::string, std::map<std::string, uint64_t>>
      door_id_by_map_door_names_;
  std::map<std::string, uint64_t> ending_id_by_name_;
  std::map<std::string, uint64_t> progressive_id_by_name_;
  std::map<std::string, uint64_t> door_group_id_by_name_;
};

}  // namespace com::fourisland::lingo2_archipelago

#endif /* TOOLS_DATAPACKER_CONTAINER_H_ */