From ae9eaccd291c4bbf2033bd2c5fa7c38b832ed9e8 Mon Sep 17 00:00:00 2001
From: Martin Michelsen <fuzziqersoftware+g@gmail.com>
Date: Tue, 8 Jul 2025 20:09:14 -0700
Subject: [PATCH] fix disconnect for websocket clients

---
 src/AsyncHTTPServer.hh | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/AsyncHTTPServer.hh b/src/AsyncHTTPServer.hh
index 8ded5cc0..1790ff20 100644
--- a/src/AsyncHTTPServer.hh
+++ b/src/AsyncHTTPServer.hh
@@ -205,9 +205,11 @@ protected:
         if (resp) {
           co_await c->send_http_response(*resp);
         }
-        auto* conn_header = req.get_header("connection");
-        if (!conn_header || (*conn_header != "keep-alive")) {
-          c->r.close();
+        if (!c->is_websocket) {
+          auto* conn_header = req.get_header("connection");
+          if (!conn_header || (*conn_header != "keep-alive")) {
+            c->r.close();
+          }
         }
       }