Separate debug cleanup

WalletConnect · Dec 16, 2022 · 52a8793 · 52a8793
1 parent bc5c5b3
commit 52a8793
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 6 deletions.
diff --git a/Example/ExampleApp/SceneDelegate.swift b/Example/ExampleApp/SceneDelegate.swift
@@ -10,7 +10,7 @@ class SceneDelegate: UIResponder, UIWindowSceneDelegate {
 
     var window: UIWindow?
 
-    func scene(_ scene: UIScene, willConnectTo session: UISceneSession, options connectionOptions: UIScene.ConnectionOptions) {
+    func scene(_ scene: UIScene, willConnectTo session: UISceneSession, options connectionOptions: UIScene.ConnectionOptions) async {
 
         let metadata = AppMetadata(
             name: "Example Wallet",
@@ -22,7 +22,7 @@ class SceneDelegate: UIResponder, UIWindowSceneDelegate {
         Pair.configure(metadata: metadata)
 #if DEBUG
         if CommandLine.arguments.contains("-cleanInstall") {
-            try? Sign.instance.cleanup()
+            try? await Sign.instance.cleanup()
         }
 #endif
 

diff --git a/Sources/WalletConnectSign/Services/SignCleanupService.swift b/Sources/WalletConnectSign/Services/SignCleanupService.swift
@@ -18,11 +18,11 @@ final class SignCleanupService {
 
     func cleanup() async throws {
         try await unsubscribe()
+        try cleanupStorages()
+    }
 
-        pairingStore.deleteAll()
-        sessionStore.deleteAll()
-        sessionToPairingTopic.deleteAll()
-        try kms.deleteAll()
+    func cleanup() throws {
+        try cleanupStorages()
     }
 }
 
@@ -34,4 +34,11 @@ private extension SignCleanupService {
 
         try await networkInteractor.batchUnsubscribe(topics: pairing + session)
     }
+
+    func cleanupStorages() throws {
+        pairingStore.deleteAll()
+        sessionStore.deleteAll()
+        sessionToPairingTopic.deleteAll()
+        try kms.deleteAll()
+    }
 }
diff --git a/Sources/WalletConnectSign/Sign/SignClient.swift b/Sources/WalletConnectSign/Sign/SignClient.swift
@@ -347,10 +347,21 @@ public final class SignClient {
     }
 
     /// Delete all stored data such as: pairings, sessions, keys
+    ///
+    /// - Note: Will unsubscribe from all topics
     public func cleanup() async throws {
         try await cleanupService.cleanup()
     }
 
+#if DEBUG
+    /// Delete all stored data such as: pairings, sessions, keys
+    ///
+    /// - Note: Doesn't unsubscribe from topics
+    public func cleanup() throws {
+        try cleanupService.cleanup()
+    }
+#endif
+
     // MARK: - Private
 
     private func setUpEnginesCallbacks() {