add users list/get/create/delete commands

Author: kevinzliu

cmd/src/format.go

@@ -0,0 +1,27 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+	"text/template"
+)
+
+func parseTemplate(text string) (*template.Template, error) {
+	tmpl := template.New("")
+	tmpl.Funcs(map[string]interface{}{
+		"json": func(v interface{}) (string, error) {
+			b, err := json.MarshalIndent(v, "", "  ")
+			return string(b), err
+		},
+	})
+	return tmpl.Parse(text)
+}
+
+func execTemplate(tmpl *template.Template, data interface{}) error {
+	if err := tmpl.Execute(os.Stdout, data); err != nil {
+		return err
+	}
+	fmt.Println()
+	return nil
+}

cmd/src/main.go

@@ -26,7 +26,8 @@ The options are:
 The commands are:
 
 	api           interact with the Sourcegraph GraphQL API
-	repos,repo    manage repositories 
+	repos,repo    manage repositories
+	users,user    manage users
 
 Use "src [command] -h" for more information about a command.
 

cmd/src/orgs.go

@@ -0,0 +1,7 @@
+package main
+
+type Org struct {
+	ID          string
+	Name        string
+	DisplayName string
+}

cmd/src/users.go

@@ -0,0 +1,76 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+)
+
+var usersCommands commander
+
+func init() {
+	usage := `'src users' is a tool that manages users on a Sourcegraph instance.
+
+Usage:
+
+	src users command [command options]
+
+The commands are:
+
+	list       lists users
+	get       get a user
+	create     creates a user account
+	delete     deletes a user account
+
+Use "src users [command] -h" for more information about a command.
+`
+
+	flagSet := flag.NewFlagSet("users", flag.ExitOnError)
+	handler := func(args []string) error {
+		usersCommands.run(flagSet, "src users", usage, args)
+		return nil
+	}
+
+	// Register the command.
+	commands = append(commands, &command{
+		flagSet: flagSet,
+		aliases: []string{"user"},
+		handler: handler,
+		usageFunc: func() {
+			fmt.Println(usage)
+		},
+	})
+}
+
+const userFragment = `
+fragment UserFields on User {
+    id
+    username
+    displayName
+    siteAdmin
+    orgs {
+        id
+        name
+        displayName
+    }
+    emails {
+        email
+        verified
+    }
+    url
+}
+`
+
+type User struct {
+	ID          string
+	Username    string
+	DisplayName string
+	SiteAdmin   bool
+	Orgs        []Org
+	Emails      []UserEmail
+	URL         string
+}
+
+type UserEmail struct {
+	Email    string
+	Verified bool
+}

cmd/src/users_create.go

@@ -0,0 +1,76 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+)
+
+func init() {
+	usage := `
+Examples:
+
+  Create a user account:
+
+    	$ src users create -username=alice -email=alice@example.com
+
+`
+
+	flagSet := flag.NewFlagSet("create", flag.ExitOnError)
+	usageFunc := func() {
+		fmt.Fprintf(flag.CommandLine.Output(), "Usage of 'src users %s':\n", flagSet.Name())
+		flagSet.PrintDefaults()
+		fmt.Println(usage)
+	}
+	var (
+		usernameFlag         = flagSet.String("username", "", `The new user's username. (required)`)
+		emailFlag            = flagSet.String("email", "", `The new user's email address. (required)`)
+		resetPasswordURLFlag = flagSet.Bool("reset-password-url", false, `Print the reset password URL to manually send to the new user.`)
+		apiFlags             = newAPIFlags(flagSet)
+	)
+
+	handler := func(args []string) error {
+		flagSet.Parse(args)
+
+		query := `mutation CreateUser(
+  $username: String!,
+  $email: String!,
+) {
+  createUserBySiteAdmin(
+    username: $username,
+    email: $email,
+  ) {
+    resetPasswordURL
+  }
+}`
+
+		var result struct {
+			CreateUserBySiteAdmin struct {
+				ResetPasswordURL string
+			}
+		}
+		return (&apiRequest{
+			query: query,
+			vars: map[string]interface{}{
+				"username": *usernameFlag,
+				"email":    *emailFlag,
+			},
+			result: &result,
+			done: func() error {
+				fmt.Printf("User %q created.\n", *usernameFlag)
+				if *resetPasswordURLFlag && result.CreateUserBySiteAdmin.ResetPasswordURL != "" {
+					fmt.Println()
+					fmt.Printf("\tReset pasword URL: %s\n", result.CreateUserBySiteAdmin.ResetPasswordURL)
+				}
+				return nil
+			},
+			flags: apiFlags,
+		}).do()
+	}
+
+	// Register the command.
+	usersCommands = append(usersCommands, &command{
+		flagSet:   flagSet,
+		handler:   handler,
+		usageFunc: usageFunc,
+	})
+}

