125 lines
3.4 KiB
Go
125 lines
3.4 KiB
Go
package controllers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
"strings"
|
|
|
|
"server/models"
|
|
|
|
beego "github.com/beego/beego/v2/server/web"
|
|
)
|
|
|
|
type PlatformLoginVerifyController struct {
|
|
beego.Controller
|
|
}
|
|
|
|
type loginVerifyPayload struct {
|
|
OpenVerifyEnabled *int8 `json:"openVerify_enabled"`
|
|
VerifyType string `json:"use_geetest"`
|
|
Geetest3ID *string `json:"geetest3_id"`
|
|
Geetest3Key *string `json:"geetest3_key"`
|
|
Geetest4ID *string `json:"geetest4_id"`
|
|
Geetest4Key *string `json:"geetest4_key"`
|
|
}
|
|
|
|
func normalizeVerifyType(v string) string {
|
|
switch strings.TrimSpace(v) {
|
|
case "sms", "geetest", "email", "captcha":
|
|
return strings.TrimSpace(v)
|
|
default:
|
|
return "captcha"
|
|
}
|
|
}
|
|
|
|
// GetLoginVerifyInfos 获取登录验证配置
|
|
// GET /platform/loginVerifyInfos
|
|
func (c *PlatformLoginVerifyController) GetLoginVerifyInfos() {
|
|
cfg, err := models.GetPlatformLoginVerify()
|
|
if err != nil {
|
|
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "获取配置失败"}
|
|
_ = c.ServeJSON()
|
|
return
|
|
}
|
|
c.Data["json"] = map[string]interface{}{
|
|
"code": 200,
|
|
"msg": "success",
|
|
"data": map[string]interface{}{
|
|
"openVerify_enabled": cfg.OpenVerifyEnabled,
|
|
"use_geetest": cfg.VerifyType,
|
|
"geetest3_id": cfg.Geetest3ID,
|
|
"geetest3_key": cfg.Geetest3Key,
|
|
"geetest4_id": cfg.Geetest4ID,
|
|
"geetest4_key": cfg.Geetest4Key,
|
|
},
|
|
}
|
|
_ = c.ServeJSON()
|
|
}
|
|
|
|
// SaveLoginVerifyInfos 保存登录验证配置
|
|
// POST /platform/saveloginVerifyInfos
|
|
func (c *PlatformLoginVerifyController) SaveLoginVerifyInfos() {
|
|
var p loginVerifyPayload
|
|
raw, _ := io.ReadAll(c.Ctx.Request.Body)
|
|
if err := json.Unmarshal(raw, &p); err != nil {
|
|
c.Data["json"] = map[string]interface{}{"code": 400, "msg": "参数错误"}
|
|
_ = c.ServeJSON()
|
|
return
|
|
}
|
|
|
|
verifyType := normalizeVerifyType(p.VerifyType)
|
|
openVerifyEnabled := int8(1)
|
|
if p.OpenVerifyEnabled != nil {
|
|
openVerifyEnabled = *p.OpenVerifyEnabled
|
|
}
|
|
if verifyType == "geetest" {
|
|
if p.Geetest4ID == nil || strings.TrimSpace(*p.Geetest4ID) == "" {
|
|
c.Data["json"] = map[string]interface{}{"code": 400, "msg": "geetest4_id 不能为空"}
|
|
_ = c.ServeJSON()
|
|
return
|
|
}
|
|
if p.Geetest4Key == nil || strings.TrimSpace(*p.Geetest4Key) == "" {
|
|
c.Data["json"] = map[string]interface{}{"code": 400, "msg": "geetest4_key 不能为空"}
|
|
_ = c.ServeJSON()
|
|
return
|
|
}
|
|
}
|
|
|
|
var existed models.PlatformLoginVerify
|
|
err := models.Orm.QueryTable(new(models.PlatformLoginVerify)).OrderBy("-id").One(&existed)
|
|
if err == nil {
|
|
update := map[string]interface{}{
|
|
"open_verify_enabled": openVerifyEnabled,
|
|
"verify_type": verifyType,
|
|
"geetest3_id": p.Geetest3ID,
|
|
"geetest3_key": p.Geetest3Key,
|
|
"geetest4_id": p.Geetest4ID,
|
|
"geetest4_key": p.Geetest4Key,
|
|
}
|
|
_, err = models.Orm.QueryTable(new(models.PlatformLoginVerify)).Filter("id", existed.ID).Update(update)
|
|
if err != nil {
|
|
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "保存失败"}
|
|
_ = c.ServeJSON()
|
|
return
|
|
}
|
|
} else {
|
|
row := &models.PlatformLoginVerify{
|
|
OpenVerifyEnabled: openVerifyEnabled,
|
|
VerifyType: verifyType,
|
|
Geetest3ID: p.Geetest3ID,
|
|
Geetest3Key: p.Geetest3Key,
|
|
Geetest4ID: p.Geetest4ID,
|
|
Geetest4Key: p.Geetest4Key,
|
|
}
|
|
if _, err := models.Orm.Insert(row); err != nil {
|
|
c.Data["json"] = map[string]interface{}{"code": 500, "msg": "保存失败"}
|
|
_ = c.ServeJSON()
|
|
return
|
|
}
|
|
}
|
|
|
|
c.Data["json"] = map[string]interface{}{"code": 200, "msg": "保存成功"}
|
|
_ = c.ServeJSON()
|
|
}
|
|
|