From 42bb3074fe9632d7aa0fee825ad30d2083c3c629 Mon Sep 17 00:00:00 2001 From: Ran Benita Date: Tue, 9 Nov 2010 01:03:27 +0200 Subject: [PATCH] add bash completion for systemctl --system I've been playing recently with systemd on Arch, and had much fun. But soon, alas, my fingers started to ache from repeatedly writing systemctl restart some-long-service.service. So, I wrote a completion script. I figured other people may want to use it, so I prepared a patch against systemd-git (attached). There are some notes/disclaimers, however: - It requires bash>=4.0, sed, grep and awk. A bash-completion package is not strictly needed; sourcing the file is enough. - It wouldn't work properly with --session, as I had no way to test it. - It uses the output of systemctl list-units directly when that's enough, but also runs systemctl show when completing on some verbs (for example, to check for AllowIsolate=yes). This /may/ be somewhat slow once there are many units, since it calls a dbus method on each one. Is there a faster way to have that information? - The code is perhaps a bit long and messy; honestly, I blame the tool ;) One way to improve on the situation is to integrate some completion code in systemctl itself, the way e.g. gdbus, gsettings and django do it. This will allow for finer grained and faster completions, and it won't be necessary to keep the verb/option tables in sync with some other file. But it does mean adding all of this code in C. If this is acceptable, I'll try to have a go at it. Finally, a couple of completion tips I run into: - If you alias systemctl to, say, sctl, you get completions on that too by running to following command: complete -F _systemctl sctl - Add the following line to your .inputrc, to have the completion show after only a single tab press: set show-all-if-ambiguous on It makes the shell quite more pleasant. Hope it's good enough! Ran --- Makefile.am | 4 + src/systemctl-bash-completion.sh | 145 +++++++++++++++++++++++++++++++ 2 files changed, 149 insertions(+) create mode 100644 src/systemctl-bash-completion.sh diff --git a/Makefile.am b/Makefile.am index 69faa5581a7..f2d20c7db32 100644 --- a/Makefile.am +++ b/Makefile.am @@ -26,6 +26,7 @@ udevrulesdir=@udevrulesdir@ pamlibdir=@pamlibdir@ pkgconfigdatadir=$(datadir)/pkgconfig polkitpolicydir=$(datadir)/polkit-1/actions +bashcompletiondir=${sysconfdir}/bash_completion.d # Our own, non-special dirs pkgsysconfdir=$(sysconfdir)/systemd @@ -161,6 +162,9 @@ dbusinterface_DATA = \ org.freedesktop.systemd1.Swap.xml \ org.freedesktop.systemd1.Path.xml +dist_bashcompletion_DATA = \ + src/systemctl-bash-completion.sh + dist_tmpfiles_DATA = \ tmpfiles.d/systemd.conf \ tmpfiles.d/x11.conf diff --git a/src/systemctl-bash-completion.sh b/src/systemctl-bash-completion.sh new file mode 100644 index 00000000000..53f8e52aaae --- /dev/null +++ b/src/systemctl-bash-completion.sh @@ -0,0 +1,145 @@ +# This file is part of systemd. +# +# Copyright 2010 Ran Benita +# +# systemd is free software; you can redistribute it and/or modify it +# under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# systemd is distributed in the hope that it will be useful, but +# WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +# General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with systemd; If not, see . + +__contains_word () { + local word=$1; shift + for w in $*; do [[ $w = $word ]] && return 0; done + return 1 +} + +__filter_units_by_property () { + local property=$1 value=$2 ; shift ; shift + local -a units=( $* ) + local -a props=( $(systemctl show --property "$property" -- ${units[*]} | grep -v ^$) ) + for ((i=0; $i < ${#units[*]}; i++)); do + if [[ "${props[i]}" = "$property=$value" ]]; then + echo "${units[i]}" + fi + done +} + +__get_all_units () { systemctl list-units --full --all | awk ' {print $1}' ; } +__get_active_units () { systemctl list-units --full | awk ' {print $1}' ; } +__get_inactive_units () { systemctl list-units --full --all | awk '$3 == "inactive" {print $1}' ; } +__get_failed_units () { systemctl list-units --full | awk '$3 == "failed" {print $1}' ; } + +_systemctl () { + local cur=${COMP_WORDS[COMP_CWORD]} prev=${COMP_WORDS[COMP_CWORD-1]} + local verb comps + + local -A OPTS=( + [STANDALONE]='--all -a --defaults --fail --force -f --full --global + --help -h --no-ask-password --no-block --no-reload --no-wall + --order --require --quiet -q --session --system --version' + [ARG]='--kill-mode --kill-who --property -p --signal -s --type -t' + ) + + if __contains_word "$prev" ${OPTS[ARG]}; then + case $prev in + --signal|-s) + comps=$(compgen -A signal | grep '^SIG' | grep -Ev 'RTMIN|RTMAX|JUNK') + ;; + --type|-t) + comps='automount device mount path service snapshot socket swap target timer' + ;; + --kill-who) + comps='all control main' + ;; + --kill-mode) + comps='control-group process process-group' + ;; + --property|-p) + comps='' + ;; + esac + COMPREPLY=( $(compgen -W "$comps" -- "$cur") ) + return 0 + fi + + + if [[ "$cur" = -* ]]; then + COMPREPLY=( $(compgen -W "${OPTS[*]}" -- "$cur") ) + return 0 + fi + + local -A VERBS=( + [ALL_UNITS]='enable disable is-active is-enabled status show' + [FAILED_UNITS]='reset-failed' + [STARTABLE_UNITS]='start restart reload-or-restart' + [STOPPABLE_UNITS]='stop kill try-restart condrestart' + [ISOLATEBLE_UNITS]='isolate' + [RELOADABLE_UNITS]='reload reload-or-try-restart force-reload' + [JOBS]='cancel' + [SNAPSHOTS]='delete' + [ENVS]='set-environment unset-environment' + [STANDALONE]='daemon-reexec daemon-reload default dot dump emergency exit halt kexec + list-jobs list-units poweroff reboot rescue show-environment' + [NAME]='snapshot load' + ) + + local verb + for ((i=0; $i <= $COMP_CWORD; i++)); do + if __contains_word "${COMP_WORDS[i]}" ${VERBS[*]} && + ! __contains_word "${COMP_WORDS[i-1]}" ${OPTS[ARG}]}; then + verb=${COMP_WORDS[i]} + break + fi + done + + if [[ -z $verb ]]; then + comps="${VERBS[*]}" + + elif __contains_word "$verb" ${VERBS[ALL_UNITS]}; then + comps=$( __get_all_units ) + + elif __contains_word "$verb" ${VERBS[STARTABLE_UNITS]}; then + comps=$( __filter_units_by_property CanStart yes \ + $( __get_inactive_units | grep -Ev '\.(device|snapshot)$' )) + + elif __contains_word "$verb" ${VERBS[STOPPABLE_UNITS]}; then + comps=$( __filter_units_by_property CanStop yes \ + $( __get_active_units ) ) + + elif __contains_word "$verb" ${VERBS[RELOADABLE_UNITS]}; then + comps=$( __filter_units_by_property CanReload yes \ + $( __get_active_units ) ) + + elif __contains_word "$verb" ${VERBS[ISOLATABLE_UNITS]}; then + comps=$( __filter_units_by_property AllowIsolate yes \ + $( __get_all_units ) ) + + elif __contains_word "$verb" ${VERBS[FAILED_UNITS]}; then + comps=$( __get_failed_units ) + + elif __contains_word "$verb" ${VERBS[STANDALONE]} ${VERBS[NAME]}; then + comps='' + + elif __contains_word "$verb" ${VERBS[JOBS]}; then + comps=$( systemctl list-jobs | awk '{print $1}' ) + + elif __contains_word "$verb" ${VERBS[SNAPSHOTS]}; then + comps=$( systemctl list-units --type snapshot --full --all | awk '{print $1}' ) + + elif __contains_word "$verb" ${VERBS[ENVS]}; then + comps=$( systemctl show-environment | sed 's_\([^=]\+=\).*_\1_' ) + compopt -o nospace + fi + + COMPREPLY=( $(compgen -W "$comps" -- "$cur") ) + return 0 +} +complete -F _systemctl systemctl