From 961e51ac9e5b1c772b2603a8d3d85c8688425616 Mon Sep 17 00:00:00 2001 From: Yanislav Igonin Date: Thu, 7 Apr 2022 11:27:49 +0300 Subject: [PATCH] connect captcha controller --- controllers/captcha_controller.go | 13 ++++++------- main.go | 4 +--- 2 files changed, 7 insertions(+), 10 deletions(-) diff --git a/controllers/captcha_controller.go b/controllers/captcha_controller.go index 8ffd763..a2dcac8 100644 --- a/controllers/captcha_controller.go +++ b/controllers/captcha_controller.go @@ -3,21 +3,20 @@ package controllers import ( "bytes" "log" - "net/http" "github.com/dchest/captcha" - "github.com/gin-gonic/gin" + "github.com/gofiber/fiber/v2" ) -func GetCaptcha(c *gin.Context) { - ID := c.Param("captchaID") +func GetCaptcha(c *fiber.Ctx) error { + ID := c.Params("captchaID") var content bytes.Buffer err := captcha.WriteImage(&content, ID, captcha.StdWidth, captcha.StdHeight) if err != nil { log.Println("error:", err) - c.HTML(http.StatusInternalServerError, "pages/500.html", nil) - return + return c.Status(fiber.StatusInternalServerError).Render("pages/500", nil) } - c.Data(200, "image/png", content.Bytes()) + c.Context().SetContentType("image/png") + return c.Send(content.Bytes()) } diff --git a/main.go b/main.go index 88e5659..5db1a44 100644 --- a/main.go +++ b/main.go @@ -73,9 +73,7 @@ func main() { app.Post("/threadID", func(c *fiber.Ctx) error { return c.SendString("create post in thread") }) - app.Get("/captcha/:captchaID", func(c *fiber.Ctx) error { - return c.SendString("get captcha by id") - }) + app.Get("/captcha/:captchaID", controllers.GetCaptcha) log.Println("app - online, port -", strconv.Itoa(config.App.Port)) log.Println("all systems nominal")