MiFaZhan
62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
// package system
|
|
// @Link https://github.com/xiujiecn/xiu-admin
|
|
// @Copyright Copyright (c) 2025 LiXiujie
|
|
// @Author Lxj <li@example.com>
|
|
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
|
package system
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
"git.5iot.com/iotlcm-public/iotlcm-app/internal/consts"
|
|
"git.5iot.com/iotlcm-public/iotlcm-app/internal/service"
|
|
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/mojocn/base64Captcha"
|
|
)
|
|
|
|
var store = base64Captcha.NewMemoryStore(100, 120*time.Second)
|
|
|
|
type sSysCaptcha struct {
|
|
}
|
|
|
|
func NewSysCaptcha() *sSysCaptcha {
|
|
return &sSysCaptcha{}
|
|
}
|
|
|
|
func init() {
|
|
service.RegisterSysCaptcha(NewSysCaptcha())
|
|
}
|
|
|
|
type CaptchaConfig struct {
|
|
KeyLong int
|
|
ImgWidth int
|
|
ImgHeight int
|
|
OpenCaptcha int
|
|
OpenCaptchaTimeout int
|
|
}
|
|
|
|
// 生成验证码
|
|
func (l *sSysCaptcha) GenerateCaptcha(ctx context.Context) (key string, image string, err error) {
|
|
var captchaConfig CaptchaConfig
|
|
g.Cfg().MustGet(ctx, "captcha").Struct(&captchaConfig)
|
|
driver := base64Captcha.NewDriverDigit(captchaConfig.ImgHeight, captchaConfig.ImgWidth, captchaConfig.KeyLong, 0.7, 80)
|
|
captcha := base64Captcha.NewCaptcha(driver, store)
|
|
key, image, _, err = captcha.Generate()
|
|
return
|
|
}
|
|
|
|
// 验证验证码
|
|
func (l *sSysCaptcha) VerifyCaptcha(ctx context.Context, key string, value string) (err error) {
|
|
v := store.Get(key, false)
|
|
if len(v) == 0 {
|
|
return gerror.NewCode(consts.CodeCaptchaError, "验证码已失效")
|
|
}
|
|
ok := store.Verify(key, value, true)
|
|
if !ok {
|
|
err = gerror.NewCode(consts.CodeCaptchaError, "验证码错误")
|
|
}
|
|
return
|
|
}
|