coder licenses add CLI command (#3632)

spikecurtis · web-flow · commit 184f0625e154 · 2022-08-23T13:55:39.000-07:00
* coder licenses add CLI command

Signed-off-by: Spike Curtis &lt;spike@coder.com&gt;

* Fix up lint

Signed-off-by: Spike Curtis &lt;spike@coder.com&gt;

* Fix t.parallel call

Signed-off-by: Spike Curtis &lt;spike@coder.com&gt;

* Code review improvements

Signed-off-by: Spike Curtis &lt;spike@coder.com&gt;

* Lint

Signed-off-by: Spike Curtis &lt;spike@coder.com&gt;

Signed-off-by: Spike Curtis &lt;spike@coder.com&gt;
diff --git a/cli/clitest/clitest.go b/cli/clitest/clitest.go
@@ -21,7 +21,13 @@ import (
 // New creates a CLI instance with a configuration pointed to a
 // temporary testing directory.
 func New(t *testing.T, args ...string) (*cobra.Command, config.Root) {
-	cmd := cli.Root(cli.AGPL())
+	return NewWithSubcommands(t, cli.AGPL(), args...)
+}
+
+func NewWithSubcommands(
+	t *testing.T, subcommands []*cobra.Command, args ...string,
+) (*cobra.Command, config.Root) {
+	cmd := cli.Root(subcommands)
 	dir := t.TempDir()
 	root := config.Root(dir)
 	cmd.SetArgs(append([]string{"--global-config", dir}, args...))
diff --git a/cli/configssh.go b/cli/configssh.go
@@ -158,7 +158,7 @@ func configSSH() *cobra.Command {
 		),
 		Args: cobra.ExactArgs(0),
 		RunE: func(cmd *cobra.Command, _ []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/create.go b/cli/create.go
@@ -27,7 +27,7 @@ func create() *cobra.Command {
 		Use:         "create [name]",
 		Short:       "Create a workspace from a template",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/delete.go b/cli/delete.go
@@ -28,7 +28,7 @@ func deleteWorkspace() *cobra.Command {
 				return err
 			}
 
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/features.go b/cli/features.go
@@ -36,7 +36,7 @@ func featuresList() *cobra.Command {
 		Use:     "list",
 		Aliases: []string{"ls"},
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/list.go b/cli/list.go
@@ -65,7 +65,7 @@ func list() *cobra.Command {
 		Aliases:     []string{"ls"},
 		Args:        cobra.ExactArgs(0),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/logout.go b/cli/logout.go
@@ -16,7 +16,7 @@ func logout() *cobra.Command {
 		Use:   "logout",
 		Short: "Remove the local authenticated session",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/parameterslist.go b/cli/parameterslist.go
@@ -22,7 +22,7 @@ func parameterList() *cobra.Command {
 		RunE: func(cmd *cobra.Command, args []string) error {
 			scope, name := args[0], args[1]
 
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/portforward.go b/cli/portforward.go
@@ -70,7 +70,7 @@ func portForward() *cobra.Command {
 				return xerrors.New("no port-forwards requested")
 			}
 
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/publickey.go b/cli/publickey.go
@@ -20,7 +20,7 @@ func publickey() *cobra.Command {
 		Aliases: []string{"pubkey"},
 		Short:   "Output your public key for Git operations",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return xerrors.Errorf("create codersdk client: %w", err)
 			}
diff --git a/cli/root.go b/cli/root.go
@@ -114,7 +114,7 @@ func Root(subcommands []*cobra.Command) *cobra.Command {
 					return nil
 				}
 
-				client, err := createClient(cmd)
+				client, err := CreateClient(cmd)
 				// If the client is unauthenticated we can ignore the check.
 				// The child commands should handle an unauthenticated client.
 				if xerrors.Is(err, errUnauthenticated) {
@@ -190,9 +190,9 @@ func isTest() bool {
 	return flag.Lookup("test.v") != nil
 }
 
-// createClient returns a new client from the command context.
+// CreateClient returns a new client from the command context.
 // It reads from global configuration files if flags are not set.
-func createClient(cmd *cobra.Command) (*codersdk.Client, error) {
+func CreateClient(cmd *cobra.Command) (*codersdk.Client, error) {
 	root := createConfig(cmd)
 	rawURL, err := cmd.Flags().GetString(varURL)
 	if err != nil || rawURL == "" {
@@ -226,7 +226,7 @@ func createClient(cmd *cobra.Command) (*codersdk.Client, error) {
 }
 
 // createAgentClient returns a new client from the command context.
-// It works just like createClient, but uses the agent token and URL instead.
+// It works just like CreateClient, but uses the agent token and URL instead.
 func createAgentClient(cmd *cobra.Command) (*codersdk.Client, error) {
 	rawURL, err := cmd.Flags().GetString(varAgentURL)
 	if err != nil {
diff --git a/cli/schedule.go b/cli/schedule.go
@@ -77,7 +77,7 @@ func scheduleShow() *cobra.Command {
 		Long:  scheduleShowDescriptionLong,
 		Args:  cobra.ExactArgs(1),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
@@ -106,7 +106,7 @@ func scheduleStart() *cobra.Command {
 		Long:  scheduleStartDescriptionLong,
 		Args:  cobra.RangeArgs(2, 4),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
@@ -156,7 +156,7 @@ func scheduleStop() *cobra.Command {
 		Short: "Edit workspace stop schedule",
 		Long:  scheduleStopDescriptionLong,
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
@@ -207,7 +207,7 @@ func scheduleOverride() *cobra.Command {
 				return err
 			}
 
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return xerrors.Errorf("create client: %w", err)
 			}
diff --git a/cli/show.go b/cli/show.go
@@ -14,7 +14,7 @@ func show() *cobra.Command {
 		Short:       "Show details of a workspace's resources and agents",
 		Args:        cobra.ExactArgs(1),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/ssh.go b/cli/ssh.go
@@ -54,7 +54,7 @@ func ssh() *cobra.Command {
 			ctx, cancel := context.WithCancel(cmd.Context())
 			defer cancel()
 
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/start.go b/cli/start.go
@@ -17,7 +17,7 @@ func start() *cobra.Command {
 		Short:       "Build a workspace with the start state",
 		Args:        cobra.ExactArgs(1),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/state.go b/cli/state.go
@@ -27,7 +27,7 @@ func statePull() *cobra.Command {
 		Use:  "pull <workspace> [file]",
 		Args: cobra.MinimumNArgs(1),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
@@ -68,7 +68,7 @@ func statePush() *cobra.Command {
 		Use:  "push <workspace> <file>",
 		Args: cobra.ExactArgs(2),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/stop.go b/cli/stop.go
@@ -25,7 +25,7 @@ func stop() *cobra.Command {
 				return err
 			}
 
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/templatecreate.go b/cli/templatecreate.go
@@ -32,7 +32,7 @@ func templateCreate() *cobra.Command {
 		Short: "Create a template from the current directory or as specified by flag",
 		Args:  cobra.MaximumNArgs(1),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/templatedelete.go b/cli/templatedelete.go
@@ -23,7 +23,7 @@ func templateDelete() *cobra.Command {
 				templates     = []codersdk.Template{}
 			)
 
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/templateedit.go b/cli/templateedit.go
@@ -25,7 +25,7 @@ func templateEdit() *cobra.Command {
 		Args:  cobra.ExactArgs(1),
 		Short: "Edit the metadata of a template by name.",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return xerrors.Errorf("create client: %w", err)
 			}
diff --git a/cli/templatelist.go b/cli/templatelist.go
@@ -16,7 +16,7 @@ func templateList() *cobra.Command {
 		Short:   "List all the templates available for the organization",
 		Aliases: []string{"ls"},
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/templatepull.go b/cli/templatepull.go
@@ -29,7 +29,7 @@ func templatePull() *cobra.Command {
 				dest = args[1]
 			}
 
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return xerrors.Errorf("create client: %w", err)
 			}
diff --git a/cli/templatepush.go b/cli/templatepush.go
@@ -29,7 +29,7 @@ func templatePush() *cobra.Command {
 		Args:  cobra.MaximumNArgs(1),
 		Short: "Push a new template version from the current directory or as specified by flag",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/templateversions.go b/cli/templateversions.go
@@ -38,7 +38,7 @@ func templateVersionsList() *cobra.Command {
 		Args:  cobra.ExactArgs(1),
 		Short: "List all the versions of the specified template",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return xerrors.Errorf("create client: %w", err)
 			}
diff --git a/cli/update.go b/cli/update.go
@@ -22,7 +22,7 @@ func update() *cobra.Command {
 		Args:        cobra.ExactArgs(1),
 		Short:       "Update a workspace to the latest template version",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/usercreate.go b/cli/usercreate.go
@@ -21,7 +21,7 @@ func userCreate() *cobra.Command {
 	cmd := &cobra.Command{
 		Use: "create",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/userlist.go b/cli/userlist.go
@@ -26,7 +26,7 @@ func userList() *cobra.Command {
 		Use:     "list",
 		Aliases: []string{"ls"},
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
@@ -76,7 +76,7 @@ func userSingle() *cobra.Command {
 		),
 		Args: cobra.ExactArgs(1),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/userstatus.go b/cli/userstatus.go
@@ -43,7 +43,7 @@ func createUserStatusCommand(sdkStatus codersdk.UserStatus) *cobra.Command {
 			},
 		),
 		RunE: func(cmd *cobra.Command, args []string) error {
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/cli/wireguardtunnel.go b/cli/wireguardtunnel.go
@@ -67,7 +67,7 @@ func wireguardPortForward() *cobra.Command {
 				return xerrors.New("no port-forwards requested")
 			}
 
-			client, err := createClient(cmd)
+			client, err := CreateClient(cmd)
 			if err != nil {
 				return err
 			}
diff --git a/enterprise/cli/licenses.go b/enterprise/cli/licenses.go
diff --git a/enterprise/cli/licenses_test.go b/enterprise/cli/licenses_test.go
diff --git a/enterprise/cli/root.go b/enterprise/cli/root.go

Original file line number	Diff line number	Diff line change
`@@ -158,7 +158,7 @@ func configSSH() *cobra.Command {`
`158`	`158`	`),`
`159`	`159`	`Args: cobra.ExactArgs(0),`
`160`	`160`	`RunE: func(cmd *cobra.Command, _ []string) error {`
`161`		`- client, err := createClient(cmd)`
	`161`	`+ client, err := CreateClient(cmd)`
`162`	`162`	`if err != nil {`
`163`	`163`	`return err`
`164`	`164`	`}`
Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ func create() *cobra.Command {`
`27`	`27`	`Use: "create [name]",`
`28`	`28`	`Short: "Create a workspace from a template",`
`29`	`29`	`RunE: func(cmd *cobra.Command, args []string) error {`
`30`		`- client, err := createClient(cmd)`
	`30`	`+ client, err := CreateClient(cmd)`
`31`	`31`	`if err != nil {`
`32`	`32`	`return err`
`33`	`33`	`}`
Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ func deleteWorkspace() *cobra.Command {`
`28`	`28`	`return err`
`29`	`29`	`}`
`30`	`30`
`31`		`- client, err := createClient(cmd)`
	`31`	`+ client, err := CreateClient(cmd)`
`32`	`32`	`if err != nil {`
`33`	`33`	`return err`
`34`	`34`	`}`
Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,7 @@ func featuresList() *cobra.Command {`
`36`	`36`	`Use: "list",`
`37`	`37`	`Aliases: []string{"ls"},`
`38`	`38`	`RunE: func(cmd *cobra.Command, args []string) error {`
`39`		`- client, err := createClient(cmd)`
	`39`	`+ client, err := CreateClient(cmd)`
`40`	`40`	`if err != nil {`
`41`	`41`	`return err`
`42`	`42`	`}`
Original file line number	Diff line number	Diff line change
`@@ -65,7 +65,7 @@ func list() *cobra.Command {`
`65`	`65`	`Aliases: []string{"ls"},`
`66`	`66`	`Args: cobra.ExactArgs(0),`
`67`	`67`	`RunE: func(cmd *cobra.Command, args []string) error {`
`68`		`- client, err := createClient(cmd)`
	`68`	`+ client, err := CreateClient(cmd)`
`69`	`69`	`if err != nil {`
`70`	`70`	`return err`
`71`	`71`	`}`
Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,7 @@ func logout() *cobra.Command {`
`16`	`16`	`Use: "logout",`
`17`	`17`	`Short: "Remove the local authenticated session",`
`18`	`18`	`RunE: func(cmd *cobra.Command, args []string) error {`
`19`		`- client, err := createClient(cmd)`
	`19`	`+ client, err := CreateClient(cmd)`
`20`	`20`	`if err != nil {`
`21`	`21`	`return err`
`22`	`22`	`}`
Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ func parameterList() *cobra.Command {`
`22`	`22`	`RunE: func(cmd *cobra.Command, args []string) error {`
`23`	`23`	`scope, name := args[0], args[1]`
`24`	`24`
`25`		`- client, err := createClient(cmd)`
	`25`	`+ client, err := CreateClient(cmd)`
`26`	`26`	`if err != nil {`
`27`	`27`	`return err`
`28`	`28`	`}`
Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ func portForward() *cobra.Command {`
`70`	`70`	`return xerrors.New("no port-forwards requested")`
`71`	`71`	`}`
`72`	`72`
`73`		`- client, err := createClient(cmd)`
	`73`	`+ client, err := CreateClient(cmd)`
`74`	`74`	`if err != nil {`
`75`	`75`	`return err`
`76`	`76`	`}`
Original file line number	Diff line number	Diff line change
`@@ -20,7 +20,7 @@ func publickey() *cobra.Command {`
`20`	`20`	`Aliases: []string{"pubkey"},`
`21`	`21`	`Short: "Output your public key for Git operations",`
`22`	`22`	`RunE: func(cmd *cobra.Command, args []string) error {`
`23`		`- client, err := createClient(cmd)`
	`23`	`+ client, err := CreateClient(cmd)`
`24`	`24`	`if err != nil {`
`25`	`25`	`return xerrors.Errorf("create codersdk client: %w", err)`
`26`	`26`	`}`