refactor(_comp_count_args): pass arguments through options

Author: akinomyoga

bash_completion

@@ -2189,23 +2189,43 @@ _comp_get_first_arg()
 }
 
 # This function counts the number of args, excluding options
-# @param $1 chars  Characters out of $COMP_WORDBREAKS which should
-#     NOT be considered word breaks. See _comp__reassemble_words.
-# @param $2 glob   Options whose following argument should not be counted
-# @param $3 glob   Options that should be counted as args
+#
+# Options:
+#     -n CHARS  Characters out of $COMP_WORDBREAKS which should
+#               NOT be considered word breaks. See
+#               _comp__reassemble_words.
+#     -a GLOB   Options whose following argument should not be counted
+#     -i GLOB   Options that should be counted as args
+#
 # @var[out] ret    Return the number of arguments
 # @since 2.12
 _comp_count_args()
 {
-    local i cword words
-    _comp__reassemble_words "${1-}<>&" words cword
+    local has_optarg="" has_exclude="" exclude="" glob_include=""
+    local OPTIND=1 OPTARG="" OPTERR=0 _opt
+    while getopts ':a:n:i:' _opt "$@"; do
+        case $_opt in
+            a) has_optarg=$OPTARG ;;
+            n) has_exclude=set exclude+=$OPTARG ;;
+            i) glob_include=$OPTARG ;;
+            *)
+                echo "bash_completion: $FUNCNAME: usage error" >&2
+                return 2
+                ;;
+        esac
+    done
+    shift "$((OPTIND - 1))"
+
+    local cword words
+    _comp__reassemble_words "$exclude<>&" words cword
 
+    local i
     ret=1
     for ((i = 1; i < cword; i++)); do
         # shellcheck disable=SC2053
-        if [[ ${2-} && ${words[i]} == ${2-} ]]; then
+        if [[ $has_optarg && ${words[i]} == $has_optarg ]]; then
             ((i++))
-        elif [[ ${words[i]} != -?* || ${3-} && ${words[i]} == ${3-} ]]; then
+        elif [[ ${words[i]} != -?* || $glob_include && ${words[i]} == $glob_include ]]; then
             ((ret++))
         elif [[ ${words[i]} == -- ]]; then
             ((ret += cword - i - 1))

completions/7z

@@ -85,7 +85,7 @@ _comp_cmd_7z()
     fi
 
     local ret
-    _comp_count_args "="
+    _comp_count_args -n "="
     if ((ret == 2)); then
         _filedir_xspec unzip "${@:2}"
         # TODO: parsing 7z i output?

completions/arp

@@ -34,7 +34,7 @@ _comp_cmd_arp()
     fi
 
     local ret
-    _comp_count_args "" "@(--device|--protocol|--file|--hw-type|-${noargopts}[iApfHt])"
+    _comp_count_args -a "@(--device|--protocol|--file|--hw-type|-${noargopts}[iApfHt])"
     case $ret in
         1)
             local ips=$("$1" -an | command sed -ne \

completions/chmod

@@ -28,7 +28,7 @@ _comp_cmd_chmod()
     fi
 
     local ret
-    _comp_count_args "" "" "$modearg"
+    _comp_count_args -i "$modearg"
 
     case $ret in
         1) ;; # mode

completions/chown

@@ -32,7 +32,7 @@ _comp_cmd_chown()
         local ret
 
         # The first argument is a usergroup; the rest are filedir.
-        _comp_count_args :
+        _comp_count_args -n :
 
         if ((ret == 1)); then
             _comp_compgen_usergroup -u

completions/cryptsetup

@@ -37,7 +37,7 @@ _comp_cmd_cryptsetup()
     local ret
     if _comp_get_first_arg; then
         local arg=$ret
-        _comp_count_args "" "-${noargopts}[chslSbopitTdM]"
+        _comp_count_args -a "-${noargopts}[chslSbopitTdM]"
         local args=$ret
         case $arg in
             open | create | luksOpen | loopaesOpen | tcryptOpen)

completions/gpgv

@@ -20,7 +20,7 @@ _comp_cmd_gpgv()
     esac
 
     local ret
-    _comp_count_args "" "--@(weak-digest|*-fd|keyring|homedir)"
+    _comp_count_args -a "--@(weak-digest|*-fd|keyring|homedir)"
     local args=$ret
 
     if [[ $cur == -* && $args -eq 1 ]]; then

completions/ifup

@@ -29,7 +29,7 @@ _comp_cmd_ifupdown()
     fi
 
     local ret
-    _comp_count_args "" "@(--allow|-i|--interfaces|--state-dir|-X|--exclude|-o)"
+    _comp_count_args -a "@(--allow|-i|--interfaces|--state-dir|-X|--exclude|-o)"
 
     if ((ret == 1)); then
         _comp_compgen_configured_interfaces

completions/jq

@@ -66,7 +66,7 @@ _comp_cmd_jq()
     local ret
     # TODO: DTRT with args taking 2 options
     # -f|--from-file are not counted here because they supply the filter
-    _comp_count_args "" "@(--arg|--arg?(json|file)|--slurpfile|--indent|--run-tests|-${noargopts}L)"
+    _comp_count_args -a "@(--arg|--arg?(json|file)|--slurpfile|--indent|--run-tests|-${noargopts}L)"
 
     # 1st arg is filter
     ((ret == 1)) && return

completions/jsonschema

@@ -21,7 +21,7 @@ _comp_cmd_jsonschema()
     fi
 
     local ret
-    _comp_count_args "" "-*"
+    _comp_count_args -a "-*"
     ((ret == 1)) || return
     _comp_compgen_filedir '@(json|schema)'
 } &&