Add shenv support

https://github.com/shenv/shenv

README.md

@@ -185,11 +185,11 @@ Usage of ./powerline-go:
          (default "patched")
   -modules string
          The list of modules to load, separated by ','
-         (valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)
+         (valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)
          (default "venv,user,host,ssh,cwd,perms,git,hg,jobs,exit,root")
   -modules-right string
          The list of modules to load anchored to the right, for shells that support it, separated by ','
-         (valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)
+         (valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)
   -newline
          Show the prompt on a new line
   -numeric-exit-codes
@@ -201,7 +201,7 @@ Usage of ./powerline-go:
          Use '~' for your home dir. You may need to escape this character to avoid shell substitution.
   -priority string
          Segments sorted by priority, if not enough space exists, the least priorized segments are removed first. Separate with ','
-         (valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)
+         (valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)
          (default "root,cwd,user,host,ssh,perms,git-branch,git-status,hg,jobs,exit,cwd-path")
   -shell string
          Set this to your shell type

defaults.go

@@ -173,6 +173,9 @@ var themes = map[string]Theme{
 		ShellVarFg: 52,
 		ShellVarBg: 11,
 
+		ShEnvFg: 15,
+		ShEnvBg: 130,
+
 		NodeFg: 15,
 		NodeBg: 40,
 
@@ -533,6 +536,9 @@ var themes = map[string]Theme{
 		TimeFg: 236,
 		TimeBg: 15,
 
+		ShEnvFg: 130,
+		ShEnvBg: 15,
+
 		LoadFg:           15,
 		LoadBg:           22,
 		LoadHighBg:       161,
@@ -861,6 +867,8 @@ var themes = map[string]Theme{
 		TimeBg:             0,
 		ShellVarFg:         1,
 		ShellVarBg:         11,
+		ShEnvFg:            15,
+		ShEnvBg:            9,
 		NodeFg:             15,
 		NodeBg:             40,
 		LoadFg:             15,
@@ -1189,6 +1197,8 @@ var themes = map[string]Theme{
 		TimeBg:             0,
 		ShellVarFg:         1,
 		ShellVarBg:         11,
+		ShEnvFg:            15,
+		ShEnvBg:            9,
 		NodeFg:             15,
 		NodeBg:             40,
 		LoadFg:             15,

main.go

@@ -109,6 +109,7 @@ var modules = map[string]func(*powerline){
 	"perms":               segmentPerms,
 	"root":                segmentRoot,
 	"shell-var":           segmentShellVar,
+	"shenv":               segmentShEnv,
 	"ssh":                 segmentSSH,
 	"termtitle":           segmentTermTitle,
 	"terraform-workspace": segmentTerraformWorkspace,
@@ -178,17 +179,17 @@ func main() {
 			"modules",
 			"venv,user,host,ssh,cwd,perms,git,hg,jobs,exit,root",
 			commentsWithDefaults("The list of modules to load, separated by ','",
-				"(valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)")),
+				"(valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)")),
 		ModulesRight: flag.String(
 			"modules-right",
 			"",
 			comments("The list of modules to load anchored to the right, for shells that support it, separated by ','",
-				"(valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)")),
+				"(valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)")),
 		Priority: flag.String(
 			"priority",
 			"root,cwd,user,host,ssh,perms,git-branch,git-status,hg,jobs,exit,cwd-path",
 			commentsWithDefaults("Segments sorted by priority, if not enough space exists, the least priorized segments are removed first. Separate with ','",
-				"(valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)")),
+				"(valid choices: aws, cwd, docker, dotenv, duration, exit, git, gitlite, hg, host, jobs, kube, load, newline, nix-shell, node, perlbrew, perms, plenv, root, shell-var, shenv, ssh, svn, termtitle, terraform-workspace, time, user, venv, vgo)")),
 		MaxWidthPercentage: flag.Int(
 			"max-width",
 			0,

segment-shenv.go

@@ -0,0 +1,23 @@
+package main
+
+import (
+	pwl "github.com/justjanne/powerline-go/powerline"
+	"os"
+	"path"
+)
+
+func segmentShEnv(p *powerline) {
+	var env string
+	if env == "" {
+		env, _ = os.LookupEnv("SHENV_VERSION")
+	}
+	if env == "" {
+		return
+	}
+	envName := path.Base(env)
+	p.appendSegment("shenv", pwl.Segment{
+		Content:    envName,
+		Foreground: p.theme.ShEnvFg,
+		Background: p.theme.ShEnvBg,
+	})
+}

themes.go

@@ -119,6 +119,9 @@ type Theme struct {
 	ShellVarFg uint8
 	ShellVarBg uint8
 
+	ShEnvFg uint8
+	ShEnvBg uint8
+
 	NodeFg uint8
 	NodeBg uint8
 

themes/default.json

@@ -58,6 +58,8 @@
   "TimeBg": 236,
   "ShellVarFg": 52,
   "ShellVarBg": 11,
+  "ShEnvFg": 15,
+  "ShEnvBg": 130,
   "NodeFg": 15,
   "NodeBg": 40,
   "LoadFg": 15,