feature: Makes connection_init payload type generic

This allows the user to define their own authorization types, but still allows us to encode/decode in a strongly typed way.

Some common default payload types are included.

Author: NeedleInAJayStack

Sources/GraphQLTransportWS/Client.swift

@@ -4,7 +4,7 @@ import Foundation
 import GraphQL
 
 /// Client is an open-ended implementation of the client side of the protocol. It parses and adds callbacks for each type of server respose.
-public class Client {
+public class Client<InitPayload: Equatable & Codable> {
     // We keep this weak because we strongly inject this object into the messenger callback
     weak var messenger: Messenger?
 
@@ -110,7 +110,7 @@ public class Client {
     }
 
     /// Send a `connection_init` request through the messenger
-    public func sendConnectionInit(payload: ConnectionInitAuth?) {
+    public func sendConnectionInit(payload: InitPayload) {
         guard let messenger = messenger else { return }
         messenger.send(
             ConnectionInitRequest(

Sources/GraphQLTransportWS/InitPayloads.swift

@@ -0,0 +1,9 @@
+// Contains convenient `connection_init` payloads for users of this package
+
+/// `connection_init` `payload` that is empty
+public struct EmptyInitPayload: Equatable & Codable { }
+
+/// `connection_init` `payload` that includes an `authToken` field
+public struct TokenInitPayload: Equatable & Codable {
+    let authToken: String
+}

Sources/GraphQLTransportWS/Requests.swift

@@ -15,24 +15,14 @@ import GraphQL
 /// ```
 
 /// A general request. This object's type is used to triage to other, more specific request objects.
-public struct Request: Equatable, JsonEncodable {
+struct Request: Equatable, JsonEncodable {
     let type: RequestMessageType
 }
 
 /// A websocket `connection_init` request from the client to the server
-public struct ConnectionInitRequest: Equatable, JsonEncodable {
+struct ConnectionInitRequest<InitPayload: Codable & Equatable>: Equatable, JsonEncodable {
     var type = RequestMessageType.connectionInit
-    public let payload: ConnectionInitAuth?
-}
-
-// TODO: Make this structure user-defined
-/// Authorization format for a websocket `connection_init` request from the client to the server
-public struct ConnectionInitAuth: Equatable, JsonEncodable {
-    public let authToken: String
-    
-    public init(authToken: String) {
-        self.authToken = authToken
-    }
+    let payload: InitPayload
 }
 
 /// A websocket `subscribe` request from the client to the server

Sources/GraphQLTransportWS/Server.swift

@@ -7,14 +7,14 @@ import NIO
 import RxSwift
 
 /// Server implements the server-side portion of the protocol, allowing a few callbacks for customization.
-public class Server {
+public class Server<InitPayload: Equatable & Codable> {
     // We keep this weak because we strongly inject this object into the messenger callback
     weak var messenger: Messenger?
 
     let onExecute: (GraphQLRequest) -> EventLoopFuture<GraphQLResult>
     let onSubscribe: (GraphQLRequest) -> EventLoopFuture<SubscriptionResult>
 
-    var auth: (ConnectionInitRequest) throws -> Void = { _ in }
+    var auth: (InitPayload) throws -> Void = { _ in }
     var onExit: () -> Void = { }
     var onMessage: (String) -> Void = { _ in }
 
@@ -64,7 +64,7 @@ public class Server {
 
             switch request.type {
                 case .connectionInit:
-                    guard let connectionInitRequest = try? self.decoder.decode(ConnectionInitRequest.self, from: data) else {
+                    guard let connectionInitRequest = try? self.decoder.decode(ConnectionInitRequest<InitPayload>.self, from: data) else {
                         self.error(.invalidRequestFormat(messageType: .connectionInit))
                         return
                     }
@@ -89,7 +89,7 @@ public class Server {
 
     /// Define the callback run during `connection_init` resolution that allows authorization using the `payload`.
     /// Throw to indicate that authorization has failed.    /// - Parameter callback: The callback to assign
-    public func auth(_ callback: @escaping (ConnectionInitRequest) throws -> Void) {
+    public func auth(_ callback: @escaping (InitPayload) throws -> Void) {
         self.auth = callback
     }
 
@@ -105,14 +105,14 @@ public class Server {
         self.onMessage = callback
     }
 
-    private func onConnectionInit(_ connectionInitRequest: ConnectionInitRequest) {
+    private func onConnectionInit(_ connectionInitRequest: ConnectionInitRequest<InitPayload>) {
         guard !initialized else {
             self.error(.tooManyInitializations())
             return
         }
 
         do {
-            try self.auth(connectionInitRequest)
+            try self.auth(connectionInitRequest.payload)
         }
         catch {
             self.error(.unauthorized())

Tests/GraphQLTransportWSTests/GraphQLTransportWSTests.swift

@@ -11,7 +11,7 @@ import XCTest
 class GraphqlTransportWSTests: XCTestCase {
     var clientMessenger: TestMessenger!
     var serverMessenger: TestMessenger!
-    var server: Server!
+    var server: Server<TokenInitPayload>!
 
     override func setUp() {
         clientMessenger = TestMessenger()
@@ -24,7 +24,7 @@ class GraphqlTransportWSTests: XCTestCase {
         let api = TestAPI()
         let context = TestContext()
 
-        server = Server(
+        server = Server<TokenInitPayload>(
             messenger: serverMessenger,
             onExecute: { graphQLRequest in
                 api.execute(
@@ -48,7 +48,7 @@ class GraphqlTransportWSTests: XCTestCase {
         var messages = [String]()
         let completeExpectation = XCTestExpectation()
 
-        let client = Client(messenger: clientMessenger)
+        let client = Client<TokenInitPayload>(messenger: clientMessenger)
         client.onMessage { message, _ in
             messages.append(message)
             completeExpectation.fulfill()
@@ -81,14 +81,14 @@ class GraphqlTransportWSTests: XCTestCase {
         var messages = [String]()
         let completeExpectation = XCTestExpectation()
 
-        let client = Client(messenger: clientMessenger)
+        let client = Client<TokenInitPayload>(messenger: clientMessenger)
         client.onMessage { message, _ in
             messages.append(message)
             completeExpectation.fulfill()
         }
 
         client.sendConnectionInit(
-            payload: ConnectionInitAuth(
+            payload: TokenInitPayload(
                 authToken: ""
             )
         )
@@ -107,7 +107,7 @@ class GraphqlTransportWSTests: XCTestCase {
         var messages = [String]()
         let completeExpectation = XCTestExpectation()
 
-        let client = Client(messenger: clientMessenger)
+        let client = Client<TokenInitPayload>(messenger: clientMessenger)
         client.onConnectionAck { _, client in
             client.sendStart(
                 payload: GraphQLRequest(
@@ -131,7 +131,7 @@ class GraphqlTransportWSTests: XCTestCase {
         }
 
         client.sendConnectionInit(
-            payload: ConnectionInitAuth(
+            payload: TokenInitPayload(
                 authToken: ""
             )
         )
@@ -154,7 +154,7 @@ class GraphqlTransportWSTests: XCTestCase {
         var dataIndex = 1
         let dataIndexMax = 3
 
-        let client = Client(messenger: clientMessenger)
+        let client = Client<TokenInitPayload>(messenger: clientMessenger)
         client.onConnectionAck { _, client in
             client.sendStart(
                 payload: GraphQLRequest(
@@ -191,7 +191,7 @@ class GraphqlTransportWSTests: XCTestCase {
         }
 
         client.sendConnectionInit(
-            payload: ConnectionInitAuth(
+            payload: TokenInitPayload(
                 authToken: ""
             )
         )