#include "Runtime/CSimplePool.hpp"

#include "Runtime/CToken.hpp"
#include "Runtime/IVParamObj.hpp"

#include <cassert>

namespace urde {

CSimplePool::CSimplePool(IFactory& factory)
: x18_factory(factory), x1c_paramXfer(new TObjOwnerParam<IObjectStore*>(this)) {}

CSimplePool::~CSimplePool() { assert(x8_resources.empty() && "Dangling CSimplePool resources detected"); }

CToken CSimplePool::GetObj(const SObjectTag& tag, const CVParamTransfer& paramXfer) {
  if (!tag) {
    return {};
  }

  const auto iter = x8_resources.find(tag);
  if (iter != x8_resources.end()) {
    return CToken(iter->second);
  }

  auto* const ret = new CObjectReference(*this, nullptr, tag, paramXfer);
  x8_resources.emplace(tag, ret);
  return CToken(ret);
}

CToken CSimplePool::GetObj(const SObjectTag& tag) { return GetObj(tag, x1c_paramXfer); }

CToken CSimplePool::GetObj(std::string_view resourceName) { return GetObj(resourceName, x1c_paramXfer); }

CToken CSimplePool::GetObj(std::string_view resourceName, const CVParamTransfer& paramXfer) {
  const SObjectTag* tag = x18_factory.GetResourceIdByName(resourceName);
  if (!tag)
    return {};
  return GetObj(*tag, paramXfer);
}

bool CSimplePool::HasObject(const SObjectTag& tag) const {
  auto iter = x8_resources.find(tag);
  if (iter != x8_resources.cend())
    return true;
  return x18_factory.CanBuild(tag);
}

bool CSimplePool::ObjectIsLive(const SObjectTag& tag) const {
  auto iter = x8_resources.find(tag);
  if (iter == x8_resources.cend())
    return false;
  return iter->second->IsLoaded();
}

void CSimplePool::Flush() {}

void CSimplePool::ObjectUnreferenced(const SObjectTag& tag) {
  auto iter = x8_resources.find(tag);
  if (iter != x8_resources.end())
    x8_resources.erase(iter);
}

std::vector<SObjectTag> CSimplePool::GetReferencedTags() const {
  std::vector<SObjectTag> ret;
  ret.reserve(x8_resources.size());
  for (const auto& obj : x8_resources)
    ret.push_back(obj.first);
  return ret;
}

} // namespace urde