#!/bin/sh

# Copyright (C) 2014-2021 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"

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

# Parameter
PARAMETER="${1}"

if [ -z "${PARAMETER}" ]
then
	echo "Usage: ${PROGRAM} COMMAND [OPTIONS]" >&2
	echo "Usage: ${PROGRAM} COMMAND1,COMMAND2,... [COMMON_OPTIONS]" >&2
	exit 1
fi

# Command
COMMANDS="${1}"

# Options
shift 1
OPTIONS="${@}"

for COMMAND in $(echo ${COMMANDS} | sed -e 's|,| |g')
do
	if [ ! -e "/usr/libexec/${PROGRAM}/${COMMAND}" ]
	then
		echo "'${COMMAND}': no such ${PROGRAM} command" >&2
		exit 1
	fi

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

	# Notification
	if echo "${OPTIONS}" | grep -qs "\--no-notification"
	then
		OPTIONS="$(echo "${OPTIONS}" | sed -e 's|--no-notification||')"
	else
		case "${COMMAND}" in
			create|cr|move|mv|remove|rm|restart|rt|start|s|stop|t)
				if [ -z "${CONTAINER_USER}" ]
				then
					CONTAINER_USER="${SUDO_USER:-${USER}}"
				fi

				DATE="$(date +%Y-%m-%d\ %H:%M:%S)"
				HOST="$(hostname -f 2> /dev/null || hostname)"

				# logfile
				echo "${DATE} ${HOST} ${CONTAINER_USER} ${PROGRAM} ${COMMAND} ${OPTIONS}" >> "/var/log/${SOFTWARE}/${PROGRAM}.log"

				# irc
				if  [ -e /usr/bin/irk ] && [ -e "/etc/${SOFTWARE}.conf" ]
				then
					. "/etc/${SOFTWARE}.conf"

					if [ -n "${IRK_TARGETS}" ]
					then
						for TARGET in ${IRK_TARGETS}
						do
							irk ${TARGET} "\x0300${CONTAINER_USER}\x03@\x0312${HOST}:\x03 \x0303${PROGRAM}\x03 \x0307${COMMAND}\x03 ${OPTIONS}"
						done
					fi
				fi
				;;
		esac
	fi

	# Run
	"/usr/libexec/${PROGRAM}/${COMMAND}" "${OPTIONS}"

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