From ed16839098d615c977a139e089e4aa354176a94d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C8=98tefan=20Talpalaru?= <stefantalpalaru@yahoo.com>
Date: Fri, 23 Aug 2019 20:24:45 +0200
Subject: [PATCH] env.sh: Zsh support

---
 scripts/env.sh | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/scripts/env.sh b/scripts/env.sh
index 6b80e32..d8aae7a 100755
--- a/scripts/env.sh
+++ b/scripts/env.sh
@@ -11,7 +11,9 @@
 ## get native Windows paths on Mingw
 #uname | grep -qi mingw && PWD_CMD="pwd -W"
 
-export REL_PATH="$(dirname $BASH_SOURCE)"
+# We use ${BASH_SOURCE[0]} instead of $0 to allow sourcing this file
+# and we fall back to a zsh-specific special var to also support zsh.
+export REL_PATH="$(dirname ${BASH_SOURCE[0]:-${(%):-%x}})"
 export ABS_PATH="$(cd ${REL_PATH}; pwd)"
 # do we still need this?
 #ABS_PATH_NATIVE="$(cd ${REL_PATH}; ${PWD_CMD})"
@@ -33,19 +35,26 @@ export NIMBLE_DIR="${ABS_PATH}/../../.nimble"
 export BUILD_OUTPUTS_DIR="${ABS_PATH}/../../../build"
 
 # change the prompt in shells that source this file
-export PS1="${PS1%\\\$ } [Nimbus env]\\$ "
+if [[ -n "$BASH_VERSION" ]]; then
+	export PS1="${PS1%\\\$ } [Nimbus env]\\$ "
+	EXPORT_FUNC="export -f"
+fi
+if [[ -n "$ZSH_VERSION" ]]; then
+	export PS1="[Nimbus env] $PS1"
+	EXPORT_FUNC="export" # doesn't actually work, because Zsh doesn't support exporting functions
+fi
 
 # functions, instead of aliases, to avoid typing long paths (aliases don't seem
 # to be expanded by default for command line arguments)
 nimble() {
 	"${ABS_PATH}/nimble.sh" "$@"
 }
-export -f nimble
+$EXPORT_FUNC nimble
 
 add_submodule() {
 	"${ABS_PATH}/add_submodule.sh" "$@"
 }
-export -f add_submodule
+$EXPORT_FUNC add_submodule
 
 if [[ $# == 1 && $1 == "bash" ]]; then
 	# the only way to change PS1 in a child shell, apparently