# ktutil completion                                        -*- shell-script -*-

proc _heimdal_principals {
    setvar COMPREPLY = ''( $( compgen -W "$( kadmin -l dump 2>/dev/null | \
        awk '{print $1}' )" -- "$cur" ) )
}

proc _heimdal_realms {
    setvar COMPREPLY = ''( $( compgen -W "( kadmin -l dump 2>/dev/null | \
        awk '{print $1}' | awk -F@ '{print $2}' )" -- "$cur" ) )
}

proc _heimdal_encodings {
    setvar COMPREPLY = ''( $( compgen -W 'des-cbc-mcrc des-cbc-md4 des-cbc-md5
        des3-cbc-sha1 arcfour-hmac-md5 aes128-cts-hmac-sha1-96
        aes256-cts-hmac-sha1-96' -- "$cur" ) )
}

proc _ktutil {
    local cur prev words cword split
    _init_completion -s || return

    local command options

    case (prev) {
        -p|--principal {
            _heimdal_principals
            return 0
            }
        -e|--enctype {
            _heimdal_encodings
            return 0
            }
        -a|--admin-server {
            _known_hosts_real $cur
            return 0
            }
        -r|--realm {
            _heimdal_realms
            return 0
            }
        -s|-k|--srvtab|--keytab {
            _filedir
            return 0
            }
    }

    $split && return 0

    setvar commands = ''add change copy get list remove rename purge srvconvert
        srv2keytab srvcreate key2srvtab''

    for (( i=1; i < cword; i++ )); do
        case ${words[i]} in
            -k|--keytab)
                i=$(($i+1))
                ;;
            -*)
                ;;
            *)
                command=${words[i]}
                break
                ;;
        esac
    done

    if [[ "$cur" == -* ]] {
        case (command) {
            add {
                setvar options = ''-p --principal -V -e --enctype -w --password -r
                    --random -s --no-salt -h --hex''
            }
            change {
                setvar options = ''-r --realm -a --admin-server -s --server-port''
            }
            get {
                setvar options = ''-p --principal -e --enctype -r --realm -a
                    --admin-server -s server --server-port''
                }
            list {
                setvar options = ''--keys --timestamp''
                }
            remove {
                setvar options = ''-p --principal -V --kvno -e --enctype''
                }
            purge {
                setvar options = ''--age''
                }
            srv2keytab|key2srvtab {
                setvar options = ''-s --srvtab''
                }
            * {
                setvar options = ''-k --keytab -v --verbose --version -v --help''
                }
        }
        setvar COMPREPLY = ''( $( compgen -W "$options" -- "$cur" ) )
    } else {
        case (command) {
            copy {
                _filedir
                }
            get {
                _heimdal_principals
                }
            rename {
                _heimdal_principals
                }
            * {
                setvar COMPREPLY = ''( $( compgen -W "$commands" -- "$cur" ) )
                }
        }
    }
} &&
complete -F _ktutil ktutil

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