#!/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

HOSTS="${*:-localhost}"

for HOST in ${HOSTS}
do
	case "${HOST}" in
		localhost)
			echo "Resetting in-memory data for all zones..."

			if systemctl status knot | grep -qs 'Active: active'
			then
				START="true"
				sudo service knot stop
			else
				START="false"
			fi

			sudo rm -rf /var/lib/knot/journal/*.mdb
			sudo rm -rf /var/lib/knot/timers/*.mdb

			if [ -e /var/lib/knot/zones/.git ]
			then
				sudo chown -R root:root /var/lib/knot/zones/

				cd /var/lib/knot/zones
				sudo git clean -dxf
				sudo git checkout -f

				sudo chown -R knot:knot /var/lib/knot/zones/
			fi

			case "${START}" in
				true)
					sudo service knot start
					;;
			esac

			echo
			echo "done."
			;;

		*)
			echo "'${HOST}': Resetting in-memory data for all zones..."

			ssh "${HOST}" \
				"sudo service knot stop && \
				sudo rm -rf /var/lib/knot/journal/*.mdb && \
				sudo rm -rf /var/lib/knot/timers/*.mdb && \
				if [ -e /var/lib/knot/zones/.git ]; \
				then \
					sudo chown -R root:root /var/lib/knot/zones/; \
					cd /var/lib/knot/zones; \
					sudo git clean -dxf; \
					sudo git checkout -f; \
					sudo chown -R knot:knot /var/lib/knot/zones/; \
				fi; \
				sudo service knot start"

			echo
			echo "done."
			;;
	esac
done