#pragma once

#include "envoy/event/timer.h"

#include "envoy/extensions/transport_sockets/tap/v3/tap.pb.h"

#include "envoy/network/transport_socket.h"

#include "source/extensions/common/tap/extension_config_base.h"

#include "source/extensions/transport_sockets/common/passthrough.h"

#include "source/extensions/transport_sockets/tap/tap_config.h"

namespace Envoy {

namespace Extensions {

namespace TransportSockets {

namespace Tap {

class TapSocket : public TransportSockets::PassthroughSocket {

public:

  TapSocket(SocketTapConfigSharedPtr config,

            const envoy::extensions::transport_sockets::tap::v3::SocketTapConfig& socket_tap_config,

            Stats::Scope& stats_scope, Network::TransportSocketPtr&& transport_socket);

  // Network::TransportSocket

  void setTransportSocketCallbacks(Network::TransportSocketCallbacks& callbacks) override;

  void closeSocket(Network::ConnectionEvent event) override;

  Network::IoResult doRead(Buffer::Instance& buffer) override;

  Network::IoResult doWrite(Buffer::Instance& buffer, bool end_stream) override;

private:

  SocketTapConfigSharedPtr config_;

  PerSocketTapperPtr tapper_;

  const envoy::extensions::transport_sockets::tap::v3::SocketTapConfig socket_tap_config_;

  TransportTapStats stats_;

  static TransportTapStats generateStats(Stats::Scope& stats_scope, const std::string& prefix);

};

class TapSocketFactory : public Common::Tap::ExtensionConfigBase, public PassthroughFactory {

public:

  TapSocketFactory(const envoy::extensions::transport_sockets::tap::v3::Tap& proto_config,

                   Common::Tap::TapConfigFactoryPtr&& config_factory, OptRef<Server::Admin> admin,

                   Singleton::Manager& singleton_manager, ThreadLocal::SlotAllocator& tls,

                   Event::Dispatcher& main_thread_dispatcher, Stats::Scope& scope,

                   Network::UpstreamTransportSocketFactoryPtr&& transport_socket_factory);

  // Network::UpstreamTransportSocketFactory

  Network::TransportSocketPtr

  createTransportSocket(Network::TransportSocketOptionsConstSharedPtr options,

                        Upstream::HostDescriptionConstSharedPtr host) const override;

private:

  const envoy::extensions::transport_sockets::tap::v3::SocketTapConfig ts_tap_config_;

  Stats::Scope& stats_scope_;

};

class DownstreamTapSocketFactory : public Common::Tap::ExtensionConfigBase,

                                   public DownstreamPassthroughFactory {

public:

  DownstreamTapSocketFactory(

      const envoy::extensions::transport_sockets::tap::v3::Tap& proto_config,

      Common::Tap::TapConfigFactoryPtr&& config_factory, OptRef<Server::Admin> admin,

      Singleton::Manager& singleton_manager, ThreadLocal::SlotAllocator& tls,

      Event::Dispatcher& main_thread_dispatcher, Stats::Scope& scope,

      Network::DownstreamTransportSocketFactoryPtr&& transport_socket_factory);

  // Network::UpstreamTransportSocketFactory

  Network::TransportSocketPtr createDownstreamTransportSocket() const override;

private:

  const envoy::extensions::transport_sockets::tap::v3::SocketTapConfig ds_ts_tap_config_;

  Stats::Scope& stats_scope_;

};

} // namespace Tap

} // namespace TransportSockets

} // namespace Extensions

} // namespace Envoy