直截了当:您必须计算变量“$0”是否等于Shell的名称。

是这样的:

#!/bin/bash

echo "First Parameter: $0"
echo
if [[ "$0" == "bash" ]] ; then
    echo "The script was sourced."
else
    echo "The script WAS NOT sourced."
fi

通过壳:

$ bash check_source.sh 
First Parameter: check_source.sh

The script WAS NOT sourced.

通过来源:

$ source check_source.sh
First Parameter: bash

The script was sourced.

很难有一种100%可移植的方法来检测脚本是否来自源代码。

根据我的经验(使用Shellscripting 7年)，唯一安全的方法(不依赖具有pid等的环境变量，这是不安全的，因为它是变量)，你应该:

扩展你的“如果”的可能性 使用开关/箱子，如果你想。

这两个选项都不能自动缩放，但这是更安全的方式。

例如:

当您通过SSH会话源脚本时，变量“$0”(当使用source时)返回的值是-bash。

#!/bin/bash

echo "First Parameter: $0"
echo
if [[ "$0" == "bash" || "$0" == "-bash" ]] ; then
    echo "The script was sourced."
else
    echo "The script WAS NOT sourced."
fi

OR

#!/bin/bash

echo "First Parameter: $0"
echo
if [[ "$0" == "bash" ]] ; then
    echo "The script was sourced."
elif [[ "$0" == "-bash" ]] ; then
    echo "The script was sourced via SSH session."
else
    echo "The script WAS NOT sourced."
fi

@mklement0答案的一个小补充。这是我在脚本中使用的自定义函数，用于确定它是否被引用:

replace_shell(){
   if [ -n "$ZSH_EVAL_CONTEXT" ]; then 
      case $ZSH_EVAL_CONTEXT in *:file*) echo "Zsh is sourced";; esac
   else
      case ${0##*/} in sh|dash|bash) echo "Bash is sourced";; esac
   fi
}

在函数中，zsh的“$ZSH_EVAL_CONTEXT”的输出是toplevel:file:shfunc，而不仅仅是toplevel:file;因此，*:file*应该解决这个问题。

编者注:这个答案的解决方案工作稳健，但只有bash。它可以简化为 (返回2 > / dev / null)。

博士TL;

尝试执行return语句。如果脚本没有来源，则会引发错误。您可以捕获该错误并按照需要进行操作。

把它放在一个文件中，并调用它，比如test.sh:

#!/usr/bin/env sh

# Try to execute a `return` statement,
# but do it in a sub-shell and catch the results.
# If this script isn't sourced, that will raise an error.
$(return >/dev/null 2>&1)

# What exit code did that give?
if [ "$?" -eq "0" ]
then
    echo "This script is sourced."
else
    echo "This script is not sourced."
fi

直接执行:

shell-prompt> sh test.sh
output: This script is not sourced.

来源:

shell-prompt> source test.sh
output: This script is sourced.

对我来说，这可以在zsh和bash中工作。

解释

如果您试图在函数之外执行return语句，或者如果脚本不是源代码，则return语句将引发错误。在shell提示符中尝试以下操作:

shell-prompt> return
output: ...can only `return` from a function or sourced script

你不需要看到错误消息，所以你可以将输出重定向到dev/null:

shell-prompt> return >/dev/null 2>&1

现在检查逃生码。0表示OK(没有发生错误)，1表示发生错误:

shell-prompt> echo $?
output: 1

您还希望在子shell中执行return语句。当return语句运行它时…嗯……的回报。如果在子shell中执行它，它将从子shell返回，而不是从脚本返回。要在子shell中执行，请将其包装在$(…)中:

shell-prompt> $(return >/dev/null 2>$1)

现在，你可以看到子shell的退出代码，它应该是1，因为在子shell内部引发了一个错误:

shell-prompt> echo $?
output: 1

$_是很脆弱的。你必须检查它作为你在脚本中做的第一件事。即使这样，它也不保证包含shell的名称(如果是源的)或脚本的名称(如果是执行的)。

例如，如果用户设置了BASH_ENV，那么在脚本的顶部，$_包含BASH_ENV脚本中执行的最后一个命令的名称。

我发现最好的方法是像这样使用0美元:

name="myscript.sh"

main()
{
    echo "Script was executed, running main..."
}

case "$0" in *$name)
    main "$@"
    ;;
esac

不幸的是，这种方式在zsh中并不能开箱使用，因为functionargzero选项的功能超出了它的名称，并且在默认情况下是打开的。

为了解决这个问题，我把unsetopt functionarg0放在我的.zshenv中。

我认为在ksh和bash中没有任何可移植的方法来做到这一点。在bash中，您可以使用调用器输出来检测它，但我认为ksh中不存在等效的输出。

使用shebang行并检查它是否正在执行。

你的脚本应该有一行shebang #!/path/to/shell表示它应该在哪个shell中运行。否则，您还会遇到其他跨shell兼容性问题。

因此，您只需要通过尝试在源时不起作用的命令来检查它是否正在执行。

如。对于Bash脚本:

#!/usr/bin/env bash

if (return 0 2>/dev/null); then
    echo "Script was sourced."
fi

这个方法也适用于zsh和sh只是改变shebang行。