#!/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"
MACHINES="/var/lib/machines"

VERSION="$(${PROGRAM} version)"

Parameters ()
{
	GETOPT_LONGOPTIONS="name:,status,os,ip,"
	GETOPT_OPTIONS="n:,"

	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
			-n|--name)
				NAME="${2}"
				shift 2
				;;

			--status)
				ACTIONS="${ACTIONS} status"
				shift 1
				;;

			--os)
				ACTIONS="${ACTIONS} os"
				shift 1
				;;

			--ip)
				ACTIONS="${ACTIONS} ip"
				shift 1
				;;

			--)
				shift 1
				break
				;;

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

Usage ()
{
	echo "Usage: ${PROGRAM} ${COMMAND} -n|--name NAME [--status] [--os] [--ip]" >&2
	echo
	echo "See ${COMMAND}(1), ${PROGRAM}(1) and ${PROJECT}(7) for more information."

	exit 1
}

Parameters "${@}"

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

ACTIONS="${ACTIONS:-status os ip}"
HOST="$(cat /etc/hostname)"

# Run

# Status
STATUS="$(machinectl show ${NAME} 2>&1 | awk -FState= '/^State=/ { print $2 }')"

if [ -e "${CONFIG}/${NAME}.conf" ]
then
	CONTAINER_SERVER="$(awk -Fcnt.container-server= '/^cnt.container-server=/ { print $2 }' ${CONFIG}/${NAME}.conf)"
	CONTAINER_SERVER="${CONTAINER_SERVER:-false}"

	case "${CONTAINER_SERVER}" in
		${HOST}|true)
			;;

		*)
			STATUS="other"
			;;
	esac
else
	STATUS="other"
fi

case "${STATUS}" in
	running)
		STATUS="started"
	;;

	other)
		;;

	*)
		STATUS="stopped"
		;;
esac

# OS
VERSION_BASH="$(chroot ${MACHINES}/${NAME} apt-cache policy bash | awk '/Installed: / { print $2 }')"

case "${VERSION_BASH}" in
	4.1-*|4.1.[0-9]*)
		OS="Debian 6 (squeeze)"
		;;

	4.2-*|4.2.[0-9]*)
		OS="Debian 7 (wheezy)"
		;;

	4.3-*|4.3.[0-9]*)
		OS="Debian 8 (jessie)"
		;;

	4.4-*|4.4.[0-9]*)
		OS="Debian 9 (stretch)"
		;;

	5.0-*|5.0.[0-9]*)
		OS="Debian 10 (buster)"
		;;

	5.1-*|5.1.[0-9]*)
		OS="Debian 11 (bullseye)"
		;;

	5.2-*|5.2.[0-9]*)
		OS="Debian 12 (bookworm)"
		;;

	*)
		OS="n/a"
		;;
esac

case "${STATUS}" in
	started)
		IP="$(cnt run -n ${NAME} -- hostname -I)"
		;;

	*)
		if ls "${MACHINES}/${NAME}/etc/systemd/network"/*.network > /dev/null 2>&1
		then
			IP="$(awk -FAddress= '/^Address/ { printf "%s ", $2 }' ${MACHINES}/${NAME}/etc/systemd/network/*.network)"
		elif [ -e "${MACHINES}/${NAME}/etc/network/interfaces" ]
		then
			IP="$(awk '/address/ { printf "%s ", $2 }' ${MACHINES}/${NAME}/etc/network/interfaces)"
		fi

		IP="${IP:-n/a}"
		;;
esac

for ACTION in ${ACTIONS}
do
	case "${ACTION}" in
		status)
			echo "${STATUS}"
			;;

		os)
			echo "${OS}"
			;;

		ip)
			echo "${IP}"
			;;
	esac
done

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