cmd/src/users_delete.go

@@ -0,0 +1,73 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+)
+
+func init() {
+	usage := `
+Examples:
+
+  Delete a user account by ID:
+
+    	$ src users delete -id=VXNlcjox
+
+  Delete a user account by username:
+
+    	$ src users delete -id=$(src users get -f='{{.ID}}' -username=alice)
+
+  Delete all user accounts that match the query:
+
+    	$ src users list -f='{{.ID}}' -query=alice | xargs -n 1 -I USERID src users delete -id=USERID
+
+`
+
+	flagSet := flag.NewFlagSet("delete", flag.ExitOnError)
+	usageFunc := func() {
+		fmt.Fprintf(flag.CommandLine.Output(), "Usage of 'src users %s':\n", flagSet.Name())
+		flagSet.PrintDefaults()
+		fmt.Println(usage)
+	}
+	var (
+		userIDFlag = flagSet.String("id", "", `The ID of the user to delete.`)
+		apiFlags   = newAPIFlags(flagSet)
+	)
+
+	handler := func(args []string) error {
+		flagSet.Parse(args)
+
+		query := `mutation DeleteUser(
+  $user: ID!
+) {
+  deleteUser(
+    user: $user
+  ) {
+    alwaysNil
+  }
+}`
+
+		var result struct {
+			DeleteUser struct{}
+		}
+		return (&apiRequest{
+			query: query,
+			vars: map[string]interface{}{
+				"user": *userIDFlag,
+			},
+			result: &result,
+			done: func() error {
+				fmt.Printf("User with ID %q deleted.\n", *userIDFlag)
+				return nil
+			},
+			flags: apiFlags,
+		}).do()
+	}
+
+	// Register the command.
+	usersCommands = append(usersCommands, &command{
+		flagSet:   flagSet,
+		handler:   handler,
+		usageFunc: usageFunc,
+	})
+}

cmd/src/users_get.go

@@ -0,0 +1,70 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+)
+
+func init() {
+	usage := `
+Examples:
+
+  Get user with username alice:
+
+    	$ src users get -username=alice
+
+`
+
+	flagSet := flag.NewFlagSet("get", flag.ExitOnError)
+	usageFunc := func() {
+		fmt.Fprintf(flag.CommandLine.Output(), "Usage of 'src users %s':\n", flagSet.Name())
+		flagSet.PrintDefaults()
+		fmt.Println(usage)
+	}
+	var (
+		usernameFlag = flagSet.String("username", "", `Look up the user by the username. (e.g. "alice")`)
+		formatFlag   = flagSet.String("f", "{{.|json}}", `Format for the output, using the syntax of Go package text/template. (e.g. "{{.ID}}: {{.Username}} ({{.DisplayName}})")`)
+		apiFlags     = newAPIFlags(flagSet)
+	)
+
+	handler := func(args []string) error {
+		flagSet.Parse(args)
+
+		tmpl, err := parseTemplate(*formatFlag)
+		if err != nil {
+			return err
+		}
+
+		query := `query User(
+  $username: String!,
+) {
+  user(
+    username: $username
+  ) {
+    ...UserFields
+  }
+}` + userFragment
+
+		var result struct {
+			User *User
+		}
+		return (&apiRequest{
+			query: query,
+			vars: map[string]interface{}{
+				"username": *usernameFlag,
+			},
+			result: &result,
+			done: func() error {
+				return execTemplate(tmpl, result.User)
+			},
+			flags: apiFlags,
+		}).do()
+	}
+
+	// Register the command.
+	usersCommands = append(usersCommands, &command{
+		flagSet:   flagSet,
+		handler:   handler,
+		usageFunc: usageFunc,
+	})
+}