# cpustat tab completion for bash.
#
# Copyright (C) 2020-2021 Canonical
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.

_cpustat()
{
	local cur prev words cword
	_init_completion || return

	#cur=${COMP_WORDS[COMP_CWORD]}
	#prev=${COMP_WORDS[COMP_CWORD-1]}
	#COMREPLY=()

	case $prev in
        '-n')
		COMPREPLY=( $(compgen -W "maxtasks" -- $cur) )
                return 0
                ;;
        '-p')	COMPREPLY=( $(compgen -W '$(command ps axo pid | sed 1d) ' $cur ) )
		return 0
		;;
        '-r')
		_filedir
		return 0
		;;
        '-t')
		COMPREPLY=( $(compgen -W "threshold" -- $cur) )
                return 0
                ;;
	esac

	case "$cur" in
                -*)
                        OPTS="-h -a -c -d -D -g -i -l -q -s -S -T -x -X"
                        COMPREPLY=( $(compgen -W "${OPTS[*]}" -- $cur) )
                        return 0
                        ;;
        esac
	return 0
}

# load the completion
complete -F _cpustat cpustat
