#!/bin/sh

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)
			NUMBER="$(systemctl | grep -c 'kresd@[0-9].service')"

			for NUMBER in $(seq 1 "${NUMBER}")
			do
				echo "Flushing localhost, resolver ${NUMBER}"
				echo "cache.clear()" | sudo socat - UNIX-CONNECT:/run/knot-resolver/control/"${NUMBER}"
			done
			;;

		*)
			NUMBER="$(ssh "${HOST}" sudo systemctl | grep -c 'kresd@[0-9].service')"

			for NUMBER in $(seq 1 "${NUMBER}")
			do
				echo "Flushing ${HOST}, resolver ${NUMBER}"
				echo "cache.clear()" | ssh "${HOST}" sudo socat - UNIX-CONNECT:/run/knot-resolver/control/"${NUMBER}"
			done
			;;
	esac
done

echo