From 638c52eb07d05e6769392c995bf77624f34e2f12 Mon Sep 17 00:00:00 2001
From: Ushitora Anqou <ushitora@anqou.net>
Date: Sun, 10 Mar 2024 18:44:17 +0900
Subject: [PATCH 1/2] cohttp-lwt: server: return connection header

---
 cohttp-lwt/src/server.ml | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/cohttp-lwt/src/server.ml b/cohttp-lwt/src/server.ml
index e6fc7678c..386788e42 100644
--- a/cohttp-lwt/src/server.ml
+++ b/cohttp-lwt/src/server.ml
@@ -146,6 +146,15 @@ module Make (IO : S.IO) = struct
             let keep_alive =
               Http.Request.is_keep_alive req && Http.Response.is_keep_alive res
             in
+            let res =
+              let headers =
+                Http.Header.add_unless_exists
+                  (Http.Response.headers res)
+                  "connection"
+                  (if keep_alive then "keep-alive" else "close")
+              in
+              { res with Http.Response.headers }
+            in
             handle_response ~keep_alive oc res body
               (fun () -> spec.conn_closed conn)
               (fun oc -> handle_client ic oc conn spec)

From 2e7f6cfee257706611d230de09d107a3afc68f8f Mon Sep 17 00:00:00 2001
From: Ushitora Anqou <ushitora@anqou.net>
Date: Sun, 10 Mar 2024 18:45:07 +0900
Subject: [PATCH 2/2] cohttp-eio: server: return connection header

---
 CHANGES.md               |  1 +
 cohttp-eio/src/server.ml | 16 +++++++++++++++-
 cohttp-eio/tests/test.ml |  8 ++++++--
 3 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/CHANGES.md b/CHANGES.md
index 8aaa01936..3e3ea0b1a 100644
--- a/CHANGES.md
+++ b/CHANGES.md
@@ -1,5 +1,6 @@
 ## Unreleased
 
+- cohttp-{lwt,eio}: server: return connection header (ushitora-anqou #1025)
 - cohttp-eio: Improve error handling in example server (talex5 #1023)
 - cohttp-eio: Don't blow up `Server.callback` on client disconnections. (mefyl #1015)
 - http: Fix assertion in `Source.to_string_trim` when `pos <> 0` (mefyl #1017)
diff --git a/cohttp-eio/src/server.ml b/cohttp-eio/src/server.ml
index 364c83385..7906e19f6 100644
--- a/cohttp-eio/src/server.ml
+++ b/cohttp-eio/src/server.ml
@@ -116,7 +116,21 @@ let callback { conn_closed; handler } ((_, peer_address) as conn) input output =
         let () =
           try
             match handler (conn, id) request body with
-            | `Response (response, body) -> write output response body
+            | `Response (response, body) ->
+                let keep_alive =
+                  Http.Request.is_keep_alive request
+                  && Http.Response.is_keep_alive response
+                in
+                let response =
+                  let headers =
+                    Http.Header.add_unless_exists
+                      (Http.Response.headers response)
+                      "connection"
+                      (if keep_alive then "keep-alive" else "close")
+                  in
+                  { response with Http.Response.headers }
+                in
+                write output response body
             | `Expert (response, handler) ->
                 let () = Io.Response.write_header response output in
                 handler input output
diff --git a/cohttp-eio/tests/test.ml b/cohttp-eio/tests/test.ml
index 2185b769e..5941cca30 100644
--- a/cohttp-eio/tests/test.ml
+++ b/cohttp-eio/tests/test.ml
@@ -44,7 +44,8 @@ let () =
         [ Cstruct.of_string "GET / HTTP/1.1\r\nconnection: close\r\n\r\n" ]
     in
     Alcotest.(check ~here:[%here] string)
-      "response" "HTTP/1.1 200 OK\r\ncontent-length: 4\r\n\r\nroot"
+      "response"
+      "HTTP/1.1 200 OK\r\nconnection: close\r\ncontent-length: 4\r\n\r\nroot"
       Eio.Buf_read.(of_flow ~max_size:max_int socket |> take_all)
   and missing socket =
     let () =
@@ -54,7 +55,8 @@ let () =
         ]
     in
     Alcotest.(check ~here:[%here] string)
-      "response" "HTTP/1.1 404 Not Found\r\ncontent-length: 0\r\n\r\n"
+      "response"
+      "HTTP/1.1 404 Not Found\r\nconnection: close\r\ncontent-length: 0\r\n\r\n"
       Eio.Buf_read.(of_flow ~max_size:max_int socket |> take_all)
   and streaming_response socket =
     let () =
@@ -66,6 +68,7 @@ let () =
     Alcotest.(check ~here:[%here] string)
       "response"
       "HTTP/1.1 200 OK\r\n\
+       connection: close\r\n\
        transfer-encoding: chunked\r\n\
        \r\n\
        5\r\n\
@@ -90,6 +93,7 @@ let () =
     Alcotest.(check ~here:[%here] string)
       "response"
       "HTTP/1.1 200 OK\r\n\
+       connection: close\r\n\
        transfer-encoding: chunked\r\n\
        \r\n\
        c\r\n\