Reuse logic from LLC

Author: nuno-vieira

Sources/StreamChatSwiftUI/ChatChannel/MessageList/Polls/PollAttachmentView.swift

@@ -55,8 +55,8 @@ public struct PollAttachmentView<Factory: ViewFactory>: View {
                 PollOptionView(
                     viewModel: viewModel,
                     option: option,
-                    optionVotes: poll.voteCountsByOption?[option.id],
-                    maxVotes: poll.voteCountsByOption?.values.max()
+                    optionVotes: poll.voteCount(for: option),
+                    maxVotes: poll.currentMaximumVoteCount
                 )
                 .layoutPriority(1) // do not compress long text
             }

Sources/StreamChatSwiftUI/ChatChannel/MessageList/Polls/PollAttachmentViewModel.swift

@@ -219,13 +219,7 @@ public class PollAttachmentViewModel: ObservableObject, PollControllerDelegate {
     ///
     /// - Note: When multiple options have the highest vote count, this function returns false.
     public func hasMostVotes(for option: PollOption) -> Bool {
-        guard let allCounts = poll.voteCountsByOption else { return false }
-        guard let optionVoteCount = allCounts[option.id], optionVoteCount > 0 else { return false }
-        guard let highestVotePerOption = allCounts.values.max() else { return false }
-        guard optionVoteCount == highestVotePerOption else { return false }
-        // Check if only one option has highest number for votes
-        let optionsByVoteCounts = Dictionary(grouping: allCounts, by: { $0.value })
-        return optionsByVoteCounts[optionVoteCount]?.count == 1
+        return poll.isOptionWithMostVotes(option)
     }
 
     // MARK: - PollControllerDelegate
@@ -244,7 +238,7 @@ public class PollAttachmentViewModel: ObservableObject, PollControllerDelegate {
     // MARK: - private
 
     private func currentUserVote(for option: PollOption) -> PollVote? {
-        currentUserVotes.first(where: { $0.optionId == option.id })
+        poll.currentUserVote(for: option)
     }
 
     private func notifySheetPresentation(shown: Bool) {

StreamChatSwiftUITests/Tests/ChatChannel/PollAttachmentViewModel_Tests.swift

@@ -58,14 +58,13 @@ final class PollAttachmentViewModel_Tests: StreamChatTestCase {
 
     func test_pollAttachmentViewModel_removeVoteCalled() {
         // Given
+        let vote = makePollVote()
         let pollController = makePollController()
         let viewModel = PollAttachmentViewModel(
             message: .mock(),
-            poll: .unique,
+            poll: .mock(ownVotes: [vote]),
             pollController: pollController
         )
-        let vote = makePollVote()
-        viewModel.currentUserVotes = [vote]
 
         // When
         viewModel.removePollVote(for: .init(id: vote.optionId!, text: ""))