# mcrypt(1) completion by Ariel Fermani <the_end@bbs.frc.utn.edu.ar>
#
# $Id: mcrypt,v 1.2 2003/12/31 08:22:37 ianmacd Exp $

_mcrypt()
{
	local cur prev i decrypt

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

	case "$prev" in
		-@(g|-openpgp-z))
			COMPREPLY=( $( compgen -W '0 1 2 3 4 5 6 7 8 9' \
				-- $cur ) )
			return 0
			;;
		-@(o|-keymode))
			COMPREPLY=( $( mcrypt --list-keymodes 2> /dev/null |
				grep "^$cur" ) )
			return 0
			;;
		-@(m|-mode))
			COMPREPLY=( $( mcrypt --list 2> /dev/null |
				sed -e 's/.*: //' -e 's/ $//' | sort |
				uniq | tr " " "\n" | grep "^$cur" ) )
			return 0
			;;
		-@(a|-algorithm))
			COMPREPLY=( $( mcrypt --list 2> /dev/null |
				awk '{print $1}' | grep "^$cur" ) )
			return 0
			;;
		-@(h|-hash))
			COMPREPLY=( $( mcrypt --list-hash 2> /dev/null |
				sed 1d | grep "^$cur" ) )
			return 0
			;;
		-@(k|s|-@(key?(size))))
			return 0
			;;
		-@(f|c|-@(keyfile|config)))
			_filedir
			return 0
			;;
		--@(algorithm|mode)s-directory)
			_filedir -d
			return 0
			;;
	esac

	if [[ "$cur" == -* ]]; then
		COMPREPLY=( $( compgen -W '-g --openpgp --openpgp-z -d \
			--decrypt -s --keysize -o --keymode -f --keyfile \
			-c --config -a --algorithm --algorithms-directory \
			-m --mode --modes-directory -h --hash -k --key \
			--noiv -b --bare -z --gzip -p --bzip2 --flush -l \
			--doublecheck -u --unlink --nodelete -t --time -F \
			--force --echo -r --random --list --list-keymodes \
			--list-hash -V --verbose -q --quiet --help -v \
			--version -L --license' -- $cur ) )
	elif [[ ${COMP_WORDS[0]} == mdecrypt ]]; then
		_filedir '@(nc)'
	else
		decrypt=0
		for (( i=1; i < ${#COMP_WORDS[@]}-1; i++ )); do
			if [[ ${COMP_WORDS[i]} == -@(d|-decrypt) ]]; then
				_filedir '@(nc)'
				decrypt=1
				break
			fi
		done
		if [[ $decrypt -eq 0 ]]; then
			_filedir
		fi
	fi
}
complete -o filenames -F _mcrypt mcrypt mdecrypt