feat(workflow-cli): add cmd for enable and disable users

Author: duanhongyi

cmd/cmd.go

@@ -89,6 +89,8 @@ type Commander interface {
 	TLSAutoEnable(string) error
 	TLSAutoDisable(string) error
 	UsersList(results int) error
+	UsersEnable(string) error
+	UsersDisable(string) error
 	WhitelistAdd(string, string) error
 	WhitelistList(string) error
 	WhitelistRemove(string, string) error

cmd/users.go

@@ -32,3 +32,38 @@ func (d *DryccCmd) UsersList(results int) error {
 	}
 	return nil
 }
+
+// UsersEnable enable user with the controller.
+func (d *DryccCmd) UsersEnable(username string) error {
+	s, err := settings.Load(d.ConfigFile)
+
+	if err != nil {
+		return err
+	}
+	d.Printf("Enabling user %s... ", username)
+	err = users.Enable(s.Client, username)
+	if d.checkAPICompatibility(s.Client, err) != nil {
+		return err
+	}
+
+	d.Println("done")
+	return nil
+}
+
+// UsersDisable disable user with the controller.
+func (d *DryccCmd) UsersDisable(username string) error {
+	s, err := settings.Load(d.ConfigFile)
+
+	if err != nil {
+		return err
+	}
+
+	d.Printf("Disabling user %s... ", username)
+	err = users.Disable(s.Client, username)
+	if d.checkAPICompatibility(s.Client, err) != nil {
+		return err
+	}
+
+	d.Println("done")
+	return nil
+}

cmd/users_test.go

