Merge pull request #351 from scottslowe/issue-350

Add support for DOCKER_CONTEXT environment variable
justjanne · Apr 19, 2022 · 400f147 · 400f147
2 parents b8eeb5a + 4964354
commit 400f147
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/segment-docker_context.go b/segment-docker_context.go
@@ -17,15 +17,20 @@ func segmentDockerContext(p *powerline) []pwl.Segment {
 	home, _ := os.LookupEnv("HOME")
 	contextFolder := filepath.Join(home, ".docker", "contexts")
 	configFile := filepath.Join(home, ".docker", "config.json")
+	contextEnvVar := os.Getenv("DOCKER_CONTEXT")
 
-	stat, err := os.Stat(contextFolder)
-	if err == nil && stat.IsDir() {
-		dockerConfigFile, err := ioutil.ReadFile(configFile)
-		if err == nil {
-			var dockerConfig DockerContextConfig
-			err = json.Unmarshal(dockerConfigFile, &dockerConfig)
-			if err == nil && dockerConfig.CurrentContext != "" {
-				context = dockerConfig.CurrentContext
+	if contextEnvVar != "" {
+		context = contextEnvVar
+	} else {
+		stat, err := os.Stat(contextFolder)
+		if err == nil && stat.IsDir() {
+			dockerConfigFile, err := ioutil.ReadFile(configFile)
+			if err == nil {
+				var dockerConfig DockerContextConfig
+				err = json.Unmarshal(dockerConfigFile, &dockerConfig)
+				if err == nil && dockerConfig.CurrentContext != "" {
+					context = dockerConfig.CurrentContext
+				}
 			}
 		}
 	}