/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/*
 *   Copyright 2020-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.
 */

#include "core/cluster.hxx"
#include "core/error_context/key_value.hxx"
#include "core/impl/observe_poll.hxx"
#include "core/impl/observe_seqno.hxx"
#include "core/operations/document_upsert.hxx"

#include <couchbase/upsert_options.hxx>

#include <utility>

namespace couchbase::core::impl
{

void
initiate_upsert_operation(std::shared_ptr<couchbase::core::cluster> core,
                          std::string bucket_name,
                          std::string scope_name,
                          std::string collection_name,
                          std::string document_key,
                          codec::encoded_value encoded,
                          upsert_options::built options,
                          upsert_handler&& handler)
{
    auto value = std::move(encoded);
    auto id = document_id{
        std::move(bucket_name),
        std::move(scope_name),
        std::move(collection_name),
        std::move(document_key),
    };
    if (options.persist_to == persist_to::none && options.replicate_to == replicate_to::none) {
        return core->execute(
          operations::upsert_request{
            std::move(id),
            std::move(value.data),
            {},
            {},
            value.flags,
            options.expiry,
            options.durability_level,
            options.timeout,
            { options.retry_strategy },
            options.preserve_expiry,
          },
          [handler = std::move(handler)](operations::upsert_response&& resp) mutable {
              return handler(std::move(resp.ctx), mutation_result{ resp.cas, std::move(resp.token) });
          });
    }

    operations::upsert_request request{
        id,
        std::move(value.data),
        {},
        {},
        value.flags,
        options.expiry,
        durability_level::none,
        options.timeout,
        { options.retry_strategy },
        options.preserve_expiry,
    };
    return core->execute(
      std::move(request), [core, id = std::move(id), options, handler = std::move(handler)](operations::upsert_response&& resp) mutable {
          if (resp.ctx.ec()) {
              return handler(std::move(resp.ctx), mutation_result{ resp.cas, std::move(resp.token) });
          }

          initiate_observe_poll(core,
                                std::move(id),
                                resp.token,
                                options.timeout,
                                options.persist_to,
                                options.replicate_to,
                                [resp = std::move(resp), handler = std::move(handler)](std::error_code ec) mutable {
                                    if (ec) {
                                        resp.ctx.override_ec(ec);
                                        return handler(std::move(resp.ctx), mutation_result{});
                                    }
                                    return handler(std::move(resp.ctx), mutation_result{ resp.cas, std::move(resp.token) });
                                });
      });
}
} // namespace couchbase::core::impl