/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/*
 * Copyright 2022-Present Couchbase, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
 * except in compliance with the License. You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under
 * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
 * ANY KIND, either express or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

#pragma once

#include "resource_units.hxx"
#include "utils/movable_function.hxx"

#include <couchbase/cas.hxx>

#include <chrono>
#include <string>
#include <system_error>
#include <vector>

namespace couchbase
{
class retry_strategy;
namespace tracing
{
class request_span;
} // namespace tracing
} // namespace couchbase

namespace couchbase::core
{
enum class key_state {
    not_persisted = 0x00,
    persisted = 0x01,
    not_found = 0x80,
    deleted = 0x81,
};

class observe_options
{
  public:
    std::vector<std::byte> key{};
    std::uint32_t replica_index{};
    std::string collection_name{};
    std::string scope_name{};
    std::uint32_t collection_id{};

    std::shared_ptr<couchbase::retry_strategy> retry_strategy{};
    std::chrono::milliseconds timeout{};
    std::shared_ptr<couchbase::tracing::request_span> parent_span{};

    struct {
        std::string user{};
    } internal{};
};

class observe_result
{
  public:
    key_state state{};
    couchbase::cas cas{};

    struct {
        std::optional<resource_unit_result> resource_units{};
    } internal{};
};

using observe_callback = utils::movable_function<void(observe_result result, std::error_code ec)>;

class observe_seqno_options
{
  public:
    std::uint16_t vbucket_id{};
    std::uint64_t vbucket_uuid{};
    std::uint32_t replica_index{};

    std::shared_ptr<couchbase::retry_strategy> retry_strategy{};
    std::chrono::milliseconds timeout{};
    std::shared_ptr<couchbase::tracing::request_span> parent_span{};

    struct {
        std::string user{};
    } internal{};
};

class observe_seqno_result
{
  public:
    bool did_failover{};
    std::uint16_t vbucket_id{};
    std::uint64_t vbucket_uuid{};
    std::uint64_t persist_sequence_number{};
    std::uint64_t current_sequence_number{};
    std::uint64_t old_vbucket_uuid{};
    std::uint64_t last_sequence_number{};

    struct {
        std::optional<resource_unit_result> resource_units{};
    } internal{};
};

using observe_seqno_callback = utils::movable_function<void(observe_seqno_result result, std::error_code ec)>;
} // namespace couchbase::core