long-agent
b6aff65975
- captcha_handler.go: Fix GenerateCaptcha/VerifyCaptcha to use {code, message, data}
- password_reset_handler.go: Fix all error responses to use {code, message}
- settings_handler.go: Add missing "code" and "message" fields
- sms_handler.go: Fix error responses to use {code, message}
- sso_handler.go: Fix all error responses to use {code, message, data}
- stats_handler.go: Add missing "message" field in success responses
- theme_handler.go: Fix error responses to use {code, message}
- totp_handler.go: Fix all responses to use {code, message, data}
Standardize all JSON responses to {code: 0, message: "success", data: ...} for success
and {code: XXX, message: "..."} for errors.
59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/user-management-system/internal/service"
|
|
)
|
|
|
|
// CaptchaHandler handles captcha requests
|
|
type CaptchaHandler struct {
|
|
captchaService *service.CaptchaService
|
|
}
|
|
|
|
// NewCaptchaHandler creates a new CaptchaHandler
|
|
func NewCaptchaHandler(captchaService *service.CaptchaService) *CaptchaHandler {
|
|
return &CaptchaHandler{captchaService: captchaService}
|
|
}
|
|
|
|
func (h *CaptchaHandler) GenerateCaptcha(c *gin.Context) {
|
|
result, err := h.captchaService.Generate(c.Request.Context())
|
|
if err != nil {
|
|
handleError(c, err)
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"message": "success",
|
|
"data": gin.H{
|
|
"captcha_id": result.CaptchaID,
|
|
"image": result.ImageData,
|
|
},
|
|
})
|
|
}
|
|
|
|
func (h *CaptchaHandler) GetCaptchaImage(c *gin.Context) {
|
|
c.JSON(http.StatusOK, gin.H{"code": 0, "message": "success"})
|
|
}
|
|
|
|
func (h *CaptchaHandler) VerifyCaptcha(c *gin.Context) {
|
|
var req struct {
|
|
CaptchaID string `json:"captcha_id" binding:"required"`
|
|
Answer string `json:"answer" binding:"required"`
|
|
}
|
|
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": err.Error()})
|
|
return
|
|
}
|
|
|
|
if h.captchaService.Verify(c.Request.Context(), req.CaptchaID, req.Answer) {
|
|
c.JSON(http.StatusOK, gin.H{"code": 0, "message": "success", "data": gin.H{"verified": true}})
|
|
} else {
|
|
c.JSON(http.StatusBadRequest, gin.H{"code": 400, "message": "invalid captcha"})
|
|
}
|
|
}
|