mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
109 lines
3.0 KiB
Go
109 lines
3.0 KiB
Go
/*
|
|
Copyright 2026 Arctel.net
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package cap
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/util/cap"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type challengeRequest struct {
|
|
Scope string `json:"scope" form:"scope"`
|
|
}
|
|
|
|
type redeemRequest struct {
|
|
Token string `json:"token" binding:"required"`
|
|
Solutions []int `json:"solutions" binding:"required"`
|
|
Scope string `json:"scope" form:"scope"`
|
|
}
|
|
|
|
// Challenge 生成 PoW 人机验证难题
|
|
// @Summary 生成人机验证难题
|
|
// @Description 客户端获取 PoW 难题和签名的 JWT Token，并在后台计算。
|
|
// @Tags cap
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param request body challengeRequest false "可选范围限制参数"
|
|
// @Success 200 {object} cap.ChallengeResponse "成功返回 PoW 难题"
|
|
// @Failure 500 {object} cap.RedeemResponse "内部服务错误"
|
|
// @Router /api/cap/challenge [post]
|
|
func Challenge(c *gin.Context) {
|
|
var req challengeRequest
|
|
_ = c.ShouldBind(&req) // 允许不传 body，默认使用 login scope
|
|
|
|
if req.Scope == "" {
|
|
req.Scope = "login"
|
|
}
|
|
|
|
mgr := cap.GetDefaultManager()
|
|
resp, err := mgr.Generate(c.Request.Context(), req.Scope)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, cap.RedeemResponse{
|
|
Success: false,
|
|
Error: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, resp)
|
|
}
|
|
|
|
// Redeem 提交 PoW 解答并兑换一次性凭证 Token
|
|
// @Summary 校验人机验证解答
|
|
// @Description 提交 PoW 解答进行核销，成功后返回一次性 X-Cap-Token 凭证
|
|
// @Tags cap
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param request body redeemRequest true "难题 Token 与解答 solutions 数组"
|
|
// @Success 200 {object} cap.RedeemResponse "核销成功，返回 X-Cap-Token"
|
|
// @Failure 400 {object} cap.RedeemResponse "参数错误或核销失败"
|
|
// @Failure 500 {object} cap.RedeemResponse "内部服务错误"
|
|
// @Router /api/cap/redeem [post]
|
|
func Redeem(c *gin.Context) {
|
|
var req redeemRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
c.JSON(http.StatusBadRequest, cap.RedeemResponse{
|
|
Success: false,
|
|
Error: "无效的参数",
|
|
})
|
|
return
|
|
}
|
|
|
|
if req.Scope == "" {
|
|
req.Scope = "login"
|
|
}
|
|
|
|
mgr := cap.GetDefaultManager()
|
|
resp, err := mgr.Redeem(c.Request.Context(), req.Token, req.Solutions, req.Scope)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, cap.RedeemResponse{
|
|
Success: false,
|
|
Error: err.Error(),
|
|
})
|
|
return
|
|
}
|
|
|
|
if !resp.Success {
|
|
c.JSON(http.StatusBadRequest, resp)
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, resp)
|
|
}
|