Changeset 329 in code for trunk/service.go

Timestamp:

Jun 8, 2020, 8:30:09 PM (5 years ago)

Author:

delthas

Message:

Add support for the user create admin service command

This adds support for user create, a new service command only accessible
to admin users. This lets users create other users on the fly and makes
soju start the user routine immediately; unlike sojuctl which currently
requires closing soju, creating the user, and starting soju again.

File:

• trunk/service.go (modified) (2 diffs)

trunk/service.go

@@ -163 +163 @@
                         },
                 },
+                "user": {
+                        children: serviceCommandSet{
+                                "create": {
+                                        usage:  "-username <username> -password <password> [-admin]",
+                                        desc:   "create a new soju user",
+                                        handle: handleUserCreate,
+                                        admin:  true,
+                                },
+                        },
+                        admin: true,
+                },
                 "change-password": {
                         usage:  "<new password>",
@@ -568 +579 @@
         return nil
 }
+
+func handleUserCreate(dc *downstreamConn, params []string) error {
+        fs := newFlagSet()
+        username := fs.String("username", "", "")
+        password := fs.String("password", "", "")
+        admin := fs.Bool("admin", false, "")
+
+        if err := fs.Parse(params); err != nil {
+                return err
+        }
+        if *username == "" {
+                return fmt.Errorf("flag -username is required")
+        }
+        if *password == "" {
+                return fmt.Errorf("flag -password is required")
+        }
+
+        hashed, err := bcrypt.GenerateFromPassword([]byte(*password), bcrypt.DefaultCost)
+        if err != nil {
+                return fmt.Errorf("failed to hash password: %v", err)
+        }
+
+        user := &User{
+                Username: *username,
+                Password: string(hashed),
+                Admin:    *admin,
+        }
+        if _, err := dc.srv.createUser(user); err != nil {
+                return fmt.Errorf("could not create user: %v", err)
+        }
+
+        sendServicePRIVMSG(dc, fmt.Sprintf("created user %q", *username))
+        return nil
+}