download images on fail

acheong08 · Jul 7, 2023 · f380b15 · f380b15
1 parent 4263227
commit f380b15
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cmd/api/main.go b/cmd/api/main.go
@@ -50,9 +50,15 @@ func captchaStart(c *gin.Context) {
 		err = session.SubmitAnswer(2)
 		if err == nil {
 			c.JSON(200, gin.H{"token": token, "status": "success"})
+			return
 		}
 		log.Println("Retrying...")
+	}
 
+	// Download the images
+	images, err = funcaptcha.DownloadChallenge(session.ConciseChallenge.URLs, true)
+	if err != nil {
+		c.JSON(500, gin.H{"error": "failed to download images"})
 	}
 	c.JSON(http.StatusNetworkAuthenticationRequired, gin.H{"token": token, "session": session, "status": "captcha", "images": images})
 }