feat: 新增图形验证码功能

Author: tangtanglove

app/tool/captcha/index.go

@@ -0,0 +1,19 @@
+package captcha
+
+import (
+	"github.com/quarkcloudio/quark-go/v3"
+	"github.com/quarkcloudio/quark-go/v3/template/tool/captcha"
+)
+
+type Index struct {
+	captcha.Template
+}
+
+// 初始化
+func (p *Index) Init(ctx *quark.Context) interface{} {
+
+	// 验证码长度
+	p.Length = 4
+
+	return p
+}

app/tool/providers.go

@@ -1,9 +1,13 @@
 package service
 
-import "github.com/quarkcloudio/quark-go/v3/app/tool/upload"
+import (
+	"github.com/quarkcloudio/quark-go/v3/app/tool/captcha"
+	"github.com/quarkcloudio/quark-go/v3/app/tool/upload"
+)
 
 // 注册服务
 var Providers = []interface{}{
 	&upload.File{},
 	&upload.Image{},
+	&captcha.Index{},
 }

app/tool/qrcode/.keep

@@ -22,7 +22,7 @@ const (
 	AppName = "QuarkGo"
 
 	// 版本号
-	Version = "3.8.6"
+	Version = "3.8.7"
 
 	// 包名
 	PkgName = "github.com/quarkcloudio/quark-go/v3"

engine.go

@@ -0,0 +1,67 @@
+package captcha
+
+import (
+	"bytes"
+	"reflect"
+	"time"
+
+	"github.com/dchest/captcha"
+	"github.com/quarkcloudio/quark-go/v3"
+	redisclient "github.com/quarkcloudio/quark-go/v3/dal/redis"
+)
+
+// 文件上传
+type Template struct {
+	quark.Template
+	Length int // 验证码长度
+}
+
+// 初始化
+func (p *Template) Init(ctx *quark.Context) interface{} {
+	return p
+}
+
+// 初始化模板
+func (p *Template) TemplateInit(ctx *quark.Context) interface{} {
+
+	// 如果启动了redis缓存，验证码使用redis缓存
+	if redisclient.Client != nil {
+		captcha.SetCustomStore(&CaptchaStore{
+			RedisClient: redisclient.Client,
+			Expiration:  time.Second * 1000,
+		})
+	}
+
+	return p
+}
+
+// 初始化路由映射
+func (p *Template) RouteInit() interface{} {
+	p.GET("/api/tool/captcha/:resource/getId", p.CaptchaId)      // 登录获取验证码ID路由
+	p.GET("/api/tool/captcha/:resource/getImage/:id", p.Captcha) // 登录验证码路由
+	p.GET("/api/captcha/:resource/getId", p.CaptchaId)           // 登录获取验证码ID路由
+	p.GET("/api/captcha/:resource/getImage/:id", p.Captcha)      // 登录验证码路由
+
+	return p
+}
+
+// 验证码ID
+func (p *Template) CaptchaId(ctx *quark.Context) error {
+	length := int(reflect.
+		ValueOf(ctx.Template).
+		Elem().
+		FieldByName("Length").Int())
+
+	return ctx.JSONOk("获取成功", map[string]string{
+		"captchaId": captcha.NewLen(length),
+	})
+}
+
+// 生成验证码
+func (p *Template) Captcha(ctx *quark.Context) error {
+	id := ctx.Param("id")
+	writer := bytes.Buffer{}
+	captcha.WriteImage(&writer, id, 110, 38)
+	ctx.Write(writer.Bytes())
+	return nil
+}

template/tool/captcha/captcha.go

@@ -0,0 +1,22 @@
+package captcha
+
+import (
+	"context"
+	"time"
+
+	"github.com/redis/go-redis/v9"
+)
+
+type CaptchaStore struct {
+	RedisClient *redis.Client
+	Expiration  time.Duration
+}
+
+func (store *CaptchaStore) Set(id string, digits []byte) {
+	store.RedisClient.Set(context.Background(), id, string(digits), store.Expiration)
+}
+
+func (store *CaptchaStore) Get(id string, clear bool) (digits []byte) {
+	bytes, _ := store.RedisClient.Get(context.Background(), id).Bytes()
+	return bytes
+}