Add support for zsh

eth-cscs · Jul 9, 2024 · 802dab4 · 802dab4
1 parent cd738e8
commit 802dab4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/activate b/activate
@@ -1,5 +1,5 @@
 # only run if bash or zsh
-[ -n "${BASH_VERSION:-}" ] || return 0
+[ -n "${BASH_VERSION:-}" ] || [ -n "${ZSH_VERSION:-}" ] || return 0
 
 export UENV_CMD='env -u PYTHONPATH -u VIRTUAL_ENV @@impl@@'
 export UENV_IMG_CMD='env -u PYTHONPATH -u VIRTUAL_ENV @@image_impl@@'
@@ -49,16 +49,16 @@ function uenv {
             uenv_usage;
         elif [ "$1" = "image" ]; then
             shift;
-            $UENV_IMG_CMD $flags "$@"
+            eval $UENV_IMG_CMD $flags "$@"
         elif [[ " $* " =~ [[:space:]](-h|--help)[[:space:]] ]]; then
-            echo "$($UENV_CMD $flags "$@")"
+            echo "$(eval $UENV_CMD $flags "$@")"
         else
-            eval "$($UENV_CMD $flags "$@")"
+            eval "$(eval $UENV_CMD $flags "$@")"
         fi
     fi
 
     unset -f uenv_usage
     return $_exitcode
 }
 
-export -f uenv
+[ -n "${BASH_VERSION:-}" ] && export -f uenv