completions

# Custom completion rules

_dirs() {
    local dir
    dir=`basename $1`
    find $1 -type d -maxdepth 1 | xargs  -II basename I | grep -v $dir
}

_list_apps() {
    # Maintains a list of apps in our system
    _dirs $HIREIQ_ROOT/py/toomah/djangoapps
}

_list_assessments() {
    # Maintains a list of apps in our system
    _dirs $HIREIQ_ROOT/py/toomah/assessments
}

_list_pois() {
    # All points of interest in our system
    _list_apps | sed 's|^|apps/|'
    _list_assessments | sed 's|^|assessments/|'
    _dirs $HIREIQ_ROOT/py/toomah/djangoprojects | sed 's|^|projects/|'
    echo root
    echo lib
    echo django
}

_poi_completion() {
    APPS=`_list_pois`
    prefix="${COMP_WORDS[COMP_CWORD]}"
    COMPREPLY=( $(compgen -W "$APPS" -- $prefix) )
    return 0
}

_manage_completion() {
    local choices APPS SUBCOMMANDS exe cmd first
    # This function is not remotely complete. Add more as needed
    exe=$COMP_WORDS
    APPS=`_list_apps`
    # This could probably be better but my awk foo just isn't that strong
    if [ -z "$_manage_subcommands" ]; then
        export _manage_subcommands="`$exe help 2>/dev/null  | awk '/.*Available.*/,/END/' | grep -v "Available" | grep -v '\['`"
    fi
    SUBCOMMANDS="$_manage_subcommands"
    SUBCOMMANDS="$SUBCOMMANDS -p" # Allow -p flag MUST BE BEFORE SUBCOMMAND
    prefix="${COMP_WORDS[COMP_CWORD]}"
    cmd=1
    if [ "-p" == "${COMP_WORDS[1]}" -a "3" -le "$COMP_CWORD" ]; then
      cmd=3
    fi
    first="${COMP_WORDS[$cmd]}"
    # If its the first choice, just use commands
    if [ "$cmd" -eq "$COMP_CWORD" ]; then
        choices=$SUBCOMMANDS
    else
        case $first in
            'test') choices=$APPS ;;
            'shell') choices='--plain' ;;
            '-p')
                if [ "2" -eq "$COMP_CWORD" ]; then # Hack to see if -p needs arg
                    choices=`_dirs $HIREIQ_ROOT/py/toomah/djangoprojects`
                fi
                ;;
            *) ;;
        esac
    fi
    COMPREPLY=( $(compgen -W "$choices" -- $prefix) )
    return 0
}

complete -F _poi_completion gt
complete -F _poi_completion edit
complete -F _manage_completion m