#pragma once

#include "envoy/buffer/buffer.h"

#include "envoy/network/connection.h"

#include "envoy/network/transport_socket.h"

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

#include "source/common/network/transport_socket_options_impl.h"

namespace Envoy {

namespace Network {

class RawBufferSocket : public TransportSocket, protected Logger::Loggable<Logger::Id::connection> {

public:

  // Network::TransportSocket

  void setTransportSocketCallbacks(TransportSocketCallbacks& callbacks) override;

  std::string protocol() const override;

  absl::string_view failureReason() const override;

  void onConnected() override;

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

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

protected:

private:

  bool shutdown_{};

  TransportSocketCallbacks* callbacks_{};

};

class RawBufferSocketFactory : public DownstreamTransportSocketFactory,

                               public CommonUpstreamTransportSocketFactory {

public:

  // Network::UpstreamTransportSocketFactory

  TransportSocketPtr createTransportSocket(TransportSocketOptionsConstSharedPtr,

                                           Upstream::HostDescriptionConstSharedPtr) const override;

  bool implementsSecureTransport() const override;

  // Network::DownstreamTransportSocketFactory

  TransportSocketPtr createDownstreamTransportSocket() const override;

};

} // namespace Network

} // namespace Envoy