Messages badge

WalletConnect · Oct 9, 2023 · 8dbdc35 · 8dbdc35
1 parent 9bab50d
commit 8dbdc35
Show file tree

Hide file tree

Showing 5 changed files with 31 additions and 7 deletions.
diff --git a/Example/WalletApp/PresentationLayer/Wallet/Notifications/Models/SubscriptionsViewModel.swift b/Example/WalletApp/PresentationLayer/Wallet/Notifications/Models/SubscriptionsViewModel.swift
@@ -5,6 +5,12 @@ typealias SubscriptionScope = [String: ScopeValue]
 
 struct SubscriptionsViewModel: Identifiable {
     let subscription: NotifySubscription
+    let messages: [NotifyMessageRecord]?
+
+    init(subscription: NotifySubscription, messages: [NotifyMessageRecord]? = nil) {
+        self.subscription = subscription
+        self.messages = messages
+    }
 
     var id: String {
         return subscription.topic
@@ -33,4 +39,12 @@ struct SubscriptionsViewModel: Identifiable {
     var scope: SubscriptionScope {
         return subscription.scope
     }
+
+    var messagesCount: Int {
+        return messages?.count ?? 0
+    }
+
+    var hasMessage: Bool {
+        return messagesCount != 0
+    }
 }
diff --git a/Example/WalletApp/PresentationLayer/Wallet/Notifications/NotificationsInteractor.swift b/Example/WalletApp/PresentationLayer/Wallet/Notifications/NotificationsInteractor.swift
@@ -39,7 +39,7 @@ final class NotificationsInteractor {
         try await Notify.instance.deleteSubscription(topic: topic)
     }
 
-    func messagesCount(subscription: NotifySubscription) -> Int {
-        return Notify.instance.getMessageHistory(topic: subscription.topic).count
+    func messages(for subscription: NotifySubscription) -> [NotifyMessageRecord] {
+        return Notify.instance.getMessageHistory(topic: subscription.topic)
     }
 }
diff --git a/Example/WalletApp/PresentationLayer/Wallet/Notifications/NotificationsPresenter.swift b/Example/WalletApp/PresentationLayer/Wallet/Notifications/NotificationsPresenter.swift
@@ -13,10 +13,8 @@ final class NotificationsPresenter: ObservableObject {
 
     var subscriptionViewModels: [SubscriptionsViewModel] {
         return subscriptions
-            .map { SubscriptionsViewModel(subscription: $0) }
-            .sorted { lhs, rhs in
-                return interactor.messagesCount(subscription: lhs.subscription) > interactor.messagesCount(subscription: rhs.subscription)
-            }
+            .map { SubscriptionsViewModel(subscription: $0, messages: interactor.messages(for: $0)) }
+            .sorted { $0.messagesCount > $1.messagesCount }
     }
 
     var listingViewModels: [ListingViewModel] {

diff --git a/Example/WalletApp/PresentationLayer/Wallet/Notifications/NotificationsView.swift b/Example/WalletApp/PresentationLayer/Wallet/Notifications/NotificationsView.swift
@@ -124,6 +124,18 @@ struct NotificationsView: View {
                             .foregroundColor(.Foreground150)
                             .font(.system(size: 14, weight: .regular, design: .rounded))
                     }
+
+                    Spacer()
+
+                    if subscription.hasMessage {
+                        Text(String(subscription.messagesCount))
+                            .foregroundColor(.Inverse100)
+                            .font(.system(size: 13, weight: .medium))
+                            .frame(width: 20, height: 20)
+                            .background {
+                                Circle().foregroundColor(.blue100)
+                            }
+                    }
                 }
             }
         }

diff --git a/Example/WalletApp/PresentationLayer/Wallet/NotifySettings/NotifyPreferencesPresenter.swift b/Example/WalletApp/PresentationLayer/Wallet/NotifySettings/NotifyPreferencesPresenter.swift
@@ -10,7 +10,7 @@ final class NotifyPreferencesPresenter: ObservableObject {
     private var disposeBag = Set<AnyCancellable>()
 
     var subscriptionViewModel: SubscriptionsViewModel {
-        return SubscriptionsViewModel(subscription: subscription)
+        return SubscriptionsViewModel(subscription: subscription, messages: [])
     }
 
     var preferences: [String] {