#pragma once

#include <string>

#include "envoy/buffer/buffer.h"

#include "source/extensions/filters/network/thrift_proxy/transport.h"

namespace Envoy {

namespace Extensions {

namespace NetworkFilters {

namespace ThriftProxy {

/**

 * FramedTransportImpl implements the Thrift Framed transport.

 * See https://github.com/apache/thrift/blob/master/doc/specs/thrift-rpc.md

 */

class FramedTransportImpl : public Transport {

public:

  FramedTransportImpl() = default;

  // Transport

  const std::string& name() const override { return TransportNames::get().FRAMED; }

  TransportType type() const override { return TransportType::Framed; }

  bool decodeFrameStart(Buffer::Instance& buffer, MessageMetadata& metadata) override;

  bool decodeFrameEnd(Buffer::Instance& buffer) override;

  void encodeFrame(Buffer::Instance& buffer, const MessageMetadata& metadata,

                   Buffer::Instance& message) override;

  static const int32_t MaxFrameSize = 0xFA0000;

};

} // namespace ThriftProxy

} // namespace NetworkFilters

} // namespace Extensions

} // namespace Envoy