【原创】可以生成树形图的shell菜单

大红机器

用cygwin编写的，没在在其它系统上运行过:wink:

生成菜单树形图比较慢呵呵，不知道还有没有更好的算法，可能真正shell环境比cygwin快一些吧，不过好在生成树形图不是经常需要做的操作吧呵呵~

1. 
   
2. #!/usr/bin/sh
   
3. 
   
4. Enter()
   
5. {
   
6. echo
   
7. printf "输入回车键继续..."
   
8. read -s Enter
   
9. echo
   
10. }
    
11. 
    
12. Chkfile()
    
13. {
    
14. if [ ! -f $1 ]
    
15.    then
    
16.         echo "ERR-0: 菜单文件 $1 不存在..."
    
17.         exit 1
    
18. fi
    
19. }
    
20. 
    
21. Chkinput()
    
22. {
    
23. if [ "x$2" = "xa" -o "x$2" = "xx" -o "x$2" = "xh" -o "x$2" = "xb" ]
    
24.    then
    
25.         return 0
    
26. fi
    
27. expr $2 + 0 >/dev/null 2>&1
    
28. if [ $? -ne 0 ]
    
29.    then
    
30.         return 1
    
31. fi
    
32. if [ $2 -le 0 -o $2 -gt `awk 'END{print NR}' $1` ]
    
33.    then
    
34.         return 2
    
35. fi
    
36. }
    
37. 
    
38. Tree()
    
39. {
    
40. menu=`expr $menu + 1`
    
41. local i=1
    
42. until [ $i -gt `awk 'END{print NR}' $1` ]
    
43. do
    
44. 
    
45.   echo $tree | awk '{for(i=1;i<="'$menu'";i++)if($i==1){printf "│  "}else{printf "    "}}'
    
46. 
    
47.   if [ $verbose -eq 1 ]
    
48.      then
    
49.           text=`awk -F"$MENUCHAR" 'NR=="'$i'"{if($2!~/'$MENUTYPE'/){print $1"     "$2}else{print $1}}' $1`
    
50.      else
    
51.           text=`awk -F"$MENUCHAR" 'NR=="'$i'"{print $1}' $1`
    
52.   fi
    
53. 
    
54.   if [ $i -eq `awk 'END{print NR}' $1` ]
    
55.      then
    
56.          echo "└─$text"
    
57.          tree=`echo $tree | awk '{for(i=1;i<=NF;i++){if(i==("'$menu'"+1))$i=0}}END{print $0}'`
    
58.      else
    
59.          echo "├─$text"
    
60.          tree=`echo $tree | awk '{for(i=1;i<=NF;i++){if(i==("'$menu'")+1)$i=1}}END{print $0}'`
    
61.   fi
    
62.   run=`awk -F"$MENUCHAR" 'NR=="'$i'"{print $2}' $1`
    
63.   if [ "`echo $run | awk -F"." '{print $NF}'`" = "$MENUTYPE" ]
    
64.      then
    
65.           tree="$tree 1"
    
66.           Tree $MENUPATH/$run
    
67.   fi
    
68.   i=`expr $i + 1`
    
69. done
    
70. menu=`expr $menu - 1`
    
71. }
    
72. 
    
73. Menu()
    
74. {
    
75. 
    
76. menu=`expr $menu + 1`
    
77. 
    
78. while true
    
79. 
    
80. do
    
81. 
    
82. if [ "x$input" = "xx" ]
    
83.    then
    
84.         exit
    
85. fi
    
86. 
    
87. clear
    
88. 
    
89. echo
    
90. echo "You can choose followed options:"
    
91. echo
    
92. echo "  ────────────────────────────"
    
93. echo
    
94. awk -F"$MENUCHAR" 'NF>1{printf "   "NR". ";if($2~/'$MENUTYPE'$/){printf "+ "}else{printf "* "}printf $1"\n\n"}' $1
    
95. 
    
96. echo "  ────────────────────────────"
    
97. echo
    
98. 
    
99. if [ $menu -gt 1 ]
    
100.    then
     
101.        echo "   b Back     --- 返回上一级菜单"
     
102.        echo
     
103. fi
     
104. echo "   a All      --- 全部执行"
     
105. echo "   h Help     --- 帮助信息"
     
106. echo "   x Exit     --- 退出"
     
107. echo
     
108. 
     
109. printf "Input your choice: "
     
110. read input
     
111. echo
     
112. 
     
113. Chkinput $1 "$input"
     
114. if [ $? -ne 0 ]
     
115.    then
     
116.         com=`echo $input | awk '{print $1}'`
     
117.         which $com >/dev/null 2>&1
     
118.         if [ $? -ne 0 ]
     
119.         then
     
120.              echo "ERR-1: 输入错误，请重新输入..."
     
121.         else
     
122.              eval $input
     
123.         fi
     
124.         Enter
     
125.         continue
     
126. fi
     
127. 
     
128. case "$input" in
     
129. 
     
130.       b) if [ $menu -ne 1 ]
     
