Zsh Mailing List Archive
Messages sorted by: Reverse Date, Date, Thread, Author

Detect if a script is being sourced vs executed

I want to detect if a script is being sourced instead of being called.

This is meant for pyenv, which uses shims to delegate execution to the selected Python version.

Currently it always uses "exec" to do so, but there is a pull request to add support for scripts that are meant to be sourced (like virtualenvwrapper.sh), and then would use "source" instead of "exec" in the shim.

The PR provides this functionality for Bash, which provides a way to detect this via $BASH_SOURCE/$BASH_LINENO: https://github.com/yyuu/pyenv/pull/100/files

My current approach so far is the following, which requires Zsh 5.0.5 (for POSIX_ARGZERO) and does not work for "zsh test_source.sh" yet (a patch has been posted to zsh-workers to fix this).

I would like to do this for older Zsh versions, and in a simpler / more elegant way.

Script to be sourced (test_source.sh):


    echo "\$0: $0"

    if [ -n "$ZSH_VERSION" ]; then
      # Use prompt expansion to get the current file.

      # Fix $0:
      if (( ${+options[posixargzero]} )); then  # added in zsh 5.0.5 (2014-06-01)
        if [[ $options[posixargzero] != "on" ]]; then
          setopt posixargzero
          [ "$0" = "$cur_file" ] || sourced=1
          setopt noposixargzero
          [ "$0" = "$cur_file" ] || sourced=1
        echo "TODO"

    elif [ -n "$BASH_VERSION" ]; then
      [ "$0" = "$BASH_SOURCE" ] || sourced=1

    echo "sourced: $sourced"

Test script (test.sh):


    echo "== CALL"

    echo "== SOURCE"
    . ./test_source.sh

    echo "== CALL: bash"
    bash ./test_source.sh

    echo "== CALL: zsh"
    zsh ./test_source.sh

    echo "== EXEC"
    exec ./test_source.sh

There must be an easier way!?

It would be really useful, if Zsh would provide a mechanism like Bash to simplify this.


Messages sorted by: Reverse Date, Date, Thread, Author