From 98816b5c66103abeb5617770799922a07e362651 Mon Sep 17 00:00:00 2001
From: Knut Ahlers <knut@ahlers.me>
Date: Tue, 23 Jun 2015 17:47:43 +0200
Subject: [PATCH] Added command "prompt"

---
 README.md    |  4 ++++
 cmdPrompt.go | 25 +++++++++++++++++++++++++
 main.go      |  1 +
 3 files changed, 30 insertions(+)
 create mode 100644 cmdPrompt.go

diff --git a/README.md b/README.md
index 1b65b1d..e09f3d2 100644
--- a/README.md
+++ b/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
diff --git a/cmdPrompt.go b/cmdPrompt.go
new file mode 100644
index 0000000..0faa5af
--- /dev/null
+++ b/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)
+		}
+	}
+}
diff --git a/main.go b/main.go
index 9f77d0a..6eaa5a1 100644
--- a/main.go
+++ b/main.go
@@ -94,6 +94,7 @@ func main() {
 		getCmdList(),
 		getCmdLock(),
 		getCmdShell(),
+		getCmdPrompt(),
 		getCmdUnlock(),
 		getCmdVersion(),
 	)