131.             then
     
132.                  menu=`expr $menu - 1`
     
133.                  return
     
134.          fi
     
135.          ;;
     
136. 
     
137.       a)
     
138.          awk -F"$MENUCHAR" 'BEGIN{
     
139.                            a=0
     
140.                           }
     
141.                          {
     
142.                            if($2!~/'$MENUTYPE'$/){
     
143.                              print $1":";print "────────────────────"
     
144.                              system($2)
     
145.                              printf "\n"
     
146.                              a+=1
     
147.                              }
     
148.                           }
     
149.                       END{
     
150.                            if(a==0)
     
151.                               print "无可执行语句/脚本,请输入相应序号打开下级菜单"
     
152.                          }' $1
     
153.          Enter
     
154.          ;;
     
155. 
     
156.       h) echo "                帮助信息"
     
157.          echo "───────────────────────"
     
158.          echo
     
159.          echo "    输入序号打开相应菜单/执行语句或脚本"
     
160.          echo
     
161.          echo "    序号后显示 + 号表示是有下级菜单"
     
162.          echo "    序号后显示 * 号表示是可执行语句或脚本"
     
163.          echo
     
164.          echo "    a 执行所有菜单选项"
     
165.          echo "    b 返回上一级菜单"
     
166.          echo "    h 显示帮助信息"
     
167.          echo "    x 退出"
     
168.          echo
     
169.          echo "───────────────────────"
     
170.          Enter
     
171.          ;;
     
172. 
     
173.       x) exit
     
174.          ;;
     
175. 
     
176.       *) run=`awk -F"$MENUCHAR" 'NR=="'$input'"{print $2}' $1`
     
177.          if [ "`echo $run | awk -F"." '{print $NF}'`" = "$MENUTYPE" ]
     
178.             then
     
179.                  if [ ! -f $MENUPATH/$run ]
     
180.                     then
     
181.                          echo "ERR-0: 菜单文件 $MENUPATH/$run 不存在..."
     
182.                          Enter
     
183.                     else
     
184.                          Menu $MENUPATH/$run
     
185.                  fi
     
186.             else
     
187.                  eval $run
     
188.                  Enter
     
189.          fi
     
190.          ;;
     
191. esac
     
192. 
     
193. done
     
194. 
     
195. }
     
196. 
     
197. 
     
198. #================================#
     
199. #            M A I N             #
     
200. #================================#
     
201. 
     
202. MENUPATH=$HOME/menu                     # 默认菜单文件存放路径
     
203. MENUTYPE=menu                           # 菜单文件后缀名
     
204. MENUFILE=$MENUPATH/TOOL.$MENUTYPE       # 默认打开的菜单文件
     
205. MENUCHAR=%                              # 默认菜单文件分隔符
     
206. 
     
207. 
     
208. menu=0 # 第几级菜单
     
209. tree=0 # 默认不显示菜单树形图
     
210. verbose=0 # 默认菜单树形图不显示详细菜单信息
     
211. 
     
212. while getopts vtc:f:h OPTION
     
213. do
     
214.         case $OPTION in
     
215. 
     
216.                 t)
     
217.                    tree=1
     
218.                    ;;
     
219.                 v)
     
220.                    verbose=1
     
221.                    ;;
     
222.                 f)
     
223.                    MENUFILE=$MENUPATH/`echo $OPTARG | sed "s/\.$MENUTYPE$//"`.$MENUTYPE
     
224.                    ;;
     
225.                 h)
     
226.                    echo
     
227.                    echo "帮助信息"
     
228.                    echo
     
229.                    echo "Usage: `basename $0` [-t[-v]] [-h] [-c char] [-f file]"
     
230.                    echo
     
231.                    echo "-t, --Tree      显示菜单树形图"
     
232.                    echo
     
233.                    echo "-v, --Verbose   显示详细菜单树形图，需跟-t参数一起使用"
     
234.                    echo
     
235.                    echo "-c char         指定菜单文件中的分隔符"
     
236.                    echo
     
237.                    echo "-f file         打开指定菜单文件"
     
238.                    echo
     
239.                    echo "-h, --Help      显示帮助信息"
     
240.                    echo
     
241.                    exit
     
242.                    ;;
     
243.                 *)
     
244.                    echo "请尝试执行\"`basename $0` -h\"来获取更多信息。"
     
245.                    exit 1
     
246.                    ;;
     
247.         esac
     
248. done
     
249. 
     
250. if [ $tree -eq 1 ]
     
251.    then
     
252.         Chkfile $MENUFILE
     
253.         tree=0
     
254.         echo
     
255.         echo "菜单列表"
     
256.         Tree $MENUFILE
     
257.    else
     
258.         Chkfile $MENUFILE
     
259.         Menu $MENUFILE
     
260. fi
     

