Merge pull request #15 from cdr/envs

Anthony Shull · web-flow · commit a02fdfef40d7 · 2020-05-12T16:16:17.000-05:00
Added ability to get all active environments
diff --git a/README.md b/README.md
@@ -29,6 +29,14 @@ $ coder sync ~/Projects/cdr/enterprise/. my-env:~/enterprise
 You can access your environment's terminal with `coder sh <env>`. You can also
 execute a command in your environment with `coder sh <env> [command] [args]`.
 
+## Environments
+
+List all active environments for the user.
+
+``
+$ coder envs
+``
+
 ## Development URLs
 
 You can retrieve the devurl of an environment.
diff --git a/cmd/coder/ceapi.go b/cmd/coder/ceapi.go
@@ -24,7 +24,8 @@ outer:
 	return uo
 }
 
-func findEnv(client *entclient.Client, name string) entclient.Environment {
+// getEnvs returns all environments for the user.
+func getEnvs(client *entclient.Client) []entclient.Environment {
 	me, err := client.Me()
 	if err != nil {
 		flog.Fatal("get self: %+v", err)
@@ -37,21 +38,34 @@ func findEnv(client *entclient.Client, name string) entclient.Environment {
 
 	orgs = userOrgs(me, orgs)
 
-	var found []string
+	var allEnvs []entclient.Environment
 
 	for _, org := range orgs {
 		envs, err := client.Envs(me, org)
 		if err != nil {
 			flog.Fatal("get envs for %v: %+v", org.Name, err)
 		}
 		for _, env := range envs {
-			found = append(found, env.Name)
-			if env.Name != name {
-				continue
-			}
+			allEnvs = append(allEnvs, env)
+		}
+	}
+
+	return allEnvs
+}
+
+// findEnv returns a single environment by name (if it exists.)
+func findEnv(client *entclient.Client, name string) entclient.Environment {
+	envs := getEnvs(client)
+
+	var found []string
+
+	for _, env := range envs {
+		found = append(found, env.Name)
+		if env.Name == name {
 			return env
 		}
 	}
+
 	flog.Info("found %q", found)
 	flog.Fatal("environment %q not found", name)
 	panic("unreachable")
diff --git a/cmd/coder/envs.go b/cmd/coder/envs.go
@@ -0,0 +1,27 @@
+package main
+
+import (
+	"fmt"
+	"github.com/spf13/pflag"
+	"go.coder.com/cli"
+)
+
+type envsCmd struct {
+}
+
+func (cmd envsCmd) Spec() cli.CommandSpec {
+	return cli.CommandSpec{
+		Name:  "envs",
+		Desc:  "get a list of active environment",
+	}
+}
+
+func (cmd envsCmd) Run(fl *pflag.FlagSet) {
+	entClient := requireAuth()
+
+	envs := getEnvs(entClient)
+
+	for _, env := range envs {
+		fmt.Println(env.Name)
+	}
+}
diff --git a/cmd/coder/main.go b/cmd/coder/main.go
@@ -26,6 +26,7 @@ func (r *rootCmd) Spec() cli.CommandSpec {
 
 func (r *rootCmd) Subcommands() []cli.Command {
 	return []cli.Command{
+		&envsCmd{},
 		&loginCmd{},
 		&logoutCmd{},
 		&shellCmd{},
