fixup! feat(virtualenv): look up VIRTUAL_ENV_PROMPT first

Co-authored-by: Janne Mareike Koschinski <[email protected]>
justjanne · Jun 17, 2024 · 3241c1b · 3241c1b
1 parent a307d69
commit 3241c1b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/segment-virtualenv.go b/segment-virtualenv.go
@@ -13,7 +13,8 @@ import (
 func segmentVirtualEnv(p *powerline) []pwl.Segment {
 	env := os.Getenv("VIRTUAL_ENV_PROMPT")
 	if strings.HasPrefix(env, "(") && strings.HasSuffix(env, ") ") {
-		env = env[1 : len(env)-2]
+		env = strings.TrimPrefix(env, "(")
+		env = strings.TrimSuffix(env, ") ")
 	}
 	if env == "" {
 		venv, _ := os.LookupEnv("VIRTUAL_ENV")