fix(update): define $ZSH if undefined (ohmyzsh#12273)

Fixes ohmyzsh#12273 Fixes topgrade-rs/topgrade#519
Kristijan · Apr 18, 2024 · c262ffb · c262ffb
1 parent eafa782
commit c262ffb
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tools/upgrade.sh b/tools/upgrade.sh
@@ -10,9 +10,14 @@ fi
 
 # Protect against unwanted sourcing
 case "$ZSH_EVAL_CONTEXT" in
-  *:file) echo "error: this file should not be sourced" && return ;;
+  *:file) echo "error: this file should not be sourced" && return 1 ;;
 esac
 
+# Define "$ZSH" if not defined -- in theory this should be `export`ed by the calling script
+if [[ -z "$ZSH" ]]; then
+  ZSH="${0:a:h:h}"
+fi
+
 cd "$ZSH"
 
 verbose_mode="default"