Merge pull request #42 from ColinNg/master

Fix zsh sourcing detection
secretGeek · Jan 8, 2025 · ffb366c · ffb366c
2 parents 0427d9a + 70cb596
commit ffb366c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ok.sh b/ok.sh
@@ -280,8 +280,9 @@ environment variables (for internal use):
 }
 
 is_sourced=""
-if [ "${ZSH_ARGZERO:+IS_SET}" = "IS_SET" ]; then
-    if [ "$ZSH_ARGZERO" = "zsh" ]; then
+if [ -n "$ZSH_VERSION" ]; then
+    # For zsh, check if script name matches $0
+    if [[ "${(%):-%N}" == *"ok.sh" ]]; then
         is_sourced="yes_indeed"
     fi
 else