clio/src/data/cassandra/impl/Result.hpp

#pragma once

#include "data/cassandra/impl/ManagedObject.hpp"
#include "data/cassandra/impl/Tuple.hpp"
#include "util/UnsupportedType.hpp"

#include <boost/uuid/string_generator.hpp>
#include <boost/uuid/uuid.hpp>
#include <cassandra.h>
#include <xrpl/basics/base_uint.h>
#include <xrpl/protocol/AccountID.h>

#include <cstddef>
#include <cstdint>
#include <functional>
#include <iterator>
#include <optional>
#include <stdexcept>
#include <string>
#include <string_view>
#include <tuple>
#include <type_traits>
#include <utility>
#include <vector>

namespace data::cassandra::impl {

template <typename Type>
inline Type
extractColumn(CassRow const* row, std::size_t idx)
{
    using std::to_string;
    Type output{};

    auto throwErrorIfNeeded = [](CassError rc, std::string_view label) {
        if (rc != CASS_OK) {
            auto const tag = '[' + std::string{label} + ']';
            throw std::logic_error(tag + ": " + cass_error_desc(rc));
        }
    };

    using DecayedType = std::decay_t<Type>;
    using UintTupleType = std::tuple<uint32_t, uint32_t>;
    using UCharVectorType = std::vector<unsigned char>;

    if constexpr (std::is_same_v<DecayedType, xrpl::uint256>) {
        cass_byte_t const* buf = nullptr;
        std::size_t bufSize = 0;
        auto const rc = cass_value_get_bytes(cass_row_get_column(row, idx), &buf, &bufSize);
        throwErrorIfNeeded(rc, "Extract xrpl::uint256");
        output = xrpl::uint256::fromVoid(buf);
    } else if constexpr (std::is_same_v<DecayedType, xrpl::AccountID>) {
        cass_byte_t const* buf = nullptr;
        std::size_t bufSize = 0;
        auto const rc = cass_value_get_bytes(cass_row_get_column(row, idx), &buf, &bufSize);
        throwErrorIfNeeded(rc, "Extract xrpl::AccountID");
        output = xrpl::AccountID::fromVoid(buf);
    } else if constexpr (std::is_same_v<DecayedType, UCharVectorType>) {
        cass_byte_t const* buf = nullptr;
        std::size_t bufSize = 0;
        auto const rc = cass_value_get_bytes(cass_row_get_column(row, idx), &buf, &bufSize);
        throwErrorIfNeeded(rc, "Extract vector<unsigned char>");
        output = UCharVectorType{buf, buf + bufSize};
    } else if constexpr (std::is_same_v<DecayedType, UintTupleType>) {
        auto const* tuple = cass_row_get_column(row, idx);
        output = TupleIterator::fromTuple(tuple).extract<uint32_t, uint32_t>();
    } else if constexpr (std::is_convertible_v<DecayedType, std::string>) {
        char const* value = nullptr;
        std::size_t len = 0;
        auto const rc = cass_value_get_string(cass_row_get_column(row, idx), &value, &len);
        throwErrorIfNeeded(rc, "Extract string");
        output = std::string{value, len};
    } else if constexpr (std::is_same_v<DecayedType, bool>) {
        cass_bool_t flag = cass_bool_t::cass_false;
        auto const rc = cass_value_get_bool(cass_row_get_column(row, idx), &flag);
        throwErrorIfNeeded(rc, "Extract bool");
        output = flag != cass_bool_t::cass_false;
    }
    // clio only uses bigint (int64_t) so we convert any incoming type
    else if constexpr (std::is_convertible_v<DecayedType, int64_t>) {
        int64_t out = 0;
        auto const rc = cass_value_get_int64(cass_row_get_column(row, idx), &out);
        throwErrorIfNeeded(rc, "Extract int64");
        output = static_cast<DecayedType>(out);
    } else if constexpr (std::is_convertible_v<DecayedType, boost::uuids::uuid>) {
        CassUuid uuid;
        auto const rc = cass_value_get_uuid(cass_row_get_column(row, idx), &uuid);
        throwErrorIfNeeded(rc, "Extract uuid");
        std::string uuidStr(CASS_UUID_STRING_LENGTH, '0');
        cass_uuid_string(uuid, uuidStr.data());
        uuidStr.pop_back();  // remove the last \0 character
        output = boost::uuids::string_generator{}(uuidStr);
    } else {
        // type not supported for extraction
        static_assert(util::Unsupported<DecayedType>);
    }

    return output;
}

struct Result : public ManagedObject<CassResult const> {
    /* implicit */ Result(CassResult const* ptr);

    [[nodiscard]] std::size_t
    numRows() const;

    [[nodiscard]] bool
    hasRows() const;

    template <typename... RowTypes>
    [[nodiscard]] std::optional<std::tuple<RowTypes...>>
    get() const
        requires(std::tuple_size<std::tuple<RowTypes...>>{} > 1)
    {
        // row managed internally by cassandra driver, hence no ManagedObject.
        auto const* row = cass_result_first_row(*this);
        if (row == nullptr)
            return std::nullopt;

        std::size_t idx = 0;
        auto advanceId = [&idx]() { return idx++; };

        return std::make_optional<std::tuple<RowTypes...>>(
            {extractColumn<RowTypes>(row, advanceId())...}
        );
    }

    template <typename RowType>
    [[nodiscard]] std::optional<RowType>
    get() const
    {
        // row managed internally by cassandra driver, hence no ManagedObject.
        auto const* row = cass_result_first_row(*this);
        if (row == nullptr)
            return std::nullopt;
        return std::make_optional<RowType>(extractColumn<RowType>(row, 0));
    }
};

class ResultIterator : public ManagedObject<CassIterator> {
    bool hasMore_ = false;

public:
    /* implicit */ ResultIterator(CassIterator* ptr);

    [[nodiscard]] static ResultIterator
    fromResult(Result const& result);

    [[maybe_unused]] bool
    moveForward();

    [[nodiscard]] bool
    hasMore() const;

    template <typename... RowTypes>
    [[nodiscard]] std::tuple<RowTypes...>
    extractCurrentRow() const
    {
        // note: row is invalidated on each iteration.
        // managed internally by cassandra driver, hence no ManagedObject.
        auto const* row = cass_iterator_get_row(*this);

        std::size_t idx = 0;
        auto advanceId = [&idx]() { return idx++; };

        return {extractColumn<RowTypes>(row, advanceId())...};
    }
};

template <typename... Types>
class ResultExtractor {
    std::reference_wrapper<Result const> ref_;

public:
    struct Sentinel {};

    struct Iterator {
        using iterator_category = std::input_iterator_tag;
        using difference_type = std::size_t;  // rows count
        using value_type = std::tuple<Types...>;

        /* implicit */ Iterator(ResultIterator iterator) : iterator_{std::move(iterator)}
        {
        }

        Iterator(Iterator const&) = delete;
        Iterator&
        operator=(Iterator const&) = delete;

        value_type
        operator*() const
        {
            return iterator_.extractCurrentRow<Types...>();
        }

        value_type
        operator->()
        {
            return iterator_.extractCurrentRow<Types...>();
        }

        Iterator&
        operator++()
        {
            iterator_.moveForward();
            return *this;
        }

        bool
        operator==(Sentinel const&) const
        {
            return not iterator_.hasMore();
        }

    private:
        ResultIterator iterator_;
    };

    ResultExtractor(Result const& result) : ref_{std::cref(result)}
    {
    }

    Iterator
    begin()
    {
        return ResultIterator::fromResult(ref_);
    }

    Sentinel
    end()
    {
        return {};
    }
};

}  // namespace data::cassandra::impl