#!/bin/bash completion for oci-hdd-maint

_oci_hdd_maint() {
	local cur prev cmd_name
	COMPREPLY=()
	cur="${COMP_WORDS[COMP_CWORD]}"
	prev="${COMP_WORDS[COMP_CWORD-1]}"
	cmd_name="${COMP_WORDS[1]}"

	opts="-h -l -s -y -u -r"

	case "${cmd_name}" in
	"-s")
		COMPREPLY=( $(compgen -W "0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16" -- ${cur}) )
		return 0
	;;
	"-y")
		COMPREPLY=( $(compgen -W "0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 /dev/sda /dev/sdb /dev/sdc /dev/sdd /dev/sde /dev/sdf /dev/sdg /dev/sdh /dev/sdi /dev/sdj /dev/sdk /dev/sdl /dev/sdm /dev/sdn /dev/sdo /dev/sdp" -- ${cur}) )
		return 0
	;;
	"-u")
		COMPREPLY=( $(compgen -W "0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16" -- ${cur}) )
		return 0
	;;
	"-r")
		case ${COMP_CWORD} in
		2)
			DEVICE_LIST=$(cat /etc/fstab | grep -E '^#UUID' | awk '{print $2}' | sed -e s@/srv/node/@@ | tr '\n' ' ')
			COMPREPLY=( $(compgen -W "${DEVICE_LIST}" -- ${cur}) )
			return 0
			;;
		*)
			return 0
			;;
		esac
	;;
	*)
	;;
	esac

	COMPREPLY=($(compgen -W "${opts}" -- "${cur}"))
}

complete -F _oci_hdd_maint oci-hdd-maint
