#!/bin/sh
##
##  postgresql-repmgr-setup.sh -- PostgreSQL Replication Manager Setup Utility
##

#   command line argument sanity check
prg="$0"
if [ $# -eq 0 ]; then
    echo "$prg:ERROR: invalid command line" 1>&2
    echo "$prg:USAGE: $prg install [<database-directory>]" 1>&2
    echo "$prg:USAGE: $prg uninstall" 1>&2
    exit 1
fi

#   database configuration
db_dir="@l_prefix@/var/postgresql-repmgr/db"
db_name="repmgr"
db_user="repmgr"
db_pass="repmgr"

#   determine RDBMS-specific details
db_sname=`grep "^superuser_database" @l_prefix@/var/postgresql/db/pg_superuser.conf |\
          sed -e 's;^ *superuser_database="\(.*\)".*;\1;'`
db_suser=`grep "^superuser_username" @l_prefix@/var/postgresql/db/pg_superuser.conf |\
          sed -e 's;^ *superuser_username="\(.*\)".*;\1;'`
db_spass=`grep "^superuser_password" @l_prefix@/var/postgresql/db/pg_superuser.conf |\
          sed -e 's;^ *superuser_password="\(.*\)".*;\1;'`

#   dispatch operation
cmd="${1:-"install"}"
shift
case "$cmd" in
    install )
        ##
        ##  create the database
        ##

        if [ $# -gt 0 ]; then
            db_dir="$1"
            shift
        fi
        ( echo "CREATE ROLE $db_user LOGIN ENCRYPTED PASSWORD '$db_pass' SUPERUSER;"
          echo "ALTER USER repmgr SET search_path TO repmgr, \"$user\", public;"
          echo "CREATE TABLESPACE $db_name OWNER $db_user LOCATION '$db_dir';"
          echo "CREATE DATABASE $db_name WITH ENCODING 'UTF8' TEMPLATE template0 OWNER $db_user TABLESPACE $db_name;"
          echo "GRANT ALL PRIVILEGES ON DATABASE $db_name TO $db_user;"
        ) | PGPASSWORD="$db_spass" @l_prefix@/bin/psql -q -U "$db_suser" -d "$db_sname" -f-
        ;;

    uninstall )
        ##
        ##  remove the database
        ##

        ( echo "DROP DATABASE $db_name;"
          echo "DROP TABLESPACE $db_name;"
          echo "DROP ROLE $db_user;"
        ) | PGPASSWORD="$db_spass" @l_prefix@/bin/psql -q -U "$db_suser" -d "$db_sname" -f-
        ;;
esac