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

#include "core/cluster.hxx"
#include "core/impl/error.hxx"
#include "core/impl/observability_recorder.hxx"
#include "core/management/bucket_settings.hxx"
#include "core/operations/management/bucket_create.hxx"
#include "core/operations/management/bucket_drop.hxx"
#include "core/operations/management/bucket_flush.hxx"
#include "core/operations/management/bucket_get.hxx"
#include "core/operations/management/bucket_get_all.hxx"
#include "core/operations/management/bucket_update.hxx"
#include "core/tracing/constants.hxx"
#include "core/tracing/tracer_wrapper.hxx"

#include <couchbase/create_bucket_options.hxx>
#include <couchbase/drop_bucket_options.hxx>
#include <couchbase/error.hxx>
#include <couchbase/flush_bucket_options.hxx>
#include <couchbase/get_all_buckets_options.hxx>
#include <couchbase/get_bucket_options.hxx>
#include <couchbase/management/bucket_settings.hxx>
#include <couchbase/update_bucket_options.hxx>

#include <future>
#include <memory>
#include <utility>
#include <vector>

namespace couchbase
{
namespace
{
auto
map_bucket_settings(const couchbase::core::management::cluster::bucket_settings& bucket)
  -> couchbase::management::cluster::bucket_settings
{
  couchbase::management::cluster::bucket_settings bucket_settings{};
  bucket_settings.name = bucket.name;
  bucket_settings.ram_quota_mb = bucket.ram_quota_mb;
  bucket_settings.minimum_durability_level = bucket.minimum_durability_level;
  bucket_settings.history_retention_collection_default =
    bucket.history_retention_collection_default;
  bucket_settings.max_expiry = bucket.max_expiry;
  bucket_settings.num_replicas = bucket.num_replicas;
  bucket_settings.replica_indexes = bucket.replica_indexes;
  bucket_settings.flush_enabled = bucket.flush_enabled;
  bucket_settings.history_retention_bytes = bucket.history_retention_bytes;
  bucket_settings.history_retention_duration = bucket.history_retention_duration;
  bucket_settings.num_vbuckets = bucket.num_vbuckets;
  switch (bucket.conflict_resolution_type) {
    case core::management::cluster::bucket_conflict_resolution::unknown:
      bucket_settings.conflict_resolution_type =
        management::cluster::bucket_conflict_resolution::unknown;
      break;
    case core::management::cluster::bucket_conflict_resolution::timestamp:
      bucket_settings.conflict_resolution_type =
        management::cluster::bucket_conflict_resolution::timestamp;
      break;
    case core::management::cluster::bucket_conflict_resolution::sequence_number:
      bucket_settings.conflict_resolution_type =
        management::cluster::bucket_conflict_resolution::sequence_number;
      break;
    case core::management::cluster::bucket_conflict_resolution::custom:
      bucket_settings.conflict_resolution_type =
        management::cluster::bucket_conflict_resolution::custom;
      break;
  }
  switch (bucket.eviction_policy) {
    case core::management::cluster::bucket_eviction_policy::unknown:
      bucket_settings.eviction_policy = management::cluster::bucket_eviction_policy::unknown;
      break;
    case core::management::cluster::bucket_eviction_policy::full:
      bucket_settings.eviction_policy = management::cluster::bucket_eviction_policy::full;
      break;
    case core::management::cluster::bucket_eviction_policy::value_only:
      bucket_settings.eviction_policy = management::cluster::bucket_eviction_policy::value_only;
      break;
    case core::management::cluster::bucket_eviction_policy::no_eviction:
      bucket_settings.eviction_policy = management::cluster::bucket_eviction_policy::no_eviction;
      break;
    case core::management::cluster::bucket_eviction_policy::not_recently_used:
      bucket_settings.eviction_policy =
        management::cluster::bucket_eviction_policy::not_recently_used;
      break;
  }
  switch (bucket.compression_mode) {
    case core::management::cluster::bucket_compression::unknown:
      bucket_settings.compression_mode = management::cluster::bucket_compression::unknown;
      break;
    case core::management::cluster::bucket_compression::off:
      bucket_settings.compression_mode = management::cluster::bucket_compression::off;
      break;
    case core::management::cluster::bucket_compression::active:
      bucket_settings.compression_mode = management::cluster::bucket_compression::active;
      break;
    case core::management::cluster::bucket_compression::passive:
      bucket_settings.compression_mode = management::cluster::bucket_compression::passive;
      break;
  }
  switch (bucket.bucket_type) {
    case core::management::cluster::bucket_type::unknown:
      bucket_settings.bucket_type = management::cluster::bucket_type::unknown;
      break;
    case core::management::cluster::bucket_type::couchbase:
      bucket_settings.bucket_type = management::cluster::bucket_type::couchbase;
      break;
    case core::management::cluster::bucket_type::memcached:
      bucket_settings.bucket_type = management::cluster::bucket_type::memcached;
      break;
    case core::management::cluster::bucket_type::ephemeral:
      bucket_settings.bucket_type = management::cluster::bucket_type::ephemeral;
      break;
  }
  switch (bucket.storage_backend) {
    case core::management::cluster::bucket_storage_backend::unknown:
      bucket_settings.storage_backend = management::cluster::bucket_storage_backend::unknown;
      break;
    case core::management::cluster::bucket_storage_backend::couchstore:
      bucket_settings.storage_backend = management::cluster::bucket_storage_backend::couchstore;
      break;
    case core::management::cluster::bucket_storage_backend::magma:
      bucket_settings.storage_backend = management::cluster::bucket_storage_backend::magma;
      break;
  }
  return bucket_settings;
}

auto
map_all_bucket_settings(const std::vector<couchbase::core::management::cluster::bucket_settings>&
                          buckets) -> std::vector<couchbase::management::cluster::bucket_settings>
{
  std::vector<couchbase::management::cluster::bucket_settings> buckets_settings{};
  for (const auto& bucket : buckets) {
    auto converted_bucket = map_bucket_settings(bucket);
    buckets_settings.emplace_back(converted_bucket);
  }
  return buckets_settings;
}

auto
map_bucket_settings(const couchbase::management::cluster::bucket_settings& bucket)
  -> couchbase::core::management::cluster::bucket_settings
{

  couchbase::core::management::cluster::bucket_settings bucket_settings{};

  bucket_settings.name = bucket.name;
  bucket_settings.ram_quota_mb = bucket.ram_quota_mb;
  bucket_settings.max_expiry = bucket.max_expiry;
  bucket_settings.minimum_durability_level = bucket.minimum_durability_level;
  bucket_settings.num_replicas = bucket.num_replicas;
  bucket_settings.replica_indexes = bucket.replica_indexes;
  bucket_settings.flush_enabled = bucket.flush_enabled;
  bucket_settings.history_retention_collection_default =
    bucket.history_retention_collection_default;
  bucket_settings.history_retention_bytes = bucket.history_retention_bytes;
  bucket_settings.history_retention_duration = bucket.history_retention_duration;
  bucket_settings.num_vbuckets = bucket.num_vbuckets;
  switch (bucket.conflict_resolution_type) {
    case management::cluster::bucket_conflict_resolution::unknown:
      bucket_settings.conflict_resolution_type =
        core::management::cluster::bucket_conflict_resolution::unknown;
      break;
    case management::cluster::bucket_conflict_resolution::timestamp:
      bucket_settings.conflict_resolution_type =
        core::management::cluster::bucket_conflict_resolution::timestamp;
      break;
    case management::cluster::bucket_conflict_resolution::sequence_number:
      bucket_settings.conflict_resolution_type =
        core::management::cluster::bucket_conflict_resolution::sequence_number;
      break;
    case management::cluster::bucket_conflict_resolution::custom:
      bucket_settings.conflict_resolution_type =
        core::management::cluster::bucket_conflict_resolution::custom;
      break;
  }
  switch (bucket.eviction_policy) {
    case management::cluster::bucket_eviction_policy::unknown:
      bucket_settings.eviction_policy = core::management::cluster::bucket_eviction_policy::unknown;
      break;
    case management::cluster::bucket_eviction_policy::full:
      bucket_settings.eviction_policy = core::management::cluster::bucket_eviction_policy::full;
      break;
    case management::cluster::bucket_eviction_policy::value_only:
      bucket_settings.eviction_policy =
        core::management::cluster::bucket_eviction_policy::value_only;
      break;
    case management::cluster::bucket_eviction_policy::no_eviction:
      bucket_settings.eviction_policy =
        core::management::cluster::bucket_eviction_policy::no_eviction;
      break;
    case management::cluster::bucket_eviction_policy::not_recently_used:
      bucket_settings.eviction_policy =
        core::management::cluster::bucket_eviction_policy::not_recently_used;
      break;
  }
  switch (bucket.compression_mode) {
    case management::cluster::bucket_compression::unknown:
      bucket_settings.compression_mode = core::management::cluster::bucket_compression::unknown;
      break;
    case management::cluster::bucket_compression::active:
      bucket_settings.compression_mode = core::management::cluster::bucket_compression::active;
      break;
    case management::cluster::bucket_compression::passive:
      bucket_settings.compression_mode = core::management::cluster::bucket_compression::passive;
      break;
    case management::cluster::bucket_compression::off:
      bucket_settings.compression_mode = core::management::cluster::bucket_compression::off;
      break;
  }
  switch (bucket.bucket_type) {
    case management::cluster::bucket_type::unknown:
      bucket_settings.bucket_type = core::management::cluster::bucket_type::unknown;
      break;
    case management::cluster::bucket_type::couchbase:
      bucket_settings.bucket_type = core::management::cluster::bucket_type::couchbase;
      break;
    case management::cluster::bucket_type::memcached:
      bucket_settings.bucket_type = core::management::cluster::bucket_type::memcached;
      break;
    case management::cluster::bucket_type::ephemeral:
      bucket_settings.bucket_type = core::management::cluster::bucket_type::ephemeral;
      break;
  }
  switch (bucket.storage_backend) {
    case management::cluster::bucket_storage_backend::unknown:
      bucket_settings.storage_backend = core::management::cluster::bucket_storage_backend::unknown;
      break;
    case management::cluster::bucket_storage_backend::magma:
      bucket_settings.storage_backend = core::management::cluster::bucket_storage_backend::magma;
      break;
    case management::cluster::bucket_storage_backend::couchstore:
      bucket_settings.storage_backend =
        core::management::cluster::bucket_storage_backend::couchstore;
      break;
  }
  return bucket_settings;
}
} // namespace

class bucket_manager_impl
{
public:
  explicit bucket_manager_impl(core::cluster core)
    : core_{ std::move(core) }
  {
  }

