package admin

import (
	"chatplus/core"
	"chatplus/handler"
	"chatplus/utils/resp"
	"github.com/gin-gonic/gin"
	"github.com/mojocn/base64Captcha"
)

type CaptchaHandler struct {
	handler.BaseHandler
}

func NewCaptchaHandler(app *core.AppServer) *CaptchaHandler {
	h := CaptchaHandler{}
	h.App = app
	return &h
}

type CaptchaVo struct {
	CaptchaId string `json:"captcha_id"`
	PicPath   string `json:"pic_path"`
}

// GetCaptcha 获取验证码
func (h *CaptchaHandler) GetCaptcha(c *gin.Context) {
	var captchaVo CaptchaVo
	driver := base64Captcha.NewDriverDigit(48, 130, 4, 0.4, 10)
	cp := base64Captcha.NewCaptcha(driver, base64Captcha.DefaultMemStore)
	// b64s是图片的base64编码
	id, b64s, err := cp.Generate()
	if err != nil {
		resp.ERROR(c, "生成验证码错误!")
		return
	}
	captchaVo.CaptchaId = id
	captchaVo.PicPath = b64s

	resp.SUCCESS(c, captchaVo)
}