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

_chronyc_command_args()
{
    local -a args=( $( compgen -W "$( $1 help 2>/dev/null | \
        awk '/^'$prev'\s[^ []/ { gsub("\\|", " ", $2); print $2 }' )" ) )
    case $args in
        \<address\>) _known_hosts_real "$cur" ;;
        \<*) ;;
        *) COMPREPLY+=( $( compgen -W '${args[@]}' -- "$cur" ) ) ;;
    esac
}

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

    case $prev in
        --help|-p)
            return
            ;;
        -h)
            _known_hosts_real "$cur"
            return
            ;;
    esac

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

    local i args=0
    for (( i=1; i < cword; i++ )); do
        [[ ${words[i]} != -* && ${words[i-1]} != @(-p|-h) ]] && (( args++ ))
    done
    
    case $args in
        0)
            COMPREPLY=( $( compgen -W "$( $1 help 2>/dev/null | \
                awk '!/(^ |: *$)/ { sub("\\|", " ", $1); print $1 }' )" \
                    -- "$cur" ) )
            ;;
        1)
            _chronyc_command_args "$1"
            if [[ ! $COMPREPLY && $prev == sources?(tats) ]]; then
                # [-v] not handled by _chronyc_command_args yet
                COMPREPLY=( $( compgen -W '-v' -- "$cur" ) )
            fi
            ;;
        2)
            [[ $prev == @(peer|server) ]] && _known_hosts_real "$cur"
            ;;
    esac
} &&
complete -F _chronyc chronyc

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