#!/bin/sh

# Copyright (C) 2014-2019 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 <http://www.gnu.org/licenses/>.

set -e

PROJECT="open-infrastructure"
SOFTWARE="storage-tools"
PROGRAM="cephfs-snap"

ACTION="${1:-manual}"

Usage ()
{
	echo "Usage: ${PROGRAM} {hourly|daily|weekly|monthly|yearly}" >&2
	exit 1
}

if [ -e "/etc/${PROJECT}/ceph-tools.conf" ]
then
	. "/etc/${PROJECT}/ceph-tools.conf"
fi

case "${CEPHFS_SNAP}" in
	true)
		;;

	*)
		exit 0
		;;
esac

if [ -z "${CEPHFS_SNAP_DIRECTORIES}" ]
then
	echo "E: no cephfs directories defined in /etc/${PROJECT}/ceph-tools.conf" >&2
	exit 1
fi

case "${ACTION}" in
	hourly|daily|weekly|monthly|yearly)
		ROTATION="$(eval "echo \${CEPHFS_SNAP_$(echo "${ACTION}" | tr [a-z] [A-Z])}")"

		if [ -z "${ROTATION}" ]
		then
			echo "E: no cephfs snapshot rotation for ${ACTION} defined in /etc/${PROJECT}/ceph-tools.conf" >&2
			exit 1
		fi
		;;

	manual)
		;;

	*)
		Usage
		;;
esac

# Run
TIME="$(date +%Y-%m-%d_%H%M%S%z)"
SNAPSHOT="${TIME}_${ACTION}"

for CEPHFS_SNAP_DIRECTORY in ${CEPHFS_SNAP_DIRECTORIES}
do
	cd "${CEPHFS_SNAP_DIRECTORY}"/.snap
	mkdir "${SNAPSHOT}"

	case "${ACTION}" in
		manual)
			;;

		*)
			if [ "$(ls -d *_"${ACTION}" | wc -l)" -gt "${ROTATION}" ]
			then
				# more snapshots available than configured to keep
				SNAPS="$(ls -d *_"${ACTION}" | sort -r | tail -n +$((${ROTATION} + 1)))"

				for SNAP in ${SNAPS}
				do
					rmdir "${SNAP}"
				done
			fi
			;;
	esac
done

DATE="$(date +%Y-%m-%d\ %H:%M:%S)"
HOST="$(hostname -f)"

# logfile
mkdir -p "/var/log/${SOFTWARE}/${PROGRAM}"
echo "${DATE} ${HOST} ${PROGRAM} creating ${ACTION} snapshots" >> "/var/log/${SOFTWARE}/${PROGRAM}/${PROGRAM}.log"

# irc
if  [ -e /usr/bin/irk ] && [ -e "/etc/${PROJECT}/ceph-tools.conf" ]
then
	if [ -n "${IRK_TARGETS}" ]
	then
		for TARGET in ${IRK_TARGETS}
		do
			irk ${TARGET} "\x0312${HOST}:\x03 \x0303${PROGRAM}\x03 creating \x0307${ACTION}\x03 snapshots"
		done
	fi
fi