Add token auth

Signed-off-by: Tomasz Pietrek <tomasz@nats.io>

Author: Jarema

Sources/NatsSwift/NatsClient/NatsClient.swift

@@ -22,11 +22,15 @@ public enum NatsState {
 public struct Auth {
     var user: String?
     var password: String?
+    var token: String?
 
     init(user: String, password: String) {
         self.user = user
         self.password = password
     }
+    init(token: String) {
+        self.token = token
+    }
 }
 
 public class Client {

Sources/NatsSwift/NatsClient/NatsClientOptions.swift

@@ -52,6 +52,15 @@ public class ClientOptions {
         return self
     }
 
+    public func token(_ token: String) -> ClientOptions {
+        if self.auth == nil {
+            self.auth = Auth(token: token)
+        } else {
+            self.auth?.token = token
+        }
+        return self
+    }
+
     public func build() -> Client {
         let client = Client()
         client.urls = urls

Sources/NatsSwift/NatsConnection.swift

@@ -149,7 +149,7 @@ class ConnectionHandler: ChannelInboundHandler {
         }
         self.serverInfo = info
         // TODO(jrm): Add rest of auth here.
-        let connect = ConnectInfo(verbose: false, pedantic: false, userJwt: nil, nkey: "", signature: nil, name: "", echo: true, lang: self.lang, version: self.version, natsProtocol: .dynamic, tlsRequired: false, user: self.auth?.user ?? "", pass: self.auth?.password ?? "", authToken: "", headers: true, noResponders: true)
+        let connect = ConnectInfo(verbose: false, pedantic: false, userJwt: nil, nkey: "", signature: nil, name: "", echo: true, lang: self.lang, version: self.version, natsProtocol: .dynamic, tlsRequired: false, user: self.auth?.user ?? "", pass: self.auth?.password ?? "", authToken: self.auth?.token ?? "", headers: true, noResponders: true)
 
         try await withCheckedThrowingContinuation { continuation in
             self.connectionEstablishedContinuation = continuation

Tests/NatsSwiftTests/Integration/ConnectionTests.swift

@@ -17,6 +17,7 @@ class CoreNatsTests: XCTestCase {
         ("testConnect", testConnect),
         ("testReconnect", testReconnect),
         ("testUsernameAndPassword", testUsernameAndPassword),
+        ("testTokenAuth", testTokenAuth)
     ]
     var natsServer = NatsServer()
 
@@ -187,4 +188,41 @@ class CoreNatsTests: XCTestCase {
         }
 
     }
+
+    func testTokenAuth() async throws {
+        logger.logLevel = .debug
+        let currentFile = URL(fileURLWithPath: #file)
+        // Navigate up to the Tests directory
+        let testsDir = currentFile.deletingLastPathComponent().deletingLastPathComponent()
+        // Construct the path to the resource
+        let resourceURL = testsDir
+            .appendingPathComponent("Integration/Resources/token.conf", isDirectory: false)
+        natsServer.start(cfg: resourceURL.path)
+        let client = ClientOptions()
+            .url(URL(string:natsServer.clientURL)!)
+            .token("s3cr3t")
+            .maxReconnects(5)
+            .build()
+        try await client.connect()
+        try client.publish("msg".data(using: .utf8)!, subject: "test")
+        try await client.flush()
+        try await client.subscribe(to: "test")
+        XCTAssertNotNil(client, "Client should not be nil")
+
+
+        // Test if client with bad credentials throws an error
+        let bad_creds_client = ClientOptions()
+            .url(URL(string:natsServer.clientURL)!)
+            .token("badtoken")
+            .maxReconnects(5)
+            .build()
+
+        do {
+            try await bad_creds_client.connect()
+            XCTFail("Should have thrown an error")
+        } catch {
+            XCTAssertNotNil(error, "Error should not be nil")
+        }
+
+    }
 }

Tests/NatsSwiftTests/Integration/Resources/token.conf

@@ -0,0 +1,4 @@
+authorization {
+  token: s3cr3t
+}
+