add zsh completion

add zsh completion for lsinitcpio and mkinitcpio

Signed-off-by: Daniel Wallace <danielwallace@gtmanfred.com>

2 files changed, 65 insertions, 0 deletions

diff --git a/Makefile b/Makefile
index 5322ff1..633dd63 100644
--- a/Makefile
+++ b/Makefile
@@ -5,6 +5,7 @@ VERSION = $(shell if test -f VERSION; then cat VERSION; else git describe | sed
 DIRS = \
 	/usr/bin \
 	/usr/share/bash-completion/completions \
+	/usr/share/zsh/site-functions \
 	/etc/mkinitcpio.d \
 	/usr/lib/initcpio/hooks \
 	/usr/lib/initcpio/install \
@@ -52,6 +53,7 @@ install: all
 	install -m644 man/lsinitcpio.1 $(DESTDIR)/usr/share/man/man1/lsinitcpio.1
 	install -m644 bash-completion $(DESTDIR)/usr/share/bash-completion/completions/mkinitcpio
 	ln -s mkinitcpio $(DESTDIR)/usr/share/bash-completion/completions/lsinitcpio
+	install -m644 zsh-completion $(DESTDIR)/usr/share/zsh/site-functions/_mkinitcpio
 
 doc: $(MANPAGES)
 man/%: man/%.txt Makefile
diff --git a/zsh-completion b/zsh-completion
new file mode 100644
index 0000000..4d1e435
--- /dev/null
+++ b/zsh-completion
@@ -0,0 +1,63 @@
+#compdef lsinitcpio mkinitcpio
+
+_detect_kver() {
+    local __ kver_validator='^[[:digit:]]+(\.[[:digit:]]+)+'
+    offset=$(hexdump -s 526 -n 2 -e '"%0d"' "$1" 2>/dev/null) || return 1
+    read kver __ < \
+        <(dd if="$1" bs=1 count=127 skip=$(( offset + 0x200 )) 2>/dev/null)
+    [[ $kver =~ $kver_validator ]] && printf "$kver"
+}
+
+_find_kernel_versions() {
+    local -a version
+    local filedata
+
+    for f in /boot/*; do
+        # only match regular files which pass validation
+        if [[ ! -L $f && -f $f ]] && kver=$(_detect_kver "$f"); then
+            version+=("$kver" "$f")
+        fi
+    done
+
+    compadd ${(u)version}
+}
+
+case $service in
+    lsinitcpio)
+        _arguments : \
+            '(-a --analyze)'{-a,--analyze}'[analyze contents of image]' \
+            '(-c --config)'{-c,--config}'[show configuration file image was built with]' \
+            '(-l --list)'{-l,--list}'[list contents of the image (default)]' \
+            '(-x --extract)'{-x,--extract}'[extract image to disk]' \
+            '(-h --help)'{-h,--help}'[display this help]' \
+            '(-n --nocolor)'{-n,--nocolor}'[disable colorized output]' \
+            '(-V --version)'{-V,--version}'[display version information]' \
+            '(-v --verbose)'{-v,--verbose}'[more verbose output]' \
+            ':files:_files'
+        ;;
+    mkinitcpio)
+        _arguments : \
+            '(-A --addhooks)'{-A,--addhooks}'[Add specified hooks, comma separated, to image]::usr hooks:_path_files -W /usr/lib/initcpio/install::lib hooks:_path_files -W /lib/initcpio/install' \
+            '(-c --config)'{-c,--config}'[Use alternate config file. (default: /etc/mkinitcpio.conf)]:config files:_files' \
+            '(-g --generate)'{-g,--generate}'[Generate cpio image and write to specified path]:config files:_files' \
+            '(-H --hookhelp)'{-H,--hookhelp}'[Display help for given hook and exit]::usr hooks:_path_files -W /usr/lib/initcpio/install::lib hooks:_path_files -W /lib/initcpio/install' \
+            '(-h --help)'{-h,--help}'[Display this message and exit]' \
+            '(-k --kernel)'{-k,--kernel}'[Use specified kernel version]:kernel versions:_find_kernel_versions' \
+            '(-L --listhooks)'{-L,--listhooks}'[List all available hooks]' \
+            '(-M --automods)'{-M,--automods}'[Display modules found via autodetection]' \
+            '(-n --nocolor)'{-n,--nocolor}'[Disable colorized output messages]' \
+            '(-p --preset)'{-p,--preset}'[Build specified preset from /etc/mkinitcpio.d]:presets:_files -g "*(\:r)" -W /etc/mkinitcpio.d' \
+            '(-r --moduleroot)'{-r,--moduleroot}'[Root directory for modules (default: /)]:directories:_files -/' \
+            '(-S --skiphooks)'{-S,--skiphooks}'[Skip specified hooks, comma-separated, during build]::usr hooks:_path_files -W /usr/lib/initcpio/install::lib hooks:_path_files -W /lib/initcpio/install' \
+            '(-s --save)'{-s,--save}'[Save build directory. (default: no)]' \
+            '(-t --builddir)'{-t,--builddir}'[Use DIR as the temporary build directory]:directories:_files -/' \
+            '(-V --version)'{-V,--version}'[Display version information and exit]' \
+            '(-v --verbose)'{-v,--verbose}'[Verbose output (default: no)]' \
+            '(-z --compress)'{-z,--compress}'[Use an alternate compressor on the image]'
+        ;;
+    *)
+        return 1;
+        ;;
+esac
+
+# vim: set et ts=4 sw=4 ft=zsh: