feat(foundation): add UserRegisterController

Author: lsy357

backend/api/api.go

@@ -90,6 +90,7 @@ func InitFoundationHandler(
 	idgen idgen.IIDGenerator,
 	db db.Provider,
 	objectStorage fileserver.BatchObjectStorage,
+	configFactory conf.IConfigLoaderFactory,
 ) (*FoundationHandler, error) {
 	wire.Build(
 		foundationSet,

backend/api/handler/coze/loop/apis/wire.go

@@ -465,6 +465,7 @@ github.com/google/pprof v0.0.0-20240827171923-fa2c70bbbfe5/go.mod h1:vavhavw2zAx
 github.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=
 github.com/google/s2a-go v0.1.8 h1:zZDs9gcbt9ZPLV0ndSyQk6Kacx2g/X+SKYovpnz3SMM=
 github.com/google/s2a-go v0.1.8/go.mod h1:6iNWHTpQ+nfNRN5E00MSdfDwVesa8hhS32PhPO8deJA=
+github.com/google/subcommands v1.2.0 h1:vWQspBTo2nEqTUFita5/KeEWlUL8kQObDFbub/EN9oE=
 github.com/google/subcommands v1.2.0/go.mod h1:ZjhPrFU+Olkh9WazFPsl27BQ4UPiG37m3yTrtFlrHVk=
 github.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=

backend/api/handler/coze/loop/apis/wire_gen.go

@@ -8,6 +8,7 @@ import (
 	"net/mail"
 	"strconv"
 
+	"github.com/bytedance/gg/gptr"
 	"github.com/bytedance/gg/gslice"
 
 	"github.com/coze-dev/coze-loop/backend/infra/middleware/session"
@@ -17,31 +18,62 @@ import (
 	"github.com/coze-dev/coze-loop/backend/modules/foundation/domain/user/entity"
 	"github.com/coze-dev/coze-loop/backend/modules/foundation/domain/user/service"
 	"github.com/coze-dev/coze-loop/backend/modules/foundation/pkg/errno"
+	"github.com/coze-dev/coze-loop/backend/pkg/conf"
 	"github.com/coze-dev/coze-loop/backend/pkg/errorx"
 	"github.com/coze-dev/coze-loop/backend/pkg/lang/conv"
 	"github.com/coze-dev/coze-loop/backend/pkg/lang/ptr"
 	"github.com/coze-dev/coze-loop/backend/pkg/lang/slices"
+	"github.com/coze-dev/coze-loop/backend/pkg/logs"
 )
 
 type UserApplicationImpl struct {
-	userService service.IUserService
+	userService        service.IUserService
+	registerController userRegisterController
+}
+
+type userRegisterController struct {
+	configLoader conf.IConfigLoader
+}
+
+type userRegisterControlConfig struct {
+	Block         bool     `mapstructure:"block"`
+	AllowedEmails []string `mapstructure:"allowed_emails"`
+}
+
+func (u *userRegisterController) allowRegister(ctx context.Context, email string) bool {
+	const keyUserRegisterControl = "user_register_control"
+
+	var config userRegisterControlConfig
+	if err := u.configLoader.UnmarshalKey(ctx, keyUserRegisterControl, &config); err != nil {
+		logs.CtxWarn(ctx, "load user_register_control config fail, err: %v", err)
+		return false
+	}
+
+	if !config.Block {
+		return true
+	}
+	return slices.Contains(config.AllowedEmails, email)
 }
 
 func NewUserApplication(
 	userService service.IUserService,
-) user.UserService {
+	configFactory conf.IConfigLoaderFactory,
+) (user.UserService, error) {
+	loader, err := configFactory.NewConfigLoader("foundation.yaml")
+	if err != nil {
+		return nil, err
+	}
 	return &UserApplicationImpl{
 		userService: userService,
-	}
+		registerController: userRegisterController{
+			configLoader: loader,
+		},
+	}, nil
 }
 
 func (u *UserApplicationImpl) Register(ctx context.Context, request *user.UserRegisterRequest) (r *user.UserRegisterResponse, err error) {
-	if request.Email == nil || request.Password == nil {
-		return nil, errorx.NewByCode(errno.CommonInvalidParamCode)
-	}
-
-	if _, err = mail.ParseAddress(*request.Email); err != nil {
-		return nil, errorx.NewByCode(errno.CommonInvalidParamCode, errorx.WithExtraMsg("email is invalid"))
+	if err := u.validateRegisterReq(ctx, request); err != nil {
+		return nil, err
 	}
 
 	userDO, err := u.userService.Create(ctx, &service.CreateUserRequest{
@@ -66,6 +98,22 @@ func (u *UserApplicationImpl) Register(ctx context.Context, request *user.UserRe
 	return r, nil
 }
 
+func (u *UserApplicationImpl) validateRegisterReq(ctx context.Context, request *user.UserRegisterRequest) error {
+	if request.Email == nil || request.Password == nil {
+		return errorx.NewByCode(errno.CommonInvalidParamCode)
+	}
+
+	if _, err := mail.ParseAddress(gptr.Indirect(request.Email)); err != nil {
+		return errorx.NewByCode(errno.CommonInvalidParamCode, errorx.WithExtraMsg("email is invalid"))
+	}
+
+	if !u.registerController.allowRegister(ctx, request.GetEmail()) {
+		return errorx.NewByCode(errno.UserRegistrationControlBlockCode)
+	}
+
+	return nil
+}
+
 func (u *UserApplicationImpl) ResetPassword(ctx context.Context, request *user.ResetPasswordRequest) (r *user.ResetPasswordResponse, err error) {
 	r = user.NewResetPasswordResponse()
 

backend/go.sum

@@ -23,6 +23,7 @@ import (
 	auth2 "github.com/coze-dev/coze-loop/backend/modules/foundation/infra/auth"
 	"github.com/coze-dev/coze-loop/backend/modules/foundation/infra/repo"
 	"github.com/coze-dev/coze-loop/backend/modules/foundation/infra/repo/mysql"
+	"github.com/coze-dev/coze-loop/backend/pkg/conf"
 )
 
 var (
@@ -92,6 +93,7 @@ func InitSpaceApplication(
 func InitUserApplication(
 	idgen idgen.IIDGenerator,
 	db db.Provider,
+	configFactory conf.IConfigLoaderFactory,
 ) (user.UserService, error) {
 	wire.Build(userSet)
 	return nil, nil

backend/modules/foundation/application/user.go

@@ -74,7 +74,7 @@ Where:
    # Generate code for evaluation domain
    ./code_gen.py evaluation --output-dir backend/module/evaluation/pkg/errno
 
-   # Or use default output directory (GOPATH/src/github.com/coze-dev/backend/module/{biz}/pkg/errno)
+   # Or use default output directory ({project_path}/backend/module/{biz}/pkg/errno)
    ./code_gen.py evaluation
    ```
 

backend/modules/foundation/application/wire.go

@@ -134,15 +134,15 @@ def generate_biz_code(biz_name: str, biz_code: int, common_errors: List[Dict], o
         biz_errors = load_yaml(biz_error_file)['error_code']
 
     # Generate and output code
-    project_dir = os.environ.get('PROJECT_DIR',
-                             os.path.join(os.environ['GOPATH'], 'src/github.com/coze-dev/coze-loop'))
+    current_file_dir = os.path.dirname(os.path.abspath(__file__))
+    project_dir = os.path.dirname(os.path.dirname(os.path.dirname(current_file_dir)))
     if not output_dir:
         output_dir = os.path.join(project_dir, 'backend/modules', biz_name, 'pkg/errno')
     else:
         output_dir = os.path.expandvars(output_dir)
         if not os.path.isabs(output_dir):
             output_dir = os.path.join(project_dir, output_dir)
-    
+
     return generate_go_code(biz_name, biz_code, common_errors, biz_errors, output_dir)