  void get_bucket(std::string bucket_name,
                  const get_bucket_options::built& options,
                  get_bucket_handler&& handler) const
  {
    auto obs_rec = create_observability_recorder(
      core::tracing::operation::mgr_buckets_get_bucket, bucket_name, options.parent_span);
    core::operations::management::bucket_get_request request{
      std::move(bucket_name),
      {},
      options.timeout,
      obs_rec->operation_span(),
    };
    core_.execute(
      std::move(request),
      [obs_rec = std::move(obs_rec), handler = std::move(handler)](const auto& resp) mutable {
        obs_rec->finish(resp.ctx.retry_attempts, resp.ctx.ec);
        return handler(core::impl::make_error(resp.ctx), map_bucket_settings(resp.bucket));
      });
  }

  void get_all_buckets(const get_all_buckets_options::built& options,
                       get_all_buckets_handler&& handler) const
  {
    auto obs_rec = create_observability_recorder(
      core::tracing::operation::mgr_buckets_get_all_buckets, std::nullopt, options.parent_span);
    core::operations::management::bucket_get_all_request request{
      {},
      options.timeout,
      obs_rec->operation_span(),
    };
    core_.execute(
      std::move(request),
      [obs_rec = std::move(obs_rec), handler = std::move(handler)](const auto& resp) mutable {
        obs_rec->finish(resp.ctx.retry_attempts, resp.ctx.ec);
        return handler(core::impl::make_error(resp.ctx), map_all_bucket_settings(resp.buckets));
      });
  }

