linux shell中的命令自动补全(compgen complete)与 命令行参数解析

很多时候,当我们写一个脚本时,我们总会提供一些可选的命令选项。当可选项比较多的时候,比如git, 如果能够提供命令自动补全,无疑是锦上添花的事。而且个人认为,这种方式,比采用将命令做成选择菜单要更好一些。


假设我们现在这样一个脚本,脚本执行命令时bsu,  类似git,脚本有两个一级主命令pushbaseline, createrepo,然后这两个一级主命令下还有若干选项

pushbaseline  --qcom  --mtk  --name=(NX01, NX02)     #要求  --qcom --mtk不能同时出现,- -name=xxx可以与前面任意一个选项同时出现

createrepo   --git  --repo  --name=xxx  --auth=(W, R, RW) # 同上,这里--git, --repo不能同时出现。 后面两个选项随意

正式行动之前我们做一个简单的需求分析,正所谓不做需求分析就干活,“加班没日夜, 累死也枉然”

以pushbaseline为例,这里要求--qcom --mtk不能同时出现,如果直接在脚本中实现,就要判断之前输入中有没有输入--qcom,如果有输入,就不能再把--mtk作为命令自动补全的候选者。这种逻辑很复杂,对shell脚本不熟的人,只能望洋兴叹——“呵呵”。

事实上,从实际情况来看的话,为什么不能同时出现,因为它们都属于某个属相的描述,所以这里简单的改为--platfrom=(qcom, mtk),问题就巧妙解决了。


下面开始写脚本,上面将--qcom --mtk 改编为--platform=(qcom,mtk),缺少了简单选项,为此再额外加一个 --multi

自动补全,linux提供了两个重要命令 compgen, complete

命令:compgen -W "aa ab bb cc" -- “a”   

表示从"aa ab bb cc"  匹配出以“a”开头的单词

这条命令的返回结果就是 “aa ab”。 


命令:complete -F  __cmd_HUB  bsu

表示当执行bsu命令时,自动补全的候选单词由函数__cmd_HUB计算得到,具体的承载容器是变量COMPREPLY

特别说明一下,像这样的complete -F XXX   ./test.sh  也是合法的。如果XXX的计算后得到COMPREPLY=(aa bb cc), 则输入 ./test.sh 之后, 按tab键,可以自动弹出候选选项aa bb cc。


解释的话说的有点多,下面直奔主题了

function __cmd_HUB() {
        #$COMP_CWORD是系统自动变量,表示当前命令单词索引。 0表示第一个单词,也就是bsu。
	case $COMP_CWORD in
	0)	#仍在完成根命令,这里不需要处理
		;;
	1)  #根命令已经完成,这里开始补充一级主命令
		#${COMP_WORDS[0]}是根命令,在这个例子中就是bsu 
		eval __cmd_${COMP_WORDS[0]}
		;;
	2|*)#一级主命令已经完成,这里开始补充一级主命令的选项
		#${COMP_WORDS[1]}是一级主命令,在这个例子中就是pushbaseline或者createrepo
		eval __cmd_${COMP_WORDS[1]}
		;;
	esac
}

complete -F __cmd_HUB bsu

如果有同学对eval命令不了解,就找度娘吧。

上面我们做了一个约定,要获取命令的选项,则总是调用函数__cmd_获得,因此下面就要实现__cmd_bsu, __cmd_pushbaseline, __cmd_createrepo

function __cmd_bsu() {
	local cur="${COMP_WORDS[COMP_CWORD]}"
	local options="$MAIN_COMMAND"
	COMPREPLY=( $(compgen -W "${options}" -- ${cur}) )
}

function __cmd_pushbaseline() {
	local cur="${COMP_WORDS[COMP_CWORD]}"
	COMPREPLY=()
    case $cur in
	--name=*)
		local options="NX01 NX02"
		COMPREPLY=( $(compgen -W "${options}" -- ${cur#--name=}) )
		;;
	--platform=*)
		local options="qcom mtk"
		COMPREPLY=( $(compgen -W "${options}" -- ${cur#--platform=}) )
		;;
	*) 
		local options="--platform= --name= --multi"
		COMPREPLY=( $(compgen -W "${options}" -- ${cur}) )
		;;
	esac	
}


命令自动补全都到此为止,篇幅有点长了,下面长话短说,直接处理命令,以命令 bsu  pushbaseline --platform=qcom   --name=NX01 --multi 为例

在bsu函数中完成命令转发,在各个子命令中详细解析参数

function bsu() {
	case $1 in
	pushbaseline)
		shift
		__parsing_pushbaseline $@
		;;
	createrepo)
		shift
		__parsing_createrepo $@
		;;
	*)
		echo "unknown command $1"
		;;
	esac	
}

function __parsing_pushbaseline() {
	while [ $# -gt 0 ]; do
	case $1 in
	--platform=*)
		local platform=${1#--platform=}
		;;
	--name=*)
		local name=${1#--name=}
		;;
	--multi)
		local multi=1
		;;
	esac
	
	shift	
	done
	
	bsu_pushbaseline $platform $name $multi
}

function bsu_pushbaseline()
{
	echo bsu_pushbaseline $@
}

解析命令的关键一点,对于--xxx=YY的选项,可以使用模式匹配运算符快速得到想要的选项。这里简单起见,没有将代码做很多的形式化,有需要的话,只要利用eval函数,相信难不倒各位看官。


over,happy new year to everyone!



你可能感兴趣的:(linux,shell)