shell-completion: support --json option for hostnamectl

yuwata · yuwata · commit f8cd3f610fc0 · 2021-05-14T09:18:29.000+09:00
diff --git a/shell-completion/bash/hostnamectl b/shell-completion/bash/hostnamectl
@@ -25,11 +25,38 @@ __contains_word () {
     done
 }
 
+__get_machines() {
+    local a b
+    machinectl list --full --no-legend --no-pager 2>/dev/null |
+        { while read a b; do echo " $a"; done; };
+}
+
 _hostnamectl() {
     local i verb comps
     local cur=${COMP_WORDS[COMP_CWORD]} prev=${COMP_WORDS[COMP_CWORD-1]}
-    local OPTS='-h --help --version --transient --static --pretty
-                --no-ask-password -H --host -M --machine'
+    local -A OPTS=(
+        [STANDALONE]='-h --help --version --transient --static --pretty --no-ask-password'
+        [ARG]='-H --host -M --machine --json'
+    )
+
+    if __contains_word "$prev" ${OPTS[ARG]} ${OPTS[ARGUNKNOWN]}; then
+        case $prev in
+            --host|-H)
+                comps=$(compgen -A hostname)
+                ;;
+            --machine|-M)
+                comps=$( __get_machines )
+                ;;
+            --json)
+                comps=$( hostnamectl --json=help 2>/dev/null )
+                ;;
+            *)
+                return 0
+                ;;
+        esac
+        COMPREPLY=( $(compgen -W '$comps' -- "$cur") )
+        return 0
+    fi
 
     if [[ $cur = -* ]]; then
         COMPREPLY=( $(compgen -W '${OPTS[*]}' -- "$cur") )
diff --git a/shell-completion/zsh/_hostnamectl b/shell-completion/zsh/_hostnamectl
@@ -75,6 +75,13 @@ _hostnamectl_commands() {
     fi
 }
 
+(( $+functions[_hostnamectl_get_json] )) || _hostnamectl_get_json()
+{
+    local -a _json_forms
+    _json_forms=( $(hostnamectl --json=help 2>/dev/null) )
+    _values 'format' $_json_forms
+}
+
 _arguments -s \
     {-h,--help}'[Show this help]' \
     '--version[Show package version]' \
@@ -84,4 +91,5 @@ _arguments -s \
     '--no-ask-password[Do not prompt for password]' \
     {-H+,--host=}'[Operate on remote host]:userathost:_sd_hosts_or_user_at_host' \
     {-M+,--machine=}'[Operate on local container]:machines:_sd_machines' \
+    '--json[Shows output formatted as JSON]:format:_hostnamectl_get_json' \
     '*::hostnamectl commands:_hostnamectl_commands'
