Add default implementation for AuthTokenProvider

aws-amplify · Apr 18, 2023 · 97794e5 · 97794e5
1 parent 9572357
commit 97794e5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/...ins/API/AWSAPICategoryPlugin/Interceptor/SubscriptionInterceptor/IAMAuthInterceptor.swift b/...ins/API/AWSAPICategoryPlugin/Interceptor/SubscriptionInterceptor/IAMAuthInterceptor.swift
@@ -28,14 +28,6 @@ class IAMAuthInterceptor: AuthInterceptor {
         self.region = region
     }
 
-    func interceptMessage(_ message: AppSyncMessage, for endpoint: URL, completion: @escaping (AppSyncMessage) -> Void) {
-
-    }
-
-    func interceptConnection(_ request: AppSyncConnectionRequest, for endpoint: URL, completion: @escaping (AppSyncConnectionRequest) -> Void) {
-
-    }
-
     func interceptMessage(_ message: AppSyncMessage, for endpoint: URL) -> AppSyncMessage {
         switch message.messageType {
         case .subscribe:

diff --git a/AmplifyPlugins/Core/AWSPluginsCore/Auth/Provider/AuthTokenProvider.swift b/AmplifyPlugins/Core/AWSPluginsCore/Auth/Provider/AuthTokenProvider.swift
@@ -15,6 +15,14 @@ public protocol AuthTokenProvider {
     func getToken(completion: @escaping (Result<String, AuthError>) -> Void)
 }
 
+public extension AuthTokenProvider {
+
+    func getToken(completion: @escaping (Result<String, AuthError>) -> Void) {
+        let result = getToken()
+        completion(result)
+    }
+}
+
 public struct BasicUserPoolTokenProvider: AuthTokenProvider {
 
     let authService: AWSAuthServiceBehavior