#!/bin/bash

echo "####################################################################"
echo "# docker_postgres_setup_script                                     #"
echo "# Dennis Buchhorn - bucde@b-eit.de                                 #"
echo "####################################################################"
echo "# WARNING: password is plain text input!                           #"
echo "#                                                                  #"
echo "# Config will be saved in '.env' file.                             #"
echo "# (everything in plain text, even the password)                    #"
#echo "#  #"
echo "####################################################################"
echo ""

CONFIG_EXISTS="false"
KEEP_CONFIG=""

POSTGRES_VERSION=""
POSTGRES_HOST_IP_ADDRESS=""
POSTGRES_HOST_PORT=""
POSTGRES_ADMIN_USER=""
POSTGRES_ADMIN_PASSWORD=""

POSTGRES_SHARED_HOSTING=""

if [ -f ".env" ]; then
    CONFIG_EXISTS="true"
    read -p "Config file '.env' already exists! Would you like to use this? (y/n)" KEEP_CONFIG
fi

if [ $CONFIG_EXISTS == "false" ] || [ $KEEP_CONFIG == "n" ]; then
    read -p "Enter postgres version which should be used: " POSTGRES_VERSION
    read -p "Enter host (vm) ip address on which postgres should be accessible: " POSTGRES_HOST_IP_ADDRESS
    read -p "Enter host (vm) port on which postgres should be accessible: " POSTGRES_HOST_PORT
    read -p "Enter admin user name for postgres: " POSTGRES_ADMIN_USER
    read -p "Enter admin password for postgres: " POSTGRES_ADMIN_PASSWORD

    rm -f .env
    touch .env
    echo "POSTGRES_VERSION="$POSTGRES_VERSION >> .env
    echo "POSTGRES_HOST_IP_ADDRESS="$POSTGRES_HOST_IP_ADDRESS >> .env
    echo "POSTGRES_HOST_PORT="$POSTGRES_HOST_PORT >> .env
    echo "POSTGRES_ADMIN_USER="$POSTGRES_ADMIN_USER >> .env
    echo "POSTGRES_ADMIN_PASSWORD="$POSTGRES_ADMIN_PASSWORD >> .env
else
    while read line; do
        if [[ $line == *"POSTGRES_ADMIN_USER="* ]]; then
            POSTGRES_ADMIN_USER=$(echo $line | cut -c 21-)
        fi
    done < ".env"
fi

read -p "Would you like to modify postgres rights for shared hosting? (y/n)" POSTGRES_SHARED_HOSTING

docker-compose up -d

echo "Wait a few seconds for the container to spin up ..."

sleep 5

if [ $POSTGRES_SHARED_HOSTING == "y" ]; then
    rm -f tmp
    touch tmp
    echo "REVOKE ALL ON DATABASE template1 FROM public;" >> tmp
    echo "REVOKE ALL ON SCHEMA public FROM public;" >> tmp
    echo "GRANT ALL ON SCHEMA public TO "$POSTGRES_ADMIN_USER";" >> tmp
    echo "REVOKE ALL ON pg_user FROM public;" >> tmp
    echo "REVOKE ALL ON pg_roles FROM public;" >> tmp
    echo "REVOKE ALL ON pg_group FROM public;" >> tmp
    echo "REVOKE ALL ON pg_authid FROM public;" >> tmp
    echo "REVOKE ALL ON pg_auth_members FROM public;" >> tmp
    echo "REVOKE ALL ON pg_database FROM public;" >> tmp
    echo "REVOKE ALL ON pg_tablespace FROM public;" >> tmp
    echo "REVOKE ALL ON pg_settings FROM public;" >> tmp

    cat tmp | docker exec -i postgres psql -U $POSTGRES_ADMIN_USER -d template1

    rm -f tmp
fi

echo "Done!"