Add a option --bot to admin user create

Partially solve #13044
go-gitea · Nov 2, 2023 · 46b6867 · 46b6867
1 parent dc52f26
commit 46b6867
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/cmd/admin_user_create.go b/cmd/admin_user_create.go
@@ -62,6 +62,10 @@ var microcmdUserCreate = &cli.Command{
 			Name:  "restricted",
 			Usage: "Make a restricted user account",
 		},
+		&cli.BoolFlag{
+			Name:  "bot",
+			Usage: "Make a bot user account",
+		},
 	},
 }
 
@@ -129,6 +133,13 @@ func runCreateUser(c *cli.Context) error {
 		restricted = util.OptionalBoolOf(c.Bool("restricted"))
 	}
 
+	userType := user_model.UserTypeIndividual
+	if c.IsSet("bot") {
+		if c.Bool("bot") {
+			userType = user_model.UserTypeBot
+		}
+	}
+
 	// default user visibility in app.ini
 	visibility := setting.Service.DefaultUserVisibilityMode
 
@@ -144,6 +155,7 @@ func runCreateUser(c *cli.Context) error {
 	overwriteDefault := &user_model.CreateUserOverwriteOptions{
 		IsActive:     util.OptionalBoolTrue,
 		IsRestricted: restricted,
+		Type:         &userType,
 	}
 
 	if err := user_model.CreateUser(ctx, u, overwriteDefault); err != nil {

diff --git a/models/user/user.go b/models/user/user.go
@@ -579,6 +579,7 @@ type CreateUserOverwriteOptions struct {
 	Theme                        *string
 	IsRestricted                 util.OptionalBool
 	IsActive                     util.OptionalBool
+	Type                         *UserType
 }
 
 // CreateUser creates record of a new user.
@@ -629,6 +630,9 @@ func CreateUser(ctx context.Context, u *User, overwriteDefault ...*CreateUserOve
 		if !overwrite.IsActive.IsNone() {
 			u.IsActive = overwrite.IsActive.IsTrue()
 		}
+		if overwrite.Type != nil {
+			u.Type = *overwrite.Type
+		}
 	}
 
 	// validate data