#!/usr/bin/env zsh
_jvn_completion() {
    local -a args
    local suggestions

    local buffer="$BUFFER"
    local cursor="$CURSOR"
    local current_word="${words[$CURRENT]}"
    local previous_word=""
    local command_name="${words[1]}"
    local word_index="$CURRENT"

    if [[ $CURRENT -gt 1 ]]; then
        previous_word="${words[$((CURRENT-1))]}"
    fi

    args=(
        -f "$buffer"
        -C "$cursor"
        -w "$current_word"
        -p "$previous_word"
        -c "$command_name"
        -i "$word_index"
        -a "${words[@]}"
    )

    suggestions=$(jvn_comp "${args[@]}" 2>/dev/null)

    if [[ $? -eq 0 ]] && [[ -n "$suggestions" ]]; then
        local -a completions
        completions=(${(f)suggestions})

        if [[ "${completions[1]}" == "_file_" ]]; then
            shift completions
            _files
        elif (( $+functions[_describe] )); then
            _describe 'jvn commands' completions
        else
            compadd -a completions
        fi
    fi
}

compdef _jvn_completion jvn

if [[ $? -ne 0 ]]; then
    compctl -K _jvn_completion jvn
fi