|
|
package system
|
|
|
|
|
|
import (
|
|
|
"pychr/plugin/upyun"
|
|
|
"time"
|
|
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
"github.com/mojocn/base64Captcha"
|
|
|
"go.uber.org/zap"
|
|
|
"pychr/global"
|
|
|
"pychr/model/common/response"
|
|
|
systemRes "pychr/model/system/response"
|
|
|
)
|
|
|
|
|
|
// 当开启多服务器部署时，替换下面的配置，使用redis共享存储验证码
|
|
|
// var store = captcha.NewDefaultRedisStore()
|
|
|
var store = base64Captcha.DefaultMemStore
|
|
|
|
|
|
type BaseApi struct{}
|
|
|
|
|
|
// Captcha
|
|
|
// @Tags Base
|
|
|
// @Summary 生成验证码
|
|
|
// @Security ApiKeyAuth
|
|
|
// @accept application/json
|
|
|
// @Produce application/json
|
|
|
// @Success 200 {object} response.Response{data=systemRes.SysCaptchaResponse,msg=string} "生成验证码,返回包括随机数id,base64,验证码长度,是否开启验证码"
|
|
|
// @Router /base/captcha [post]
|
|
|
func (b *BaseApi) Captcha(c *gin.Context) {
|
|
|
// 判断验证码是否开启
|
|
|
openCaptcha := global.GVA_CONFIG.Captcha.OpenCaptcha // 是否开启防爆次数
|
|
|
openCaptchaTimeOut := global.GVA_CONFIG.Captcha.OpenCaptchaTimeOut // 缓存超时时间
|
|
|
key := c.ClientIP()
|
|
|
v, ok := global.BlackCache.Get(key)
|
|
|
if !ok {
|
|
|
global.BlackCache.Set(key, 1, time.Second*time.Duration(openCaptchaTimeOut))
|
|
|
}
|
|
|
|
|
|
var oc bool
|
|
|
if openCaptcha == 0 || openCaptcha < interfaceToInt(v) {
|
|
|
oc = true
|
|
|
}
|
|
|
// 字符,公式,验证码配置
|
|
|
// 生成默认数字的driver
|
|
|
driver := base64Captcha.NewDriverDigit(global.GVA_CONFIG.Captcha.ImgHeight, global.GVA_CONFIG.Captcha.ImgWidth, global.GVA_CONFIG.Captcha.KeyLong, 0.7, 80)
|
|
|
// cp := base64Captcha.NewCaptcha(driver, store.UseWithCtx(c)) // v8下使用redis
|
|
|
cp := base64Captcha.NewCaptcha(driver, store)
|
|
|
id, b64s, answer, err := cp.Generate()
|
|
|
if err != nil {
|
|
|
global.GVA_LOG.Error("验证码获取失败!", zap.Error(err))
|
|
|
response.FailWithMessage("验证码获取失败", c)
|
|
|
return
|
|
|
}
|
|
|
global.GVA_LOG.Debug(answer)
|
|
|
response.OkWithDetailed(systemRes.SysCaptchaResponse{
|
|
|
CaptchaId: id,
|
|
|
PicPath: b64s,
|
|
|
CaptchaLength: global.GVA_CONFIG.Captcha.KeyLong,
|
|
|
OpenCaptcha: oc,
|
|
|
}, "验证码获取成功", c)
|
|
|
}
|
|
|
|
|
|
type GetCDNTokenQuery struct {
|
|
|
Method string `form:"method"`
|
|
|
SaveKey string `form:"saveKey"`
|
|
|
}
|
|
|
|
|
|
type GetCDNTokenData struct {
|
|
|
Authorization string `json:"authorization"`
|
|
|
Bucket string `json:"bucket"`
|
|
|
Method string `json:"method"`
|
|
|
Policy string `json:"policy"`
|
|
|
Expiration int64 `json:"expiration"`
|
|
|
}
|
|
|
|
|
|
// @Tags CDN
|
|
|
// @Security ApiKeyAuth
|
|
|
// @Summary 获取CDN操作令牌
|
|
|
// @Produce json
|
|
|
// @Param method query int false "方法"
|
|
|
// @Param saveKey query int false "保存关键字"
|
|
|
// @Success 200 {object} structure.GetLoginInformationData{}
|
|
|
// @Router /base/cdn/token [get]
|
|
|
func (b *BaseApi) GetCDNToken(c *gin.Context) {
|
|
|
// method := c.Query("method")
|
|
|
// saveKey := c.Query("saveKey")
|
|
|
var (
|
|
|
q = GetCDNTokenQuery{}
|
|
|
data = GetCDNTokenData{}
|
|
|
token upyun.FormAPIToken
|
|
|
)
|
|
|
err := c.ShouldBindQuery(&q)
|
|
|
if err != nil {
|
|
|
response.FailWithMessage(err.Error(), c)
|
|
|
return
|
|
|
}
|
|
|
|
|
|
token, err = upyun.GetUpyunToken(q.Method, q.SaveKey)
|
|
|
if err != nil {
|
|
|
global.GVA_LOG.Error("获取失败!", zap.Error(err))
|
|
|
response.FailWithMessage("获取失败", c)
|
|
|
return
|
|
|
} else {
|
|
|
data.Authorization = token.Authorization
|
|
|
data.Bucket = token.Bucket
|
|
|
data.Method = token.Method
|
|
|
data.Policy = token.Policy
|
|
|
data.Expiration = token.XUpYunExpire
|
|
|
response.OkWithData(data, c)
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
// 类型转换
|
|
|
func interfaceToInt(v interface{}) (i int) {
|
|
|
switch v := v.(type) {
|
|
|
case int:
|
|
|
i = v
|
|
|
default:
|
|
|
i = 0
|
|
|
}
|
|
|
return
|
|
|
}
|