bugfix: unvote

wishkit · Sep 26, 2024 · 863bd6f · 863bd6f
1 parent 780a4b2
commit 863bd6f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/Sources/WishKit/Configuration/Configuration.swift b/Sources/WishKit/Configuration/Configuration.swift
@@ -28,6 +28,8 @@ public struct Configuration {
     public var emailField: EmailField = .optional
 
     public var commentSection: Display = .show
+
+    public var allowUndoVote: Bool = false
 
     init(
         statusBadgeDisplay: Display = .hide,

diff --git a/Sources/WishKit/SwiftUI/WishView.swift b/Sources/WishKit/SwiftUI/WishView.swift
@@ -26,9 +26,6 @@ struct WishView: View {
     @Binding
     private var voteCount: Int
 
-    @State
-    private var hasVoted = false
-
     private let wishResponse: WishResponse
 
     private let voteActionCompletion: () -> Void
@@ -164,12 +161,21 @@ struct WishView: View {
     }
 
     private func voteAction() {
-
         if wishResponse.state == .implemented {
             alertModel.alertReason = .alreadyImplemented
             alertModel.showAlert = true
             return
         }
+
+        let userUUID = UUIDManager.getUUID()
+
+        let hasVoted = wishResponse.votingUsers.contains(where: { user in user.uuid == userUUID })
+
+        if (hasVoted) && WishKit.config.allowUndoVote == false {
+            alertModel.alertReason = .alreadyVoted
+            alertModel.showAlert = true
+            return
+        }
 
         let request = VoteWishRequest(wishId: wishResponse.id)
 
@@ -192,7 +198,7 @@ struct WishView: View {
 extension WishView {
     var arrowColor: Color {
         let userUUID = UUIDManager.getUUID()
-        if wishResponse.votingUsers.contains(where: { user in user.uuid == userUUID }) || hasVoted {
+        if wishResponse.votingUsers.contains(where: { user in user.uuid == userUUID }) {
             return WishKit.theme.primaryColor
         }