  void create_bucket(const management::cluster::bucket_settings& bucket_settings,
                     const create_bucket_options::built& options,
                     create_bucket_handler&& handler) const
  {
    auto obs_rec =
      create_observability_recorder(core::tracing::operation::mgr_buckets_create_bucket,
                                    bucket_settings.name,
                                    options.parent_span);
    core::operations::management::bucket_create_request request{
      map_bucket_settings(bucket_settings),
      {},
      options.timeout,
      obs_rec->operation_span(),
    };
    core_.execute(
      std::move(request),
      [obs_rec = std::move(obs_rec), handler = std::move(handler)](const auto& resp) mutable {
        obs_rec->finish(resp.ctx.retry_attempts, resp.ctx.ec);
        return handler(core::impl::make_error(resp.ctx));
      });
  }

  void update_bucket(const management::cluster::bucket_settings& bucket_settings,
                     const update_bucket_options::built& options,
                     update_bucket_handler&& handler) const
  {
    auto obs_rec =
      create_observability_recorder(core::tracing::operation::mgr_buckets_update_bucket,
                                    bucket_settings.name,
                                    options.parent_span);
    core::operations::management::bucket_update_request request{
      map_bucket_settings(bucket_settings),
      {},
      options.timeout,
      obs_rec->operation_span(),
    };
    core_.execute(
      std::move(request),
      [obs_rec = std::move(obs_rec), handler = std::move(handler)](const auto& resp) mutable {
        obs_rec->finish(resp.ctx.retry_attempts, resp.ctx.ec);
        return handler(core::impl::make_error(resp.ctx));
      });
  }

