shell创建文本菜单

创建文本菜单

核心是case命令，根据用户的选择来执行特定的命令。

创新互联建站是网站建设专家,致力于互联网品牌建设与网络营销，专业领域包括成都做网站、网站设计、电商网站制作开发、小程序开发、微信营销、系统平台开发，与其他网站设计及系统开发公司不同，我们的整合解决方案结合了恒基网络品牌建设经验和互联网整合营销的理念，并将策略和执行紧密结合，且不断评估并优化我们的方案，为客户提供全方位的互联网品牌整合方案！

创建菜单布局

使用echo命令打印字符，生成一个菜单，还可以包括标题等信息：

clear
echo
# -e 选项，打印特殊字符
echo -e "\t\t\tSys Admin Menu\n"
echo -e "\t1. Display disk space"
echo -e "\t2. Display logged on users"
echo -e "\t3. Display memory usage"
echo -e "\t0. Exit program\n\n"
# -en 会去掉末尾的换行符，光标会留在行尾
echo -en "\t\tEnter option: "
# 获取用户输入，只期望获取到单个字符，-n选择限制只读1个字符，并且用户不用回车
read -n 1 option

clear命令，先清空当前会话的内容。
echo -e选项，可以打印特殊字符。
echo -en选项，会去掉末尾的换行符。这让菜单看上去更专业，光标会一直行尾等待用户输入。

获取用户输入
打印出菜单后，就要等待并获取用户输入。这步使用read命令。这里期望只要单个字符，所以用 -n 选项限制只读取一个字符。这样用户只需要输入一个数字，并且不用按回车。

创建菜单函数
把上面的部分封装成一个函数，这样，在任何时候只要调用函数，就能重现菜单。

创建桩函数

桩函数（stub function），是一个空函数，或者只有一个echo语句，说明最终这里需要什么内容：

function diskspace {
    clear
    echo "Display disk space"
}

这样，就不需要事先写出所有函数。菜单能够直接投入使用，之后在来实现具体的操作。

添加菜单逻辑

菜单布局和函数都有了，下面需要创建程序逻辑将二者结合起来。这里就需要case命令。
case命令根据菜单中输入的字符来调用相应的函数。用默认的case命令字符星号来处理所有不正确的菜单项。
下面就是这个完整的菜单脚本的例子：

#!/bin/bash
# 为脚本创建文本菜单

# --------------------
# 定义函数
# --------------------

# 打印菜单
function menu {
    clear
    echo
    # -e 选项，打印特殊字符
    echo -e "\t\t\tSys Admin Menu\n"
    echo -e "\t1. Display disk space"
    echo -e "\t2. Display logged on users"
    echo -e "\t3. Display memory usage"
    echo -e "\t0. Exit program\n\n"
    # -en 会去掉末尾的换行符，光标会留在行尾
    echo -en "\t\tEnter option: "
    # 获取用户输入，只期望获取到单个字符，-n选择限制只读1个字符，并且用户不用回车
    read -n 1 option
}

function diskspace {
    clear
    df -k
}

function whoseon {
    clear
    who
}

function menusage {
    clear
    cat /proc/meminfo
}

# --------------------
# 函数主体
# --------------------
while [ 1 ]
do
    # 菜单逻辑
    menu
    case $option in
    0)
        break ;;
    1)
        diskspace ;;
    2)
        whoseon ;;
    3)
        menusage ;;
    *)
        clear
        echo "Wrong selection";;
    esac
    echo -en "\n\n\t\tHit any key to continue"
    read -n 1 line
done
clear

菜单的显示效果如下：

                        Sys Admin Menu

        1. Display disk space
        2. Display logged on users
        3. Display memory usage
        0. Exit program

                Enter option: 

使用 select 命令

创建文本菜单的过程中，花了一半的精力在建立菜单布局和获取用户输入上。bash shell 提供了一个很容易上手的小工具，可以帮助自动完成这些工作。

select命令只需要一条命令就可以创建出菜单，然后获取输入并自动处理。命令格式如下：

select 选项变量 in "选项1" "选项2" "选项3"
do
    命令
done

select命令会将每个选项自动加上编号，然后为选项显示一个由PS3环境变量定义的特殊提示符。所以还要定义PS3环境变量。

