Added module for docker machine names

defaults.go

@@ -25,24 +25,24 @@ var symbolTemplates = map[string]Symbols{
 	},
 }
 
-var shellInfos = map[string]ShellInfo {
+var shellInfos = map[string]ShellInfo{
 	"bash": {
-		colorTemplate: "\\[\\e%s\\]",
-		rootIndicator: " \\$ ",
+		colorTemplate:    "\\[\\e%s\\]",
+		rootIndicator:    " \\$ ",
 		escapedBackslash: `\\\\`,
-		escapedDollar: `\$`,
+		escapedDollar:    `\$`,
 	},
 	"zsh": {
-		colorTemplate: "%%{\u001b%s%%}",
-		rootIndicator: " %# ",
+		colorTemplate:    "%%{\u001b%s%%}",
+		rootIndicator:    " %# ",
 		escapedBackslash: `\\`,
-		escapedDollar: `\$`,
+		escapedDollar:    `\$`,
 	},
 	"bare": {
-		colorTemplate: "%s",
-		rootIndicator: " $ ",
+		colorTemplate:    "%s",
+		rootIndicator:    " $ ",
 		escapedBackslash: `\`,
-		escapedDollar: `$`,
+		escapedDollar:    `$`,
 	},
 }
 
@@ -70,6 +70,9 @@ var defaultTheme = Theme{
 	SshFg: 254,
 	SshBg: 166, // medium orange
 
+	DockerMachineFg: 177, // light purple
+	DockerMachineBg: 55,  // purple
+
 	RepoCleanFg: 0,   // black
 	RepoCleanBg: 148, // a light green color
 	RepoDirtyFg: 15,  // white

main.go

@@ -63,6 +63,7 @@ func getValidCwd() string {
 
 var modules = map[string](func(*powerline)){
 	"cwd":   segmentCwd,
+	"docker": segmentDocker,
 	"exit":  segmentExitCode,
 	"git":   segmentGit,
 	"hg":    segmentHg,
@@ -101,7 +102,7 @@ func main() {
 		Modules: flag.String("modules",
 			"venv,user,ssh,host,cwd,perms,git,hg,jobs,exit,root",
 			"The list of modules to load. Separate with ','\n"+
-				"    	(valid choices: cwd, exit, git, hg, host, jobs, perms, root, ssh, user, venv)\n"+
+				"    	(valid choices: cwd, docker, exit, git, hg, host, jobs, perms, root, ssh, user, venv)\n"+
 				"       "),
 		PrevError: flag.Int("error", 0,
 			"Exit code of previously executed command"),

segment-docker.go

@@ -0,0 +1,17 @@
+package main
+
+import (
+	"os"
+	"fmt"
+)
+
+func segmentDocker(p *powerline) {
+	dockerMachineName, _ := os.LookupEnv("DOCKER_MACHINE_NAME")
+	if dockerMachineName != "" {
+		p.appendSegment(segment{
+			content:    fmt.Sprintf(" %s ", dockerMachineName),
+			foreground: p.theme.DockerMachineFg,
+			background: p.theme.DockerMachineBg,
+		})
+	}
+}

themes.go

@@ -38,6 +38,9 @@ type Theme struct {
 	SshFg uint8
 	SshBg uint8
 
+	DockerMachineFg uint8
+	DockerMachineBg uint8
+
 	RepoCleanFg uint8
 	RepoCleanBg uint8
 	RepoDirtyFg uint8