  void drop_bucket(std::string bucket_name,
                   const drop_bucket_options::built& options,
                   drop_bucket_handler&& handler) const
  {
    auto obs_rec = create_observability_recorder(
      core::tracing::operation::mgr_buckets_drop_bucket, bucket_name, options.parent_span);
    core::operations::management::bucket_drop_request request{
      std::move(bucket_name),
      {},
      options.timeout,
      obs_rec->operation_span(),
    };
    core_.execute(
      std::move(request),
      [obs_rec = std::move(obs_rec), handler = std::move(handler)](const auto& resp) mutable {
        obs_rec->finish(resp.ctx.retry_attempts, resp.ctx.ec);
        return handler(core::impl::make_error(resp.ctx));
      });
  }

  void flush_bucket(std::string bucket_name,
                    const flush_bucket_options::built& options,
                    flush_bucket_handler&& handler) const
  {
    auto obs_rec = create_observability_recorder(
      core::tracing::operation::mgr_buckets_flush_bucket, bucket_name, options.parent_span);
    core::operations::management::bucket_flush_request request{
      std::move(bucket_name),
      {},
      options.timeout,
      obs_rec->operation_span(),
    };
    core_.execute(
      std::move(request),
      [obs_rec = std::move(obs_rec), handler = std::move(handler)](const auto& resp) mutable {
        obs_rec->finish(resp.ctx.retry_attempts, resp.ctx.ec);
        return handler(core::impl::make_error(resp.ctx));
      });
  }

private:
  [[nodiscard]] auto create_observability_recorder(
    const std::string& operation_name,
    const std::optional<std::string>& bucket_name,
    const std::shared_ptr<tracing::request_span>& parent_span) const
    -> std::unique_ptr<core::impl::observability_recorder>
  {
    auto rec = core::impl::observability_recorder::create(
      operation_name, parent_span, core_.tracer(), core_.meter());
    rec->with_service(core::tracing::service::management);
    if (bucket_name.has_value()) {
      rec->with_bucket_name(bucket_name.value());
    }
    return rec;
  }

