Reactions: Add primary reactions as message actions

2026-04-19 16:13:42 +02:00 · 2019-05-15 15:58:38 +02:00
parent 8557a501db
commit 090597f461
7 changed files with 382 additions and 0 deletions
@@ -0,0 +1,99 @@
+/*
+ Copyright 2019 New Vector Ltd
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+ */
+
+import UIKit
+
+final class ReactionsMenuViewModel: ReactionsMenuViewModelType {
+
+    // MARK: - Properties
+
+    // MARK: Private
+    private let aggregations: MXAggregations
+    private let roomId: String
+    private let eventId: String
+
+    // MARK: Public
+
+    var isAgreeButtonSelected: Bool = false
+    var isDisagreeButtonSelected: Bool = false
+    var isLikeButtonSelected: Bool = false
+    var isDislikeButtonSelected: Bool = false
+
+    weak var viewDelegate: ReactionsMenuViewModelDelegate?
+    weak var coordinatorDelegate: ReactionsMenuViewModelCoordinatorDelegate?
+
+    // MARK: - Setup
+
+    init(aggregations: MXAggregations, roomId: String, eventId: String) {
+
+        self.aggregations = aggregations
+        self.roomId = roomId
+        self.eventId = eventId
+
+        self.loadData()
+        self.listenToDataUpdate()
+    }
+
+    // MARK: - Private
+
+
+    private func resetData() {
+        self.isAgreeButtonSelected = false
+        self.isDisagreeButtonSelected = false
+        self.isLikeButtonSelected = false
+        self.isDislikeButtonSelected = false
+    }
+
+    private func loadData() {
+        guard let reactionCounts = self.aggregations.reactions(onEvent: self.eventId, inRoom: self.roomId) else {
+            return
+        }
+
+        self.resetData()
+        reactionCounts.forEach { (reaction) in
+            if let reaction = ReactionsMenuReactions(rawValue: reaction.reaction) {
+                switch reaction {
+                case .agree:
+                    self.isAgreeButtonSelected = true
+                case .disagree:
+                    self.isDisagreeButtonSelected = true
+                case .like:
+                    self.isLikeButtonSelected = true
+                case .dislike:
+                    self.isDislikeButtonSelected = true
+                }
+            }
+        }
+
+        if let viewDelegate = self.viewDelegate {
+            viewDelegate.reactionsMenuViewModelDidUpdate(self)
+        }
+    }
+
+    private func listenToDataUpdate() {
+        self.aggregations.listenToReactionCountUpdate(inRoom: self.roomId) { [weak self] (changes) in
+
+            guard let sself = self else {
+                return
+            }
+
+            if changes[sself.eventId] != nil {
+                sself.loadData()
+            }
+        }
+    }
+
+}