Add createHeadResponse

hummingbird-project · Apr 23, 2024 · 590492b · 590492b
1 parent 12f5696
commit 590492b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/Sources/Hummingbird/Router/EndpointResponder.swift b/Sources/Hummingbird/Router/EndpointResponder.swift
@@ -35,9 +35,8 @@ struct EndpointResponders<Context: BaseRequestContext>: Sendable {
     mutating func autoGenerateHeadEndpoint() {
         if self.methods[.head] == nil, let get = methods[.get] {
             self.methods[.head] = CallbackResponder { request, context in
-                var response = try await get.respond(to: request, context: context)
-                response.body = .init()
-                return response
+                let response = try await get.respond(to: request, context: context)
+                return response.createHeadResponse()
             }
         }
     }

diff --git a/Sources/HummingbirdCore/Response/Response.swift b/Sources/HummingbirdCore/Response/Response.swift
@@ -42,6 +42,11 @@ public struct Response: Sendable {
         get { self.head.headerFields }
         set { self.head.headerFields = newValue }
     }
+
+    /// Return HEAD response based off this response
+    public func createHeadResponse() -> Response {
+        .init(status: self.status, headers: self.headers, body: .init())
+    }
 }
 
 extension Response: CustomStringConvertible {