HOME

# shellcheck shell=bash
# shellcheck disable=SC2207
# Based on the completion from the Mandriva dkms package.

# This function completes available kernels
_kernels()
{
	COMPREPLY=( $( cd /lib/modules && compgen -d -- "$cur" ) )
}

# complete on full directory names under $1
_subdirectories()
{
	COMPREPLY=( $( cd "$1" && compgen -d -- "$cur" ) )
}

# complete on $2 part of filenames matching pattern $1 under /usr/src
_filename_parts()
{
    # shellcheck disable=SC1003
    # TODO: figure out what is going on here
	COMPREPLY=( $( command ls -F /usr/src/ 2>/dev/null | grep -E '^'"$1"'/$' \
		| sed -r -e 's/^([^-]+)-(.+)\/$/\'"$2"'/' | grep "^$cur" ) )
}

_dkms()
{
	local cur prev command module i

	COMPREPLY=()
	cur=${COMP_WORDS[COMP_CWORD]}

	if [[ $COMP_CWORD -eq 1 ]] ; then
		COMPREPLY=( $( compgen -W "add remove build unbuild install uninstall autoinstall \
			match mktarball ldtarball generate_mok \
			status" -- "$cur" ) )
	else
		prev=${COMP_WORDS[COMP_CWORD-1]}
		command=${COMP_WORDS[1]}
		case $prev in
			-a)
				COMPREPLY=( $( compgen -W "$(uname -m)" -- "$cur" ) )
				return 0
				;;
			-m)
				if [ "$command" = 'add' ]; then
					_filename_parts '.*-.*' 1
				else
					_subdirectories /var/lib/dkms
				fi
				return 0
				;;
			-v)
				for (( i=1; i < COMP_CWORD; i++ )); do
					if [[ "${COMP_WORDS[i]}" == -m ]]; then
						module=${COMP_WORDS[i+1]}
						break
					fi
				done
				if [ -n "$module" ]; then
					if [ "$command" = 'add' ]; then
						_filename_parts "$module-.*" 2
					else
						_subdirectories "/var/lib/dkms/$module"
					fi
					return 0
				fi
				;;
			-k|--templatekernel)
				_kernels
				return 0
				;;
			-c|--spec|--archive|--config)
				_filedir
				return 0
				;;
			--kernelsourcedir|--dkmstree|--sourcetree|--installtree)
				_filedir -d
				return 0
				;;
		esac


		if [[ "$cur" == -* ]]; then
			case $command in
				add)
					options='-c --rpm_safe_upgrade'
					;;
				remove)
					options='--rpm_safe_upgrade'
					;;
				build)
					options='--config --force'
					;;
				unbuild)
					options=''
					;;
				install)
					options='--force'
					;;
				uninstall)
					options=''
					;;
				autoinstall)
					options=''
					;;
				match)
					options='--templatekernel'
					;;
				mktarball)
					options='--source-only --binaries-only'
					;;
				ldtarball)
					options='--archive --force'
					;;
				status)
					options=''
				    ;;
				generate_mok)
					options=''
				    ;;
			esac

			options="$options -m -v -k -a --arch -q --quiet -V \
				--version --all --kernelsourcedir \
				--directive"

			COMPREPLY=( $( compgen -W "$options" -- "$cur" ) )
		fi
	fi
}
complete -F _dkms dkms