| #!/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. |
| |
| _BREEZE_ALLOWED_PYTHON_VERSIONS=" 2.7 3.5 3.6 " |
| _BREEZE_ALLOWED_ENVS=" docker kubernetes " |
| _BREEZE_ALLOWED_BACKENDS=" sqlite mysql postgres " |
| _BREEZE_ALLOWED_KUBERNETES_VERSIONS=" v1.13.0 " |
| _BREEZE_ALLOWED_KUBERNETES_MODES=" persistent_mode git_mode " |
| _BREEZE_ALLOWED_STATIC_CHECKS=" all check-apache-license check-executables-have-shebangs check-hooks-apply check-merge-conflict check-xml debug-statements doctoc detect-private-key end-of-file-fixer flake8 forbid-tabs insert-license lint-dockerfile mixed-line-ending mypy setup-order shellcheck" |
| _BREEZE_DEFAULT_DOCKERHUB_USER="apache" |
| _BREEZE_DEFAULT_DOCKERHUB_REPO="airflow" |
| |
| _BREEZE_SHORT_OPTIONS=" |
| h P: E: B: K: M: |
| s b O |
| v y n C A |
| r R p u |
| c D: H: e a |
| t: d: k x: S: F: |
| " |
| |
| _BREEZE_LONG_OPTIONS=" |
| help python: env: backend: kubernetes-version: kubernetes-mode: |
| skip-mounting-local-sources build-only build-docs |
| verbose assume-yes assume-no toggle-suppress-cheatsheet toggle-suppress-asciiart |
| force-build-images force-build-images-clean force-pull-images push-images |
| cleanup-images dockerhub-user: dockerhub-repo: initialize-local-virtualenv setup-autocomplete |
| test-target: docker-compose: stop-environment execute-command: static-check: static-check-all-files: |
| " |
| |
| # Note on OSX bash has no associative arrays (Bash 3.2) so we have to fake it |
| |
| _BREEZE_KNOWN_VALUES="" |
| |
| function _get_known_values_breeze { |
| case "$1" in |
| -P | --python ) |
| _BREEZE_KNOWN_VALUES=${_BREEZE_ALLOWED_PYTHON_VERSIONS} ;; |
| -E | --env ) |
| _BREEZE_KNOWN_VALUES=${_BREEZE_ALLOWED_ENVS} ;; |
| -B | --backend ) |
| _BREEZE_KNOWN_VALUES=${_BREEZE_ALLOWED_BACKENDS} ;; |
| -K | --kubernetes-version ) |
| _BREEZE_KNOWN_VALUES=${_BREEZE_ALLOWED_KUBERNETES_VERSIONS} ;; |
| -M | --kubernetes-mode ) |
| _BREEZE_KNOWN_VALUES=${_BREEZE_ALLOWED_KUBERNETES_MODES} ;; |
| -S | --static-check ) |
| _BREEZE_KNOWN_VALUES=${_BREEZE_ALLOWED_STATIC_CHECKS} ;; |
| -F | --static-check-all-files ) |
| _BREEZE_KNOWN_VALUES=${_BREEZE_ALLOWED_STATIC_CHECKS} ;; |
| -d | --docker-compose ) |
| # shellcheck disable=SC2034 |
| if typeset -f "_docker_compose" > /dev/null; then |
| _docker_compose |
| fi |
| _BREEZE_KNOWN_VALUES="" ;; |
| -D | --dockerhub-user ) |
| _BREEZE_KNOWN_VALUES="${_BREEZE_DEFAULT_DOCKERHUB_USER}" ;; |
| -H | --dockerhub-repo ) |
| _BREEZE_KNOWN_VALUES="${_BREEZE_DEFAULT_DOCKERHUB_REPO}" ;; |
| *) |
| _BREEZE_KNOWN_VALUES="" |
| esac |
| } |
| |
| _BREEZE_GETOPT_SHORT_OPTIONS="" |
| _BREEZE_GETOPT_LONG_OPTIONS="" |
| |
| function _build_options_breeze { |
| local SEPARATOR="" |
| local OPTION |
| |
| for OPTION in ${_BREEZE_SHORT_OPTIONS} |
| do |
| _BREEZE_GETOPT_SHORT_OPTIONS="${_BREEZE_GETOPT_SHORT_OPTIONS}${SEPARATOR}${OPTION}" |
| SEPARATOR="," |
| done |
| |
| SEPARATOR="" |
| for OPTION in ${_BREEZE_LONG_OPTIONS} |
| do |
| _BREEZE_GETOPT_LONG_OPTIONS="${_BREEZE_GETOPT_LONG_OPTIONS}${SEPARATOR}${OPTION}" |
| SEPARATOR="," |
| done |
| } |
| |
| function _listcontains_breeze { |
| local WORD |
| for WORD in $1; do |
| [[ ${WORD} = "$2" ]] && return 0 |
| done |
| return 1 |
| } |
| |
| # A completion function for breeze |
| function _comp_breeze { |
| local ALL_OPTIONS="" |
| local EXTRA_ARG_OPTIONS="" |
| local OPTION |
| local GETOPT_OPTION |
| local LAST_COMMAND_PREFIX |
| local PREVIOUS_COMMAND |
| |
| for OPTION in ${_BREEZE_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 ${_BREEZE_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_breeze "${EXTRA_ARG_OPTIONS}" "${PREVIOUS_COMMAND}"; then |
| COMPREPLY=() |
| _get_known_values_breeze "${PREVIOUS_COMMAND}" |
| while IFS='' read -r LINE; do COMPREPLY+=("$LINE"); done \ |
| < <(compgen -W "${_BREEZE_KNOWN_VALUES}" -- "${LAST_COMMAND_PREFIX}") |
| else |
| COMPREPLY=() |
| while IFS='' read -r LINE; do COMPREPLY+=("$LINE"); done \ |
| < <(compgen -W "${ALL_OPTIONS}" -- "${LAST_COMMAND_PREFIX}") |
| fi |
| } |
| |
| _build_options_breeze |
| |
| complete -F _comp_breeze breeze |