Don't let people make votes for options that don't exist

1 files changed, 13 insertions, 0 deletions

diff --git a/main.go b/main.go
index b78de27..4e00a82 100644
--- a/main.go
+++ b/main.go
@@ -183,6 +183,10 @@ func main() {
 			return
 		}
 
+		if !hasOption(poll, c.PostForm("option")) {
+			c.JSON(500, gin.H{"error": err.Error()})
+		}
+
 		vote := &database.Vote{
 			Id:     "",
 			PollId: pId,
@@ -305,3 +309,12 @@ func containsPoll(polls []*database.Poll, poll *database.Poll) bool {
 	}
 	return false
 }
+
+func hasOption(poll *database.Poll, option string) bool {
+	for _, opt := range poll.Options {
+		if opt == option {
+			return true
+		}
+	}
+	return false
+}