#!/bin/sh
#
# kas - setup tool for bitbake based projects
#
# Copyright (c) Siemens AG, 2018
#
# Authors:
#  Jan Kiszka <jan.kiszka@siemens.com>
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.

set -e

usage()
{
	printf "%b" "Usage: $0 [OPTIONS] { build | shell } [KASOPTIONS] KASFILE\n"
	printf "%b" "       $0 [OPTIONS] clean\n"
	printf "%b" "\nPositional arguments:\n"
	printf "%b" "build\t\t\tCheck out repositories and build target.\n"
	printf "%b" "shell\t\t\tRun a shell in the build environment.\n"
	printf "%b" "clean\t\t\tClean build artifacts, keep downloads.\n"
	printf "%b" "\nOptional arguments:\n"
	printf "%b" "--isar\t\t\tUse kas-isar container to build Isar image.\n"
	printf "%b" "--docker-args\t\tAdditional arguments to pass to docker for" \
		"running the build.\n"
	printf "%b" "-v\t\t\tPrint operations.\n"
	printf "%b" "--ssh-dir\t\tDirectory containing SSH configurations, " \
		"avoid $HOME/.ssh unless you fully trust the container.\n"
	printf "%b" "--no-proxy-from-env\tDo not inherit proxy settings from environment.\n"
	exit 1
}

trace()
{
	[ -n "${VERBOSE}" ] && echo "+ ""$@"
	eval "$@"
}

if [ -z "${KAS_IMAGE_VERSION}" ]; then
	KAS_IMAGE_VERSION="latest"
fi

DOCKER_IMAGE=kasproject/kas:${KAS_IMAGE_VERSION}

if [ -n "${KAS_WORK_DIR}" ]; then
	KAS_WORK_DIR=$(readlink -f ${KAS_WORK_DIR})
else
	KAS_WORK_DIR=$(pwd)
fi

# parse kas-docker options
while [ $# -gt 0 ]; do
	case "$1" in
	--isar)
		DOCKER_IMAGE=kasproject/kas-isar:${KAS_IMAGE_VERSION}
		if ! LOOP_DEV=$(/sbin/losetup -f 2>/dev/null); then
			if [ $(id -u) -eq 0 ]; then
				echo "Error: loop device not available!"
				exit 1
			fi
			sudo_command="/sbin/losetup -f"
			sudo_message="[sudo] enter password to setup loop"
			sudo_message="$sudo_message devices by calling"
			sudo_message="$sudo_message '$sudo_command': " 
			if ! LOOP_DEV=$(sudo -p "$sudo_message" $sudo_command \
				2>/dev/null); then
				echo "Error: loop device setup unsuccessful!"
				echo "try calling '$sudo_command' with root" \
					"permissions manually."
				exit 1
			fi
		fi
		ISAR_ARGS="--cap-add=SYS_ADMIN --cap-add=MKNOD --privileged \
			--device ${LOOP_DEV}"
		shift 1
		;;
	--docker-args)
		[ $# -gt 0 ] || usage
		USER_ARGS=$2
		shift 2
		;;
	--ssh-dir)
		[ $# -gt 2 ] || usage
		SSH_DIR=$2
		shift 2
		;;
	--no-proxy-from-env)
		NO_PROXY_FROM_ENV=1
		shift 1
		;;
	-v)
		VERBOSE=1
		shift 1
		;;
	--*)
		usage
		;;
	clean)
		[ $# -eq 1 ] || usage
		CLEAN_DIR=build/tmp
		if [ -n "${ISAR_ARGS}" ]; then
			trace docker run -v ${KAS_WORK_DIR}:/work:rw \
					 --workdir=/work --rm ${ISAR_ARGS} \
					 ${DOCKER_IMAGE} \
					 sudo rm -rf ${CLEAN_DIR}
		else
			trace rm -rf ${KAS_WORK_DIR}/{CLEAN_DIR}
		fi
		exit 0
		;;
	build|shell)
		CMD=$1
		shift 1
		break
		;;
	*)
		usage
		;;
	esac
done

[ -n "${CMD}" ] || usage

