xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-12 12:13:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add support for registing use force use invite code

Browse Source
This commit is contained in:
RockYang
2023-11-24 12:02:28 +08:00
parent 67d30353f0
commit d68f2ef12c
9 changed files with 83 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/handler/user_handler.go
View File

@@ -71,7 +71,12 @@ func (h *UserHandler) Register(c *gin.Context) {
// 验证邀请码
inviteCode := model.InviteCode{}
if data.InviteCode != "" {
if data.InviteCode == "" {
if h.App.SysConfig.ForceInvite {
resp.ERROR(c, "当前系统设定必须使用邀请码才能注册")
return
}
} else {
res := h.db.Where("code = ?", data.InviteCode).First(&inviteCode)
if res.Error != nil {
resp.ERROR(c, "无效的邀请码")