incubator-airflow/run-tests-complete

#!/usr/bin/env bash
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

_RUN_TESTS_SHORT_OPTIONS="
h i x s v f:"

_RUN_TESTS_LONG_OPTIONS="
help with-db-init with-xunit nocapture verbose xunit-file:
"

# Note on OSX bash has no associative arrays (Bash 3.2) so we have to fake it

_RUN_TESTS_KNOWN_VALUES=""

function _get_known_values_run_tests {
    case "$1" in
        *)
            _RUN_TESTS_KNOWN_VALUES=""
    esac
}

_RUN_TESTS_GETOPT_SHORT_OPTIONS=""
_RUN_TESTS_GETOPT_LONG_OPTIONS=""

function _build_options_run_tests {
    local SEPARATOR=""
    local OPTION

    for OPTION in ${_RUN_TESTS_SHORT_OPTIONS}
    do
        _RUN_TESTS_GETOPT_SHORT_OPTIONS="${_RUN_TESTS_GETOPT_SHORT_OPTIONS}${SEPARATOR}${OPTION}"
        SEPARATOR=","
    done

    SEPARATOR=""
    for OPTION in ${_RUN_TESTS_LONG_OPTIONS}
    do
        _RUN_TESTS_GETOPT_LONG_OPTIONS="${_RUN_TESTS_GETOPT_LONG_OPTIONS}${SEPARATOR}${OPTION}"
        SEPARATOR=","
    done
}

function _listcontains_run_tests {
  local WORD
  for WORD in $1; do
    [[ ${WORD} = "$2" ]] && return 0
  done
  return 1
}

# A completion function for run_tests
function _comp_run_tests {
    local ALL_OPTIONS=""
    local EXTRA_ARG_OPTIONS=""
    local OPTION
    local GETOPT_OPTION
    local LAST_COMMAND_PREFIX
    local PREVIOUS_COMMAND
    local ALL_TESTS_FILE="${HOME}/all_tests.txt"

    for OPTION in ${_RUN_TESTS_SHORT_OPTIONS}
    do
        LAST_CHAR="${OPTION:$((${#OPTION}-1)):1}"
        GETOPT_OPTION='-'${OPTION//:/}
        if [[ "${LAST_CHAR}" == ":" ]]; then
            EXTRA_ARG_OPTIONS="${EXTRA_ARG_OPTIONS} ${GETOPT_OPTION}"
        fi
        ALL_OPTIONS="${ALL_OPTIONS} ${GETOPT_OPTION}"
    done
    for OPTION in ${_RUN_TESTS_LONG_OPTIONS}
    do
        LAST_CHAR="${OPTION:$((${#OPTION}-1)):1}"
        GETOPT_OPTION='--'${OPTION//:/}
        ALL_OPTIONS="${ALL_OPTIONS} ${GETOPT_OPTION}"
        if [[ "${LAST_CHAR}" == ":" ]]; then
            EXTRA_ARG_OPTIONS="${EXTRA_ARG_OPTIONS} ${GETOPT_OPTION}"
        fi
    done

    LAST_COMMAND_PREFIX="${COMP_WORDS[${#COMP_WORDS[@]}-1]}"
    if [[ ${#COMP_WORDS[@]} -gt 1 ]]; then
        PREVIOUS_COMMAND="${COMP_WORDS[${#COMP_WORDS[@]}-2]}"
    else
        PREVIOUS_COMMAND=""
    fi

    if _listcontains_run_tests "${EXTRA_ARG_OPTIONS}" "${PREVIOUS_COMMAND}"; then
        COMPREPLY=()
        _get_known_values_run_tests "${PREVIOUS_COMMAND}"
        while IFS='' read -r LINE; do COMPREPLY+=("$LINE"); done \
           < <(compgen -W "${_RUN_TESTS_KNOWN_VALUES}" -- "${LAST_COMMAND_PREFIX}")
    else
        COMPREPLY=()
        while IFS='' read -r LINE; do COMPREPLY+=("$LINE"); done \
           < <(compgen -W "${ALL_OPTIONS}" -- "${LAST_COMMAND_PREFIX}")

        if [[ -f ${ALL_TESTS_FILE} ]]; then
            while IFS='' read -r LINE; do COMPREPLY+=("$LINE"); done \
                < <(compgen -W "$(cat "${ALL_TESTS_FILE}")" -- "${LAST_COMMAND_PREFIX}")
        fi
    fi
}

_build_options_run_tests

# allow completion to contain colon (see http://tiswww.case.edu/php/chet/bash/FAQ)
COMP_WORDBREAKS=${COMP_WORDBREAKS//:}

complete -F _comp_run_tests run-tests