#pragma once

#include <map>

#include <memory>

#include <string>

#include "envoy/common/time.h"

#include "envoy/data/core/v3/health_check_event.pb.h"

#include "envoy/server/health_checker_config.h"

#include "envoy/upstream/health_check_event_sink.h"

#include "source/common/common/thread.h"

#include "source/common/protobuf/protobuf.h"

#include "absl/base/thread_annotations.h"

#include "cilium/api/health_check_sink.pb.h"

#include "cilium/api/health_check_sink.pb.validate.h" // IWYU pragma: keep

#include "cilium/uds_client.h"

namespace Envoy {

namespace Cilium {

class HealthCheckEventPipeSinkFactory;

class HealthCheckEventPipeSink : public Upstream::HealthCheckEventSink {

public:

  void log(envoy::data::core::v3::HealthCheckEvent event) override;

protected:

  friend class HealthCheckEventPipeSinkFactory;

  friend class HealthCheckEventPipeSink_logTest_Test;

  explicit HealthCheckEventPipeSink(const cilium::HealthCheckEventPipeSink& config,

                                    TimeSource& time_source);

private:

  static Thread::MutexBasicLockable udss_mutex;

  static std::map<std::string, std::weak_ptr<UDSClient>> udss ABSL_GUARDED_BY(udss_mutex);

  std::shared_ptr<UDSClient> uds_client_;

};

class HealthCheckEventPipeSinkFactory : public Upstream::HealthCheckEventSinkFactory {

public:

  Upstream::HealthCheckEventSinkPtr

  createHealthCheckEventSink(const ProtobufWkt::Any& config,

                             Server::Configuration::HealthCheckerFactoryContext& context) override;

};

} // namespace Cilium

} // namespace Envoy