Added command "prompt"

README.md

@@ -124,6 +124,10 @@ AWS_ACCESS_KEY_ID=myaccesskey
 AWS_ACCESS_KEY=myaccesskey
 AWS_SECRET_KEY=mysecretkey
 
+# Now the prompt command can tell you which env is set
+$ awsenv prompt
+demoenv
+
 # Lets try to unlock with a wrong password
 $ awsenv lock
 $ awsenv unlock

cmdPrompt.go

@@ -0,0 +1,25 @@
+package main
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+)
+
+func getCmdPrompt() *cobra.Command {
+	cmd := cobra.Command{
+		Use:   "prompt",
+		Short: "echos the name of the currently set env for use in prompts",
+		Run:   actionCmdPrompt,
+	}
+	return &cmd
+}
+
+func actionCmdPrompt(cmd *cobra.Command, args []string) {
+	for k, v := range awsCredentials.Credentials {
+		if v.AWSAccessKeyID == os.Getenv("AWS_ACCESS_KEY") {
+			fmt.Printf(k)
+		}
+	}
+}

main.go

@@ -94,6 +94,7 @@ func main() {
 		getCmdList(),
 		getCmdLock(),
 		getCmdShell(),
+		getCmdPrompt(),
 		getCmdUnlock(),
 		getCmdVersion(),
 	)