Rather cool. I've seen it before and had it on my todo list.
! command -v vault >/dev/null || complete -C vault vault