  core::cluster core_;
};

bucket_manager::bucket_manager(core::cluster core)
  : impl_(std::make_shared<bucket_manager_impl>(std::move(core)))
{
}

void
bucket_manager::get_bucket(std::string bucket_name,
                           const get_bucket_options& options,
                           get_bucket_handler&& handler) const
{
  return impl_->get_bucket(std::move(bucket_name), options.build(), std::move(handler));
}

auto
bucket_manager::get_bucket(std::string bucket_name, const get_bucket_options& options) const
  -> std::future<std::pair<error, management::cluster::bucket_settings>>
{
  auto barrier =
    std::make_shared<std::promise<std::pair<error, management::cluster::bucket_settings>>>();
  get_bucket(std::move(bucket_name), options, [barrier](auto err, auto result) mutable {
    barrier->set_value(std::make_pair(std::move(err), std::move(result)));
  });
  return barrier->get_future();
}

void
bucket_manager::get_all_buckets(const get_all_buckets_options& options,
                                get_all_buckets_handler&& handler) const
{
  return impl_->get_all_buckets(options.build(), std::move(handler));
}

auto
bucket_manager::get_all_buckets(const get_all_buckets_options& options) const
  -> std::future<std::pair<error, std::vector<management::cluster::bucket_settings>>>
{
  auto barrier = std::make_shared<
    std::promise<std::pair<error, std::vector<management::cluster::bucket_settings>>>>();
  get_all_buckets(options, [barrier](auto err, auto result) mutable {
    barrier->set_value(std::make_pair(std::move(err), std::move(result)));
  });
  return barrier->get_future();
}

void
bucket_manager::create_bucket(const management::cluster::bucket_settings& bucket_settings,
                              const create_bucket_options& options,
                              create_bucket_handler&& handler) const
{
  return impl_->create_bucket(bucket_settings, options.build(), std::move(handler));
}

auto
bucket_manager::create_bucket(const management::cluster::bucket_settings& bucket_settings,
                              const create_bucket_options& options) const -> std::future<error>
{
  auto barrier = std::make_shared<std::promise<error>>();
  create_bucket(bucket_settings, options, [barrier](auto err) mutable {
    barrier->set_value(std::move(err));
  });
  return barrier->get_future();
}

void
bucket_manager::update_bucket(const management::cluster::bucket_settings& bucket_settings,
                              const update_bucket_options& options,
                              update_bucket_handler&& handler) const
{
  return impl_->update_bucket(bucket_settings, options.build(), std::move(handler));
}

auto
bucket_manager::update_bucket(const management::cluster::bucket_settings& bucket_settings,
                              const update_bucket_options& options) const -> std::future<error>
{
  auto barrier = std::make_shared<std::promise<error>>();
  update_bucket(bucket_settings, options, [barrier](auto err) mutable {
    barrier->set_value(std::move(err));
  });
  return barrier->get_future();
}

void
bucket_manager::drop_bucket(std::string bucket_name,
                            const drop_bucket_options& options,
                            drop_bucket_handler&& handler) const
{
  return impl_->drop_bucket(std::move(bucket_name), options.build(), std::move(handler));
}

auto
bucket_manager::drop_bucket(std::string bucket_name,
                            const drop_bucket_options& options) const -> std::future<error>
{
  auto barrier = std::make_shared<std::promise<error>>();
  drop_bucket(std::move(bucket_name), options, [barrier](auto err) mutable {
    barrier->set_value(std::move(err));
  });
  return barrier->get_future();
}

void
bucket_manager::flush_bucket(std::string bucket_name,
                             const flush_bucket_options& options,
                             flush_bucket_handler&& handler) const
{
  return impl_->flush_bucket(std::move(bucket_name), options.build(), std::move(handler));
}

auto
bucket_manager::flush_bucket(std::string bucket_name,
                             const flush_bucket_options& options) const -> std::future<error>
{
  auto barrier = std::make_shared<std::promise<error>>();
  flush_bucket(std::move(bucket_name), options, [barrier](auto err) mutable {
    barrier->set_value(std::move(err));
  });
  return barrier->get_future();
}
} // namespace couchbase