#pragma once

#include <assets/handle.h>
#include <mutex>
#include <shared_mutex>
#include <string>
#include <unordered_map>

namespace extra2d {

template <typename T> class AssetCache {
public:
  Handle<T> find(const std::string &key) const {
    std::shared_lock<std::shared_mutex> lock(mutex_);
    auto it = map_.find(key);
    if (it == map_.end()) {
      return Handle<T>::invalid();
    }
    return it->second;
  }

  void set(const std::string &key, Handle<T> handle) {
    std::unique_lock<std::shared_mutex> lock(mutex_);
    map_[key] = handle;
  }

  void erase(const std::string &key) {
    std::unique_lock<std::shared_mutex> lock(mutex_);
    map_.erase(key);
  }

  void clear() {
    std::unique_lock<std::shared_mutex> lock(mutex_);
    map_.clear();
  }

  std::string findKeyByHandle(Handle<T> handle) const {
    std::shared_lock<std::shared_mutex> lock(mutex_);
    for (const auto &pair : map_) {
      if (pair.second == handle) {
        return pair.first;
      }
    }
    return "";
  }

private:
  std::unordered_map<std::string, Handle<T>> map_;
  mutable std::shared_mutex mutex_;
};

} // namespace extra2d