Authenticator fix

WalletConnect · Sep 20, 2023 · 6fdb97c · 6fdb97c
1 parent 8c7fe98
commit 6fdb97c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Sources/WalletConnectRelay/ClientAuth/ClientIdAuthenticator.swift b/Sources/WalletConnectRelay/ClientAuth/ClientIdAuthenticator.swift
@@ -1,7 +1,7 @@
 import Foundation
 
 public protocol ClientIdAuthenticating {
-    func createAuthToken() throws -> String
+    func createAuthToken(url: String?) throws -> String
 }
 
 public struct ClientIdAuthenticator: ClientIdAuthenticating {
@@ -13,9 +13,9 @@ public struct ClientIdAuthenticator: ClientIdAuthenticating {
         self.url = url
     }
 
-    public func createAuthToken() throws -> String {
+    public func createAuthToken(url: String? = nil) throws -> String {
         let keyPair = try clientIdStorage.getOrCreateKeyPair()
-        let payload = RelayAuthPayload(subject: getSubject(), audience: url)
+        let payload = RelayAuthPayload(subject: getSubject(), audience: url ?? self.url)
         return try payload.signAndCreateWrapper(keyPair: keyPair).jwtString
     }
 

diff --git a/Sources/WalletConnectRelay/RelayURLFactory.swift b/Sources/WalletConnectRelay/RelayURLFactory.swift
@@ -23,7 +23,7 @@ struct RelayUrlFactory {
             URLQueryItem(name: "projectId", value: projectId)
         ]
         do {
-            let authToken = try socketAuthenticator.createAuthToken()
+            let authToken = try socketAuthenticator.createAuthToken(url: fallback ? "wss://" + NetworkConstants.fallbackUrl : "wss://" + relayHost)
             components.queryItems?.append(URLQueryItem(name: "auth", value: authToken))
         } catch {
             // TODO: Handle token creation errors