local_reply_config : support content-type in SubstitutionFormatString

api/envoy/config/core/v3/substitution_format_string.proto

@@ -65,4 +65,12 @@ message SubstitutionFormatString {
   //   empty string, so that empty values are omitted entirely.
   // * for ``json_format`` the keys with null values are omitted in the output structure.
   bool omit_empty_values = 3;
+
+  // Specify a content_type for text_format. This will be ignored for json_format
+  //
+  // .. code-block::
+  //
+  //   content_type: "text/html; charset=UTF-8"
+  //
+  string content_type = 4 [(validate.rules).string = {min_bytes: 1}];
 }

docs/root/version_history/current.rst

@@ -64,6 +64,7 @@ New Features
 * access log: added support for :ref:`%DOWNSTREAM_PEER_FINGERPRINT_1% <config_access_log_format_response_flags>` as a response flag.
 * access log: added support for nested objects in :ref:`JSON logging mode <config_access_log_format_dictionaries>`.
 * access log: added :ref:`omit_empty_values<envoy_v3_api_field_config.core.v3.SubstitutionFormatString.omit_empty_values>` option to omit unset value from formatted log.
+* access log: added :ref:`content_type<envoy_v3_api_field_config.core.v3.SubstitutionFormatString.content_type>` option to set content-type for textformat.
 * build: enable building envoy :ref:`arm64 images <arm_binaries>` by buildx tool in x86 CI platform.
 * cluster: added new :ref:`connection_pool_per_downstream_connection <envoy_v3_api_field_config.cluster.v3.Cluster.connection_pool_per_downstream_connection>` flag, which enable creation of a new connection pool for each downstream connection.
 * decompressor filter: reports compressed and uncompressed bytes in trailers.

source/common/formatter/substitution_format_string.cc

@@ -1,6 +1,7 @@
 #include "common/formatter/substitution_format_string.h"
 
 #include "common/formatter/substitution_formatter.h"
+#include "common/http/headers.h"
 
 namespace Envoy {
 namespace Formatter {
@@ -25,5 +26,18 @@ FormatterPtr SubstitutionFormatStringUtils::fromProtoConfig(
   return nullptr;
 }
 
+absl::string_view SubstitutionFormatStringUtils::getContentType(
+    const envoy::config::core::v3::SubstitutionFormatString& config) {
+  const std::string contentType = config.content_type();
+  if (contentType == Http::Headers::get().ContentTypeValues.TextEventStream)
+    return Http::Headers::get().ContentTypeValues.TextEventStream;
+  else if (contentType == Http::Headers::get().ContentTypeValues.TextUtf8)
+    return Http::Headers::get().ContentTypeValues.TextUtf8;
+  else if (contentType == Http::Headers::get().ContentTypeValues.Html)
+    return Http::Headers::get().ContentTypeValues.Html;
+  else
+    return Http::Headers::get().ContentTypeValues.Text;
+}
+
 } // namespace Formatter
 } // namespace Envoy

source/common/formatter/substitution_format_string.h

@@ -26,6 +26,12 @@ class SubstitutionFormatStringUtils {
    */
   static FormatterPtr createJsonFormatter(const ProtobufWkt::Struct& struct_format,
                                           bool preserve_types, bool omit_empty_values);
+
+  /**
+   * Returns content-type from config SubstitutionFormatString
+   */
+  static absl::string_view
+  getContentType(const envoy::config::core::v3::SubstitutionFormatString& config);
 };
 
 } // namespace Formatter

source/common/local_reply/local_reply.cc

@@ -26,7 +26,7 @@ class BodyFormatter {
             config.format_case() ==
                     envoy::config::core::v3::SubstitutionFormatString::FormatCase::kJsonFormat
                 ? Http::Headers::get().ContentTypeValues.Json
-                : Http::Headers::get().ContentTypeValues.Text) {}
+                : Formatter::SubstitutionFormatStringUtils::getContentType(config)) {}
 
   void format(const Http::RequestHeaderMap& request_headers,
               const Http::ResponseHeaderMap& response_headers,

test/common/formatter/substitution_format_string_test.cc

@@ -94,5 +94,15 @@ TEST_F(SubstitutionFormatStringUtilsTest, TestInvalidConfigs) {
   }
 }
 
+TEST_F(SubstitutionFormatStringUtilsTest, TestFromProtoConfigContentType) {
+  const std::string yaml = R"EOF(
+  text_format: "<h1>Sample html</h1>"
+  content_type: "text/html; charset=UTF-8"
+)EOF";
+  TestUtility::loadFromYaml(yaml, config_);
+  const absl::string_view contentType = SubstitutionFormatStringUtils::getContentType(config_);
+  EXPECT_EQ("text/html; charset=UTF-8", contentType);
+}
+
 } // namespace Formatter
 } // namespace Envoy

test/common/local_reply/local_reply_test.cc

@@ -336,5 +336,62 @@ TEST_F(LocalReplyTest, TestHeaderAddition) {
   ASSERT_EQ(out[1], "append-bar3");
 }
 
+TEST_F(LocalReplyTest, TestMapperWithContentType) {
+  // Match with response_code, and rewrite the code and body.
+  const std::string yaml = R"(
+    mappers:
+    - filter:
+        status_code_filter:
+          comparison:
+            op: EQ
+            value:
+              default_value: 400
+              runtime_key: key_b
+      status_code: 401
+      body:
+        inline_string: "401 body text"
+      body_format_override:
+        text_format: "<h1>%LOCAL_REPLY_BODY%</h1>"
+        content_type: "text/html; charset=UTF-8"
+    - filter:
+        status_code_filter:
+          comparison:
+            op: EQ
+            value:
+              default_value: 410
+              runtime_key: key_b
+      status_code: 411
+      body:
+        inline_string: "411 body text"
+    body_format:
+      text_format: "<h1>%LOCAL_REPLY_BODY%</h1> %RESPONSE_CODE% default formatter"
+      content_type: "text/html; charset=UTF-8"
+)";
+  TestUtility::loadFromYaml(yaml, config_);
+  auto local = Factory::create(config_, context_);
+
+  // code=400 matches the first filter; rewrite code and body
+  // has its own formatter.
+  resetData(400);
+  local->rewrite(&request_headers_, response_headers_, stream_info_, code_, body_, content_type_);
+  EXPECT_EQ(code_, static_cast<Http::Code>(401));
+  EXPECT_EQ(stream_info_.response_code_, 401U);
+  EXPECT_EQ(response_headers_.Status()->value().getStringView(), "401");
+  EXPECT_EQ(content_type_, "text/html; charset=UTF-8");
+
+  const std::string expected = R"(<h1>401 body text</h1>)";
+  EXPECT_EQ(body_, expected);
+
+  // code=410 matches the second filter; rewrite code and body
+  // but using default formatter.
+  resetData(410);
+  local->rewrite(&request_headers_, response_headers_, stream_info_, code_, body_, content_type_);
+  EXPECT_EQ(code_, static_cast<Http::Code>(411));
+  EXPECT_EQ(stream_info_.response_code_, 411U);
+  EXPECT_EQ(response_headers_.Status()->value().getStringView(), "411");
+  EXPECT_EQ(body_, "<h1>411 body text</h1> 411 default formatter");
+  EXPECT_EQ(content_type_, "text/html; charset=UTF-8");
+}
+
 } // namespace LocalReply
 } // namespace Envoy