fix: throw bad request for negative count value

D3vd · Jun 14, 2021 · f523369 · f523369
1 parent 214900c
commit f523369
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/api/gimme/n_random_memes.go b/api/gimme/n_random_memes.go
@@ -19,6 +19,14 @@ func GetNRandomMemes(c *gin.Context) {
 
 	count, _ := strconv.Atoi(c.Param("interface"))
 
+    if count <= 0 {
+        c.JSON(http.StatusBadRequest, response.Error{
+            Code: http.StatusBadRequest,
+            Message: "Invalid Count Value",
+        })
+        return
+    }
+
 	// Check if the count is less than 50
 	if count > 50 {
 		count = 50

diff --git a/api/gimme/n_random_posts_from_sub.go b/api/gimme/n_random_posts_from_sub.go
@@ -23,7 +23,7 @@ func GetNPostsFromSub(c *gin.Context) {
 	sub := strings.ToLower(c.Param("interface"))
 	count, err := strconv.Atoi(c.Param("count"))
 
-	if err != nil {
+	if err != nil || count <= 0 {
 		res := response.Error{
 			Code:    http.StatusBadRequest,
 			Message: "Invalid Count Value",