From 297cc97cdaa9dcf5b15d24a6aa3abee4be5c1e42 Mon Sep 17 00:00:00 2001
From: Lucas Bremgartner <lucas@bremis.ch>
Date: Thu, 10 May 2018 20:33:20 +0200
Subject: [PATCH] Allow .envrc for dotenv segment

[direnv] uses `.envrc` as filename.

[direnv]: https://github.com/direnv/direnv
---
 segment-dotenv.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/segment-dotenv.go b/segment-dotenv.go
index c465910e..49dc8fc4 100644
--- 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,