add bash/zsh completion

4 files changed, 81 insertions, 5 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 24bcc9f1bc4c..56ac25260019 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,21 +1,26 @@
 pkgbase = prime
 	pkgdesc = PRIME Render Offload Launcher. Please visit https://download.nvidia.com/XFree86/Linux-x86_64/435.21/README/primerenderoffload.html to manually configure your Device.
 	pkgver = 0.1
-	pkgrel = 6
+	pkgrel = 7
 	url = https://download.nvidia.com/XFree86/Linux-x86_64/435.21/README/primerenderoffload.html
 	install = prime.install
 	arch = any
 	depends = sh
 	optdepends = vulkan-intel: for vulkan support
 	optdepends = lib32-vulkan-intel: for vulkan support
+	optdepends = zsh: zsh completion
 	conflicts = bumblebee
 	conflicts = optimus-manager
 	source = prime
 	source = prime.install
 	source = 10-nvidia.conf
+	source = zsh_prime
+	source = bash_prime
 	md5sums = e19cc7136f1e0c070de7da4a65799351
 	md5sums = d5fbf5400bdd0705fcda1ce78ec12077
 	md5sums = 6bc9250bd74dcfd23a9ad4e2039681d4
+	md5sums = bbc3d71244a5fc5d4d6caeea941ff91f
+	md5sums = 5451a4458227a2ffe534713e7fba38f8
 
 pkgname = prime
 
diff --git a/PKGBUILD b/PKGBUILD
index 8ebc245711ed..e5a13a725c99 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -4,21 +4,26 @@ pkgname=prime
 pkgdesc="PRIME Render Offload Launcher. Please visit https://download.nvidia.com/XFree86/Linux-x86_64/435.21/README/primerenderoffload.html to manually configure your Device."
 url="https://download.nvidia.com/XFree86/Linux-x86_64/435.21/README/primerenderoffload.html"
 pkgver=0.1
-pkgrel=6
+pkgrel=7
 arch=('any')
 # license=("nothing")
 conflicts=(bumblebee optimus-manager)
 depends=(sh)
 optdepends=(
 	"vulkan-intel: for vulkan support"
-	"lib32-vulkan-intel: for vulkan support")
+	"lib32-vulkan-intel: for vulkan support"
+	"zsh: zsh completion")
 
 source=(prime
 		prime.install
-		10-nvidia.conf)
+		10-nvidia.conf
+		zsh_prime
+		bash_prime)
 md5sums=('e19cc7136f1e0c070de7da4a65799351'
          'd5fbf5400bdd0705fcda1ce78ec12077'
-         '6bc9250bd74dcfd23a9ad4e2039681d4')
+         '6bc9250bd74dcfd23a9ad4e2039681d4'
+         'bbc3d71244a5fc5d4d6caeea941ff91f'
+         '5451a4458227a2ffe534713e7fba38f8')
 
 install=prime.install
 
@@ -26,7 +31,12 @@ package()
 {
 	mkdir -p "${pkgdir}"/usr/bin/
 	mkdir -p "${pkgdir}"/usr/share/X11/xorg.conf.d/
+	mkdir -p "${pkgdir}"/usr/share/zsh/site-functions/
+	mkdir -p "${pkgdir}"/usr/share/bash-completion/completions
+
 	install -m a+xr ./prime "${pkgdir}"/usr/bin/
 	install 10-nvidia.conf "${pkgdir}"/usr/share/X11/xorg.conf.d/
+	install zsh_prime "${pkgdir}"/usr/share/zsh/site-functions/_prime
+	install bash_prime "${pkgdir}"/usr/share/zsh/site-functions/prime
 }
 
diff --git a/bash_prime b/bash_prime
new file mode 100644
index 000000000000..6aed6867d3a1
--- /dev/null
+++ b/bash_prime
@@ -0,0 +1,57 @@
+# bash completion for prime
+
+_prime()
+{
+    local cur prev words cword split
+    _init_completion -s || return
+    #
+    # local i mode=normal
+    # [[ $1 == *sudoedit ]] && mode=edit
+    #
+    # [[ $mode == normal ]] &&
+    for (( i=1; i <= cword; i++ )); do
+        if [[ ${words[i]} != -* ]]; then
+            # local PATH=$PATH:/sbin:/usr/sbin:/usr/local/sbin
+            local root_command=${words[i]}
+            _command_offset $i
+            return
+        fi
+	done
+    #     if [[ ${words[i]} == -@(!(-*)e*|-edit) ]]; then
+    #         mode=edit
+    #         break
+    #     fi
+    #     [[ ${words[i]} == \
+    #         -@(user|other-user|group|close-from|prompt|!(-*)[uUgCp]) ]] \
+    #         && ((i++))
+    # done
+    #
+    # case "$prev" in
+    #     --user|--other-user|-!(-*)[uU])
+    #         COMPREPLY=( $(compgen -u -- "$cur") )
+    #         return
+    #         ;;
+    #     --group|-!(-*)g)
+    #         COMPREPLY=( $(compgen -g -- "$cur") )
+    #         return
+    #         ;;
+    #     --close-from|--prompt|-!(-*)[Cp])
+    #         return
+    #         ;;
+    # esac
+    #
+    # $split && return
+    #
+    # if [[ "$cur" == -* ]]; then
+    #     local opts=$(_parse_help "$1")
+    #     COMPREPLY=( $(compgen -W '${opts:-$(_parse_usage "$1")}' -- "$cur") )
+    #     [[ $COMPREPLY == *= ]] && compopt -o nospace
+    #     return
+    # fi
+    # if [[ $mode == edit ]]; then
+    #     _filedir
+    # fi
+} &&
+complete -F _prime prime
+
+# ex: filetype=sh
diff --git a/zsh_prime b/zsh_prime
new file mode 100644
index 000000000000..8773482534d4
--- /dev/null
+++ b/zsh_prime
@@ -0,0 +1,4 @@
+#compdef prime
+
+_arguments -s -S '*::arguments:{ _normal }'
+