【学习分享】仿火狐插件hostAdmin插件（切换host）

Diao_Cow

最近一直在学习shell, 但是写脚本的机会并不是很多，但是学习shell的最好的方式就是多写脚本，多用不同的方式去解决同一个问题。

为了提高自己的shell水平，尝试把每一个的重复性工作，都用shell来完成，作为自己使用的小工具（毕竟别人不一定会觉得用的爽 ）

这是我的文章链接：http://diaocow.iteye.com/blog/1674074

1. #!/bin/bash
   
2. 
   
3. ##############################
   
4. #  切换hosts的工具类 swich hosts
   
5. #
   
6. #  --------hosts 文件格式--------
   
7. #  ==offline
   
8. #  ip list...
   
9. #  ==offline
   
10. #
    
11. #  ==online
    
12. #  ip list...
    
13. #  ==online
    
14. #  --------hosts 文件格式--------
    
15. #  
    
16. #  操作：sh shosts.sh offline 开启offline这组host绑定
    
17. #
    
18. #  kang.zhouk 2012-9-8
    
19. #
    
20. ##############################
    
21. 
    
22. # 检测用户是否输入了group tag
    
23. if [[ -z $1 ]] ; then
    
24.         echo "Error: Please input a group tag! eg. offline"
    
25. fi
    
26. 
    
27. HOSTS="testdata"         # hosts文件(这里，你需要替换成/etc/hosts)
    
28. HOSTS_TEMP="host.tmp"    # hosts临时文件
    
29. GROUP_TAG="==$1"         # 需要打开的组
    
30. 
    
31. # 检测group tag有效性
    
32. if ! grep -q  "$GROUP_TAG"  "$HOSTS" ; then
    
33.         echo "Error: there is no group tag named $GROUP_TAG in $HOSTS"
    
34.         exit -111
    
35. fi
    
36. 
    
37. # 函数：判断字符串是否为IP
    
38. isIp() {
    
39.         echo $1 | grep -E -q '([0-9]+\.){3}[0-9]'
    
40.     ip_result=`echo $?`
    
41. }
    
42. 
    
43. # clean临时文件
    
44. : > $HOSTS_TEMP
    
45. # group tag 计数
    
46. start_flag=0
    
47. 
    
48. # 处理hosts
    
49. cat $HOSTS | while read line ; do
    
50. 
    
51.         if echo $line | grep -q $GROUP_TAG; then
    
52.                 start_flag=$((start_flag + 1))
    
53.         fi
    
54. 
    
55.         isIp "$line"
    
56. 
    
57.         if [[ $start_flag == 1 && $ip_result == 0 ]] ; then
    
58.                 # 去掉开头的注释符
    
59.                 line=${line/#\#/}
    
60.         elif [[ $ip_result == 0 ]] ; then
    
61.                 # 在行开头添加注释符
    
62.                 if ! echo $line | grep -q '^ *#' ; then
    
63.                      line=\#$line
    
64.                 fi
    
65.         fi
    
66.         echo "$line" >> $HOSTS_TEMP
    
67. 
    
68. done
    
69. 
    
70. # 重新生成hosts
    
71. cat $HOSTS_TEMP > $HOSTS
    
72. 
    
73. rm -rf $HOSTS_TEMP

复制代码

现在我们来测试下这个脚本：

我们准备下测试数据（脚本中的HOSTS变量指定hosts文件，目前我们暂定义为testdata)：
[diaocow@ubuntu]$ cat testdata   
==offline  
72.51.30.13    offline.test1.com   
72.20.123.321  offline.test2.com  
  
==offline  
  
==online  
12.56.92.97 online.test1.com      
12.56.38.84 online.test2.com  
==online  

将host切换到线上环境

[diaocow@ubuntu]$ sh shosts.sh  online  
[diaocow@ubuntu]$ cat testdata  
==offline  
#72.51.30.13    offline.test1.com  
#72.20.123.321  offline.test2.com  
  
==offline  
  
==online  
12.56.92.97 online.test1.com  
12.56.38.84 online.test2.com  
==online  

重复执行  sh shosts.sh  online 不会有任何问题

我们在尝试切换成线下环境  
[diaocow@ubuntu]$ sh shosts.sh  offline  
[diaocow@ubuntu]$ cat testdata  
==offline  
72.51.30.13    offline.test1.com  
72.20.123.321  offline.test2.com  
  
==offline  
  
==online  
#12.56.92.97 online.test1.com  
#12.56.38.84 online.test2.com  
==online  

一切OK  

在脚本的编写过程中，我经常使用这两个命令

sh -n shosts.sh online                 # 检测脚本语法错误
sh -x shosts.sh online                 # 详细打出脚本每一步执行过程

kk5234

谢谢分享！:wink:

ziyunfei

火狐粉

Diao_Cow

谢谢ls各位的支持