publish

#!/bin/bash
set -e

SELFDIR=$(dirname "$0")
SELFDIR=$(cd "$SELFDIR" && pwd)
# shellcheck source=internal/lib/library.sh
source "$SELFDIR/internal/lib/library.sh"
# shellcheck source=internal/lib/distro_info.sh
source "$SELFDIR/internal/lib/distro_info.sh"

OUTPUT_DIR=
REPO_SERVER_API_USERNAME=
REPO_SERVER_API_TOKEN_FILE=
REPOSITORY=
LOG_DIR=
YANK=false
YANK_ALL=false
CONCURRENCY=1
SHOW_BACKTRACES=false
SHOW_TASKS=false
ARGV=

function usage()
{
	echo "Usage: ./publish [OPTIONS] <TASK NAMES...>"
	echo "Publish built RPM packages to the Passenger RPM repos server."
	echo
	echo "Required options:"
	echo "  -d DIR     Path in which build products are stored"
	echo "  -u NAME    Repo server API username"
	echo "  -c PATH    Path to repo server API token file"
	echo "  -r NAME    Repository name: yum-repo-oss(.staging) or"
	echo "             yum-repo-enterprise(.staging)"
	echo
	echo "Optional options:"
	echo "  -l DIR     Write logs to the given directory"
	echo "  -j NUM     Set concurrency. Default: 1"
	echo "  -y         Yank old packages"
	echo "  -Y         Yank all packages from repository before publishing"
	echo "  -O         Periodically show progress overview"
	echo "  -t         Show backtraces on error"
	echo
	echo "  -T         Show all tasks"
	echo "  -h         Show usage"
}

function parse_options()
{
	local OPTIND=1
	local ORIG_ARGV
	local opt
	while getopts "d:u:c:r:D:C:l:j:yYOtTh" opt; do
		case "$opt" in
		d)
			OUTPUT_DIR="$OPTARG"
			;;
		u)
			REPO_SERVER_API_USERNAME="$OPTARG"
			;;
		c)
			REPO_SERVER_API_TOKEN_FILE="$OPTARG"
			;;
		r)
			REPOSITORY="$OPTARG"
			;;
		l)
			LOG_DIR="$OPTARG"
			;;
		j)
			CONCURRENCY=$OPTARG
			;;
		y)
			YANK=true
			;;
		Y)
			YANK_ALL=true
			;;
		O)
			SHOW_OVERVIEW_PERIODICALLY=true
			;;
		t)
			SHOW_BACKTRACES=true
			;;
		T)
			SHOW_TASKS=true
			;;
		h)
			usage
			exit
			;;
		*)
			return 1
			;;
		esac
	done

	(( OPTIND -= 1 )) || true
	shift $OPTIND || true
	ORIG_ARGV=("$@")

	BUILDBOX_IMAGE=`get_buildbox_image`

	if [[ ${#ORIG_ARGV[@]} = 0 ]]; then
		SHOW_TASKS=true
	else
		ARGV=("${ORIG_ARGV[@]}" finish)
		if $SHOW_BACKTRACES; then
			ARGV+=(--trace)
		fi
	fi

	if [[ "$OUTPUT_DIR" = "" ]]; then
		echo "ERROR: please specify a build products directory with -d."
		exit 1
	fi
	if ! $SHOW_TASKS; then
		if [[ "$REPO_SERVER_API_USERNAME" = "" ]]; then
			echo "ERROR: please specify a repo server API username with -u."
			exit 1
		fi
		if [[ "$REPO_SERVER_API_TOKEN_FILE" = "" ]]; then
			echo "ERROR: please specify a repo server API token file with -c."
			exit 1
		fi
		if [[ "$REPOSITORY" = "" ]]; then
			echo "ERROR: please specify a repository name with -r."
			exit 1
		fi
	fi
}

parse_options "$@"
OUTPUT_DIR="`absolute_path \"$OUTPUT_DIR\"`"
if [[ "$REPO_SERVER_API_TOKEN_FILE" = "" ]]; then
	REPO_SERVER_API_TOKEN_FILE="`absolute_path \"$REPO_SERVER_API_TOKEN_FILE\"`"
fi

EXTRA_MOUNTS=()
if [[ "$LOG_DIR" != "" ]]; then
	LOG_DIR="`absolute_path \"$LOG_DIR\"`"
	EXTRA_MOUNTS+=(-v "$LOG_DIR:/work")
	run mkdir -p "$LOG_DIR"
fi

if tty -s; then
	TTY_ARGS="-t -i"
else
	TTY_ARGS=
fi

if $SHOW_TASKS; then
	exec docker run \
		--rm $TTY_ARGS \
		-v "$SELFDIR/internal:/system/internal:ro" \
		-v "$OUTPUT_DIR:/output:ro" \
		-e "APP_UID=`/usr/bin/id -u`" \
		-e "APP_GID=`/usr/bin/id -g`" \
		-e "SHOW_TASKS=true" \
		-e "LC_CTYPE=en_US.UTF-8" \
		$BUILDBOX_IMAGE \
		/system/internal/scripts/my_init --quiet --skip-runit --skip-startup-files -- \
		/system/internal/scripts/inituidgid.sh \
		/system/internal/publish/preinit.sh \
		/system/internal/scripts/setuser app \
		rake -f /system/internal/publish/Rakefile -T --trace
else
	echo "-------- Entering Docker container --------"
	exec docker run \
		--rm $TTY_ARGS \
		-v "$SELFDIR/internal:/system/internal:ro" \
		-v "$OUTPUT_DIR:/output:ro" \
		-v "$REPO_SERVER_API_TOKEN_FILE:/repo_server_api_token.txt:ro" \
		"${EXTRA_MOUNTS[@]}" \
		-e "REPOSITORY=$REPOSITORY" \
		-e "YANK=$YANK" \
		-e "YANK_ALL=$YANK_ALL" \
		-e "REPO_SERVER_API_USERNAME=$REPO_SERVER_API_USERNAME" \
		-e "SHOW_OVERVIEW_PERIODICALLY=$SHOW_OVERVIEW_PERIODICALLY" \
		-e "APP_UID=`/usr/bin/id -u`" \
		-e "APP_GID=`/usr/bin/id -g`" \
		-e "LC_CTYPE=en_US.UTF-8" \
		$BUILDBOX_IMAGE \
		/system/internal/scripts/my_init --quiet --skip-runit --skip-startup-files -- \
		/system/internal/scripts/inituidgid.sh \
		/system/internal/publish/preinit.sh \
		/system/internal/scripts/setuser app \
		rake -f /system/internal/publish/Rakefile -j $CONCURRENCY "${ARGV[@]}"
fi