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

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

proc _chronyc
{
    var cur = '', prev = '', words = '', cword = ''
    _init_completion || return

    match $prev {
        with --help|-p
            return
            
        with -h
            _known_hosts_real $cur
            return
            
    }

    if [[ $cur == -* ]] {
        setglobal COMPREPLY = '( '$( compgen -W '$( _parse_usage "$1" ) -6' -- "$cur" ) )
        return
    }

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

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