Merge pull request #135 from d-exclaimation/acception-combined-protocols

fix: Allow combined protocol with , in headers
d-exclaimation · Jun 21, 2023 · c0556ff · c0556ff
2 parents ca7d455 + 2a217fb
commit c0556ff
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Sources/Pioneer/Vapor/WebSocket/Pioneer+WebSocket.swift b/Sources/Pioneer/Vapor/WebSocket/Pioneer+WebSocket.swift
@@ -21,6 +21,10 @@ public extension Pioneer {
     func webSocketHandler(req: Request, context: @escaping VaporWebSocketContext, guard: @escaping VaporWebSocketGuard) async throws -> Response {
         /// Explicitly handle Websocket upgrade with sub-protocol
         let protocolHeader: [String] = req.headers[.secWebSocketProtocol]
+            .flatMap { $0.split(separator: ",") }
+            .map { String($0) }
+            .map { $0.trimmingCharacters(in: .whitespaces) }
+
         guard let _ = protocolHeader.first(where: websocketProtocol.isValid) else {
             return try GraphQLError(
                 message: "Unrecognized websocket protocol. Specify the 'sec-websocket-protocol' header with '\(websocketProtocol.name)'"