# parse kas sub-command (build or shell) options
while [ $# -gt 0 ]; do
	case "$1" in
	-h|--help)
		trace docker run ${DOCKER_IMAGE} ${CMD} --help
		exit 0
		;;
	--skip|--target|--task|-c|--command)
		KAS_OPTIONS="${KAS_OPTIONS} $1 '$2'"
		shift 2
		;;
	-*)
		KAS_OPTIONS="${KAS_OPTIONS} $1"
		shift 1
		;;
	*)
		KAS_FILES=
		for FILE in $(IFS=':'; echo $1); do
			if ! REAL_FILE=$(realpath -qe $FILE); then
				echo "Error: configuration file '${FILE}' not found"
				exit 1
			fi
			if [ -z "${KAS_FILES}" ]; then
				FIRST_KAS_FILE=${REAL_FILE}
				KAS_FILES=${REAL_FILE}
			else
				KAS_FILES=${KAS_FILES}:${REAL_FILE}
			fi
		done
		shift 1
		;;
	esac
done

[ -n "${FIRST_KAS_FILE}" ] || usage

KAS_FILE_DIR=$(dirname ${FIRST_KAS_FILE})

REPO_DIR=$(git -C ${KAS_FILE_DIR} rev-parse --show-toplevel 2>/dev/null) \
	|| REPO_DIR=$(hg --cwd ${KAS_FILE_DIR} root 2>/dev/null) \
	|| REPO_DIR=${KAS_FILE_DIR}

KAS_FILES=/repo/$(echo ${KAS_FILES} | sed 's|'${REPO_DIR}'/||g;s|:|:/repo/|g')

trace mkdir -p ${KAS_WORK_DIR}

DOCKER_ARGS="-v ${REPO_DIR}:/repo:ro \
	     -v ${KAS_WORK_DIR}:/work:rw --workdir=/work \
	     -e USER_ID=$(id -u) -e GROUP_ID=$(id -g) --rm"

if [ -n "${SSH_DIR}" ] ; then
	if [ ! -d "${SSH_DIR}" ]; then
		echo "Passed SSH_VALUE '${SSH_DIR}' is not a directory"
		exit 1
	fi
	DOCKER_ARGS="${DOCKER_ARGS} -v $(readlink -f ${SSH_DIR}):/etc/skel/.ssh:ro"
fi

if [ -t 1 ]; then
	DOCKER_ARGS="${DOCKER_ARGS} -t -i"
fi

if [ -n "${DL_DIR}" ]; then
	trace mkdir -p ${DL_DIR}
	DOCKER_ARGS="${DOCKER_ARGS} \
		-v $(readlink -f ${DL_DIR}):/downloads:rw \
		-e DL_DIR=/downloads"
fi

if [ -n "${SSTATE_DIR}" ]; then
	trace mkdir -p ${SSTATE_DIR}
	DOCKER_ARGS="${DOCKER_ARGS} \
		-v $(readlink -f ${SSTATE_DIR}):/sstate:rw \
		-e SSTATE_DIR=/sstate"
fi

if [ -n "${KAS_REPO_REF_DIR}" ]; then
	DOCKER_ARGS="${DOCKER_ARGS} \
		-v $(readlink -f ${KAS_REPO_REF_DIR}):/repo-ref:ro \
		-e KAS_REPO_REF_DIR=${KAS_REPO_REF_DIR}"
fi

set -- ${DOCKER_ARGS}

for var in TERM KAS_DISTRO KAS_MACHINE KAS_TARGET KAS_TASK \
           KAS_PREMIRRORS; do
	if [ -n "$(eval echo \$${var})" ]; then
		set -- "$@" -e "${var}='$(eval echo \"\$${var}\")'"
	fi
done

# propagate only supported SHELL settings
case "$SHELL" in
/bin/sh|/bin/bash|/bin/dash)
	set -- "$@" -e "SHELL='$(eval echo \"\$SHELL\")'"
	;;
esac

if [ -z "${NO_PROXY_FROM_ENV+x}" ]; then
	for var in http_proxy https_proxy ftp_proxy no_proxy NO_PROXY; do
		if [ -n "$(eval echo \$${var})" ]; then
			set -- "$@" -e "${var}='$(eval echo \$${var})'"
		fi
	done
fi

trace docker run "$@" ${ISAR_ARGS} ${USER_ARGS} \
		 ${DOCKER_IMAGE} ${CMD} ${KAS_OPTIONS} ${KAS_FILES}