#!/bin/sh

# container-tools - Manage systemd-nspawn containers
# Copyright (C) 2014-2016 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

COMMAND="$(basename ${0})"

CONFIG="/etc/container-tools/config"
MACHINES="/var/lib/machines"

Parameters ()
{
	LONG_OPTIONS="all,format:,host:,started,stopped"
	OPTIONS="a,f:,h:,s,t"

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

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

	eval set -- "${PARAMETERS}"

	while true
	do
		case "${1}" in
			-a|--all)
				LIST="all"
				shift 1
				;;

			-f|--format)
				FORMAT="${2}"
				shift 2
				;;

			-h|--host)
				HOST="${2}"
				shift 2
				;;

			-s|--started)
				LIST="started"
				shift 1
				;;

			-t|--stopped)
				LIST="stopped"
				shift 1
				;;

			--)
				shift 1
				break
				;;

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

Usage ()
{
	echo "Usage: container ${COMMAND} -a|--all --format FORMAT -s|--started -t|--stopped" >&2
	exit 1
}

Parameters "${@}"

LIST="${LIST:-all}"
FORMAT="${FORMAT:-full}"
HOST="${HOST:-$(hostname -f)}"

# Run
case "${FORMAT}" in
	full)
		RED="$(tput setaf 1)$(tput bold)"
		GREEN="$(tput setaf 2)$(tput bold)"
		YELLOW="$(tput setaf 3)$(tput bold)"
		BLUE="$(tput setaf 4)$(tput bold)"
		NORMAL="$(tput sgr0)"

cat << EOF
  Container                                          IPv4 Address(es)
--------------------------------------------------------------------------------
EOF

		;;

	short)
		;;
esac

CONTAINERS="$(cd "${MACHINES}" 2>/dev/null && find -maxdepth 1 -type d -and -not -name '.container-tools' -and -not -name 'container-tools' -and -not -name 'lost+found' -and -not -name '.snap' -and -not -name '.snapshot' -printf '%P\n' | sort)"

for CONTAINER in ${CONTAINERS}
do
	if [ "${HOST}" != "all" ]
	then
		if [ -e "${CONFIG}/${CONTAINER}.conf" ]
		then
			AUTO="$(awk -F= '/^cnt.auto=/ { print $2 }' ${CONFIG}/${CONTAINER}.conf)"
			AUTO="${AUTO:-false}"

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

				*)
					continue
					;;
			esac
		else
			continue
		fi
	fi

	# FIXME: ignore lxc container for now
	if [ -e "${MACHINES}/${CONTAINER}/rootfs" ]
	then
		continue
	fi

	STATE="$(machinectl show ${CONTAINER} 2>&1 | awk -F= '/^State=/ { print $2 }')"

	if [ -e "${MACHINES}/${CONTAINER}/etc/network/interfaces" ]
	then
		ADDRESSES="$(awk '/address/ { print $2 }' ${MACHINES}/${CONTAINER}/etc/network/interfaces)"
	fi

	ADDRESSES="${ADDRESSES:-n/a}"

	case "${STATE}" in
		running)
			STATUS="${GREEN}✓${NORMAL}"

			;;

		*)
			STATUS="${RED}✕${NORMAL}"
			;;
	esac

	case "${LIST}" in
		all)
			case "${FORMAT}" in
				short)
					printf "${CONTAINER}\n"
					;;

				full)
					FIRST_LINE="true"

					for ADDRESS in ${ADDRESSES}
					do
						case "${FIRST_LINE}" in
							true)
								FIRST_LINE="false"
								printf "%-80s %-29s\n" "${STATUS} ${BLUE}${CONTAINER}${NORMAL}" "${YELLOW}${ADDRESS}${NORMAL}"
								;;

							*)
								printf "%-80s %-29s\n" "" "${ADDRESS}"
								;;
						esac
					done
					;;
			esac
			;;

		started)
			case "${STATE}" in
				running)
					case "${FORMAT}" in
						short)
							printf "${CONTAINER}\n"
							;;

						full)
							FIRST_LINE="true"

							for ADDRESS in ${ADDRESSES}
							do
								case "${FIRST_LINE}" in
									true)
										FIRST_LINE="false"
										printf "%-80s %-29s\n" "${STATUS} ${BLUE}${CONTAINER}${NORMAL}" "${YELLOW}${ADDRESS}${NORMAL}"
										;;

									*)
										printf "%-80s %-29s\n" "" "${ADDRESS}"
										;;
								esac
							done
							;;
					esac
					;;
			esac
			;;

		stopped)
			case "${STATE}" in
				running)
					;;

				*)
					case "${FORMAT}" in
						short)
							printf "${CONTAINER}\n"
							;;

						full)
							FIRST_LINE="true"

							for ADDRESS in ${ADDRESSES}
							do
								case "${FIRST_LINE}" in
									true)
										FIRST_LINE="false"
										printf "%-80s %-29s\n" "${STATUS} ${BLUE}${CONTAINER}${NORMAL}" "${YELLOW}${ADDRESS}${NORMAL}"
										;;

									*)
										printf "%-80s %-29s\n" "" "${ADDRESS}"
										;;
								esac
							done
							;;
					esac
					;;
			esac
			;;
	esac
done