欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  资讯频道

我从其他Shell脚本中学到了什么?

程序员文章站 2022-03-25 12:31:44
...
作者Fizer Khan是一位Shell脚本迷,他对有关Shell脚本新奇有趣的东西是如此的痴迷。最近他遇到了authy-ssh脚本,为了缓解ssh服务器双重认证问题,他学到了许多有用且很酷的东西。对此,他想分享给大家。

1.  为输出着色

大多数情况下,你希望输出带颜色的结果,比如绿色代表成功,红色代表失败,黄色代表警告。

NORMAL=$(tput sgr0)
GREEN=$(tput setaf 2; tput bold)
YELLOW=$(tput setaf 3)
RED=$(tput setaf 1)
 
function red() {
    echo -e "$RED$*$NORMAL"
}
 
function green() {
    echo -e "$GREEN$*$NORMAL"
}
 
function yellow() {
    echo -e "$YELLOW$*$NORMAL"
}
 
# To print success
green "Task has been completed"
 
# To print error
red "The configuration file does not exist"
 
# To print warning
yellow "You have to use higher version."


这里使用tput来设置颜色、文本设置并重置到正常颜色。想更多了解tput,请参阅prompt-color-using-tput

2.  输出调试信息

输出调试信息只需调试设置flag。

function debug() {
    if [[ $DEBUG ]]
    then
        echo ">>> $*"
    fi
}
 
# For any debug message
debug "Trying to find config file"


某些极客还会提供在线调试功能:

# From cool geeks at hacker news
function debug() { ((DEBUG)) && echo ">>> $*"; }
function debug() { [ "$DEBUG" ] && echo ">>> $*"; }


3.  检查特定可执行的文件是否存在

OK=0
FAIL=1
 
function require_curl() {
    which curl &>/dev/null
    if [ $? -eq 0 ]
    then
      return $OK
    fi
 
    return $FAIL
}


这里使用which来命令查找可执行的curl 路径。如果成功,那么可执行的文件存在,反之则不存在。将&>/dev/null设置在输出流中,错误流会显示to /dev/null (这就意味着在控制板上没有任何东西可打印)。

有些极客会建议直接通过返回which来返回代码。

# From cool geeks at hacker news
function require_curl() { which "curl" &>/dev/null; }
function require_curl() { which -s "curl"; }


4.  输出脚本Usage

当我开始编写shell 脚本,我会用echo来命令打印已使用的脚本。当有大量的文本在使用时, echo命令会变得凌乱,那么可以利用cat来设置命令。

cat << EOF
Usage: myscript <command> <arguments>
VERSION: 1.0
Available Commands
    install - Install package
    uninstall - Uninstall package
    update - Update package
    list - List packages
EOF


这里的<<被称为<<here document,字符串在两个EOF中。

5.  用户配置值vs. 默认值

有时,如果用户没有设置值,那么会使用默认值。

URL=${URL:-http://localhost:8080}


检查URL环境变量。如果不存在,可指定为localhost。

6.  检查字符串长度

if [ ${#authy_api_key} != 32 ]
then
  red "you have entered a wrong API key"
  return $FAIL
fi


利用 ${#VARIABLE_NAME} 定义变量值的长度。

7.  读取超时输入

READ_TIMEOUT=60
read -t "$READ_TIMEOUT" input
 
# if you do not want quotes, then escape it
input=$(sed "s/[;\`\"\$\' ]//g" <<< $input)
 
# For reading number, then you can escape other characters
input=$(sed 's/[^0-9]*//g' <<< $input)


8.  获取目录名和文件名

# To find base directory
APP_ROOT=`dirname "$0"`
 
# To find the file name
filename=`basename "$filepath"`
 
# To find the file name without extension
filename=`basename "$filepath" .html`


英文出自:FizerKhan