Ansible-playbook bash complete for specific directory

ansible-playbook.bash

#!/bin/env bash

_ansible_playbook() {
    local current_word=${COMP_WORDS[COMP_CWORD]}
    local previous_word=${COMP_WORDS[COMP_CWORD - 1]}

    if [[ "$current_word" == -* ]]; then
        _ansible_complete_options "$current_word"
    elif [[ "$current_word" == @* ]]; then
        _ansible_target "$current_word"
    else
        _ansible_playbook_yml "$current_word"
    fi
}

complete -F _ansible_playbook ansible-playbook

# Compute completion for the generics options
_ansible_complete_options() {
    local current_word=$1
    local options="-h --help -e --extra-vars -t --tags -v --verbose"

    COMPREPLY=( $( compgen -W "$options" -- "$current_word" ) )
}

_ansible_playbook_yml(){
   local word=`echo $1 | sed -e 's/\/etc\/ansible\/ansible-docker\/playbooks\///g'`
   COMPREPLY=( $(
     cd /etc/ansible/ansible-docker/playbooks && \
     compgen -f -X '!*.yml' -- "$word" | awk '{print "/etc/ansible/ansible-docker/playbooks/"$word}'
   ) )
}

_ansible_target(){
   local word=`echo $1 | sed -e 's/^@//g'`
   word=`echo $word | sed -e 's/\/etc\/ansible\/target\///g'`
   COMPREPLY=( $(
     cd /etc/ansible/target && \
     compgen -o plusdirs  -f -X '!*' -- "$word" | awk '{print "@/etc/ansible/target/"$1}'
   ) )
}