Add support for resetting Remote Messages for debugging purposes

duckduckgo · Jul 12, 2024 · 1caeeeb · 1caeeeb
1 parent 9ee9b37
commit 1caeeeb
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/Sources/RemoteMessaging/RemoteMessagingStore.swift b/Sources/RemoteMessaging/RemoteMessagingStore.swift
@@ -360,6 +360,26 @@ extension RemoteMessagingStore {
             }
         }
     }
+
+    public func resetRemoteMessages() {
+        guard remoteMessagingAvailabilityProvider.isRemoteMessagingAvailable else {
+            return
+        }
+
+        let context = database.makeContext(concurrencyType: .privateQueueConcurrencyType, name: Constants.privateContextName)
+        context.performAndWait {
+            context.deleteAll(entityDescriptions: [
+                RemoteMessageManagedObject.entity(in: context),
+                RemoteMessagingConfigManagedObject.entity(in: context)
+            ])
+
+            do {
+                try context.save()
+            } catch {
+                os_log("Failed to reset remote messages", log: log, type: .error)
+            }
+        }
+    }
 }
 
 // MARK: - RemoteMessageManagedObject Private Interface

diff --git a/Sources/RemoteMessaging/RemoteMessagingStoring.swift b/Sources/RemoteMessaging/RemoteMessagingStoring.swift
@@ -18,7 +18,11 @@
 
 import Foundation
 
-public protocol RemoteMessagingStoring {
+public protocol RemoteMessagingStoringDebuggingSupport {
+    func resetRemoteMessages()
+}
+
+public protocol RemoteMessagingStoring: RemoteMessagingStoringDebuggingSupport {
 
     func saveProcessedResult(_ processorResult: RemoteMessagingConfigProcessor.ProcessorResult)
     func fetchRemoteMessagingConfig() -> RemoteMessagingConfig?