很多時候，當我們寫一個指令碼時，我們總會提供一些可選的命令選項。當可選項比較多的時候，比如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

上面我們做了一個約定，要獲取命令<command>的選項，則總是呼叫函式__cmd_<command>獲得，因此下面就要實現__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 [email protected]
		;;
	createrepo)
		shift
		__parsing_createrepo [email protected]
		;;
	*)
		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 [email protected]
}

over，happy new year to everyone！