#include "core_v2/core.h"

#include <cassert>
#include <vector>

#include "core_v2/options.h"
#include "platform_v2/public/count_down_latch.h"
#include "platform_v2/public/logging.h"
#include "absl/time/clock.h"

namespace location {
namespace nearby {
namespace connections {

constexpr absl::Duration Core::kWaitForDisconnect;

Core::~Core() {
  CountDownLatch latch(1);
  router_.ClientDisconnecting(
      &client_, {
                    .result_cb = [&latch](Status) { latch.CountDown(); },
                });
  if (!latch.Await(kWaitForDisconnect).result()) {
    NEARBY_LOG(FATAL, "Unable to shutdown");
  }
}

void Core::StartAdvertising(absl::string_view service_id,
                            ConnectionOptions options,
                            ConnectionRequestInfo info,
                            ResultCallback callback) {
  assert(!service_id.empty());
  assert(options.strategy.IsValid());

  router_.StartAdvertising(&client_, service_id, options, info, callback);
}

void Core::StopAdvertising(const ResultCallback callback) {
  router_.StopAdvertising(&client_, callback);
}

void Core::StartDiscovery(absl::string_view service_id,
                          ConnectionOptions options, DiscoveryListener listener,
                          ResultCallback callback) {
  assert(!service_id.empty());
  assert(options.strategy.IsValid());

  router_.StartDiscovery(&client_, service_id, options, listener, callback);
}

void Core::StopDiscovery(ResultCallback callback) {
  router_.StopDiscovery(&client_, callback);
}

void Core::RequestConnection(absl::string_view endpoint_id,
                             ConnectionRequestInfo info,
                             ResultCallback callback) {
  assert(!endpoint_id.empty());

  router_.RequestConnection(&client_, endpoint_id, info, callback);
}

void Core::AcceptConnection(absl::string_view endpoint_id,
                            PayloadListener listener, ResultCallback callback) {
  assert(!endpoint_id.empty());

  router_.AcceptConnection(&client_, endpoint_id, listener, callback);
}

void Core::RejectConnection(absl::string_view endpoint_id,
                            ResultCallback callback) {
  assert(!endpoint_id.empty());

  router_.RejectConnection(&client_, endpoint_id, callback);
}

void Core::InitiateBandwidthUpgrade(absl::string_view endpoint_id,
                                    ResultCallback callback) {
  router_.InitiateBandwidthUpgrade(&client_, endpoint_id, callback);
}

void Core::SendPayload(absl::Span<const std::string> endpoint_ids,
                       Payload payload, ResultCallback callback) {
  assert(payload.GetType() != Payload::Type::kUnknown);
  assert(!endpoint_ids.empty());

  router_.SendPayload(&client_, endpoint_ids, std::move(payload), callback);
}

void Core::CancelPayload(std::int64_t payload_id, ResultCallback callback) {
  assert(payload_id != 0);

  router_.CancelPayload(&client_, payload_id, callback);
}

void Core::DisconnectFromEndpoint(absl::string_view endpoint_id,
                                  ResultCallback callback) {
  assert(!endpoint_id.empty());

  router_.DisconnectFromEndpoint(&client_, endpoint_id, callback);
}

void Core::StopAllEndpoints(ResultCallback callback) {
  router_.StopAllEndpoints(&client_, callback);
}

}  // namespace connections
}  // namespace nearby
}  // namespace location