#!/bin/sh

# Open Infrastructure: service-tools

# 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

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

Parameters ()
{
	GETOPT_LONGOPTIONS="add,init,remove,update"
	GETOPT_OPTIONS="a,i,r,u,"

	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
			-a|--add)
				ACTION="add"
				shift 1
				;;

			-i|--init)
				ACTION="init"
				shift 1
				;;

			-r|--remove)
				ACTION="remove"
				shift 1
				;;

			-u|--update)
				ACTION="update"
				shift 1
				;;

			--)
				shift 1
				break
				;;

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

Usage ()
{
	echo "Usage: ${PROGRAM} [-i|--init] [-a|--add LINE] [-r|--remove LINE] [-u|--update]" >&2
	echo
	echo "See ${PROGRAM}(1) for more information."

	exit 1
}

Parameters "${@}"

Run_add()
{
	echo "'add' not implemented yet"
	exit 0
}

Run_init()
{
	echo "'init' not implemented yet"
	exit 0
}

Run_remove()
{
	echo "'remove' not implemented yet"
	exit 0
}

Run_update()
{
	# Check for pg_hba.conf
	if ! ls /etc/postgresql/*/*/pg_hba.conf > /dev/null 2>&1
	then
		echo "W: no pg_hba.conf in /etc/postgresql found." >&2
		exit 1
	fi

	# Check for managed pg_hba.conf
	if ! ls /etc/postgresql/*/*/pg_hba.conf.g > /dev/null 2>&1
	then
		echo "W: no managed pg_hba.conf in /etc/postgresql found, maybe use '${PROGRAM} --init' first." >&2
		exit 1
	fi

	# Update pg_hba.conf from pg_hba.conf.g directories
	for DIRECTORY in /etc/postgresql/*/*/pg_hba.conf.g
	do
		if ls "${DIRECTORY}"/*.conf > /dev/null 2>&1
		then
			CONFIG="$(basename "${DIRECTORY}" .g)"

			echo -n "Updating ${CONFIG}..."

			rm -f "${CONFIG}"

			for FILE in "${DIRECTORY}"/*.conf
			do
				cat "${FILE}" >> "${CONFIG}"
			done

			echo " done."
		fi
	done
}

case "${PROGRAM}" in
	update-pg_hba.conf)
		ACTION="update"
		;;
esac

case "${ACTION}" in
	add)
		Run_add
		;;

	init)
		Run_init
		;;

	remove)
		Run_remove
		;;

	update)
		Run_update
		;;

	*)
		Usage
		;;
esac