abra/abra

#!/usr/bin/env bash

###### Global help

DOC="
The cooperative cloud utility belt 🎩🐇

Usage:
  abra [options] deploy
  abra [options] upgrade
  abra [options] logs [<service>]
  abra [options] multilogs
  abra [options] cp <src> <dst>
  abra [options] run [--user=<user>] <app> <args>...
  abra [options] run <app> <args>...
  abra [options] secret generate <secret> <version> [<cmd>]
  abra [options] secret insert <secret> <version> <data>
  abra [options] server add <host> [<user>] [<port>]
  abra [options] server rm <host>
  abra [options] server use <host>
  abra [options] server init <host>
  abra [options] server ls
  abra [options] <command> <args>...

Options:
  -c, --config=<path>  Stack configuration to use
  -e, --env=<path>     Environment variables to load
  -h, --help           Show this message and exit
  -s, --stack=<path>   Name of the target stack
  -v, --version        Show program version
"

# docopt parser below, refresh this parser with `docopt.sh abra`
# shellcheck disable=2016,1075,2154
docopt() { parse() { if ${DOCOPT_DOC_CHECK:-true}; then local doc_hash
if doc_hash=$(printf "%s" "$DOC" | (sha256sum 2>/dev/null || shasum -a 256)); then
if [[ ${doc_hash:0:5} != "$digest" ]]; then
stderr "The current usage doc (${doc_hash:0:5}) does not match \
what the parser was generated with (${digest})
Run \`docopt.sh\` to refresh the parser."; _return 70; fi; fi; fi
local root_idx=$1; shift; argv=("$@"); parsed_params=(); parsed_values=()
left=(); testdepth=0; local arg; while [[ ${#argv[@]} -gt 0 ]]; do
if [[ ${argv[0]} = "--" ]]; then for arg in "${argv[@]}"; do
parsed_params+=('a'); parsed_values+=("$arg"); done; break
elif [[ ${argv[0]} = --* ]]; then parse_long
elif [[ ${argv[0]} = -* && ${argv[0]} != "-" ]]; then parse_shorts
elif ${DOCOPT_OPTIONS_FIRST:-false}; then for arg in "${argv[@]}"; do
parsed_params+=('a'); parsed_values+=("$arg"); done; break; else
parsed_params+=('a'); parsed_values+=("${argv[0]}"); argv=("${argv[@]:1}"); fi
done; local idx; if ${DOCOPT_ADD_HELP:-true}; then
for idx in "${parsed_params[@]}"; do [[ $idx = 'a' ]] && continue
if [[ ${shorts[$idx]} = "-h" || ${longs[$idx]} = "--help" ]]; then
stdout "$trimmed_doc"; _return 0; fi; done; fi
if [[ ${DOCOPT_PROGRAM_VERSION:-false} != 'false' ]]; then
for idx in "${parsed_params[@]}"; do [[ $idx = 'a' ]] && continue
if [[ ${longs[$idx]} = "--version" ]]; then stdout "$DOCOPT_PROGRAM_VERSION"
_return 0; fi; done; fi; local i=0; while [[ $i -lt ${#parsed_params[@]} ]]; do
left+=("$i"); ((i++)) || true; done
if ! required "$root_idx" || [ ${#left[@]} -gt 0 ]; then error; fi; return 0; }
parse_shorts() { local token=${argv[0]}; local value; argv=("${argv[@]:1}")
[[ $token = -* && $token != --* ]] || _return 88; local remaining=${token#-}
while [[ -n $remaining ]]; do local short="-${remaining:0:1}"
remaining="${remaining:1}"; local i=0; local similar=(); local match=false
for o in "${shorts[@]}"; do if [[ $o = "$short" ]]; then similar+=("$short")
[[ $match = false ]] && match=$i; fi; ((i++)) || true; done
if [[ ${#similar[@]} -gt 1 ]]; then
error "${short} is specified ambiguously ${#similar[@]} times"
elif [[ ${#similar[@]} -lt 1 ]]; then match=${#shorts[@]}; value=true
shorts+=("$short"); longs+=(''); argcounts+=(0); else value=false
if [[ ${argcounts[$match]} -ne 0 ]]; then if [[ $remaining = '' ]]; then
if [[ ${#argv[@]} -eq 0 || ${argv[0]} = '--' ]]; then
error "${short} requires argument"; fi; value=${argv[0]}; argv=("${argv[@]:1}")
else value=$remaining; remaining=''; fi; fi; if [[ $value = false ]]; then
value=true; fi; fi; parsed_params+=("$match"); parsed_values+=("$value"); done
}; parse_long() { local token=${argv[0]}; local long=${token%%=*}
local value=${token#*=}; local argcount; argv=("${argv[@]:1}")
[[ $token = --* ]] || _return 88; if [[ $token = *=* ]]; then eq='='; else eq=''
value=false; fi; local i=0; local similar=(); local match=false
for o in "${longs[@]}"; do if [[ $o = "$long" ]]; then similar+=("$long")
[[ $match = false ]] && match=$i; fi; ((i++)) || true; done
if [[ $match = false ]]; then i=0; for o in "${longs[@]}"; do
if [[ $o = $long* ]]; then similar+=("$long"); [[ $match = false ]] && match=$i
fi; ((i++)) || true; done; fi; if [[ ${#similar[@]} -gt 1 ]]; then
error "${long} is not a unique prefix: ${similar[*]}?"
elif [[ ${#similar[@]} -lt 1 ]]; then
[[ $eq = '=' ]] && argcount=1 || argcount=0; match=${#shorts[@]}
[[ $argcount -eq 0 ]] && value=true; shorts+=(''); longs+=("$long")
argcounts+=("$argcount"); else if [[ ${argcounts[$match]} -eq 0 ]]; then
if [[ $value != false ]]; then
error "${longs[$match]} must not have an argument"; fi
elif [[ $value = false ]]; then
if [[ ${#argv[@]} -eq 0 || ${argv[0]} = '--' ]]; then
error "${long} requires argument"; fi; value=${argv[0]}; argv=("${argv[@]:1}")
fi; if [[ $value = false ]]; then value=true; fi; fi; parsed_params+=("$match")
parsed_values+=("$value"); }; required() { local initial_left=("${left[@]}")
local node_idx; ((testdepth++)) || true; for node_idx in "$@"; do
if ! "node_$node_idx"; then left=("${initial_left[@]}"); ((testdepth--)) || true
return 1; fi; done; if [[ $((--testdepth)) -eq 0 ]]; then
left=("${initial_left[@]}"); for node_idx in "$@"; do "node_$node_idx"; done; fi
return 0; }; either() { local initial_left=("${left[@]}"); local best_match_idx
local match_count; local node_idx; ((testdepth++)) || true
for node_idx in "$@"; do if "node_$node_idx"; then
if [[ -z $match_count || ${#left[@]} -lt $match_count ]]; then
best_match_idx=$node_idx; match_count=${#left[@]}; fi; fi
left=("${initial_left[@]}"); done; ((testdepth--)) || true
if [[ -n $best_match_idx ]]; then "node_$best_match_idx"; return 0; fi
left=("${initial_left[@]}"); return 1; }; optional() { local node_idx
for node_idx in "$@"; do "node_$node_idx"; done; return 0; }; oneormore() {
local i=0; local prev=${#left[@]}; while "node_$1"; do ((i++)) || true
[[ $prev -eq ${#left[@]} ]] && break; prev=${#left[@]}; done
if [[ $i -ge 1 ]]; then return 0; fi; return 1; }; _command() { local i
local name=${2:-$1}; for i in "${!left[@]}"; do local l=${left[$i]}
if [[ ${parsed_params[$l]} = 'a' ]]; then
if [[ ${parsed_values[$l]} != "$name" ]]; then return 1; fi
left=("${left[@]:0:$i}" "${left[@]:((i+1))}")
[[ $testdepth -gt 0 ]] && return 0; if [[ $3 = true ]]; then
eval "((var_$1++)) || true"; else eval "var_$1=true"; fi; return 0; fi; done
return 1; }; switch() { local i; for i in "${!left[@]}"; do local l=${left[$i]}
if [[ ${parsed_params[$l]} = "$2" ]]; then
left=("${left[@]:0:$i}" "${left[@]:((i+1))}")
[[ $testdepth -gt 0 ]] && return 0; if [[ $3 = true ]]; then
eval "((var_$1++))" || true; else eval "var_$1=true"; fi; return 0; fi; done
return 1; }; value() { local i; for i in "${!left[@]}"; do local l=${left[$i]}
if [[ ${parsed_params[$l]} = "$2" ]]; then
left=("${left[@]:0:$i}" "${left[@]:((i+1))}")
[[ $testdepth -gt 0 ]] && return 0; local value
value=$(printf -- "%q" "${parsed_values[$l]}"); if [[ $3 = true ]]; then
eval "var_$1+=($value)"; else eval "var_$1=$value"; fi; return 0; fi; done
return 1; }; stdout() { printf -- "cat <<'EOM'\n%s\nEOM\n" "$1"; }; stderr() {
printf -- "cat <<'EOM' >&2\n%s\nEOM\n" "$1"; }; error() {
[[ -n $1 ]] && stderr "$1"; stderr "$usage"; _return 1; }; _return() {
printf -- "exit %d\n" "$1"; exit "$1"; }; set -e; trimmed_doc=${DOC:1:871}
usage=${DOC:40:577}; digest=d484a; shorts=(-v -c -s -e -h '')
longs=(--version --config --stack --env --help --user); argcounts=(0 1 1 1 0 1)
node_0(){ switch __version 0; }; node_1(){ value __config 1; }; node_2(){
value __stack 2; }; node_3(){ value __env 3; }; node_4(){ switch __help 4; }
node_5(){ value __user 5; }; node_6(){ value _service_ a; }; node_7(){
value _src_ a; }; node_8(){ value _dst_ a; }; node_9(){ value _app_ a; }
node_10(){ value _args_ a true; }; node_11(){ value _secret_ a; }; node_12(){
value _version_ a; }; node_13(){ value _cmd_ a; }; node_14(){ value _data_ a; }
node_15(){ value _host_ a; }; node_16(){ value _user_ a; }; node_17(){
value _port_ a; }; node_18(){ value _command_ a; }; node_19(){ _command deploy
}; node_20(){ _command upgrade; }; node_21(){ _command logs; }; node_22(){
_command multilogs; }; node_23(){ _command cp; }; node_24(){ _command run; }
node_25(){ _command secret; }; node_26(){ _command generate; }; node_27(){
_command insert; }; node_28(){ _command server; }; node_29(){ _command add; }
node_30(){ _command rm; }; node_31(){ _command use; }; node_32(){ _command init
}; node_33(){ _command ls; }; node_34(){ optional 0 1 2 3 4; }; node_35(){
optional 34; }; node_36(){ required 35 19; }; node_37(){ required 35 20; }
node_38(){ optional 6; }; node_39(){ required 35 21 38; }; node_40(){
required 35 22; }; node_41(){ required 35 23 7 8; }; node_42(){ optional 5; }
node_43(){ oneormore 10; }; node_44(){ required 35 24 42 9 43; }; node_45(){
required 35 24 9 43; }; node_46(){ optional 13; }; node_47(){
required 35 25 26 11 12 46; }; node_48(){ required 35 25 27 11 12 14; }
node_49(){ optional 16; }; node_50(){ optional 17; }; node_51(){
required 35 28 29 15 49 50; }; node_52(){ required 35 28 30 15; }; node_53(){
required 35 28 31 15; }; node_54(){ required 35 28 32 15; }; node_55(){
required 35 28 33; }; node_56(){ required 35 18 43; }; node_57(){
either 36 37 39 40 41 44 45 47 48 51 52 53 54 55 56; }; node_58(){ required 57
}; cat <<<' docopt_exit() { [[ -n $1 ]] && printf "%s\n" "$1" >&2
printf "%s\n" "${DOC:40:577}" >&2; exit 1; }'; unset var___version \
var___config var___stack var___env var___help var___user var__service_ \
var__src_ var__dst_ var__app_ var__args_ var__secret_ var__version_ var__cmd_ \
var__data_ var__host_ var__user_ var__port_ var__command_ var_deploy \
var_upgrade var_logs var_multilogs var_cp var_run var_secret var_generate \
var_insert var_server var_add var_rm var_use var_init var_ls; parse 58 "$@"
local prefix=${DOCOPT_PREFIX:-''}; unset "${prefix}__version" \
"${prefix}__config" "${prefix}__stack" "${prefix}__env" "${prefix}__help" \
"${prefix}__user" "${prefix}_service_" "${prefix}_src_" "${prefix}_dst_" \
"${prefix}_app_" "${prefix}_args_" "${prefix}_secret_" "${prefix}_version_" \
"${prefix}_cmd_" "${prefix}_data_" "${prefix}_host_" "${prefix}_user_" \
"${prefix}_port_" "${prefix}_command_" "${prefix}deploy" "${prefix}upgrade" \
"${prefix}logs" "${prefix}multilogs" "${prefix}cp" "${prefix}run" \
"${prefix}secret" "${prefix}generate" "${prefix}insert" "${prefix}server" \
"${prefix}add" "${prefix}rm" "${prefix}use" "${prefix}init" "${prefix}ls"
eval "${prefix}"'__version=${var___version:-false}'
eval "${prefix}"'__config=${var___config:-}'
eval "${prefix}"'__stack=${var___stack:-}'
eval "${prefix}"'__env=${var___env:-}'
eval "${prefix}"'__help=${var___help:-false}'
eval "${prefix}"'__user=${var___user:-}'
eval "${prefix}"'_service_=${var__service_:-}'
eval "${prefix}"'_src_=${var__src_:-}'; eval "${prefix}"'_dst_=${var__dst_:-}'
eval "${prefix}"'_app_=${var__app_:-}'
if declare -p var__args_ >/dev/null 2>&1; then
eval "${prefix}"'_args_=("${var__args_[@]}")'; else eval "${prefix}"'_args_=()'
fi; eval "${prefix}"'_secret_=${var__secret_:-}'
eval "${prefix}"'_version_=${var__version_:-}'
eval "${prefix}"'_cmd_=${var__cmd_:-}'; eval "${prefix}"'_data_=${var__data_:-}'
eval "${prefix}"'_host_=${var__host_:-}'
eval "${prefix}"'_user_=${var__user_:-}'
eval "${prefix}"'_port_=${var__port_:-}'
eval "${prefix}"'_command_=${var__command_:-}'
eval "${prefix}"'deploy=${var_deploy:-false}'
eval "${prefix}"'upgrade=${var_upgrade:-false}'
eval "${prefix}"'logs=${var_logs:-false}'
eval "${prefix}"'multilogs=${var_multilogs:-false}'
eval "${prefix}"'cp=${var_cp:-false}'; eval "${prefix}"'run=${var_run:-false}'
eval "${prefix}"'secret=${var_secret:-false}'
eval "${prefix}"'generate=${var_generate:-false}'
eval "${prefix}"'insert=${var_insert:-false}'
eval "${prefix}"'server=${var_server:-false}'
eval "${prefix}"'add=${var_add:-false}'; eval "${prefix}"'rm=${var_rm:-false}'
eval "${prefix}"'use=${var_use:-false}'
eval "${prefix}"'init=${var_init:-false}'; eval "${prefix}"'ls=${var_ls:-false}'
local docopt_i=1; [[ $BASH_VERSION =~ ^4.3 ]] && docopt_i=2
for ((;docopt_i>0;docopt_i--)); do declare -p "${prefix}__version" \
"${prefix}__config" "${prefix}__stack" "${prefix}__env" "${prefix}__help" \
"${prefix}__user" "${prefix}_service_" "${prefix}_src_" "${prefix}_dst_" \
"${prefix}_app_" "${prefix}_args_" "${prefix}_secret_" "${prefix}_version_" \
"${prefix}_cmd_" "${prefix}_data_" "${prefix}_host_" "${prefix}_user_" \
"${prefix}_port_" "${prefix}_command_" "${prefix}deploy" "${prefix}upgrade" \
"${prefix}logs" "${prefix}multilogs" "${prefix}cp" "${prefix}run" \
"${prefix}secret" "${prefix}generate" "${prefix}insert" "${prefix}server" \
"${prefix}add" "${prefix}rm" "${prefix}use" "${prefix}init" "${prefix}ls"; done
}
# docopt parser above, complete command for generating this parser is `docopt.sh abra`

PROGRAM_NAME=$(basename "$0")

###### Utility functions

error() {
  echo "$(tput setaf 1)ERROR: $*$(tput sgr0)"
  exit 1
}

warning() {
  echo "$(tput setaf 3)WARNING: $*$(tput sgr0)"
}

success() {
  echo "$(tput setaf 2)$*$(tput sgr0)"
}

###### Default settings

if [ -z "$COMPOSE_FILE" ]; then
  COMPOSE_FILE="compose.yml"
fi

if [ -z "$ABRA_STACK_DIR" ]; then
  ABRA_STACK_DIR="stacks/$SERVICE"
fi

###### Safety checks

require_multitail() {
  if ! type multitail > /dev/null 2>&1; then
    error "multitail program is not installed"
  fi
}

require_stack() {
  if [ -z "$STACK_NAME" ]; then
    error "no stack_name, export \$STACK_NAME=my_cool_app"
  fi
}

require_stack_dir() {
  if [ -z "$ABRA_STACK_DIR" ] || [ ! -d "$ABRA_STACK_DIR" ]; then
    error "can't find \$ABRA_STACK_DIR '$ABRA_STACK_DIR'"
  fi
}

if [ -z "$ABRA_ENV" ] && [ -f .envrc ] && type direnv > /dev/null 2>&1 && ! direnv status | grep -q 'Found RC allowed true'; then
  error "direnv is blocked, run direnv allow"
fi

###### Run-time loading

load_custom_commands() {
  if [ -f abra-commands.sh ]; then
    # shellcheck disable=SC1091
    source abra-commands.sh
  fi

  if [ -f "$ABRA_STACK_DIR/abra-commands.sh" ]; then
    # shellcheck disable=SC1091
    source "$ABRA_STACK_DIR/abra-commands.sh"
  fi
}

load_context() {
  # Load current context from env or Docker
  if [ -z "$DOCKER_CONTEXT" ]; then
    warning "\$DOCKER_CONTEXT not set, (slowly) looking it up"
    # shellcheck disable=SC2063
    DOCKER_CONTEXT=$(docker context ls | grep '*' | cut -d' ' -f1)
    # make sure grep doesn't parse this, we want a literal '*'
  fi
}

###### Subcommand `secret`

sub_secret_insert() {
  require_stack
  load_context

  SECRET="$abra__secret_"
  VERSION="$abra__version_"
  PW="$abra__data_"

  if [ -z "$SECRET" ] || [ -z "$VERSION" ] || [ -z "$PW" ]; then
    error "Required arguments missing"
  fi

  echo "$PW" | docker secret create "${STACK_NAME}_${SECRET}_${VERSION}" - > /dev/null
  if type pass > /dev/null 2>&1; then
    echo "$PW" | pass insert "hosts/$DOCKER_CONTEXT/${STACK_NAME}/${SECRET}" -m > /dev/null
    success "pass: hosts/$DOCKER_CONTEXT/${STACK_NAME}/${SECRET}"
  fi
}

sub_secret_generate(){
  SECRET="$abra__secret_"
  VERSION="$abra__version_"
  PWGEN=${abra__cmd_:-pwqgen}

  if [ -z "$SECRET" ] || [ -z "$VERSION" ]; then
    error "Required arguments missing"
    echo
    echo "Usage: $PROGRAM_NAME secret generate SECRET VERSION [PW]"
    exit
  fi

  PW=$($PWGEN)

  success "Password: $PW"

  # FIXME 3wc: this is a little janky, might be better to make a
  # util_secret_insert function which this and sub_secret_insert can call
  abra__data_="$PW"

  sub_secret_insert
}

###### Subcommand `run`

sub_run(){
  require_stack

  if [ -n "$abra___user" ]; then
    RUN_USER="-u $abra___user"
  fi

  CONTAINER=$(docker container ls --format "table {{.ID}},{{.Names}}" \
    | grep "${STACK_NAME}_${abra__app_}" | head -n1 | cut -d',' -f1)

  if [ -z "$CONTAINER" ]; then
    error "Can't find a container for ${STACK_NAME}_${abra__app_}"
    exit
  fi

  # shellcheck disable=SC2086
  docker exec $RUN_USER -it "$CONTAINER" "$@"

  return
}

###### Subcommand `deploy`

sub_deploy (){
  require_stack
  require_stack_dir
  load_context

  echo "About to deploy:"
  echo "  Context: $(tput setaf 4)${DOCKER_CONTEXT}$(tput sgr0)"
  if [ "${COMPOSE_FILE/:/}" == "${COMPOSE_FILE}" ]; then
    echo "  Compose: $(tput setaf 3)${ABRA_STACK_DIR}/${COMPOSE_FILE}$(tput sgr0)"
  else
    echo "  Compose: $(tput setaf 3)${ABRA_STACK_DIR}/"
    IFS=':' read -ra COMPOSE_FILES <<< "$COMPOSE_FILE"
    for COMPOSE in "${COMPOSE_FILES[@]}"; do
      echo "   - ${COMPOSE}"
    done
    tput sgr0
  fi
  if [ -n "$DOMAIN" ]; then
    echo "  Domain: $(tput setaf 2)${DOMAIN}$(tput sgr0)"
  fi
  echo "  Stack: $(tput setaf 1)${STACK_NAME}$(tput sgr0)"

  read -rp "Continue? (y/[n])? " choice

  case "$choice" in
    y|Y ) ;;
    n|N ) return;;
    * ) return;;
  esac

  (
    cd "$ABRA_STACK_DIR" || error "\$ABRA_STACK_DIR '$ABRA_STACK_DIR' not found"
    # shellcheck disable=SC2086
    if docker stack deploy -c ${COMPOSE_FILE/:/ -c } "$STACK_NAME"; then
      if [ -n "$DOMAIN" ]; then
        success "Yay! App should be available at https://${DOMAIN}"
      else
        success "Yay! That worked. No \$DOMAIN defined, check logs."
      fi
    else
      error "Oh no! Something went wrong 😕 Check errors above"
    fi
  )
}

###### Subcommand `logs`

# Inspired by https://github.com/moby/moby/issues/31458#issuecomment-475411564
sub_multilogs() {
  require_stack
  require_multitail

  # Get a list of the service names
  SERVICES=$(docker stack services --format "{{.Name}}" "${STACK_NAME}")
  # Sort the service names
  SERVICES=$(echo "${SERVICES}" | sort)
  # Create the command to run
  COMMAND='multitail --mergeall'
  for SERVICE in ${SERVICES}; do
      COMMAND="${COMMAND} -L 'docker service logs --tail 20 -f ${SERVICE}'"
  done
  # Run the command
  bash -c "${COMMAND}"
}

sub_logs (){
  require_stack

  SERVICE="${abra__service_}"

  if [ -z "$SERVICE" ]; then
    warning "No \$SERVICE provided, running multilogs"
    sub_multilogs
  fi

  shift

  if [ $# -eq 0 ]; then
    LOGS_ARGS="\
    --follow \
    --tail 20 \
    --no-trunc \
    --details \
    --timestamps"
  else
    # shellcheck disable=SC2124
    LOGS_ARGS=$@
  fi

  # shellcheck disable=SC2086
  docker service logs "${STACK_NAME}_${SERVICE}" $LOGS_ARGS
}

###### Subcommand `cp`

sub_cp() {
  require_stack

  SOURCE="${abra__src_}"
  DEST="${abra__dst_}"

  SERVICE=$(echo "$SOURCE" | grep -o '^[^:]\+:' || echo "$DEST" | grep -o '^[^:]\+:')
  SERVICE=$(echo "$SERVICE" | tr -d ':')

  if [ -z "$SERVICE" ]; then
    echo "Usage: $PROGRAM_NAME cp SERVICE:SRC_PATH DEST_PATH"
    echo "       $PROGRAM_NAME cp SRC_PATH SERVICE:DEST_PATH"
    echo ""
    error "Can't find SERVICE in either SRC or DEST"
  fi

  CONTAINER=$(docker container ls --format "table {{.ID}},{{.Names}}" \
    | grep "${STACK_NAME}_${SERVICE}" | cut -d',' -f1)

  if [ -z "$CONTAINER" ]; then
    error "Can't find a container for ${STACK_NAME}_${SERVICE}"
    exit
  fi

  CP_ARGS=$(echo "$SOURCE $DEST" | sed "s/$SERVICE:/$CONTAINER:/")

  # shellcheck disable=SC2086
  docker cp ${CP_ARGS}
}

###### Subcommand `server`

sub_server_init() {
  export DOCKER_CONTEXT="${abra__host_}"

  if [ -n "$DOCKER_CONTEXT" ]; then
    load_context
  fi

  docker swarm init || true
  docker network create --driver=overlay proxy --scope swarm || true
}

sub_server_add() {
  HOST="$abra__host_"
  USERNAME="$abra__user_"
  PORT="$abra__port_"

  if [ -n "$PORT" ]; then
    PORT=":$PORT"
  fi

  if [ -n "$USERNAME" ]; then
    USERNAME="$USERNAME@"
  fi

  docker context create "$HOST" \
    --docker "host=ssh://$USERNAME$HOST$PORT"
}

sub_server_rm() {
  docker context rm  "$abra__host_"
}

sub_server_use() {
  docker context use "$abra__host_"
}

sub_server_ls() {
  docker context ls
}

###### Subcommand `upgrade`

sub_upgrade() {
  curl https://install.abra.autonomic.zone | bash
}

###### Docker alias subcommands

sub_stack() {
  # shellcheck disable=SC2068
  docker stack $@
}

sub_volume() {
  # shellcheck disable=SC2068
  docker volume $@
}

###### Main

abra() {
  # TODO (3wc): we either need to do this, or add 'shellcheck disable' all over
  # the place to handle the dynamically-defined vars
  declare abra___stack abra___env abra___config  abra__command_ abra__args_ \
    abra__secret_ abra__version_ abra__data_ abra___user abra__host_ \
    abra__app_ abra__port_ abra__user_ abra__service_ abra__src_ abra__dst_

  DOCOPT_PREFIX=abra_
  eval "$(docopt "$@")"

  STACK_NAME=$abra___stack

  if [ -n "$abra___env" ]; then
    # shellcheck disable=SC1090
    source "$abra___env" || error "Unable to load env from '$abra___env'"
  fi

  load_custom_commands

  # Search for sub_* functions, and check if any of them matches enabled
  # arguments (i.e. is a command and is specified)
  for SUBCOMMAND in $(declare -Ff | grep 'sub_' | cut -d' ' -f3 | sort -r); do
    IFS='_' read -r -a PARTS <<< "$SUBCOMMAND"
    for PART in "${PARTS[@]:1}"; do
      VAR=$(eval "echo \$abra_$PART")
      if [ ! "$VAR" == "true" ]; then
        continue 2
      fi
    done
    abra__command_=$(IFS="_"; echo "${PARTS[*]:1}")
    break
  done

  # Fallback for non-docopt-defined options
  # shellcheck disable=SC2086
  "sub_${abra__command_}" ${abra__args_[*]}
}

abra "$@"