Merge pull request #82 from breml/envrc

Allow .envrc for dotenv segment
justjanne · May 11, 2018 · c934340 · c934340
2 parents 2d68816 + 297cc97
commit c934340
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/segment-dotenv.go b/segment-dotenv.go
@@ -5,8 +5,16 @@ import (
 )
 
 func segmentDotEnv(p *powerline) {
-	stat, err := os.Stat(".env")
-	if err == nil && !stat.IsDir() {
+	files := []string{".env", ".envrc"}
+	dotEnv := false
+	for _, file := range files {
+		stat, err := os.Stat(file)
+		if err == nil && !stat.IsDir() {
+			dotEnv = true
+			break
+		}
+	}
+	if dotEnv {
 		p.appendSegment("dotenv", segment{
 			content:    "\u2235",
 			foreground: p.theme.DotEnvFg,