Ensure Abstain is always present in simple custom votes

1 files changed, 12 insertions, 0 deletions

diff --git a/main.go b/main.go
index d395a6d..2ddf04e 100644
--- a/main.go
+++ b/main.go
@@ -130,6 +130,9 @@ func main() {
 			poll.Options = []string{}
 			for _, opt := range strings.Split(c.PostForm("customOptions"), ",") {
 				poll.Options = append(poll.Options, strings.TrimSpace(opt))
+				if !containsString(poll.Options, "Abstain") && (poll.VoteType == database.POLL_TYPE_SIMPLE) {
+					poll.Options = append(poll.Options, "Abstain")
+				}
 			}
 		case "pass-fail":
 		default:
@@ -403,3 +406,12 @@ func hasOption(poll *database.Poll, option string) bool {
 	}
 	return false
 }
+
+func containsString(arr []string, val string) bool {
+	for _, a := range arr {
+		if a == val {
+			return true
+		}
+	}
+	return false
+}