add a list command (#67)

* add a `list` command
segmentio · Jan 18, 2019 · ab9818e · ab9818e
1 parent 270de5b
commit ab9818e
Show file tree

Hide file tree

Showing 2 changed files with 69 additions and 1 deletion.
diff --git a/cmd/exec.go b/cmd/exec.go
@@ -113,7 +113,7 @@ func execRun(cmd *cobra.Command, args []string) error {
 	}
 
 	if _, ok := profiles[profile]; !ok {
-		return fmt.Errorf("Profile '%s' not found in your aws config", profile)
+		return fmt.Errorf("Profile '%s' not found in your aws config. Use list command to see configured profiles.", profile)
 	}
 
 	// check for an assume_role_ttl in the profile if we don't have a more explicit one

diff --git a/cmd/list.go b/cmd/list.go
@@ -0,0 +1,68 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+	"sort"
+	"text/tabwriter"
+
+	analytics "github.com/segmentio/analytics-go"
+	"github.com/segmentio/aws-okta/lib"
+	"github.com/spf13/cobra"
+)
+
+// listCmd represents the list command
+var listCmd = &cobra.Command{
+	Use:   "list",
+	Short: "list will show you the profiles currently configured",
+	RunE:  listRun,
+}
+
+func init() {
+	RootCmd.AddCommand(listCmd)
+}
+
+func listRun(cmd *cobra.Command, args []string) error {
+	config, err := lib.NewConfigFromEnv()
+	if err != nil {
+		return err
+	}
+
+	profiles, err := config.Parse()
+	if err != nil {
+		return err
+	}
+
+	// Let's sort this list of profiles so we can have some more deterministic output:
+	var profileNames []string
+
+	for profile := range profiles {
+		profileNames = append(profileNames, profile)
+	}
+
+	sort.Strings(profileNames)
+
+	w := new(tabwriter.Writer)
+	w.Init(os.Stdout, 0, 8, 2, '\t', 0)
+	fmt.Fprintln(w, "PROFILE\tARN\tSOURCE_ROLE\t")
+	for _, profile := range profileNames {
+		v := profiles[profile]
+		if role, exist := v["role_arn"]; exist {
+			fmt.Fprintf(w, "%s\t%s\t%s\n", profile, role, v["source_profile"])
+		}
+	}
+	w.Flush()
+
+	if analyticsEnabled && analyticsClient != nil {
+		analyticsClient.Enqueue(analytics.Track{
+			UserId: username,
+			Event:  "Listed Profiles",
+			Properties: analytics.NewProperties().
+				Set("backend", backend).
+				Set("aws-okta-version", version).
+				Set("profile-count", len(profiles)).
+				Set("command", "list"),
+		})
+	}
+	return nil
+}