shop/pkg/captcha/CaptchaUtils.go

package captcha

import (
	"github.com/mojocn/base64Captcha"
	"image/color"
	"time"
	"ulink-admin/config"
)

var result = base64Captcha.NewMemoryStore(200240, 3*time.Minute)

type CaptchaInfo struct {
	Id    string `json:"id" `    //id
	Image string `json:"image" ` //验证码
}

// mathConfig 生成图形化算术验证码配置
func mathConfig() *base64Captcha.DriverMath {
	mathType := &base64Captcha.DriverMath{
		Height:          50,
		Width:           150,
		NoiseCount:      0,
		ShowLineOptions: base64Captcha.OptionShowHollowLine,
		/*		BgColor: &color.RGBA{
				R: 40,
				G: 30,
				B: 89,
				A: 29,
			},*/
		Fonts: nil,
	}
	return mathType
}

// digitConfig 生成图形化数字验证码配置
func digitConfig() *base64Captcha.DriverDigit {
	digitType := &base64Captcha.DriverDigit{
		Height:   50,
		Width:    100,
		Length:   5,
		MaxSkew:  0.45,
		DotCount: 80,
	}
	return digitType
}

// stringConfig 生成图形化字符串验证码配置
func stringConfig() *base64Captcha.DriverString {
	stringType := &base64Captcha.DriverString{
		Height:     50,
		Width:      150,
		NoiseCount: 0,
		//ShowLineOptions: base64Captcha.OptionShowHollowLine | base64Captcha.OptionShowSlimeLine,
		Length: 4,
		Source: "23456789qwertyupasdfghjkzxcvb",

		BgColor: &color.RGBA{
			R: 40,
			G: 30,
			B: 89,
			A: 29,
		},
		Fonts: []string{"wqy-microhei.ttc"},
	}
	return stringType
}

// chineseConfig 生成图形化汉字验证码配置
func chineseConfig() *base64Captcha.DriverChinese {
	chineseType := &base64Captcha.DriverChinese{
		Height:          50,
		Width:           100,
		NoiseCount:      0,
		ShowLineOptions: base64Captcha.OptionShowSlimeLine,
		Length:          2,
		Source:          "设想,你在,处理,消费者,的音,频输,出音,频可,能无,论什,么都,没有,任何,输出,或者,它可,能是,单声道,立体声,或是,环绕立,体声的,不想要,的值",
		BgColor: &color.RGBA{
			R: 40,
			G: 30,
			B: 89,
			A: 29,
		},
		Fonts: nil,
	}
	return chineseType
}

// autoConfig 生成图形化数字音频验证码配置
func autoConfig() *base64Captcha.DriverAudio {
	chineseType := &base64Captcha.DriverAudio{
		Length:   4,
		Language: "zh",
	}
	return chineseType
}

// @Result id 验证码id
// @Result bse64s 图片base64编码
// @Result err 错误
func CreateCode() (string, string, error) {
	var driver base64Captcha.Driver
	// switch case分支中的方法为目录3的配置
	// switch case分支中的方法为目录3的配置
	// switch case分支中的方法为目录3的配置
	captchaConfig := config.GetCaptchaConfig()
	switch captchaConfig.Type {
	case "audio":
		driver = autoConfig()
	case "string":
		driver = stringConfig().ConvertFonts()
	case "math":
		driver = mathConfig()
	case "chinese":
		driver = chineseConfig()

	case "digit":
		driver = digitConfig()
	}
	if driver == nil {
		panic("生成验证码的类型没有配置，请在yaml文件中配置完再次重试启动项目")
	}
	// 创建验证码并传入创建的类型的配置，以及存储的对象
	//driver = driver.ConvertFonts()
	c := base64Captcha.NewCaptcha(driver, result)
	id, b64s, err := c.Generate()
	return id, b64s, err
}

// @Pram id 验证码id
// @Pram VerifyValue 用户输入的答案
// @Result true：正确，false：失败
func VerifyCaptcha(id, VerifyValue string) bool {
	// result 为步骤1 创建的图片验证码存储对象
	return result.Verify(id, VerifyValue, true)
}

// @Pram codeId 验证码id
// @Result 验证码答案
func GetCodeAnswer(codeId string) string {
	// result 为步骤1 创建的图片验证码存储对象
	return result.Get(codeId, false)
}