#!/bin/sh

# Copyright (c) 2002, 2003, 2004, 2005 Peter Palfrader <peter@palfrader.org>
#
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish,
# distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so, subject to
# the following conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

umask 037

set -e
set -u

DATE=`date "+%Y%m%d-%H%M%S"`
CLIENTBASE=/srv/www/vhosts
SYSBASE=/var/backups/local/pg

sudo -u postgres psql -l -t | awk '$1 != "" {print $1}' |
while read db; do
	if [ "$db" = "template0" -o "$db" = "template1" ] ; then
		continue;
	fi

	CLIENT=`echo $db | sed -e 's/_.*//'`;
	if [ -d "$CLIENTBASE/$CLIENT" ] ; then
		BASE="$CLIENTBASE/$CLIENT/pg"
		if [ ! -e "$BASE" ]; then
			mkdir "$BASE"
			chmod 02750 "$BASE"
		else
			if [ ! -d "$BASE" ]; then
				echo "$BASE exists but is not a directory" >&2
				continue
			fi
		fi

		file="$BASE/$DATE-$db.plain"
		sudo -u postgres pg_dump --create --format=p "$db" > "$file"
		nice bzip2 -- "$file"

		md5=`md5sum -- "$file.bz2" | awk '{print $1}'`
		if [ -e "$BASE/MD5-$md5" ]; then
			rm -- "$file.bz2"
			ln "$BASE/MD5-$md5" "$file.bz2"
		else
			ln "$file.bz2" "$BASE/MD5-$md5"
		fi


		file="$BASE/$DATE-$db.custom"
		sudo -u postgres pg_dump --blobs --create --format=c "$db" > "$file"
		nice bzip2 -- "$file"
	elif [ "$db" = "postgres" ]; then
		file="$SYSBASE/$DATE-$db.plain"
		sudo -u postgres pg_dump --create --format=p "$db" > "$file"
		nice bzip2 -- "$file"

		md5=`md5sum -- "$file.bz2" | awk '{print $1}'`
		if [ -e "$SYSBASE/MD5-$md5" ]; then
			rm -- "$file.bz2"
			ln "$SYSBASE/MD5-$md5" "$file.bz2"
		else
			ln "$file.bz2" "$SYSBASE/MD5-$md5"
		fi

		file="$SYSBASE/$DATE-$db.custom"
		sudo -u postgres pg_dump --blobs --create --format=c "$db" > "$file"
		nice bzip2 -- "$file"
	else
		echo "Not doing a backup of $db" >&2
	fi
done