#!/bin/bash
# ************************************************************************* 
# Copyright (c) 2014-2015, SUSE LLC
# 
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# 
# 1. Redistributions of source code must retain the above copyright notice,
# this list of conditions and the following disclaimer.
# 
# 2. Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# 
# 3. Neither the name of SUSE LLC nor the names of its contributors may be
# used to endorse or promote products derived from this software without
# specific prior written permission.
# 
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
# ************************************************************************* 
#
# - destroy existing database, user and create a new one from scratch
set -e

help() {
    echo
    echo "Sets up a dochazka database and user. Needs to be run on the DB PostgreSQL server."
    echo
    echo "Options:"
    echo "-d <database> The database name. Default: dochazka-test"
    echo "-u <user>     The database user. Default: dochazka"
    echo "-p <password> The database user's password. Default: dochazka"
    echo
}

[[ $1 == '--help' ]] && help && exit 0

DBNAME="dochazka-test"
DBUSER=dochazka
DBPASS=dochazka

while getopts d:u:p:h arg; do
    case ${arg} in
        d) DBNAME=${OPTARG} ;;
        u) DBUSER=${OPTARG} ;;
        p) DBPASS=${OPTARG} ;;
        h) help ;;
        *) help && exit 1 ;;
    esac
done

[[ -z $DBNAME ]] && help && exit 1
[[ -z $DBUSER ]] && help && exit 1
[[ -z $DBPASS ]] && help && exit 1

psql -U postgres <<EOD
DROP DATABASE IF EXISTS "$DBNAME";
DROP ROLE IF EXISTS "$DBUSER";
CREATE ROLE "$DBUSER" WITH LOGIN PASSWORD '$DBPASS';
CREATE DATABASE "$DBNAME" WITH OWNER "$DBUSER";
REVOKE CONNECT ON DATABASE "$DBNAME" FROM PUBLIC;
GRANT ALL PRIVILEGES ON DATABASE "$DBNAME" TO "$DBUSER";
\\connect $DBNAME
CREATE EXTENSION hstore;
CREATE EXTENSION btree_gist;
EOD

echo "Database $DBNAME reset to pristine state. Now run 'dochazka-dbinit'."
