From 2fc03da7ea3a87b77111ebc598e248b863fbd600 Mon Sep 17 00:00:00 2001
From: Martin Englund <martin@englund.nu>
Date: Sun, 8 Oct 2017 10:16:08 -0700
Subject: [PATCH] add support for displaying aws profile

---
 aws.go      | 17 +++++++++++++++++
 defaults.go |  3 +++
 main.go     |  5 +++--
 themes.go   |  3 +++
 4 files changed, 26 insertions(+), 2 deletions(-)
 create mode 100644 aws.go

diff --git a/aws.go b/aws.go
new file mode 100644
index 00000000..7a1b4e1e
--- /dev/null
+++ b/aws.go
@@ -0,0 +1,17 @@
+package main
+
+import (
+	"fmt"
+	"os"
+)
+
+func segmentAWS(p *powerline) {
+	profile, _ := os.LookupEnv("AWS_PROFILE")
+	if profile != "" {
+		p.appendSegment("aws", segment{
+			content:    fmt.Sprintf(" %s ", profile),
+			foreground: p.theme.AWSFg,
+			background: p.theme.AWSBg,
+		})
+	}
+}
diff --git a/defaults.go b/defaults.go
index bc2fc7a6..4357e091 100644
--- a/defaults.go
+++ b/defaults.go
@@ -95,6 +95,9 @@ var themes = map[string]Theme{
 		DotEnvFg: 15, // white
 		DotEnvBg: 55, // purple
 
+		AWSFg: 15, // white
+		AWSBg: 22, // dark green
+
 		RepoCleanFg: 0,   // black
 		RepoCleanBg: 148, // a light green color
 		RepoDirtyFg: 15,  // white
diff --git a/main.go b/main.go
index dea32837..807f7246 100644
--- a/main.go
+++ b/main.go
@@ -76,6 +76,7 @@ func getValidCwd() string {
 }
 
 var modules = map[string](func(*powerline)){
+	"aws":      segmentAWS,
 	"cwd":      segmentCwd,
 	"docker":   segmentDocker,
 	"dotenv":   segmentDotEnv,
@@ -127,12 +128,12 @@ func main() {
 		Modules: flag.String("modules",
 			"venv,user,host,ssh,cwd,perms,git,hg,jobs,exit,root",
 			"The list of modules to load, separated by ','\n"+
-				"    	(valid choices: cwd, docker, dotenv, exit, git, gitlite, hg, host, jobs, perlbrew, perms, root, ssh, time, user, venv)\n"+
+				"    	(valid choices: aws, cwd, docker, dotenv, exit, git, gitlite, hg, host, jobs, perlbrew, perms, root, ssh, time, user, venv)\n"+
 				"       "),
 		Priority: flag.String("priority",
 			"root,cwd,user,host,ssh,perms,git-branch,git-status,hg,jobs,exit",
 			"Segments sorted by priority, if not enough space exists, the least priorized segments are removed first. Separate with ','\n"+
-				"    	(valid choices: cwd, docker, exit, git-branch, git-status, hg, host, jobs, perlbrew, perms, root, ssh, time, user, venv)\n"+
+				"    	(valid choices: aws, cwd, docker, exit, git-branch, git-status, hg, host, jobs, perlbrew, perms, root, ssh, time, user, venv)\n"+
 				"       "),
 		MaxWidthPercentage: flag.Int("max-width",
 			50,
diff --git a/themes.go b/themes.go
index 08f1f06f..546009cb 100644
--- a/themes.go
+++ b/themes.go
@@ -48,6 +48,9 @@ type Theme struct {
 	DotEnvFg uint8
 	DotEnvBg uint8
 
+	AWSFg uint8
+	AWSBg uint8
+
 	RepoCleanFg uint8
 	RepoCleanBg uint8
 	RepoDirtyFg uint8