add bash completion for gobgpcli

Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>

1 files changed, 180 insertions, 0 deletions

diff --git a/tools/completion/gobgpcli-completion.bash b/tools/completion/gobgpcli-completion.bash
new file mode 100644
index 00000000..4b92292e
--- /dev/null
+++ b/tools/completion/gobgpcli-completion.bash
@@ -0,0 +1,180 @@
+#!bash
+
+__gobgpcli_q() {
+    gobgpcli 2>/dev/null "$@"
+}
+
+__search_target() {
+    local word target c=1
+
+    while [ $c -lt $cword ]; do
+	word="${words[c]}"
+	for target in $1; do
+	    if [ "$target" = "$word" ]; then
+		echo "$target"
+		return
+	    fi
+	done
+	((c++))
+    done
+}    
+
+__gobgpcli_table_list() {
+    local table_list=("local adj-in adj-out")
+    local table="$(__search_target "${table_list}")"
+    if [ -z "$table" ]; then
+	case "$cur" in
+	    *)
+		COMPREPLY=( $( compgen -W "${table_list}" -- "$cur") )
+		;;
+	esac
+	return
+    fi
+    COMPREPLY=( $( compgen -W "ipv4 ipv6 evpn" -- "$cur") )
+    return
+}
+
+__gobgpcli_neighbr_list() {
+    local neighbor_list=( $(__gobgpcli_q show --quiet neighbors) )
+    local target="$(__search_target "${neighbor_list[*]}")"
+    if [ -z "$target" ]; then
+	case "$cur" in
+	    *)
+		COMPREPLY=( $( compgen -W "${neighbor_list[*]}" -- "$cur") )
+		;;
+	esac
+	__ltrim_colon_completions "$cur"
+	return
+    fi
+}
+
+_gobgpcli_show_neighbor() {
+   __gobgpcli_neighbr_list
+    if [ -z ${COMPREPLY} ]; then
+	__gobgpcli_table_list
+	return
+    fi
+}
+
+_gobgpcli_show_neighbors() {
+    case "$cur" in
+	*)
+	    ;;
+    esac
+    return
+}
+
+_gobgpcli_show_global() {
+    local targets="ipv4 ipv6 evpn"
+    local target="$(__search_target "$targets")"
+    if [ -z "$target" ]; then
+	case "$cur" in
+	    *)
+		COMPREPLY=( $( compgen -W "${targets[*]}" -- "$cur" ) )
+		;;
+	esac
+	return
+    fi
+}
+
+_gobgpcli_show() {
+    local targets="neighbor neighbors global"
+    local target="$(__search_target "$targets")"
+    if [ -z "$target" ]; then
+	case "$cur" in
+	    *)
+		COMPREPLY=( $( compgen -W "${targets[*]}" -- "$cur" ) )
+		;;
+	esac
+	return
+    fi
+    _gobgpcli_show_${target}
+}
+
+__gobgpcli_generic_reset() {
+    local targets="neighbor"
+    local target="$(__search_target "$targets")"
+    if [ -z "$target" ]; then
+	case "$cur" in
+	    *)
+		COMPREPLY=( $( compgen -W "${targets[*]}" -- "$cur" ) )
+		;;
+	esac
+	return
+    fi
+    __gobgpcli_neighbr_list
+}
+
+_gobgpcli_reset() {
+    __gobgpcli_generic_reset
+}
+
+_gobgpcli_softresetin() {
+    __gobgpcli_generic_reset
+}
+
+_gobgpcli_softresetout() {
+    __gobgpcli_generic_reset
+}
+
+_gobgpcli_shutdown() {
+    __gobgpcli_generic_reset
+}
+
+_gobgpcli_gobgpcli() {
+    case "$prev" in
+	-h)
+	    return
+	    ;;
+	*)
+	    ;;
+	esac
+
+	case "$cur" in
+	    -*)
+		COMPREPLY=( $( compgen -W "-h" -- "$cur" ) )
+		;;
+	    *)
+		COMPREPLY=( $( compgen -W "${commands[*]} help" -- "$cur" ) )
+		;;
+	esac
+}
+
+_gobgpcli() {
+    local commands=(
+	show
+	reset
+	softresetin
+	softresetout
+	shutdown
+    )
+
+    COMPREPLY=()
+    local cur prev words cword
+    _get_comp_words_by_ref -n : cur prev words cword
+
+    local command='gobgpcli'
+    local counter=1
+    while [ $counter -lt $cword ]; do
+	case "${words[$counter]}" in
+	    -h)
+		(( counter++ ))
+		;;
+	    -*)
+		;;
+	    *)
+		command="${words[$counter]}"
+		cpos=$counter
+		(( cpos++ ))
+		break
+		;;
+	esac
+	(( counter++ ))
+    done
+    local completions_func=_gobgpcli_${command}
+    declare -F $completions_func > /dev/null && $completions_func
+
+    return 0
+}
+
+complete -F _gobgpcli gobgpcli