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
$