# mcrypt(1) completion                                     -*- shell-script -*-
# by Ariel Fermani <the_end@bbs.frc.utn.edu.ar>

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

    case $prev in
        -g|--openpgp-z)
            COMPREPLY=( $( compgen -W '{0..9}' -- "$cur" ) )
            return 0
            ;;
        -o|--keymode)
            COMPREPLY=( $( compgen -W '$( mcrypt --list-keymodes \
                2>/dev/null )' -- "$cur" ) )
            return 0
            ;;
        -m|--mode)
            COMPREPLY=( $( compgen -W "$( mcrypt --list \
                2>/dev/null | sed -e 's/.*: //' -e 's/ $//' | \
                sort -u )" -- "$cur" ) )
            return 0
            ;;
        -a|--algorithm)
            COMPREPLY=( $( compgen -W "$( mcrypt --list 2>/dev/null | \
                awk '{print $1}' )" -- "$cur" ) )
            return 0
            ;;
        -h|--hash)
            COMPREPLY=( $( compgen -W '$( mcrypt --list-hash 2>/dev/null | \
                sed -e 1d )' -- "$cur" ) )
            return 0
            ;;
        -k|-s|--key|--keysize)
            return 0
            ;;
        -f|-c|--keyfile|--config)
            _filedir
            return 0
            ;;
        --algorithms-directory|--modes-directory)
            _filedir -d
            return 0
            ;;
    esac

    if [[ "$cur" == -* ]]; then
        COMPREPLY=( $( compgen -W '$( _parse_help "$1" )' -- "$cur" ) )
    elif [[ ${words[0]} == mdecrypt ]]; then
        _filedir nc
    else
        local i decrypt=0
        for (( i=1; i < ${#words[@]}-1; i++ )); do
            if [[ ${words[i]} == -@(d|-decrypt) ]]; then
                _filedir nc
                decrypt=1
                break
            fi
        done
        if [[ $decrypt -eq 0 ]]; then
            _filedir
        fi
    fi
} &&
complete -F _mcrypt mcrypt mdecrypt

# ex: ts=4 sw=4 et filetype=sh
