#!/bin/bash
#   Software Updater - autocompletion script
#
#	  Copyright © 2018 Intel Corporation.
#
#   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, version 2 or later of the License.
#
#   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, see <http://www.gnu.org/licenses/>.
#


#declares the completion function
_swupd()
{
	# $1 is the command being completed, $2 is the current word being expanded
	local opts IFS=$' \t\n'
	local -i i installed
	local global="--help --url --contenturl --versionurl --port --path --format --nosigcheck --ignore-time --statedir --certpath  --time --no-scripts --no-boot-update --max-parallel-downloads --max-retries --retry-delay --json-output  --allow-insecure-http --debug --verbose --quiet --no-progress --wait-for-scripts"
	COMPREPLY=()
	for ((i=COMP_CWORD-1;i>=0;i--))
	do case "${COMP_WORDS[$i]}" in
		("$1")
		opts="--help --version autoupdate bundle-add bundle-remove
		bundle-list bundle-info hashdump update diagnose check-update search
		search-file info clean mirror os-install repair 3rd-party"
		break;;
		("info")
		opts="$global "
		break;;
		("autoupdate")
		opts="$global --help --enable --disable "
		break;;
		("check-update")
		opts="$global "
		break;;
		("update")
		opts="$global --download --status --force --migrate --allow-mix-collisions --keepcache --update-search-file-index "
		break;;
		("bundle-add")
		opts="$global --skip-diskspace-check --skip-optional "
		break;;
		("bundle-remove")
		opts="$global --force --recursive "
		break;;
		("bundle-list")
		opts="$global --all --deps --has-dep --status "
		break;;
		("bundle-info")
		opts="$global --dependencies --files --version "
		break;;
		("search")
		opts="--help --all --quiet --verbose "
		break;;
		("search-file")
		opts="$global --version --library --binary --top --csv --init --order "
		break;;
		("diagnose")
		opts="$global --version --picky --picky-tree --picky-whitelist --quick --force --extra-files-only --bundles --file "
		break;;
		("repair")
		opts="$global --version --picky --picky-tree --picky-whitelist --quick --force --extra-files-only --bundles --file "
		break;;
		("os-install")
		opts="$global --version --force --bundles --statedir-cache --download --skip-optional"
		break;;
		("mirror")
		opts="$global --set --unset "
		break;;
		("clean")
		opts="$global --all --dry-run "
		break;;
		("hashdump")
		opts="--help --no-xattrs --path --debug --quiet "
		break;;
		("3rd-party")
		opts="$global add remove list bundle-add bundle-list bundle-remove bundle-info update diagnose repair check-update clean "
		break;;
		("add")
		opts="$global --repo"
		break;;
		("remove")
		opts="$global --repo"
		break;;
	esac
	done
	# Add in additional completion options if we need to
	if (( i >= 0 ))
	then
	case "${COMP_WORDS[$i]}" in
		("bundle-add")
		# only show the list of upstream bundles if not using "3rd-party bundle-add"
		if [ "${COMP_WORDS[$i - 1]}" != "3rd-party" ]; then
			MoM=""
			if [ -r /var/tmp/swupd/Manifest.MoM ]
			then MoM=/var/tmp/swupd/Manifest.MoM
			elif [ -r /var/lib/swupd/version ] &&
				   installed=$(</var/lib/swupd/version) &&
				   [ -r "/var/lib/swupd/$installed/Manifest.MoM" ]
			then
				MoM=/var/lib/swupd/$installed/Manifest.MoM
			fi
			if [ -n "$MoM" ]
			then
				opts+="$( sed '/^[^M]/d' "$MoM" | cut -f4 | LC_ALL=C sort )"
			fi
		else
			opts+="--repo"
		fi
		;;
		("bundle-remove")
		#
		if [ "${COMP_WORDS[$i - 1]}" != "3rd-party" ]; then
			opts+=" $(unset CDPATH; test -d /usr/share/clear/bundles && \
				find /usr/share/clear/bundles/ -maxdepth 1 -type f ! -name os-core -printf '%f ')"
		else
			opts+="--repo"
		fi
		;;
		("bundle-list")
		if [ "${COMP_WORDS[$i - 1]}" = "3rd-party" ]; then
			opts+="--repo"
		fi
		;;
		("bundle-info")
		if [ "${COMP_WORDS[$i - 1]}" = "3rd-party" ]; then
			opts+="--repo"
		fi
		;;
		("update")
		if [ "${COMP_WORDS[$i - 1]}" = "3rd-party" ]; then
			opts+="--repo"
		fi
		;;
		("diagnose")
		if [ "${COMP_WORDS[$i - 1]}" = "3rd-party" ]; then
			opts+="--repo"
		fi
		;;
		("repair")
		if [ "${COMP_WORDS[$i - 1]}" = "3rd-party" ]; then
			opts+="--repo"
		fi
		;;
		("check-update")
		if [ "${COMP_WORDS[$i - 1]}" = "3rd-party" ]; then
			opts+="--repo"
		fi
		;;
		("clean")
		if [ "${COMP_WORDS[$i - 1]}" = "3rd-party" ]; then
			opts+="--repo"
		fi
		;;
		("hashdump")
		# Add in filenames. TODO add in directory completion
		opts+=" $( compgen -f -- "$2" )"
		;;
	esac
	fi

	# Ignore SC2207 because that's the standard way to fill COMPREPLY
	# shellcheck disable=SC2207
	COMPREPLY=($(compgen -W "${opts}" -- "${2}"));
	return 0
}
if [ "${BASH_VERSINFO[0]}" -gt 4 ] || { [ "${BASH_VERSINFO[0]}" -eq 4 ] && [ "${BASH_VERSINFO[1]}" -ge 4 ]; }
then
	complete -F _swupd -o nosort swupd
else
	complete -F _swupd swupd
fi