#pragma once

#include "envoy/extensions/http/header_formatters/preserve_case/v3/preserve_case.pb.h"

#include "envoy/http/header_formatter.h"

#include "envoy/registry/registry.h"

#include "source/common/protobuf/utility.h"

#include "source/extensions/http/header_formatters/preserve_case/preserve_case_formatter.h"

namespace Envoy {

namespace Extensions {

namespace Http {

namespace HeaderFormatters {

namespace PreserveCase {

class PreserveCaseFormatterFactory : public Envoy::Http::StatefulHeaderKeyFormatterFactory {

public:

  PreserveCaseFormatterFactory(

      const bool forward_reason_phrase,

      const envoy::extensions::http::header_formatters::preserve_case::v3::

          PreserveCaseFormatterConfig::FormatterTypeOnEnvoyHeaders formatter_type_on_envoy_headers)

  // Envoy::Http::StatefulHeaderKeyFormatterFactory

private:

  const bool forward_reason_phrase_;

  const envoy::extensions::http::header_formatters::preserve_case::v3::PreserveCaseFormatterConfig::

      FormatterTypeOnEnvoyHeaders formatter_type_on_envoy_headers_;

};

class PreserveCaseFormatterFactoryConfig

    : public Envoy::Http::StatefulHeaderKeyFormatterFactoryConfig {

public:

  // Envoy::Http::StatefulHeaderKeyFormatterFactoryConfig

  Envoy::Http::StatefulHeaderKeyFormatterFactorySharedPtr

  createFromProto(const Protobuf::Message& message) override;

};

DECLARE_FACTORY(PreserveCaseFormatterFactoryConfig);

} // namespace PreserveCase

} // namespace HeaderFormatters

} // namespace Http

} // namespace Extensions

} // namespace Envoy