Bash脚本被source执行时如何安全退出并避免关闭用户Shell？

这个坑我踩过好多次！当你用source my-script.sh（或者.缩写）加载脚本时，脚本会在当前Shell进程里执行，这时候直接用exit 1会直接关闭用户的Shell，还经常因为Shell退出太快，用户连错误提示都看不到。下面给你一套完整的解决方案：

一、判断脚本是否被source执行 #

有两种可靠的方法可以检测脚本的执行方式：

方法1：对比${BASH_SOURCE[0]}和${0} #

Bash里的${BASH_SOURCE[0]}变量永远指向当前脚本的路径，而${0}在直接执行脚本时等于脚本路径，在source执行时等于当前Shell的名称（比如bash或zsh）。这个方法兼容性最好，哪怕脚本有别名或符号链接都能准确判断：

if [[ "${BASH_SOURCE[0]}" != "${0}" ]]; then
    # 脚本被source执行
    IS_SOURCED=true
else
    # 脚本被直接执行
    IS_SOURCED=false
fi

方法2：检查$0是否为Shell进程名 #

如果不需要兼容复杂的脚本路径场景，也可以直接判断$0是不是当前Shell的进程名：

if [[ "$0" == "bash" ]]; then  # 用zsh的话就换成zsh
    echo "脚本正在被source执行"
fi

二、以非零状态码退出被source的脚本（不关闭Shell） #

当脚本被source时，不能用exit，要用return N——它会终止脚本的执行，返回状态码N，但不会关闭当前Shell。我们可以写一个通用的安全退出函数，自动适配两种执行方式：

# 安全退出函数：自动判断执行方式，避免source时关闭Shell
safe_exit() {
    local exit_code=${1:-1}  # 默认退出码为1，可自定义
    # 把错误信息输出到标准错误流（>&2），避免和正常输出混淆
    if [[ $# -gt 1 ]]; then
        echo "${@:2}" >&2
    fi
    
    if [[ "${BASH_SOURCE[0]}" != "${0}" ]]; then
        # 被source时用return返回状态码
        return $exit_code
    else
        # 直接执行时用exit退出
        exit $exit_code
    fi
}

然后把你原来的检查逻辑替换成这个函数：

# 原来的代码
# if [ something bad ] ; then echo "error message" exit 1 fi

# 替换后的代码
if [ something bad ]; then
    safe_exit 1 "错误：something bad 发生了！"
fi

三、通用最佳实践 #

1. 强制脚本只能直接执行：如果你的脚本设计为不能被source（比如修改环境变量会搞乱用户Shell），可以在脚本开头加检查：

if [[ "${BASH_SOURCE[0]}" != "${0}" ]]; then
    echo "错误：该脚本不支持source执行，请使用 ./my-script.sh 运行" >&2
    return 1  # 这里用return，因为已经确定是source场景
fi

2. 错误信息输出到stderr：永远把错误提示输出到标准错误流（>&2），这样即使用户重定向了标准输出，也能看到错误信息。

3. 结合set -e的注意事项：如果你用set -e让脚本在出错时自动退出，要确保它和safe_exit兼容；或者用trap处理退出逻辑时，记得判断执行方式，避免在source时触发exit。

内容的提问来源于stack exchange，提问作者1.84467E+19