示例代码

这里是一个select命令的示例：

#!/bin/bash
# 为脚本创建文本菜单

function diskspace {
    clear
    df -k
}

function whoseon {
    clear
    who
}

function menusage {
    clear
    cat /proc/meminfo
}

PS3="Enter option: "
select option in "Display disk space" "Display logged on users" \
"Display memory usage" "Exit program"
do
    case $option in
    "Exit program")
        break ;;
    "Display disk space")
        diskspace ;;
    "Display logged on users")
        whoseon ;;
    "Display memory usage")
        menusage ;;
    *)
        clear
        echo "Wrong selection";;
    esac
done

菜单效果如下：

$ menu2.sh 
1) Display disk space       3) Display memory usage
2) Display logged on users  4) Exit program
Enter option: 

使用这个工具可以快速的创建一个简易的菜单，不过视觉效果就差很多。

制作窗口（dialog包）

dialog包能够用ANSI转义控制字符在文本环境中创建标准的窗口对话框。

窗口部件

dialog包提供了很多窗口部件（widget），使用的时候命令格式如下：

dialog --widget parameters

常用的部件如下：

1. msgbox部件：在窗口中显示一条简单的消息，会有一个OK按钮
2. yesno部件：允许用户对窗口中显示的问题选择yes或no，会有两个按钮
3. inputbox部件：提供一个简单的文本框区域来输入文本
4. textbox部件：可以在窗口中显示大量信息，会生成一个滚动窗口
5. menu部件：创建文本菜单，需要为每个选项提供一个选择标号和显示文本
6. fselect部件：可以用来浏览文件和文件夹

更多窗口部件，后面会详细列出。

获取部件的输出

每个dialog部件都提供了两种形式的输出：

• 使用退出状态码
• 使用STDERR

返回选项
如果选择了YES或OK，返回退出状态码0。如果选择了Cancel或No，返回退出状态码1。可以用标准的$?变量来判断具体选择了哪个按钮。

返回数据
如果是返回数据，则会将数据发送到STDERR。可以用标准的bash shell方法来将STDERR输出重定向到另一个文件或文件描述符中：

dialog --inputbox "Enter your age: " 10 20 2>age.txt

具体怎么用可以参考下面的示例。

示例代码

内容比较多，感觉也不一定需要制作这么好的窗口来交换。编写脚本时记住两件事：

• 如果有按钮，检查dialog命令的退出状态码
• 否则就重定向STDERR来获取输出的值

前面的例子使用dialog包实现的示例代码：

#!/bin/bash
# 为脚本创建文本窗口菜单

temp=$(mktemp -t test.XXXXXX)
temp2=$(mktemp -t test2.XXXXXX)

function diskspace {
    df -k > $temp
    dialog --textbox $temp 20 60
}

function whoseon {
    who > $temp
    dialog --textbox $temp 20 50
}

function menusage {
    cat /proc/meminfo > $temp
    dialog --textbox $temp 20 50
}

while [ 1 ]
do
dialog --menu "Sys Admin Menu" 20 30 10 \
1 "Display disk space" \
2 "Display logged on users" \
3 "Display memory usage" \
0 "Exit program" 2> $temp2
if [ $? -eq 1 ]
then
    break
fi

selection=$(cat $temp2)
case $selection in
0)
    break ;;
1)
    diskspace ;;
2)
    whoseon ;;
3)
    menusage ;;
*)
    dialog --msgbox "Wrong selection" 10 30
esac
done
rm -f $temp 2> /dev/null
rm -f $temp2 2> /dev/null

这里用到了临时文件，并且使用mktemp命令来创建临时文件，看着挺专业。

安装dialog包

另外系统可能默认没有安装dialog包，要运行这个脚本，需要先安装dialog包：

[root@Ansible ~]# yum info dialog
已加载插件：fastestmirror
Loading mirror speeds from cached hostfile
 * base: mirrors.aliyun.com
 * extras: mirrors.aliyun.com
 * updates: mirrors.aliyun.com
