区分 bash shell 脚本中的运行和源代码？

Question 1

是的 - $0 变量给出了脚本运行时的名称：

$ cat example.sh
#!/bin/bash
script_name=$( basename ${0#-} ) #- needed if sourced no path
this_script=$( basename ${BASH_SOURCE} )
if [[ ${script_name} = ${this_script} ]] ; then
    echo "running me directly"
else
    echo "sourced from ${script_name}"
fi 

$ cat example2.sh
#!/bin/bash
. ./example.sh

运行如下：

$ ./example.sh
running me directly
$ ./example2.sh
example.sh sourced from example2.sh

这并不适合从交互式 shell 获取源代码，但您会明白这个想法（我希望）。

已更新以包含 BASH_SOURCE - 感谢 hjk

Answer

是的 - $0 变量给出了脚本运行时的名称：

$ cat example.sh
#!/bin/bash
script_name=$( basename ${0#-} ) #- needed if sourced no path
this_script=$( basename ${BASH_SOURCE} )
if [[ ${script_name} = ${this_script} ]] ; then
    echo "running me directly"
else
    echo "sourced from ${script_name}"
fi 

$ cat example2.sh
#!/bin/bash
. ./example.sh

运行如下：

$ ./example.sh
running me directly
$ ./example2.sh
example.sh sourced from example2.sh

这并不适合从交互式 shell 获取源代码，但您会明白这个想法（我希望）。

已更新以包含 BASH_SOURCE - 感谢 hjk

Question 2

组合@黑暗之心使用环境变量回答BASH_SOURCE似乎可以解决问题：

$ head example*.sh
==> example2.sh <==
#!/bin/bash
. ./example.sh

==> example.sh <==
#!/bin/bash
if [ "$(basename $0)" = "$(basename $BASH_SOURCE)" ]; then
    echo "running directly"
else
    echo "sourced from $0"
fi
$ ./example2.sh
sourced from ./example2.sh
$ ./example.sh
running directly

编辑BASH_SOURCE如果我仅计算数组中元素的数量，这似乎是一个更简单的解决方案：

if [ ${#BASH_SOURCE[@]} -eq 1 ]; then echo "running directly"; else echo "sourced from $0"; fi

Answer

组合@黑暗之心使用环境变量回答BASH_SOURCE似乎可以解决问题：

$ head example*.sh
==> example2.sh <==
#!/bin/bash
. ./example.sh

==> example.sh <==
#!/bin/bash
if [ "$(basename $0)" = "$(basename $BASH_SOURCE)" ]; then
    echo "running directly"
else
    echo "sourced from $0"
fi
$ ./example2.sh
sourced from ./example2.sh
$ ./example.sh
running directly

编辑BASH_SOURCE如果我仅计算数组中元素的数量，这似乎是一个更简单的解决方案：

if [ ${#BASH_SOURCE[@]} -eq 1 ]; then echo "running directly"; else echo "sourced from $0"; fi

Question 3

我刚刚创建了与 BusyBox 类似的相同类型的库脚本。在其中，我使用以下函数来测试它是否正在被获取......

function isSourced () {
  [[ "${FUNCNAME[1]}" == "source" ]]  && return 0
  return 1
}

Bash 维护的 FUNCNAME 数组本质上是一个函数调用堆栈。$FUNCNAME(或${FUNCNAME[0]}) 是当前正在执行的函数的名称。${FUNCNAME[1]}是调用它的函数的名称，等等。

最上面的项目是脚本本身的特殊值。它将包含...

如果脚本是来源的，则“来源”一词
如果正在执行脚本并且测试是在函数内完成的，则为“main”一词
""(null) 如果正在执行脚本并且测试是在任何函数之外完成的，即......在脚本本身的级别。

上面的函数实际上仅在脚本级别调用时才有效（这就是我所需要的）。如果从另一个函数内部调用它将会失败，因为数组项编号将是错误的。要使其在任何地方都能工作，需要找到堆栈顶部并测试该值，这比较复杂。

如果您需要的话，您可以使用以下命令获取“堆栈顶部”的数组项编号...

  local _top_of_stack=$(( ${#FUNCNAME[@]} - 1 ))

${#FUNCNAME[@]}是数组中的项目数。作为从零开始的数组，我们减去 1 以获得最后一项#。

这三个函数一起使用来生成类似于 Python 的函数堆栈跟踪，它们可能会让您更好地了解这一切是如何工作的......

function inspFnStack () {
  local T+="  "
  local _at=
  local _text="\n"
  local _top=$(inspFnStackTop)
  local _fn=${FUNCNAME[1]}; [[ $_fn =~ source|main ]]  || _fn+="()"
  local i=_top; ((--i))
  #
  _text+="$i item function call stack for $_fn ...\n"
  _text+="| L   BASH_SOURCE{BASH_LINENO called from}.FUNCNAME  \n"
  _text+="| ---------------------------------------------------\n"
  while (( $i > 0 ))
  do
    _text+="| $i ${T}$(inspFnStackItem $i)\n"
    T+="  "
    ((--i))
  done
  #
  printf "$_text\n"
  #
  return 0
}

function inspFnStackItem ()  {
  local _i=$1
  local _fn=${FUNCNAME[$_i]}; [[ $_fn =~ source|main ]]  || _fn+="()"
  local _at="${BASH_LINENO[$_i-1]}"; [[ $_at == 1 ]]  && _at="trap"
  local _item="${BASH_SOURCE[$_i]}{${_at}}.$_fn"
  #
  printf "%s" "$_item"
  return 0
}

function inspFnStackTop () {
  # top stack item is 1 less than length of FUNCNAME array stack
  printf "%d\n" $(( ${#FUNCNAME[@]} - 1 ))
  #
  return 0
}

请注意，FUNCNAME、BASH_SOURCE 和 BASH_LINENO 是 bash 维护的 3 个数组，就好像它们是一个三维数组一样。

Answer