package controllers import ( model "github.com/HFO4/cloudreve/models" "github.com/HFO4/cloudreve/pkg/conf" "github.com/HFO4/cloudreve/pkg/serializer" "github.com/HFO4/cloudreve/pkg/util" "github.com/gin-gonic/gin" "github.com/mojocn/base64Captcha" ) // SiteConfig 获取站点全局配置 func SiteConfig(c *gin.Context) { siteConfig := model.GetSettingByNames([]string{ "siteName", "login_captcha", "qq_login", "reg_captcha", "email_active", "forget_captcha", "email_active", "themes", "defaultTheme", }) // 如果已登录,则同时返回用户信息 user, _ := c.Get("user") if user, ok := user.(*model.User); ok { c.JSON(200, serializer.BuildSiteConfig(siteConfig, user)) return } c.JSON(200, serializer.BuildSiteConfig(siteConfig, nil)) } // Ping 状态检查页面 func Ping(c *gin.Context) { c.JSON(200, serializer.Response{ Code: 0, Msg: "Pong", }) } // Captcha 获取验证码 func Captcha(c *gin.Context) { // 验证码配置 var configD = base64Captcha.ConfigCharacter{ Height: conf.CaptchaConfig.Height, Width: conf.CaptchaConfig.Width, //const CaptchaModeNumber:数字,CaptchaModeAlphabet:字母,CaptchaModeArithmetic:算术,CaptchaModeNumberAlphabet:数字字母混合. Mode: conf.CaptchaConfig.Mode, ComplexOfNoiseText: conf.CaptchaConfig.ComplexOfNoiseText, ComplexOfNoiseDot: conf.CaptchaConfig.ComplexOfNoiseDot, IsShowHollowLine: conf.CaptchaConfig.IsShowHollowLine, IsShowNoiseDot: conf.CaptchaConfig.IsShowNoiseDot, IsShowNoiseText: conf.CaptchaConfig.IsShowNoiseText, IsShowSlimeLine: conf.CaptchaConfig.IsShowSlimeLine, IsShowSineLine: conf.CaptchaConfig.IsShowSineLine, CaptchaLen: conf.CaptchaConfig.CaptchaLen, } // 生成验证码 idKeyD, capD := base64Captcha.GenerateCaptcha("", configD) // 将验证码UID存入Session以便后续验证 util.SetSession(c, map[string]interface{}{ "captchaID": idKeyD, }) // 将验证码图像编码为Base64 base64stringD := base64Captcha.CaptchaWriteToBase64Encoding(capD) c.JSON(200, serializer.Response{ Code: 0, Data: base64stringD, }) }