#pragma once

#include "data/Types.hpp"
#include "etl/Models.hpp"

#include <xrpl/basics/base_uint.h>
#include <xrpl/basics/hardened_hash.h>

#include <cstddef>
#include <cstdint>
#include <expected>
#include <optional>
#include <string>
#include <vector>

namespace data {

/**
 * @brief Cache for an entire ledger.
 */
class LedgerCacheInterface {
public:
    virtual ~LedgerCacheInterface() = default;
    LedgerCacheInterface() = default;
    LedgerCacheInterface(LedgerCacheInterface&&) = delete;
    LedgerCacheInterface(LedgerCacheInterface const&) = delete;
    LedgerCacheInterface&
    operator=(LedgerCacheInterface&&) = delete;
    LedgerCacheInterface&
    operator=(LedgerCacheInterface const&) = delete;

    /**
     * @brief Update the cache with new ledger objects.
     *
     * @param objs The ledger objects to update cache with
     * @param seq The sequence to update cache for
     * @param isBackground Should be set to true when writing old data from a background thread
     */
    virtual void
    update(std::vector<LedgerObject> const& objs, uint32_t seq, bool isBackground = false) = 0;

    /**
     * @brief Update the cache with new ledger objects.
     *
     * @param objs The ledger objects to update cache with
     * @param seq The sequence to update cache for
     */
    virtual void
    update(std::vector<etl::model::Object> const& objs, uint32_t seq) = 0;

    /**
     * @brief Fetch a cached object by its key and sequence number.
     *
     * @param key The key to fetch for
     * @param seq The sequence to fetch for
     * @return If found in cache, will return the cached Blob; otherwise nullopt is returned
     */
    [[nodiscard]] virtual std::optional<Blob>
    get(xrpl::uint256 const& key, uint32_t seq) const = 0;

    /**
     * @brief Fetch a recently deleted object by its key and sequence number.
     *
     * @param key The key to fetch for
     * @param seq The sequence to fetch for
     * @return If found in deleted cache, will return the cached Blob; otherwise nullopt is returned
     */
    [[nodiscard]] virtual std::optional<Blob>
    getDeleted(xrpl::uint256 const& key, uint32_t seq) const = 0;

    /**
     * @brief Gets a cached successor.
     *
     * Note: This function always returns std::nullopt when @ref isFull() returns false.
     *
     * @param key The key to fetch for
     * @param seq The sequence to fetch for
     * @return If found in cache, will return the cached successor; otherwise nullopt is returned
     */
    [[nodiscard]] virtual std::optional<LedgerObject>
    getSuccessor(xrpl::uint256 const& key, uint32_t seq) const = 0;

    /**
     * @brief Gets a cached predcessor.
     *
     * Note: This function always returns std::nullopt when @ref isFull() returns false.
     *
     * @param key The key to fetch for
     * @param seq The sequence to fetch for
     * @return If found in cache, will return the cached predcessor; otherwise nullopt is returned
     */
    [[nodiscard]] virtual std::optional<LedgerObject>
    getPredecessor(xrpl::uint256 const& key, uint32_t seq) const = 0;

    /**
     * @brief Disables the cache.
     */
    virtual void
    setDisabled() = 0;

    /**
     * @return true if the cache is disabled; false otherwise
     */
    [[nodiscard]] virtual bool
    isDisabled() const = 0;

    /**
     * @brief Sets the full flag to true.
     *
     * This is used when cache loaded in its entirety at startup of the application. This can be
     * either loaded from DB, populated together with initial ledger download (on first run) or
     * downloaded from a peer node (specified in config).
     */
    virtual void
    setFull() = 0;

    /**
     * @return The latest ledger sequence for which cache is available.
     */
    [[nodiscard]] virtual uint32_t
    latestLedgerSequence() const = 0;

    /**
     * @return true if the cache has all data for the most recent ledger; false otherwise
     */
    [[nodiscard]] virtual bool
    isFull() const = 0;

    /**
     * @return The total size of the cache.
     */
    [[nodiscard]] virtual size_t
    size() const = 0;

    /**
     * @return A number representing the success rate of hitting an object in the cache versus
     * missing it.
     */
    [[nodiscard]] virtual float
    getObjectHitRate() const = 0;

    /**
     * @return A number representing the success rate of hitting a successor in the cache versus
     * missing it.
     */
    [[nodiscard]] virtual float
    getSuccessorHitRate() const = 0;

    /**
     * @brief Waits until the cache contains a specific sequence.
     *
     * @param seq The sequence to wait for
     */
    virtual void
    waitUntilCacheContainsSeq(uint32_t seq) = 0;

    /**
     * @brief Save the cache to file
     * @note This operation takes about 7 seconds and it keeps a shared lock of mtx_
     *
     * @param path The file path to save the cache to
     * @return An error as a string if any
     */
    [[nodiscard]] virtual std::expected<void, std::string>
    saveToFile(std::string const& path) const = 0;

    /**
     * @brief Load the cache from file
     * @note This operation takes about 7 seconds and it keeps mtx_ exclusively locked
     *
     * @param path The file path to load data from
     * @param minLatestSequence The minimum allowed value of the latestLedgerSequence in cache file
     * @return An error as a string if any
     */
    [[nodiscard]] virtual std::expected<void, std::string>
    loadFromFile(std::string const& path, uint32_t minLatestSequence) = 0;

    /**
     * @brief Mark the cache as currently loading from the backend.
     * @note Should be called before initiating a backend-based cache load. The flag is
     *       automatically cleared when setFull() is called.
     */
    virtual void
    startLoading() = 0;

    /**
     * @brief Check whether the cache is currently being loaded from the backend.
     * @return true if startLoading() has been called and setFull() has not yet been called
     */
    [[nodiscard]] virtual bool
    isCurrentlyLoading() const = 0;
};

}  // namespace data