support captcha

providers/captcha/captcha.go

@@ -0,0 +1,55 @@
+package captcha
+
+import (
+	"atom/container"
+	"atom/providers/config"
+	"errors"
+	"log"
+
+	"github.com/mojocn/base64Captcha"
+)
+
+func init() {
+	if err := container.Container.Provide(NewCaptcha); err != nil {
+		log.Fatal(err)
+	}
+}
+
+type CaptchaResponse struct {
+	CaptchaId     string `json:"captcha_id,omitempty"`
+	PicPath       string `json:"pic_path,omitempty"`
+	CaptchaLength uint   `json:"captcha_length,omitempty"`
+	OpenCaptcha   uint   `json:"open_captcha,omitempty"`
+}
+
+type Captcha struct {
+	conf    *config.Config
+	captcha *base64Captcha.Captcha
+}
+
+func NewCaptcha(conf *config.Config, driver base64Captcha.Driver) (*Captcha, error) {
+	var store = base64Captcha.DefaultMemStore
+	return &Captcha{
+		conf:    conf,
+		captcha: base64Captcha.NewCaptcha(driver, store),
+	}, nil
+
+}
+
+func (c *Captcha) Generate() (*CaptchaResponse, error) {
+	id, b64s, err := c.captcha.Generate()
+	if err != nil {
+		return nil, errors.New("验证码获取失败")
+	}
+
+	return &CaptchaResponse{
+		CaptchaId:     id,
+		PicPath:       b64s,
+		CaptchaLength: c.conf.Captcha.KeyLong,
+		OpenCaptcha:   c.conf.Captcha.OpenCaptcha,
+	}, nil
+}
+
+func (c *Captcha) Verify(id, answer string) bool {
+	return c.captcha.Verify(id, answer, false)
+}

providers/captcha/driver/digit.go

@@ -0,0 +1,26 @@
+package storage
+
+import (
+	"atom/container"
+	"atom/providers/config"
+	"log"
+
+	"github.com/mojocn/base64Captcha"
+)
+
+func init() {
+	if err := container.Container.Provide(NewCaptchaDriverDigit); err != nil {
+		log.Fatal(err)
+	}
+}
+
+func NewCaptchaDriverDigit(conf *config.Config) (base64Captcha.Driver, error) {
+	// 字符,公式,验证码配置
+	// 生成默认数字的driver
+	return base64Captcha.NewDriverDigit(
+		int(conf.Captcha.ImgHeight),
+		int(conf.Captcha.ImgWidth),
+		int(conf.Captcha.KeyLong),
+		0.7,
+		80), nil
+}

providers/config/section_captcha.go

@@ -6,10 +6,10 @@ import (
 )
 
 type Captcha struct {
-	KeyLong            int    // 验证码长度
-	ImgWidth           int    // 验证码宽度
-	ImgHeight          int    // 验证码高度
-	OpenCaptcha        int    // 防爆破验证码开启此数，0代表每次登录都需要验证码，其他数字代表错误密码此数，如3代表错误三次后出现验证码
+	KeyLong            uint   // 验证码长度
+	ImgWidth           uint   // 验证码宽度
+	ImgHeight          uint   // 验证码高度
+	OpenCaptcha        uint   // 防爆破验证码开启此数，0代表每次登录都需要验证码，其他数字代表错误密码此数，如3代表错误三次后出现验证码
 	OpenCaptchaTimeOut string // 防爆破验证码超时时间，单位：s(秒)
 }
 

providers/provider.go

@@ -1,6 +1,8 @@
 package providers
 
 import (
+	_ "atom/providers/captcha"
+	_ "atom/providers/captcha/driver"
 	_ "atom/providers/config"
 	_ "atom/providers/database"
 	_ "atom/providers/faker"