#pragma once

#include <array>

#include "Runtime/RetroTypes.hpp"
#include "Runtime/World/CEntity.hpp"

namespace metaforce {

enum class EGameObjectList {
  Invalid = -1,
  All,
  Actor,
  PhysicsActor,
  GameCamera,
  GameLight,
  ListeningAi,
  AiWaypoint,
  PlatformAndDoor,
};

class CObjectList {
  friend class CGameArea;

  struct SObjectListEntry {
    CEntity* entity = nullptr;
    s16 next = -1;
    s16 prev = -1;
  };
  std::array<SObjectListEntry, kMaxEntities> x0_list; // was an rstl::reserved_vector
  EGameObjectList x2004_listEnum;
  s16 x2008_firstId = -1;
  u16 x200a_count = 0;

public:
  class iterator {
    friend class CObjectList;
    CObjectList& m_list;
    s16 m_id;
    iterator(CObjectList& list, s16 id) : m_list(list), m_id(id) {}

  public:
    iterator& operator++() {
      m_id = m_list.GetNextObjectIndex(m_id);
      return *this;
    }
    bool operator==(const iterator& other) const { return m_id == other.m_id; }
    bool operator!=(const iterator& other) const { return !operator==(other); }
    CEntity* operator*() const { return m_list.GetObjectByIndex(m_id); }
  };

  class const_iterator {
    friend class CObjectList;
    const CObjectList& m_list;
    s16 m_id;
    const_iterator(const CObjectList& list, s16 id) : m_list(list), m_id(id) {}

  public:
    const_iterator& operator++() {
      m_id = m_list.GetNextObjectIndex(m_id);
      return *this;
    }
    bool operator==(const iterator& other) const { return m_id == other.m_id; }
    bool operator!=(const iterator& other) const { return !operator==(other); }
    const CEntity* operator*() const { return m_list.GetObjectByIndex(m_id); }
  };

  [[nodiscard]] iterator begin() { return iterator(*this, x2008_firstId); }
  [[nodiscard]] iterator end() { return iterator(*this, -1); }
  [[nodiscard]] const_iterator begin() const { return const_iterator(*this, x2008_firstId); }
  [[nodiscard]] const_iterator end() const { return const_iterator(*this, -1); }
  [[nodiscard]] const_iterator cbegin() const { return begin(); }
  [[nodiscard]] const_iterator cend() const { return end(); }

  explicit CObjectList(EGameObjectList listEnum);
  virtual ~CObjectList() = default;

  void AddObject(CEntity& entity);
  void RemoveObject(TUniqueId uid);
  const CEntity* operator[](size_t i) const;
  CEntity* operator[](size_t i);
  const CEntity* GetObjectById(TUniqueId uid) const;
  const CEntity* GetObjectByIndex(s16 index) const { return x0_list[index].entity; }
  CEntity* GetObjectByIndex(s16 index) { return x0_list[index].entity; }
  CEntity* GetObjectById(TUniqueId uid);
  const CEntity* GetValidObjectById(TUniqueId uid) const;
  CEntity* GetValidObjectById(TUniqueId uid);
  s16 GetFirstObjectIndex() const { return x2008_firstId; }
  s16 GetNextObjectIndex(s16 prev) const { return x0_list[prev].next; }
  virtual bool IsQualified(const CEntity&) const;
  u16 size() const { return x200a_count; }
};

} // namespace metaforce