# dict(1) completion                                       -*- shell-script -*-

_dictdata()
{
    dict $host $port $1 2>/dev/null | command sed -ne \
        's/^[[:blank:]]\{1,\}\([^[:blank:]]*\).*$/\1/p'
}

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

    local host port db i

    for ((i = 1; i < cword; i++)); do
        case ${words[i]} in
            --host | -!(-*)h)
                host=${words[++i]}
                [[ -n $host ]] && host="-h $host"
                ;;
            --port | -!(-*)p)
                port=${words[++i]}
                [[ -n $port ]] && port="-p $port"
                ;;
            --database | -!(-*)d)
                db=${words[++i]}
                [[ -n $db ]] && host="-d $db"
                ;;
        esac
    done

    if [[ $cur == -* ]]; then
        COMPREPLY=($(compgen -W '$(_parse_help "$1")' -- "$cur"))
        return
    fi

    case $prev in
        --database | -info | -!(-*)[di])
            COMPREPLY=($(compgen -W '$(_dictdata -D)' -- "$cur"))
            return
            ;;
        --strategy | -!(-*)s)
            COMPREPLY=($(compgen -W '$(_dictdata -S)' -- "$cur"))
            return
            ;;
    esac

    local dictfile=/usr/share/dict/words
    if [[ -r $dictfile ]]; then
        # Dictfile may be too large for practical compgen -W usage, so narrow
        # it down with grep if $cur looks like something that's safe to embed
        # in a pattern instead.
        if [[ $cur == +([-A-Za-z0-9/.]) ]]; then
            COMPREPLY=($(compgen -W \
                '$(command grep "^${cur//./\\.}" $dictfile)' -- "$cur"))
        else
            COMPREPLY=($(compgen -W '$(cat $dictfile)' -- "$cur"))
        fi
    fi
} &&
    complete -F _dict -o default dict rdict

# ex: filetype=sh