Add coder envs rm command for removing environments

cmoog · cmoog · commit 0d0c495e104b · 2020-10-23T10:29:06.000-05:00
diff --git a/docs/coder_envs.md b/docs/coder_envs.md
@@ -23,5 +23,6 @@ Perform operations on the Coder environments owned by the active user.
 
 * [coder](coder.md)	 - coder provides a CLI for working with an existing Coder Enterprise installation
 * [coder envs ls](coder_envs_ls.md)	 - list all environments owned by the active user
+* [coder envs rm](coder_envs_rm.md)	 - remove Coder environments by name
 * [coder envs stop](coder_envs_stop.md)	 - stop Coder environments by name
 
diff --git a/docs/coder_envs_rm.md b/docs/coder_envs_rm.md
@@ -0,0 +1,29 @@
+## coder envs rm
+
+remove Coder environments by name
+
+### Synopsis
+
+remove Coder environments by name
+
+```
+coder envs rm [...environment_names] [flags]
+```
+
+### Options
+
+```
+  -f, --force   force remove the specified environments without prompting first
+  -h, --help    help for rm
+```
+
+### Options inherited from parent commands
+
+```
+      --user string   Specify the user whose resources to target (default "me")
+  -v, --verbose       show verbose output
+```
+
+### SEE ALSO
+
+* [coder envs](coder_envs.md)	 - Interact with Coder environments
diff --git a/internal/cmd/envs.go b/internal/cmd/envs.go
@@ -9,8 +9,10 @@ import (
 	"cdr.dev/coder-cli/coder-sdk"
 	"cdr.dev/coder-cli/internal/clog"
 	"cdr.dev/coder-cli/internal/x/xtabwriter"
+	"github.com/manifoldco/promptui"
 	"github.com/spf13/cobra"
 	"golang.org/x/sync/errgroup"
+	"golang.org/x/sync/semaphore"
 	"golang.org/x/xerrors"
 )
 
@@ -24,7 +26,6 @@ const (
 )
 
 func envsCommand() *cobra.Command {
-	var outputFmt string
 	var user string
 	cmd := &cobra.Command{
 		Use:   "envs",
@@ -33,7 +34,22 @@ func envsCommand() *cobra.Command {
 	}
 	cmd.PersistentFlags().StringVar(&user, "user", coder.Me, "Specify the user whose resources to target")
 
-	lsCmd := &cobra.Command{
+	cmd.AddCommand(
+		lsEnvsCommand(&user),
+		stopEnvsCommand(&user),
+		rmEnvsCommand(&user),
+		watchBuildLogCommand(),
+		rebuildEnvCommand(),
+		createEnvCommand(),
+		editEnvCommand(&user),
+	)
+	return cmd
+}
+
+func lsEnvsCommand(user *string) *cobra.Command {
+	var outputFmt string
+
+	cmd := &cobra.Command{
 		Use:   "ls",
 		Short: "list all environments owned by the active user",
 		Long:  "List all Coder environments owned by the active user.",
@@ -42,7 +58,7 @@ func envsCommand() *cobra.Command {
 			if err != nil {
 				return err
 			}
-			envs, err := getEnvs(cmd.Context(), client, user)
+			envs, err := getEnvs(cmd.Context(), client, *user)
 			if err != nil {
 				return err
 			}
@@ -70,17 +86,13 @@ func envsCommand() *cobra.Command {
 			return nil
 		},
 	}
-	lsCmd.Flags().StringVarP(&outputFmt, "output", "o", "human", "human | json")
-	cmd.AddCommand(lsCmd)
-	cmd.AddCommand(editEnvCommand(&user))
-	cmd.AddCommand(stopEnvCommand(&user))
-	cmd.AddCommand(watchBuildLogCommand())
-	cmd.AddCommand(rebuildEnvCommand())
-	cmd.AddCommand(createEnvCommand())
+
+	cmd.Flags().StringVarP(&outputFmt, "output", "o", "human", "human | json")
+
 	return cmd
 }
 
-func stopEnvCommand(user *string) *cobra.Command {
+func stopEnvsCommand(user *string) *cobra.Command {
 	return &cobra.Command{
 		Use:   "stop [...environment_names]",
 		Short: "stop Coder environments by name",
@@ -310,3 +322,62 @@ coder envs edit back-end-env --disk 20`,
 	cmd.Flags().BoolVar(&follow, "follow", false, "follow buildlog after initiating rebuild")
 	return cmd
 }
+
+func rmEnvsCommand(user *string) *cobra.Command {
+	var force bool
+	cmd := &cobra.Command{
+		Use:   "rm [...environment_names]",
+		Short: "remove Coder environments by name",
+		Args:  cobra.MinimumNArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx := cmd.Context()
+			client, err := newClient()
+			if err != nil {
+				return err
+			}
+
+			// only show one confirmation dialogue at a time
+			confirmLimiter := semaphore.NewWeighted(1)
+
+			var egroup errgroup.Group
+			for _, envName := range args {
+				envName := envName
+				egroup.Go(func() error {
+					env, err := findEnv(ctx, client, envName, *user)
+					if err != nil {
+						clog.Log(err)
+						return err
+					}
+					if !force {
+						confirm := promptui.Prompt{
+							Label:     fmt.Sprintf("Delete environment \"%s\"? (all data will be lost)", env.Name),
+							IsConfirm: true,
+						}
+						if err := confirmLimiter.Acquire(ctx, 1); err != nil {
+							return err
+						}
+
+						if _, err = confirm.Run(); err != nil {
+							confirmLimiter.Release(1)
+							return xerrors.Errorf("confirm deletion of environment \"%s\"", env.Name)
+						}
+						confirmLimiter.Release(1)
+					}
+					if err = client.DeleteEnvironment(cmd.Context(), env.ID); err != nil {
+						err = clog.Error(fmt.Sprintf(`failed to delete environment "%s"`, env.Name), clog.Cause(err.Error()))
+						clog.Log(err)
+						return err
+					}
+					return nil
+				})
+			}
+
+			if err = egroup.Wait(); err != nil {
+				return xerrors.Errorf("some environment remove operations failed")
+			}
+			return nil
+		},
+	}
+	cmd.Flags().BoolVarP(&force, "force", "f", false, "force remove the specified environments without prompting first")
+	return cmd
+}
