#pragma once

#include "envoy/network/udp_packet_writer_handler.h"

#include "quiche/quic/core/quic_packet_writer.h"

namespace Envoy {

namespace Quic {

class EnvoyQuicPacketWriter : public quic::QuicPacketWriter {

public:

  EnvoyQuicPacketWriter(Network::UdpPacketWriterPtr envoy_udp_packet_writer);

  quic::WriteResult WritePacket(const char* buffer, size_t buf_len,

                                const quic::QuicIpAddress& self_address,

                                const quic::QuicSocketAddress& peer_address,

                                quic::PerPacketOptions* options,

                                const quic::QuicPacketWriterParams& params) override;

  // quic::QuicPacketWriter

  // Currently this writer doesn't support pacing offload.

  // Currently this writer doesn't support Explicit Congestion Notification.

  absl::optional<int> MessageTooBigErrorCode() const override;

  quic::QuicByteCount GetMaxPacketSize(const quic::QuicSocketAddress& peer_address) const override;

  quic::QuicPacketBuffer GetNextWriteLocation(const quic::QuicIpAddress& self_address,

                                              const quic::QuicSocketAddress& peer_address) override;

  quic::WriteResult Flush() override;

private:

  Network::UdpPacketWriterPtr envoy_udp_packet_writer_;

};

} // namespace Quic

} // namespace Envoy