@@ -110,3 +110,46 @@ func TestUsersListLimit(t *testing.T) {
 test
 `, "output")
 }
+
+func TestUsersEnable(t *testing.T) {
+	t.Parallel()
+	cf, server, err := testutil.NewTestServerAndClient()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer server.Close()
+	var b bytes.Buffer
+	cmdr := DryccCmd{WOut: &b, ConfigFile: cf}
+
+	server.Mux.HandleFunc("/v2/users/test/enable/", func(w http.ResponseWriter, r *http.Request) {
+		testutil.SetHeaders(w)
+		fmt.Fprintf(w, ``)
+	})
+
+	err = cmdr.UsersEnable("test")
+	assert.NoErr(t, err)
+
+	assert.Equal(t, b.String(), `Enabling user test... done
+`, "output")
+}
+
+func TestUsersDisable(t *testing.T) {
+	t.Parallel()
+	cf, server, err := testutil.NewTestServerAndClient()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer server.Close()
+	var b bytes.Buffer
+	cmdr := DryccCmd{WOut: &b, ConfigFile: cf}
+
+	server.Mux.HandleFunc("/v2/users/test/disable/", func(w http.ResponseWriter, r *http.Request) {
+		testutil.SetHeaders(w)
+		fmt.Fprintf(w, ``)
+	})
+
+	err = cmdr.UsersDisable("test")
+	assert.NoErr(t, err)
+	assert.Equal(t, b.String(), `Disabling user test... done
+`, "output")
+}

go.mod

@@ -5,7 +5,7 @@ go 1.13
 require (
 	github.com/arschles/assert v1.0.1-0.20191213221312-71f210f9375a
 	github.com/docopt/docopt-go v0.0.0-20180111231733-ee0de3bc6815
-	github.com/drycc/controller-sdk-go v0.0.0-20200914062048-dcf027e6dd1a
+	github.com/drycc/controller-sdk-go v0.0.0-20201013084049-82230b1dc54f
 	github.com/drycc/pkg v0.0.0-20200811173146-1f2b2781a852
 	github.com/olekukonko/tablewriter v0.0.4
 	golang.org/x/crypto v0.0.0-20200728195943-123391ffb6de

go.sum

@@ -40,6 +40,8 @@ github.com/drycc/controller-sdk-go v0.0.0-20200910051502-0b84631ba397 h1:MbpNwDO
 github.com/drycc/controller-sdk-go v0.0.0-20200910051502-0b84631ba397/go.mod h1:mF8/kZ/E4oDHCtfUVS790Frm8oVzZkoCGCKYT2eWQro=
 github.com/drycc/controller-sdk-go v0.0.0-20200914062048-dcf027e6dd1a h1:sQEQ8KUgm3X1y8XVsGXqeAkeowMGdOtu+PwBgd4V2eU=
 github.com/drycc/controller-sdk-go v0.0.0-20200914062048-dcf027e6dd1a/go.mod h1:mF8/kZ/E4oDHCtfUVS790Frm8oVzZkoCGCKYT2eWQro=
+github.com/drycc/controller-sdk-go v0.0.0-20201013084049-82230b1dc54f h1:Lwm3ijm8W+STLHOOmumLSQRvCkUTuya+XkGM+VNFfGA=
+github.com/drycc/controller-sdk-go v0.0.0-20201013084049-82230b1dc54f/go.mod h1:mF8/kZ/E4oDHCtfUVS790Frm8oVzZkoCGCKYT2eWQro=
 github.com/drycc/pkg v0.0.0-20190129033019-bcdffff0fcb3 h1:QRAbL97/hJecXQzaSONMCiwt1JjBKAAMKK0qJW5sgIM=
 github.com/drycc/pkg v0.0.0-20190129033019-bcdffff0fcb3/go.mod h1:zNgTT6kuGZOKMIdSrjskIw5fctdbbszqiTPEt3ffxHw=
 github.com/drycc/pkg v0.0.0-20200811173146-1f2b2781a852 h1:OsWH7jrv2i32rkvHFuj3yVpGMR+A6heRge9tm01XR/U=

parser/users.go

@@ -18,6 +18,10 @@ Use 'drycc help [command]' to learn more.
 	switch argv[0] {
 	case "users:list":
 		return usersList(argv, cmdr)
+	case "users:enable":
+		return usersEnable(argv, cmdr)
+	case "users:disable":
+		return usersDisable(argv, cmdr)
 	default:
 		if printHelp(argv, usage) {
 			return nil
@@ -58,3 +62,53 @@ Options:
 
 	return cmdr.UsersList(results)
 }
+
+func usersEnable(argv []string, cmdr cmd.Commander) error {
+	usage := `
+Enable a user when his status is disabled.
+Requires admin privileges.
+
+Usage: drycc users:enable <username>
+
+Arguments:
+  <username>
+  the username you want to enable.
+`
+
+	args, err := docopt.Parse(usage, argv, true, "", false, true)
+	if err != nil {
+		return err
+	}
+	username := safeGetValue(args, "<username>")
+
+	if err != nil {
+		return err
+	}
+
+	return cmdr.UsersEnable(username)
+}
+
+func usersDisable(argv []string, cmdr cmd.Commander) error {
+	usage := `
+Disable a user when his status is enabled.
+Requires admin privileges.
+
+Usage: drycc users:disable <username>
+
+Arguments:
+  <username>
+  the username you want to disable.
+`
+
+	args, err := docopt.Parse(usage, argv, true, "", false, true)
+	if err != nil {
+		return err
+	}
+	username := safeGetValue(args, "<username>")
+
+	if err != nil {
+		return err
+	}
+
+	return cmdr.UsersDisable(username)
+}

parser/users_test.go

@@ -16,6 +16,14 @@ func (d FakeDryccCmd) UsersList(results int) error {
 	return errors.New("users:list")
 }
 
+func (d FakeDryccCmd) UsersEnable(string) error {
+	return errors.New("users:enable")
+}
+
+func (d FakeDryccCmd) UsersDisable(string) error {
+	return errors.New("users:disable")
+}
+
 func TestUsers(t *testing.T) {
 	t.Parallel()
 
@@ -41,6 +49,14 @@ func TestUsers(t *testing.T) {
 			args:     []string{"users"},
 			expected: "users:list",
 		},
+		{
+			args:     []string{"users:enable", "test"},
+			expected: "",
+		},
+		{
+			args:     []string{"users:disable", "test"},
+			expected: "",
+		},
 	}
 
 	// For each case, check that calling the route with the arguments