Base64Captcha 登录验证码

Captcha
GitHub 地址： github.com/dchest/captcha
简介： Captcha 是一个功能强大的验证码生成库，支持生成图片和音频验证码。它能够生成数字、字母、数字字母组合等各种类型的验证码，并且使用简单方便。

Gin-Utils
GitHub 地址： github.com/gin-contrib/gin-utils
简介： Gin-Utils 是一个针对 Gin 框架的实用工具集，其中包含了生成验证码的功能。它能够轻松地生成图片验证码，并且与 Gin 框架很好地集成。

BaseCaptcha
GitHub 地址： github.com/mojocn/base64Captcha
简介： BaseCaptcha 是一个基于 Golang 的验证码生成库，不过它提供了针对多种语言的接口。虽然是用 Golang 编写的，但可以通过其提供的接口在 Python 中使用。

 

正式开始Captcha

Captcha Go-Gin
获取库：
go get github.com/mojocn/base64Captcha

源码：

package login

import (
    "fmt"
    "go-admin/service/errcode"
    "net/http"

    "github.com/gin-gonic/gin"
    "github.com/mojocn/base64Captcha"
)

var (
    CaptchaStore base64Captcha.Store
)

func InitCatcha() {
    // 创建验证码的配置
    CaptchaStore = base64Captcha.DefaultMemStore
}

// @Summary 获取验证码
// @Schemes http
// @Description 登录-获取验证码
// @Tags 用户
// @Produce json
// @Success 200 {object} CaptchaReply  "successfully"
// @Router /login/captcha [get]
func Captcha(c *gin.Context) {
    defer func() {
        if r := recover(); r != nil {
            c.JSON(http.StatusInternalServerError, gin.H{
                "code": errcode.ERRInternalServerError,
                "msg":  fmt.Sprintf("%v", r),
            })
        }
    }()
    // driver := base64Captcha.NewDriverDigit(80, 240, 6, 0.7, 80)
    driver := base64Captcha.NewDriverString(
        80, 240, 6, 1, 4,
        "123456789abcdefghjkmnpqrstuvwxyzABCDEFGHJKLMNPQRSTUVWXYZ", // 包含数字和字母的字符集
        nil, nil, nil)
    captcha := base64Captcha.NewCaptcha(driver, CaptchaStore)
    id, b64s, answer, err := captcha.Generate()
    fmt.Println("####################", answer, " ", id)
    if err != nil {
        c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to generate captcha"})
        return
    }

    c.JSON(http.StatusOK, gin.H{
        "code": 0,
        "msg":  "",
        "result": CaptchaReply{
            CaptchaID: id,
            ImageData: b64s,
        },
    })
}

// @Summary 手动校验验证码
// @Schemes http
// @Description 登录-获取验证码
// @Tags 用户
// @Accept json
// @Produce json
// @Param request body VerifyRequest true "Verify request object"
// @Success 200 {object} comm.CommReply  "successfully"
// @Router /login/verify [post]
func Verify(c *gin.Context) {
    defer func() {
        if r := recover(); r != nil {
            c.JSON(http.StatusInternalServerError, gin.H{
                "code": errcode.ERRInternalServerError,
                "msg":  fmt.Sprintf("%v", r),
            })
        }
    }()

    in := new(VerifyRequest)
    err := c.ShouldBindJSON(in)
    if err != nil {
        c.JSON(http.StatusOK, gin.H{
            "code": -1,
            "msg":  err.Error(),
        })
        return
    }
    captchaId := in.CaptchaID
    captchaSolution := in.CaptchaAnswer

    if captchaId == "" || captchaSolution == "" {
        c.JSON(http.StatusBadRequest, gin.H{
            "code": http.StatusBadRequest,
            "msg":  "captcha_id and captcha_solution are required",
        })
        return
    }
    driver := base64Captcha.NewDriverString(
        80, 240, 6, 1, 4,
        "123456789abcdefghjkmnpqrstuvwxyzABCDEFGHJKLMNPQRSTUVWXYZ", // 包含数字和字母的字符集
        nil, nil, nil)
    captcha := base64Captcha.NewCaptcha(driver, CaptchaStore)

    // 验证用户输入的验证码
    if captcha.Verify(captchaId, captchaSolution, true) {
        c.JSON(http.StatusOK, gin.H{
            "code": 0,
            "msg":  "successfully",
        })
    } else {
        c.JSON(http.StatusUnauthorized, gin.H{
            "code": http.StatusUnauthorized,
            "msg":  "Invalid captcha",
        })
    }
}

func verify_(data *LoginPasswordRequest) (res bool) {
    res = false
    driver := base64Captcha.NewDriverString(
        80, 240, 6, 1, 4,
        "123456789abcdefghjkmnpqrstuvwxyzABCDEFGHJKLMNPQRSTUVWXYZ", // 包含数字和字母的字符集
        nil, nil, nil)
    captcha := base64Captcha.NewCaptcha(driver, CaptchaStore)

    return captcha.Verify(data.CaptchaID, data.CaptchaAnswer, true)
}