#pragma once

#include <memory>

#include "envoy/network/connection.h"

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

namespace Envoy {

namespace Quic {

// Read ~32k bytes per connection by default, which is about the same as TCP.

static const uint32_t DEFAULT_PACKETS_TO_READ_PER_CONNECTION = 32u;

class QuicWriteEventCallback {

public:

  // Called when QUIC finishes a write.

  virtual void onWriteEventDone() PURE;

};

// A base class of both the client and server connections which keeps stats and

// connection socket.

class QuicNetworkConnection : protected Logger::Loggable<Logger::Id::connection> {

public:

  QuicNetworkConnection(Network::ConnectionSocketPtr&& connection_socket);

  virtual ~QuicNetworkConnection();

  // Called by EnvoyQuicSession::setConnectionStats().

  // Called in session Initialize().

  void setEnvoyConnection(Network::Connection& connection, QuicWriteEventCallback& write_callback);

  // Needed for ENVOY_CONN_LOG.

  uint64_t id() const;

protected:

  // REQUIRES: hasConnectionStats() == true.

  void onWriteEventDone();

private:

  // TODO(danzh): populate stats.

  std::unique_ptr<Network::Connection::ConnectionStats> connection_stats_;

  // Hosts a list of active sockets, while only the last one is used for writing data.

  // Hosts a single default socket upon construction. New sockets can be pushed in later as a result

  // of QUIC connection migration.

  std::vector<Network::ConnectionSocketPtr> connection_sockets_;

  // Points to an instance of EnvoyQuicServerSession or EnvoyQuicClientSession.

  Network::Connection* envoy_connection_{nullptr};

  QuicWriteEventCallback* write_callback_{nullptr};

};

} // namespace Quic

} // namespace Envoy