mirror of https://github.com/1Panel-dev/1Panel
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package captcha
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/1Panel-dev/1Panel/backend/app/dto"
|
|
"github.com/1Panel-dev/1Panel/backend/constant"
|
|
"github.com/mojocn/base64Captcha"
|
|
)
|
|
|
|
var store = base64Captcha.DefaultMemStore
|
|
|
|
func VerifyCode(codeID string, code string) error {
|
|
if codeID == "" {
|
|
return constant.ErrCaptchaCode
|
|
}
|
|
vv := store.Get(codeID, true)
|
|
vv = strings.TrimSpace(vv)
|
|
code = strings.TrimSpace(code)
|
|
|
|
if strings.EqualFold(vv, code) {
|
|
return nil
|
|
}
|
|
return constant.ErrCaptchaCode
|
|
}
|
|
|
|
func CreateCaptcha() (*dto.CaptchaResponse, error) {
|
|
var driverString base64Captcha.DriverString
|
|
driverString.Source = "1234567890QWERTYUPLKJHGFDSAZXCVBNMqwertyupkjhgfdsazxcvbnm"
|
|
driverString.Width = 120
|
|
driverString.Height = 50
|
|
driverString.NoiseCount = 0
|
|
driverString.Length = 4
|
|
driverString.Fonts = []string{"RitaSmith.ttf", "actionj.ttf", "chromohv.ttf"}
|
|
driver := driverString.ConvertFonts()
|
|
c := base64Captcha.NewCaptcha(driver, store)
|
|
id, b64s, err := c.Generate()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &dto.CaptchaResponse{
|
|
CaptchaID: id,
|
|
ImagePath: b64s,
|
|
}, nil
|
|
}
|