#!/bin/sh

# container-tools - Manage systemd-nspawn containers
# Copyright (C) 2014-2017 Daniel Baumann <daniel.baumann@open-infrastructure.net>
#
# 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 <http://www.gnu.org/licenses/>.

set -e

SOFTWARE="container-tools"
PROGRAM="container"

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

# Parameter
PARAMETER="${1}"

if [ -z "${PARAMETER}" ]
then
	echo "Usage: ${PROGRAM} COMMAND [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/lib/${SOFTWARE}/${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|remove|rm|restart|start|stop)
				USER="${SUDO_USER:-${USER}}"
				DATE="$(date +%Y-%m-%d\ %H:%M:%S)"
				HOST="$(hostname -f)"

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

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

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

	# Run
	exec "/usr/lib/${SOFTWARE}/${PROGRAM}/${COMMAND}" "${OPTIONS}"

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