From ffbbce9557a204917e47450f6464858dbb21e6a8 Mon Sep 17 00:00:00 2001
From: Nic Patterson <arasureynn@gmail.com>
Date: Thu, 23 May 2019 22:28:55 -0700
Subject: [PATCH] allow go_version prompt to work with go modules

---
 segments/go_version/go_version.p9k | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/segments/go_version/go_version.p9k b/segments/go_version/go_version.p9k
index ea163adf2..cc278835e 100644
--- a/segments/go_version/go_version.p9k
+++ b/segments/go_version/go_version.p9k
@@ -30,7 +30,7 @@ prompt_go_version() {
   go_version=$(go version 2>/dev/null | sed -E "s/.*(go[0-9.]*).*/\1/")
   go_path=$(go env GOPATH 2>/dev/null)
 
-  if [[ -n "$go_version" && "${PWD##$go_path}" != "$PWD" ]]; then
+  if [[ -n "$go_version" && ("${PWD##$go_path}" != "$PWD" || -f "$PWD/go.mod") ]]; then
     p9k::prepare_segment "$0" "" $1 "$2" $3 "$go_version"
   fi
 }