# bash completion for nslookup                             -*- shell-script -*-

proc _bind_queryclass {
    setvar COMPREPLY = ''( $( compgen -W 'IN CH HS ANY' -- "$cur" ) )
}

proc _bind_querytype {
    # http://en.wikipedia.org/wiki/List_of_DNS_record_types
    setvar COMPREPLY = ''( $( compgen -W 'A AAAA AFSDB APL CERT CNAME DHCID DLV DNAME
        DNSKEY DS HIP IPSECKEY KEY KX LOC MX NAPTR NS NSEC NSEC3 NSEC3PARAM PTR
        RRSIG RP SIG SOA SPF SRV SSHFP TXT' -- "$cur" ) )
}

proc _nslookup {
    local cur prev words cword
    _init_completion -n = || return

    case (cur) {
        -class=*|-cl=* {
            setvar cur = ${cur#*=}
            _bind_queryclass
            return
            }
        -querytype=*|-type=*|-q=*|-ty=* {
            setvar cur = ${cur#*=}
            _bind_querytype
            return
            }
        -?*=* {
            return
            }
    }

    if [[ $cur == -* ]] {
        setvar COMPREPLY = ''( $( compgen -W '-all -class= -debug -nodebug -d2 -nod2
            -domain= -search -nosearch -port= -querytype= -recurse -norecurse
            -retry= -timeout= -vc -novc -fail -nofail' -- "$cur" ) )
        [[ $COMPREPLY == *= ]] && compopt -o nospace
        return
    }

    local args
    _count_args =
    if [[ $args -le 2 ]] {
        _known_hosts_real $cur
        [[ $args -eq 1 && $cur == @(|-) ]] && setvar COMPREPLY = ''( - )
    }
} &&
complete -F _nslookup nslookup

proc _host {
    local cur prev words cword
    _init_completion || return

    case (prev) {
        -c {
            _bind_queryclass
            return
            }
        -t {
            _bind_querytype
            return
            }
        -m {
            setvar COMPREPLY = ''( $( compgen -W 'trace record usage' -- "$cur" ) )
            return
            }
        -N|-R|-W {
            return
            }
    }

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

    _known_hosts_real $cur
} &&
complete -F _host host

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