feat: add shell completions for atomic-upgrade

Author: linux_name2

Makefile

@@ -42,10 +42,14 @@ install:
 		$(DESTDIR)$(ZSH_COMPDIR)/_atomic-gc
 	install -Dm644 completions/_atomic-rebuild-uki \
 		$(DESTDIR)$(ZSH_COMPDIR)/_atomic-rebuild-uki
+	install -Dm644 completions/_atomic-upgrade \
+		$(DESTDIR)$(ZSH_COMPDIR)/_atomic-upgrade
 	install -Dm644 completions/atomic-gc.bash \
 		$(DESTDIR)$(BASH_COMPDIR)/atomic-gc
 	install -Dm644 completions/atomic-rebuild-uki.bash \
 		$(DESTDIR)$(BASH_COMPDIR)/atomic-rebuild-uki
+	install -Dm644 completions/atomic-upgrade.bash \
+		$(DESTDIR)$(BASH_COMPDIR)/atomic-upgrade
 
 	install -Dm644 hooks/00-block-direct-upgrade.hook \
 		$(DESTDIR)$(HOOKSDIR)/00-block-direct-upgrade.hook
@@ -73,8 +77,10 @@ uninstall:
 	rm -rf $(DESTDIR)$(LIBDIR)/
 	rm -f  $(DESTDIR)$(ZSH_COMPDIR)/_atomic-gc
 	rm -f  $(DESTDIR)$(ZSH_COMPDIR)/_atomic-rebuild-uki
+	rm -f  $(DESTDIR)$(ZSH_COMPDIR)/_atomic-upgrade
 	rm -f  $(DESTDIR)$(BASH_COMPDIR)/atomic-gc
 	rm -f  $(DESTDIR)$(BASH_COMPDIR)/atomic-rebuild-uki
+	rm -f  $(DESTDIR)$(BASH_COMPDIR)/atomic-upgrade
 	rm -f  $(DESTDIR)$(HOOKSDIR)/00-block-direct-upgrade.hook
 	rm -f  $(DESTDIR)$(PREFIX)/local/bin/pacman
 	rm -f  $(DESTDIR)$(MANDIR)/man8/atomic-upgrade.8

completions/_atomic-upgrade

@@ -0,0 +1,16 @@
+#compdef atomic-upgrade
+
+_atomic-upgrade() {
+    _arguments \
+        '(- *)'{-h,--help}'[Show help]' \
+        '(- *)'{-V,--version}'[Show version]' \
+        '(-n --dry-run)'{-n,--dry-run}'[Show what would be done]' \
+        '(-t --tag)'{-t,--tag}'[Custom tag for snapshot]:tag:' \
+        '--no-gc[Skip garbage collection]' \
+        '--separate-home[Create isolated /home subvolume (requires --tag)]' \
+        '--copy-files[Copy files into new home]:file list:' \
+        '(-)--[Command separator]' \
+        '*:command:_command_names'
+}
+
+_atomic-upgrade "$@"

completions/atomic-upgrade.bash

@@ -0,0 +1,26 @@
+# bash completion for atomic-upgrade
+
+_atomic_upgrade() {
+    local cur prev words cword
+    _init_completion || return
+
+    local i has_dashdash=0
+    for ((i=1; i < cword; i++)); do
+        [[ "${words[i]}" == "--" ]] && { has_dashdash=1; break; }
+    done
+
+    if [[ $has_dashdash -eq 1 ]]; then
+        COMPREPLY=( $(compgen -c -- "$cur") )
+        return
+    fi
+
+    case "$prev" in
+        -t|--tag|--copy-files) return ;;
+    esac
+
+    if [[ "$cur" == -* ]]; then
+        COMPREPLY=( $(compgen -W "-h --help -V --version -n --dry-run -t --tag --no-gc --separate-home --copy-files --" -- "$cur") )
+    fi
+}
+
+complete -F _atomic_upgrade atomic-upgrade