复制代码

执行效果：

1. 
   
2. 
   
3. You can choose followed options:
   
4. 
   
5.   ────────────────────────────
   
6. 
   
7.    1. * PLSQL Developer
   
8. 
   
9.    2. * SecureCRT
   
10. 
    
11.    3. * UltraEdit
    
12. 
    
13.   ────────────────────────────
    
14. 
    
15.    a All      --- 全部执行
    
16.    h Help     --- 帮助信息
    
17.    x Exit     --- 退出
    
18. 
    
19. Input your choice:
    
20. 
    

复制代码

使用说明：

修改脚本中以下变量定义菜单文件：

MENUPATH=$HOME/menu                               # 默认菜单文件存放路径
MENUTYPE=menu                                           # 菜单文件后缀名
MENUFILE=$MENUPATH/TOOL.$MENUTYPE       # 默认打开的菜单文件
MENUCHAR=%                                              # 默认菜单文件分隔符

然后按以上例子就是编辑菜单文件$HOME/menu/TOOL.menu，内容如下：

1. 
   
2. PLSQL Developer%/cygdrive/d/Program\ Files/PLSQL\ Developer/plsqldev.exe &
   
3. SecureCRT%/cygdrive/d/Tools/SecureCRT/SecureCRT.exe &
   
4. UltraEdit%/cygdrive/d/Tools/UltraEdit/Uedit32.exe &
   

复制代码

%符号就是作为菜单标题和要执行命令（或脚本）的分隔符

如果想要加入子菜单，就再编写一个.menu文件，然后在这个%后面写上.menu文件即可

菜单文件全部编写好后，使用-t参数，可以生成菜单树形图，效果如下：

1. 
   
2. 菜单列表
   
3.     ├─系统状态检查
   
4.     │  ├─系统状态检查
   
5.     │  │  ├─文件系统使用率
   
6.     │  │  ├─网络状态
   
7.     │  │  ├─系统进程状态
   
8.     │  │  └─占用内存进程TOP5
   
9.     ├─数据库状态检查
   
10.     │  ├─IMEX数据库
    
11.     │  │  ├─数据库状态
    
12.     │  │  ├─表空间使用率
    
13.     │  │  ├─数据库session阻塞
    
14.     │  │  └─导出Execl文件
    
15.     │  └─影像数据库
    
16.     │      ├─表空间使用率
    
17.     │      ├─数据库锁
    
18.     │      ├─正在运行的事务
    
19.     │      ├─全表扫描的等待事件
    
20.     │      └─数据库无效对象
    
21.     └─生成报告
    
22.         ├─晨检
    
23.         ├─周报
    
24.         └─月报
    

复制代码

-v 参数可以在树形图中将执行的命令（或脚本）显示出来(需要跟 -t 参数同时使用)

-f file 可以指定使用菜单文件

-c char 可以指定分隔符

-h 可以看使用帮助

大红机器

哦对了，选项前是 + 号说明是子菜单，* 号是可执行语句（或脚本）

另外选择的时候直接输入执行语句也可以~

大红机器

生成树状菜单的代码改成awk，速度提高近10倍啊~

以下是tree.awk脚本

1. #!/usr/bin/awk -f
   
2. 
   
3. BEGIN{
   
4. FS=MENUCHAR
   
5. "awk 'END{print NR}' "filename |getline
   
6. line=$1
   
7. menu++
   
8. }
   
9. 
   
10. {
    
11.   split(t,col,"#")
    
12.   for(i=1;i<=menu;i++){
    
13.       if(col[i]>0)
    
14.          printf "│  "
    
15.       if(col[i]==0)
    
16.          printf "    "
    
17.   }
    
18.   if(NR==line){
    
19.   print "└─"$1
    
20.   col[menu+1]=0
    
21.   t=col[1]
    
22.   for(i=2;i<=menu+1;i++)t=t"#"col[i]
    
23.   }
    
24.   else{
    
25.   print "├─"$1
    
26.   col[menu+1]=1
    
27.   t=col[1]
    
28.   for(i=2;i<=menu+1;i++)t=t"#"col[i]
    
29.   }
    
30.   if($2~MENUTYPE"$"){
    
31.   t=t"#"menu
    
32.   menu=system("awk -v MENUTYPE="MENUTYPE" -v MENUCHAR="MENUCHAR" -v MENUPATH="MENUPATH" -v menu="menu" -v t="t" -v filename="MENUPATH"/"$2" -f tree.awk "MENUPATH"/"$2)
    
33.   }
    
34. }
    
35. 
    
36. END{
    
37. menu--
    
38. exit menu
    
39. }
    

复制代码

主脚本中Tree()函数改为

1. awk -v MENUTYPE=$MENUTYPE -v MENUCHAR="$MENUCHAR" -v MENUPATH=$MENUPATH -v menu=$menu -v t=0 -v filename=$1 -f tree.awk $1

复制代码