#pragma once

#include "ICacheDatabase.h"

#include <spdlog/fwd.h>

#include <cstddef>
#include <memory>
#include <optional>
#include <string>

namespace CesiumAsync {

/**
 * @brief Cache storage using SQLITE to store completed response.
 */
class CESIUMASYNC_API SqliteCache : public ICacheDatabase {
public:
  /**
   * @brief Constructs a new instance with a given `databaseName` pointing to a
   * database.
   *
   * The instance will connect to the existing database or create a new one if
   * it doesn't exist
   *
   * @param pLogger The logger that receives error messages.
   * @param databaseName the database path.
   * @param maxItems the maximum number of items should be kept in the database
   * after prunning.
   */
  SqliteCache(
      const std::shared_ptr<spdlog::logger>& pLogger,
      const std::string& databaseName,
      uint64_t maxItems = 4096);
  ~SqliteCache();

  /** @copydoc ICacheDatabase::getEntry*/
  virtual std::optional<CacheItem>
  getEntry(const std::string& key) const override;

  /** @copydoc ICacheDatabase::storeEntry*/
  virtual bool storeEntry(
      const std::string& key,
      std::time_t expiryTime,
      const std::string& url,
      const std::string& requestMethod,
      const HttpHeaders& requestHeaders,
      uint16_t statusCode,
      const HttpHeaders& responseHeaders,
      const std::span<const std::byte>& responseData) override;

  /** @copydoc ICacheDatabase::prune*/
  virtual bool prune() override;

  /** @copydoc ICacheDatabase::clearAll*/
  virtual bool clearAll() override;

private:
  struct Impl;
  std::unique_ptr<Impl> _pImpl;
  void createConnection() const;
  void destroyDatabase();
};
} // namespace CesiumAsync