package v1

import (
	"aps_crm/conf"
	"aps_crm/constvar"
	"aps_crm/model/response"
	"aps_crm/pkg/contextx"
	"aps_crm/pkg/convertx"
	"aps_crm/pkg/ecode"
	"aps_crm/pkg/logx"
	"github.com/gin-gonic/gin"
	"github.com/mojocn/base64Captcha"
	"time"
)

type BaseApi struct{}

// Captcha
//	@Tags		Base
//	@Summary	获取验证码
//	@Produce	application/json
//	@Success	200	{object}	contextx.Response{data=response.LoginResponse}	"成功"
//	@Router		/api/base/captcha [post]
func (slf *BaseApi) Captcha(c *gin.Context) {
	ctx, ok := contextx.NewContext(c, nil)
	if !ok {
		return
	}

	// 判断验证码是否开启
	openCaptcha := conf.Conf.Captcha.OpenCaptcha               // 是否开启防爆次数
	openCaptchaTimeOut := conf.Conf.Captcha.OpenCaptchaTimeOut // 缓存超时时间
	key := c.ClientIP()
	v, ok := constvar.BlackCache.Get(key)
	if !ok {
		constvar.BlackCache.Set(key, 1, time.Second*time.Duration(openCaptchaTimeOut))
	}

	var oc bool
	if openCaptcha == 0 || convertx.InterfaceToInt(v) > openCaptcha { // 0 表示每次登录都需要验证码 或者当前次数已超过防爆次数
		oc = true
	}
	// 字符,公式,验证码配置
	// 生成默认数字的driver
	driver := base64Captcha.NewDriverDigit(conf.Conf.Captcha.ImgHeight, conf.Conf.Captcha.ImgWidth, conf.Conf.Captcha.KeyLong, 0.7, 80)
	cp := base64Captcha.NewCaptcha(driver, store)
	id, b64s, err := cp.Generate()
	captcha := store.Get(id, false)
	logx.Infof("Captcha Generate captchaId:%v captcha:%v", id, captcha)
	if err != nil {
		logx.Errorf("Captcha Generate err:%v", err)
		ctx.Fail(ecode.CaptchaGenerateFailed)
		return
	}

	ctx.OkWithDetailed(response.CaptchaResponse{
		CaptchaId:     id,
		PicPath:       b64s,
		CaptchaLength: conf.Conf.Captcha.KeyLong,
		OpenCaptcha:   oc,
	})
}