使用If / Then / Else语句合并Bash脚本

Question

我对Bash脚本世界有些陌生，并且仍在尝试了解大多数人认为最基本的东西。 我找到了这份有用的指南 ，我从中汲取了大部分信息，尽管它指的是“高级”，但对于像我这样的新手来说，它实际上很好地涵盖了基础知识。

在这一点上，我已经写了一个脚本，该脚本可以按我的意愿工作，但是我想我是一些可能认为是代码意识的人。 因此，尽管我的脚本确实可以工作，但我认为我没有适当地利用Bash的if / then / else语句。

这是我编写的小脚本，主要用于通过API查询来检查用户名以及用户名是否存在。 它允许用户通过./script.sh list定义./script.sh list ，或者，如果他们未定义列表，则只需尝试字母的所有变体（ {a..z} ）：

#!/bin/bash

# variables
host='https://api.example.com/api/users/'
sleep=1

# check if jq install
if ! jq_loc="$(type -p "jq")" || [ -z "$jq_loc" ]; then
  echo "no jq installed."
  sleep 1
  echo "we're going to install it now."
  sleep 2
  sudo apt-get install jq
fi

# specified list
if [ -n "$1" ] ; then
  list=`cat $1`
  for username in $list
  do
     #echo $username
     sleep $sleep
     VAL=`curl -s "$host$username" | jq -r ".id" | grep null`
     #echo $VAL
     if [[ $VAL == null ]]
     then
       echo -e "$username is available"
       echo "$username" >> username.free
     else
       echo -e "$username is taken"
     fi
  done
fi

# no list specified
if [ $# -eq 0 ] ; then
list=`echo {a..z}{a..z}{a..z}`
  for username in $list
  do
     #echo $username
     sleep $sleep
     VAL=`curl -s "$host$username" | jq -r ".id" | grep null`
     #echo $VAL
     if [[ $VAL == null ]]
     then
       echo -e "$username is available"
       echo "$username" >> username.free
     else
       echo -e "$username is taken"
     fi
  done
fi

就像我说的那样，该脚本完全按照我想要的方式工作。 但是我担心的是if [ $# -eq 0 ] ; then if [ $# -eq 0 ] ; then ， if [ -n "$1" ] ; then if [ -n "$1" ] ; then 。 我觉得我在回馈它不必要地执行相同的功能。 例如，不必告诉它在每个if执行curl命令，有没有办法将这两个if合并为一个？

希望我已经对此做了足够的解释，如果没有，请发表评论，我可以相应地更新这个问题。

提前致谢！

Answer 1

您的if语句涵盖了所有逻辑分支。 因此，以下代码块将以任何一种方式执行。 因此，没有理由重复它。 只需将其移动到if语句下方。

for username in $list
do
   #echo $username
   sleep $sleep
   VAL=`curl -s "$host$username" | jq -r ".id" | grep null`
   #echo $VAL
   if [[ $VAL == null ]]
   then
     echo -e "$username is available"
     echo "$username" >> username.free
   else
     echo -e "$username is taken"
   fi
done

下面是重新编写代码以减少重复的代码：

#!/bin/bash

# variables
host='https://api.example.com/api/users/'
sleep=1

# check if jq install
if ! jq_loc="$(type -p "jq")" || [ -z "$jq_loc" ]; then
  echo "no jq installed."
  sleep 1
  echo "we're going to install it now."
  sleep 2
  sudo apt-get install jq
fi

if [ -n "$1" ] ; then
  # specified list
  list=`cat $1`
else
  # no list specified
  list=`echo {a..z}{a..z}{a..z}`
fi

for username in $list
do
   #echo $username
   sleep $sleep
   VAL=`curl -s "$host$username" | jq -r ".id" | grep null`
   #echo $VAL
   if [[ $VAL == null ]]
   then
     echo -e "$username is available"
     echo "$username" >> username.free
   else
     echo -e "$username is taken"
   fi
done

Answer 2

不是将数据读入变量然后对该变量进行操作，而是一种常用技术是在读取数据时对其进行处理。 为此，我将对Micah的脚本进行以下编辑：

#!/bin/bash

# variables
host='https://api.example.com/api/users/'
sleep=1

# check if jq install
type jq > /dev/null || sudo apt-get install jq

# Here is a bit of "magic".  We create a process to generate the data
# that we want to use as input and redirect its output to be the input
# of the script so that the subsequent read handles it. (Note that it
# is totally unnecessary to do this, and we could just as easily pipe
# this data directly to the while/read loop, but for some reason I 
# thought it would be instructive to see this technique.) 
exec < <(
if [ -n "$1" ] ; then
  cat $1
else
  echo {a..z}{a..z}{a..z} | tr ' ' \\n
fi
)

while read username 
do
   sleep $sleep
   if curl -s "$host$username" | jq -r ".id" | grep -q null
   then
     echo "$username is available"
     echo "$username" >> username.free
   else
     echo "$username is taken"
   fi
done