#!/bin/sh

# Copyright (C) 2014-2022 Daniel Baumann <daniel.baumann@open-infrastructure.net>
#
# SPDX-License-Identifier: GPL-3.0+
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.

set -e

PROJECT="open-infrastructure"
SOFTWARE="compute-tools"
PROGRAM="container"
COMMAND="$(basename ${0})"

CONFIG="/etc/${SOFTWARE}/config"
HOOKS="/etc/${SOFTWARE}/hooks"

Parameters ()
{
	GETOPT_LONGOPTIONS="force,start,stop,"
	GETOPT_OPTIONS="f,s,t,"

	PARAMETERS="$(getopt --longoptions ${GETOPT_LONGOPTIONS} --name=${COMMAND} --options ${GETOPT_OPTIONS} --shell sh -- ${@})"

	if [ "${?}" != "0" ]
	then
		echo "'${COMMAND}': getopt exit" >&2
		exit 1
	fi

	eval set -- "${PARAMETERS}"

	while true
	do
		case "${1}" in
			-f|--force)
				FORCE="true"
				shift 1
				;;

			-s|--start)
				ACTION="start"
				shift 1
				;;

			-t|--stop)
				ACTION="stop"
				shift 1
				;;

			--)
				shift 1
				break
				;;

			*)
				echo "'${COMMAND}': getopt error" >&2
				exit 1
				;;
		esac
	done
}

Usage ()
{
	echo "Usage: ${PROGRAM} ${COMMAND} -f|--force -s|--start -t|--stop" >&2
	echo
	echo "See ${COMMAND}(1), ${PROGRAM}(1) and ${PROJECT}(7) for more information."

	exit 1
}

Parameters "${@}"

if [ -z "${ACTION}" ]
then
	Usage
fi

HOST="$(hostname -f 2> /dev/null || hostname)"

# Pre hooks
for FILE in "${HOOKS}/pre-${COMMAND}".* "${HOOKS}/${NAME}.pre-${COMMAND}"
do
	if [ -x "${FILE}" ]
	then
		"${FILE}"
	fi
done

# Run
OPTIONS=""

case "${FORCE}" in
	true)
		OPTIONS="${OPTIONS} -f"
		;;
esac

for FILE in "${CONFIG}"/*.conf
do
	if ! grep -Eqs "^ *cnt.container-server=${HOST}" "${FILE}"
	then
		continue
	fi

	CONTAINER="$(basename ${FILE} .conf)"
	CNT_AUTO="$(grep -Es "^ *cnt.auto=" ${FILE} | awk -F= '{ print $2 }')"

	case "${ACTION}" in
		start)
			case "${CNT_AUTO}" in
				force-true)
					OPTIONS="${OPTIONS} -f"

					cnt ${ACTION} -n ${CONTAINER} ${OPTIONS} || true
					;;

				last-on)
					if grep -qs start "/var/lib/${SOFTWARE}/state/${CONTAINER}.run" || \
					   [ ! -e "/var/lib/${SOFTWARE}/state/${CONTAINER}.run" ]
					then
						cnt start -n ${CONTAINER} ${OPTIONS} -f || true
					fi
					;;

				last-off)
					if grep -qs start "/var/lib/${SOFTWARE}/state/${CONTAINER}.run"
					then
						cnt start -n ${CONTAINER} ${OPTIONS} -f || true
					fi
					;;

				true)
					cnt ${ACTION} -n ${CONTAINER} ${OPTIONS} || true
					;;
			esac
			;;

		stop)
			OPTIONS="${OPTIONS} -f --stateless"

			cnt ${ACTION} -n ${CONTAINER} ${OPTIONS} || true
			;;
	esac
done

# Post hooks
for FILE in "${HOOKS}/post-${COMMAND}".* "${HOOKS}/${NAME}.post-${COMMAND}"
do
	if [ -x "${FILE}" ]
	then
		"${FILE}"
	fi
done