mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
103 lines
3.0 KiB
Go
103 lines
3.0 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package controller provides HTTP handlers and middlewares for the auth plugin.
|
|
package controller
|
|
|
|
import (
|
|
"Wavelet/pkg/logger"
|
|
"Wavelet/pkg/response"
|
|
"Wavelet/plugins/domain/auth/consts"
|
|
"Wavelet/plugins/domain/auth/model/dto"
|
|
"Wavelet/plugins/domain/auth/service"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// CaptchaHandler handles CAPTCHA challenge and redeem endpoints.
|
|
type CaptchaHandler struct {
|
|
capMgr *service.CaptchaManager
|
|
}
|
|
|
|
// NewCaptchaHandler creates a new CaptchaHandler.
|
|
func NewCaptchaHandler(mgr *service.CaptchaManager) *CaptchaHandler {
|
|
return &CaptchaHandler{
|
|
capMgr: mgr,
|
|
}
|
|
}
|
|
|
|
// Challenge 生成 PoW 人机验证难题
|
|
// @Summary 生成人机验证难题
|
|
// @Description 客户端获取 PoW 难题和签名的 JWT Token，并在后台计算。
|
|
// @Tags cap
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param request body dto.ChallengeRequest false "可选范围限制参数"
|
|
// @Success 200 {object} response.Any{data=dto.ChallengeResponse} "成功返回 PoW 难题"
|
|
// @Failure 500 {object} response.Any "内部服务错误"
|
|
// @Router /api/v1/cap/challenge [get]
|
|
// @Router /api/v1/cap/challenge [post]
|
|
func (h *CaptchaHandler) Challenge(c *gin.Context) {
|
|
var req dto.ChallengeRequest
|
|
_ = c.ShouldBind(&req) // 允许不传 body，默认使用 login scope
|
|
|
|
if req.Scope == "" {
|
|
req.Scope = "login"
|
|
}
|
|
|
|
if h.capMgr == nil {
|
|
response.AbortInternal(c, consts.ErrCapNotConfigured)
|
|
return
|
|
}
|
|
resp, err := h.capMgr.Generate(c.Request.Context(), req.Scope)
|
|
if err != nil {
|
|
logger.ErrorF(c.Request.Context(), "Generate cap challenge failed: %v", err)
|
|
response.AbortInternal(c, consts.ErrChallengeGenerateFailed)
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, response.OK(resp))
|
|
}
|
|
|
|
// Redeem 提交 PoW 解答并兑换一次性凭证 Token
|
|
// @Summary 校验人机验证解答
|
|
// @Description 提交 PoW 解答进行核销，成功后返回一次性 X-Cap-Token 凭证
|
|
// @Tags cap
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param request body dto.RedeemRequest true "难题 Token 与解答 solutions 数组"
|
|
// @Success 200 {object} response.Any{data=dto.RedeemResponse} "核销成功，返回 X-Cap-Token"
|
|
// @Failure 400 {object} response.Any "参数错误或核销失败"
|
|
// @Failure 500 {object} response.Any "内部服务错误"
|
|
// @Router /api/v1/cap/redeem [post]
|
|
func (h *CaptchaHandler) Redeem(c *gin.Context) {
|
|
var req dto.RedeemRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.AbortBadRequest(c, consts.ErrInvalidRequestParams)
|
|
return
|
|
}
|
|
|
|
if req.Scope == "" {
|
|
req.Scope = "login"
|
|
}
|
|
|
|
if h.capMgr == nil {
|
|
response.AbortInternal(c, consts.ErrCapNotConfigured)
|
|
return
|
|
}
|
|
resp, err := h.capMgr.Redeem(c.Request.Context(), req.Token, req.Solutions, req.Scope)
|
|
if err != nil {
|
|
logger.ErrorF(c.Request.Context(), "Redeem cap solutions failed: %v", err)
|
|
response.AbortInternal(c, consts.ErrSolutionVerifyFailed)
|
|
return
|
|
}
|
|
|
|
if !resp.Success {
|
|
response.AbortBadRequest(c, resp.Error)
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, response.OK(resp))
|
|
}
|