add siwe publishers

WalletConnect · Jun 13, 2024 · a6e7b85 · a6e7b85
1 parent be72b87
commit a6e7b85
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 3 deletions.
diff --git a/Sources/Web3Modal/Core/Web3ModalClient.swift b/Sources/Web3Modal/Core/Web3ModalClient.swift
@@ -80,7 +80,11 @@ public class Web3ModalClient {
         return analyticsService.isAnalyticsEnabled
     }
 
-//    add user authenticated publisher authResponsePublisher+SIWE
+    public var SIWEAuthenticationPublisher: AnyPublisher<Result<(message: String, signature: String), SIWEAuthenticationError>, Never> {
+        return SIWEAuthenticationPublisherSubject.eraseToAnyPublisher()
+    }
+
+    internal let SIWEAuthenticationPublisherSubject = PassthroughSubject<Result<(message: String, signature: String), SIWEAuthenticationError>, Never>()
 
     // MARK: - Private Properties
 

diff --git a/Sources/Web3Modal/Sheets/Web3ModalViewModel.swift b/Sources/Web3Modal/Sheets/Web3ModalViewModel.swift
@@ -1,6 +1,11 @@
 import Combine
 import SwiftUI
 
+public enum SIWEAuthenticationError: Error {
+    case requestRejected
+    case messageVerificationFailed
+}
+
 class Web3ModalViewModel: ObservableObject {
     private(set) var router: Router
     private(set) var store: Store
@@ -257,14 +262,25 @@ class Web3ModalViewModel: ObservableObject {
                         } catch {
                             guard let self = self else { return }
 
+                            Web3Modal.instance.SIWEAuthenticationPublisherSubject.send(Result.failure(
+                                .messageVerificationFailed))
                             DispatchQueue.main.async {
                                 self.store.toast = .init(style: .error, message: error.localizedDescription)
+                                guard let topic = self.store.session?.topic else { return }
+                                Task {try await Web3Modal.instance.disconnect(topic: topic)}
+
                             }
                         }
                     }
                 case .error(let error):
-                    guard let self = self else { return }
-                    self.store.toast = .init(style: .error, message: error.message)
+                    DispatchQueue.main.async {
+                        Web3Modal.instance.SIWEAuthenticationPublisherSubject.send(Result.failure(
+                            .requestRejected))
+                        guard let self = self else { return }
+                        self.store.SIWEFallbackState = false
+                        guard let topic = self.store.session?.topic else { return }
+                        Task {try await Web3Modal.instance.disconnect(topic: topic)}
+                    }
                 }
             }
         }.store(in: &disposeBag)