已安装的软件包
名称    ：dialog
架构    ：x86_64
版本    ：1.2
发布    ：5.20130523.el7
大小    ：505 k
源    ：installed
来自源：base
简介    ： A utility for creating TTY dialog boxes
网址    ：http://invisible-island.net/dialog/dialog.html
协议    ： LGPLv2
描述    ： Dialog is a utility that allows you to show dialog boxes (containing
         : questions or messages) in TTY (text mode) interfaces.  Dialog is called
         : from within a shell script.  The following dialog boxes are implemented:
         : yes/no, menu, input, message, text, info, checklist, radiolist, and
         : gauge.
         : 
         : Install dialog if you would like to create TTY dialog boxes.

dialog部件详表

dialog部件

dialog选项

除了标准部件，还可以在dialog命令中定制不同的选项。这些选项可以让你全面定制窗口外观和操作。

dialog命令选项

dialog命令选项2
按钮的选项功能都差不多，单独列出在这里。可以重写对话窗口中的任意按钮标签：

举例说明

• --title选项，允许你设置出现在窗口顶部的部件标题。
• --backtitle选项，是为脚本中的菜单创建公共标题的简便办法。
• --create-rc选项，dialog命令支持运行时配置。该命令会根据配置文件模板创建一份配置文件。

创建本地临时文件（mktemp）

在需要临时将内容保存到文件的时候，有个特殊命令可以用来创建临时文件。mktemp命令可以在/tmp目录中创建一个唯一的临时文件。shell会创建这个文件，但不用默认的umask值。它会将当前用户设置为文件的属主，并且只有属主有读写权限。

创建文件

默认情况下，mktemp会在当前目录中创建一个文件。使用命令的时候，需要指定一个文件名模板。模板可以包含任意文本文件名，在文件名末尾加上6个X（几个X都没关系，书上建议6位）：

$ mktemp test1.XXXXXX
test1.Acrugq
$ ls -al test1*
-rw-------. 1 steed steed 0 12月 12 14:26 test1.Acrugq
$ 

mktemp命令会用6位字符码替换这6个X，从而保证文件名在目录中是唯一的。

在脚本中使用

在脚本中使用mktemp命令时，需要将文件名保存到变量中，这样就能在后面的脚本中引用了：

$ mktemp test2.XXXXXX
test2.leOFBZ
$ mktemp test2.XXXXXX
test2.5kDbKn
$ mktemp test2.XXXXXX
test2.domeOC
$ mktemp test2.XXXXXX
test2.CJX702
$ tempfile=$(mktemp test2.XXXXXX)
$ exec 3>>$tempfile
$ echo "test2 Line1" >&3
$ echo "TEST2 LINE2" >&3
$ exec 3>&-
$ cat $tempfile
test2 Line1
TEST2 LINE2
$ ls -al test2*
-rw-------. 1 steed steed  0 12月 12 14:41 test2.5kDbKn
-rw-------. 1 steed steed  0 12月 12 14:41 test2.CJX702
-rw-------. 1 steed steed  0 12月 12 14:41 test2.domeOC
-rw-------. 1 steed steed  0 12月 12 14:41 test2.leOFBZ
-rw-------. 1 steed steed 24 12月 12 14:41 test2.tMVTwN
$ rm -f $tempfile 2> /dev/null
$ ls -al test2*
-rw-------. 1 steed steed 0 12月 12 14:41 test2.5kDbKn
-rw-------. 1 steed steed 0 12月 12 14:41 test2.CJX702
-rw-------. 1 steed steed 0 12月 12 14:41 test2.domeOC
-rw-------. 1 steed steed 0 12月 12 14:41 test2.leOFBZ
$ 

这里先创建了几个临时文件，干扰一下。使用的时候，创建了文件描述符3来使用，使用完之后关闭了文件描述符。最后精确的把用完的临时文件给删除了。

在/tmp目录创建临时文件

-t选项会强制mktemp命令在系统的临时目录创建文件。此时返回的就是创建的文件的全路径：

$ mktemp -t test3.XXXXXX
/tmp/test3.aPIXIy
$ 

创建临时目录

-d选项则是创建一个临时目录。如果要在/tmp目录下创建临时目录，就是-dt：

$ mktemp -d dir.XXXXXX
dir.aBDmsd
$ mktemp -dt dir.XXXXXX
/tmp/dir.pqW927
$