1 files changed, 169 insertions, 0 deletions
diff --git a/postgresql/bin/pg_hba.conf b/postgresql/bin/pg_hba.conf
new file mode 100755
index 0000000..2b6c7cb
--- /dev/null
+++ b/postgresql/bin/pg_hba.conf
@@ -0,0 +1,169 @@
+#!/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