feat: add no-need-invite-code feature for specific email suffix

JingYiJun · Feb 29, 2024 · 8569faa · 8569faa
1 parent 12720d8
commit 8569faa
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/apis/account/account.go b/apis/account/account.go
@@ -2,6 +2,7 @@ package account
 
 import (
 	"errors"
+	"strings"
 	"time"
 
 	"MOSS_backend/config"
@@ -68,7 +69,17 @@ func Register(c *fiber.Ctx) error {
 	}
 
 	// check Invite code
-	if configObject.InviteRequired {
+	var inviteRequired = configObject.InviteRequired
+	if body.EmailModel != nil {
+		// check email suffix in no need invite code
+		for _, emailSuffix := range config.Config.NoNeedInviteCodeEmailSuffix {
+			if strings.HasSuffix(body.Email, emailSuffix) {
+				inviteRequired = false
+				break
+			}
+		}
+	}
+	if inviteRequired {
 		if body.InviteCode == nil {
 			return errCollection.ErrNeedInviteCode
 		}

diff --git a/config/config.go b/config/config.go
@@ -63,7 +63,8 @@ var Config struct {
 
 	// InnerThoughtsPostprocess bool `env:"INNER_THOUGHTS_POSTPROCESS" envDefault:"false"`
 
-	DefaultModelID int `env:"DEFAULT_MODEL_ID" envDefault:"1"`
+	DefaultModelID              int      `env:"DEFAULT_MODEL_ID" envDefault:"1"`
+	NoNeedInviteCodeEmailSuffix []string `env:"NO_NEED_INVITE_CODE_EMAIL_SUFFIX" envSeparator:"," envDefault:"fudan.edu.cn"`
 }
 
 func InitConfig() {