From 1b23bcfeaa27736a84d846cd96eb5695d997a246 Mon Sep 17 00:00:00 2001
From: qicosmos <qicosmos@linux.alibaba.com>
Date: Tue, 30 Jan 2024 11:24:44 +0800
Subject: [PATCH] support gzip

---
 .../thirdparty/cinatra/coro_http_client.hpp   |  2 +-
 .../thirdparty/cinatra/coro_http_response.hpp | 27 ++++++++++++-
 src/coro_http/examples/example.cpp            | 38 +++++++++++++++++++
 3 files changed, 64 insertions(+), 3 deletions(-)

diff --git a/include/ylt/thirdparty/cinatra/coro_http_client.hpp b/include/ylt/thirdparty/cinatra/coro_http_client.hpp
index 3d8b46b29..34219e57e 100644
--- a/include/ylt/thirdparty/cinatra/coro_http_client.hpp
+++ b/include/ylt/thirdparty/cinatra/coro_http_client.hpp
@@ -246,7 +246,7 @@ class coro_http_client : public std::enable_shared_from_this<coro_http_client> {
     return true;
   }
 
-  [[nodiscard]] bool init_ssl(int verify_mode = asio::ssl::verify_peer,
+  [[nodiscard]] bool init_ssl(int verify_mode = asio::ssl::verify_none,
                               std::string full_path = "",
                               const std::string &sni_hostname = "") {
     std::string base_path;
diff --git a/include/ylt/thirdparty/cinatra/coro_http_response.hpp b/include/ylt/thirdparty/cinatra/coro_http_response.hpp
index 4b8c3b88e..d035a466c 100644
--- a/include/ylt/thirdparty/cinatra/coro_http_response.hpp
+++ b/include/ylt/thirdparty/cinatra/coro_http_response.hpp
@@ -12,6 +12,9 @@
 #include "async_simple/coro/SyncAwait.h"
 #include "cookie.hpp"
 #include "define.h"
+#ifdef CINATRA_ENABLE_GZIP
+#include "gzip.hpp"
+#endif
 #include "response_cv.hpp"
 #include "time_util.hpp"
 #include "utils.hpp"
@@ -46,9 +49,29 @@ class coro_http_response {
     content_ = std::move(content);
     has_set_content_ = true;
   }
-  void set_status_and_content(status_type status, std::string content = "") {
+  void set_status_and_content(
+      status_type status, std::string content = "",
+      content_encoding encoding = content_encoding::none) {
     status_ = status;
-    content_ = std::move(content);
+#ifdef CINATRA_ENABLE_GZIP
+    if (encoding == content_encoding::gzip) {
+      std::string encode_str;
+      bool r = gzip_codec::compress(
+          std::string_view(content.data(), content.length()), encode_str, true);
+      if (!r) {
+        set_status_and_content(status_type::internal_server_error,
+                               "gzip compress error");
+      }
+      else {
+        add_header("Content-Encoding", "gzip");
+        set_content(std::move(encode_str));
+      }
+    }
+    else
+#endif
+    {
+      content_ = std::move(content);
+    }
     has_set_content_ = true;
   }
   void set_delay(bool r) { delay_ = r; }
diff --git a/src/coro_http/examples/example.cpp b/src/coro_http/examples/example.cpp
index 9ba4fb353..607310e44 100644
--- a/src/coro_http/examples/example.cpp
+++ b/src/coro_http/examples/example.cpp
@@ -405,6 +405,41 @@ async_simple::coro::Lazy<void> basic_usage() {
 #endif
 }
 
+#ifdef CINATRA_ENABLE_GZIP
+std::string_view get_header_value(auto &resp_headers, std::string_view key) {
+  for (const auto &[k, v] : resp_headers) {
+    if (k == key)
+      return v;
+  }
+  return {};
+}
+
+void test_gzip() {
+  coro_http_server server(1, 8090);
+  server.set_http_handler<GET, POST>(
+      "/gzip", [](coro_http_request &req, coro_http_response &res) {
+        assert(req.get_header_value("Content-Encoding") == "gzip");
+        res.set_status_and_content(status_type::ok, "hello world",
+                                   content_encoding::gzip);
+      });
+  server.async_start();
+
+  std::this_thread::sleep_for(std::chrono::milliseconds(100));
+
+  coro_http_client client{};
+  std::string uri = "http://127.0.0.1:8090/gzip";
+  client.add_header("Content-Encoding", "gzip");
+  auto result = async_simple::coro::syncAwait(client.async_get(uri));
+  auto content = get_header_value(result.resp_headers, "Content-Encoding");
+  assert(get_header_value(result.resp_headers, "Content-Encoding") == "gzip");
+  std::string decompress_data;
+  bool ret = gzip_codec::uncompress(result.resp_body, decompress_data);
+  assert(ret == true);
+  assert(decompress_data == "hello world");
+  server.stop();
+}
+#endif
+
 int main() {
   async_simple::coro::syncAwait(basic_usage());
   async_simple::coro::syncAwait(use_aspects());
@@ -412,5 +447,8 @@ int main() {
   async_simple::coro::syncAwait(use_websocket());
   async_simple::coro::syncAwait(chunked_upload_download());
   async_simple::coro::syncAwait(byte_ranges_download());
+#ifdef CINATRA_ENABLE_GZIP
+  test_gzip();
+#endif
   return 0;
 }
\ No newline at end of file