abra-bash/abra

#!/bin/bash

PROGRAM_NAME=$(basename "$0")

###### Utility functions

yml_pattern_exists() {
	PATTERN=$1

	if [ -f "$ABRA_CONFIG" ]; then
			RESULT=$(yq read "$ABRA_CONFIG" "$PATTERN")

			if [ "$RESULT" != 0 ]; then
				return 0
			fi
	fi

	return 1
}

parse_subcommand() {
	SUBCOMMAND="$1"
	PREFIX=$2

	if [ -n "$PREFIX" ]; then
		PPREFIX="_$2"
		SPREFIX="$2 "
		SSPREFIX=" $2"
	fi

	case $SUBCOMMAND in
		"" | "-h" | "--help")
			"sub${PPREFIX}_help"
			;;
		*)
			shift 2
			"sub${PPREFIX}_${SUBCOMMAND}" "$@"
			if [ $? = 127 ]; then
				echo "Error: '$SPREFIX$SUBCOMMAND' is not a known subcommand." >&2
				echo "       Run '$PROGRAM_NAME$SSPREFIX --help' for a list of known subcommands." >&2
				exit 1
			fi
			;;
	esac
}

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)"
}

###### Top-level arguments

ABRA_CONFIG=abra.yml
if [ "$1" == "-c" ]; then
	ABRA_CONFIG=$2
	shift 2
fi

if [ "$1" == "-e" ]; then
	ABRA_ENV=$2
	shift 2
fi

if [ "$1" == "-a" ]; then
	STACK_NAME=$2
	shift 2
fi

###### Load config

if [ -f "$ABRA_CONFIG" ]; then
	require_yq

	if yml_pattern_exists stack_name; then
		STACK_NAME=$(yq read "$ABRA_CONFIG" stack_name)
	fi
	# FIXME load other variables somehow
fi

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

###### Default settings

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

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

load_context() {
	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)
		# FIXME 3wc: make sure grep doesn't parse this, we're want a literal '*'
	fi
}

###### Safety checks

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

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 or add it to abra.yml"
	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

###### 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=SC1090
	source "$ABRA_STACK_DIR/abra-commands.sh"
fi

###### Global help

sub_help() {
	echo "Usage: $PROGRAM_NAME [OPTIONS] COMMAND [ARGS]..."
	echo ""
	echo "Options:"
	echo "  --config      Stack configuration to use"
	echo "  --env         Environment variables to load"
	echo "  --help        Show this message and exit"
	echo "  --stack       Name of the target stack"
	echo "  --version     Show program version"
	echo ""
	echo "Commands:"
	echo "  context     manage remote swarm contexts"
	echo "  cp          copy files to a container"
	echo "  deploy      let 'em rip"
	echo "  logs        tail logs from a deployed service"
	echo "  multilogs   tail logs from a whole stackk"
	echo "  run         run a command in the specified service's container"
	echo "  secret      manage secrets"
	echo "  upgrade     upgrade to the latest version"
}

###### Subcommand `secret`

sub_secret_help() {
	echo "Usage: $PROGRAM_NAME [global opts] secret <subcommand> [sub opts]"
	echo ""
	echo "Subcommands:"
	echo "    generate SECRET VERSION [PWGEN]       generate & store secret"
	echo "    insert SECRET VERSION PW              save PW in docker and pass"
}

sub_secret_insert() {
	require_stack
	load_context

	SECRET=$1
	VERSION=$2
	PW=$3

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

	echo "$PW" | docker secret create "${STACK_NAME}_${SECRET}_${VERSION}" - > /dev/null
	echo "$PW" | pass insert "hosts/$DOCKER_CONTEXT/${STACK_NAME}/${SECRET}" -m > /dev/null
}

sub_secret_generate(){
	SECRET=$1
	VERSION=$2
	PWGEN=${3:-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"

	sub_secret_insert "$SECRET" "$VERSION" "$PW"
}

sub_secret() {
	SUBCOMMAND=$1
	shift
	# shellcheck disable=SC2068
	parse_subcommand "$SUBCOMMAND" "secret" $@
}

###### Subcommand `run`

sub_run(){
	require_stack

	if [ "$1" == "-u" ]; then
		RUN_USER="-u $2"
		shift 2
	fi

	SERVICE=$1

	shift

	if [ -z "$SERVICE" ]; then
		echo "Usage: $PROGRAM_NAME run [-u USER] SERVICE [CMD]"
		exit
	fi

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

	if [ -z "$CONTAINER" ]; then
		error "Can't find a container for ${STACK_NAME}_${SERVICE}"
		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=$1

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

	shift

	if [ $# -eq 0 ]; then
		LOGS_ARGS="\
		--follow \
		--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=$1
	DEST=$2

	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 `context`

sub_context_help() {
	echo "Usage: $PROGRAM_NAME [global opts] context <subcommand> [sub opts]"
	echo ""
	echo "Subcommands:"
	echo "    create HOST [USER] [PORT]             set up remote Docker context"
	echo "    use HOST                              activate remote Docker context"
	echo "    init HOST                             activate swarm mode"
}

sub_context_init() {
	load_context

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

sub_context_create() {
	HOST="$1"
	USERNAME="$2"
	PORT="$3"

	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_context_use() {
	docker context use "$1"
}

sub_context() {
	SUBCOMMAND2=$1
	shift
	# shellcheck disable=SC2068
	parse_subcommand "$SUBCOMMAND2" "context" $@
}


###### Subcommand `upgrade`

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

###### Docker alias subcommands

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

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

###### Main

SUBCOMMAND=$1
shift
# shellcheck disable=SC2086,SC2068
parse_subcommand $SUBCOMMAND "" $@