#pragma once

#include "data/Types.hpp"
#include "util/Mutex.hpp"

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

#include <cstddef>
#include <cstdint>
#include <map>
#include <optional>
#include <thread>
#include <vector>

struct DiffProvider {
    std::vector<data::LedgerObject>
    getLatestDiff()  // NOLINT(readability-convert-member-functions-to-static)
    {
        using namespace ripple;

        return {
            {.key = uint256{"05E1EAC2574BE082B00B16F907CE32E6058DEB8F9E81CF34A00E80A5D71FA4FE"},
             .blob = Blob{'s'}},
            {.key = uint256{"110872C7196EE6EF7032952F1852B11BB461A96FF2D7E06A8003B4BB30FD130B"},
             .blob = Blob{'s'}},
            {.key = uint256{"3B3A84E850C724E914293271785A31D0BFC8B9DD1B6332E527B149AD72E80E18"},
             .blob = Blob{'s'}},
            {.key = uint256{"4EC98C5C3F34C44409BC058998CBD64F6AED3FF6C0CAAEC15F7F42DF14EE9F04"},
             .blob = Blob{'s'}},
            {.key = uint256{"58CEC9F17733EA7BA68C88E6179B8F207D001EE04D4E0366F958CC04FF6AB834"},
             .blob = Blob{'s'}},
            {.key = uint256{"64FB1712146BA604C274CC335C5DE7ADFE52D1F8C3E904A9F9765FE8158A3E01"},
             .blob = Blob{'s'}},
            {.key = uint256{"700BE23B1D9EE3E6BF52543D05843D5345B85D9EDB3D33BBD6B4C3A13C54B38E"},
             .blob = Blob{'s'}},
            {.key = uint256{"82C297FCBCD634C4424F263D17480AA2F13975DF5846A5BB57246022CEEBE441"},
             .blob = Blob{'s'}},
            {.key = uint256{"A2AA4C212DC2CA2C49BF58805F7C63363BC981018A01AC9609A7CBAB2A02CEDF"},
             .blob = Blob{'s'}},
            {.key = uint256{"BC0DAE09C0BFBC4A49AA94B849266588BFD6E1F554B184B5788AC55D6E07EB95"},
             .blob = Blob{'s'}},
            {.key = uint256{"DCC8759A35CB946511763AA5553A82AA25F20B901C98C9BB74D423BCFAFF5F9D"},
             .blob = Blob{'s'}},
        };
    }

    std::optional<ripple::uint256>
    nextKey(std::size_t keysSize)
    {
        // mock the result from doFetchSuccessorKey, be aware this function will be called from
        // multiple threads
        auto keysMap = threadKeysMap_.lock();
        keysMap->operator[](std::this_thread::get_id())++;

        if (keysMap->operator[](std::this_thread::get_id()) == keysSize - 1) {
            return data::kLAST_KEY;
        }
        if (keysMap->operator[](std::this_thread::get_id()) == keysSize) {
            keysMap->operator[](std::this_thread::get_id()) = 0;
            return std::nullopt;
        }

        return ripple::uint256{"E6DBAFC99223B42257915A63DFC6B0C032D4070F9A574B255AD97466726FC321"};
    }

private:
    util::Mutex<std::map<std::thread::id, uint32_t>> threadKeysMap_;
};