From efba3be1636b0dd534859ba2816fcc231c4f8ca4 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@open-infrastructure.net>
Date: Sun, 30 Oct 2022 11:28:07 +0100
Subject: Reworking knot-zones-reset script.

Signed-off-by: Daniel Baumann <daniel.baumann@open-infrastructure.net>
---
 knot/bin/knot-reset-zones        | 58 -------------------------------
 knot/bin/knot-zones-reset        | 75 ++++++++++++++++++++++++++++++++++++++++
 knot/share/cron/knot-reset-zones |  3 --
 knot/share/cron/knot-zones-reset |  3 ++
 4 files changed, 78 insertions(+), 61 deletions(-)
 delete mode 100755 knot/bin/knot-reset-zones
 create mode 100755 knot/bin/knot-zones-reset
 delete mode 100755 knot/share/cron/knot-reset-zones
 create mode 100755 knot/share/cron/knot-zones-reset

diff --git a/knot/bin/knot-reset-zones b/knot/bin/knot-reset-zones
deleted file mode 100755
index 40779cf..0000000
--- a/knot/bin/knot-reset-zones
+++ /dev/null
@@ -1,58 +0,0 @@
-#!/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="${*}"
-
-if [ -z "${HOSTS}" ]
-then
-	echo "Usage: ${0} localhost|[HOST1 HOST2 ...]" >&2
-	exit 1
-fi
-
-for HOST in ${HOSTS}
-do
-	case "${HOST}" in
-		localhost)
-			echo -n "Resetting in-memory data for all zones..."
-
-			service knot stop
-			rm -rf /var/lib/knot/journal/*.mdb
-			rm -rf /var/lib/knot/timers/*.mdb
-			service knot start
-
-			echo " done."
-			;;
-
-		*)
-			echo -n "'${HOST}': Resetting in-memory data for all zones..."
-
-			ssh "${HOST}" \
-				"sudo service knot stop && \
-				rm -rf /var/lib/knot/journal/*.mdb && \
-				rm -rf /var/lib/knot/timers/*.mdb && \
-				sudo service knot start"
-
-			echo " done."
-			;;
-	esac
-done
diff --git a/knot/bin/knot-zones-reset b/knot/bin/knot-zones-reset
new file mode 100755
index 0000000..6471cf4
--- /dev/null
+++ b/knot/bin/knot-zones-reset
@@ -0,0 +1,75 @@
+#!/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..."
+
+			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."
+			;;
+
+		*)
+			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
diff --git a/knot/share/cron/knot-reset-zones b/knot/share/cron/knot-reset-zones
deleted file mode 100755
index 9762da4..0000000
--- a/knot/share/cron/knot-reset-zones
+++ /dev/null
@@ -1,3 +0,0 @@
-# /etc/cron.d/knot-reset-zone
-
-0 0 * * *	root	/usr/bin/knot-reset-zones localhost > /dev/null 2>&1
diff --git a/knot/share/cron/knot-zones-reset b/knot/share/cron/knot-zones-reset
new file mode 100755
index 0000000..13dfd44
--- /dev/null
+++ b/knot/share/cron/knot-zones-reset
@@ -0,0 +1,3 @@
+# /etc/cron.d/knot-zones-reset
+
+0 0 * * *	root	/usr/bin/knot-zones-reset > /dev/null 2>&1
-- 
cgit v1.2.3