#!/bin/bash
source /bin/sx-lib.sh

set -e

set_listen_addresses() {
	sedEscapedValue="$(echo "$1" | sed 's/[\/&]/\\&/g')"
	sed -ri "s/^#?(listen_addresses\s*=\s*)\S+/\1'$sedEscapedValue'/" "$POSTGRESQL_CONF"
}



function check_postgresql_environment {
    check_environment
    if [ ! -v DATA_PATH ]; then
        DATA_PATH="/data"
        export DATA_PATH
    fi
    if [ ! -v LOG_PATH ]; then
        LOG_PATH="/logs"
        export LOG_PATH
    fi
}

function display_container_postgresql_header {
    echo "+====================================================="
    echo "| Container   : $HOSTNAME"
    echo "| OS          : $(</etc/redhat-release)"
    echo "| Engine      : $(postgres --version)"
    if [ -v CONTAINER_TYPE ]; then
        echo "| Type        : $CONTAINER_TYPE"
    fi
    if [ -v CONTAINER_SERVICE ]; then
        echo "| Service     : $CONTAINER_SERVICE"
    fi
    if [ -v CONTAINER_INSTANCE ]; then
        echo "| Instance    : $CONTAINER_INSTANCE"
    fi
    if [ -v DATA_PATH ]; then
        echo "| Data path   : $DATA_PATH"
    fi
    if [ -v LOG_PATH ]; then
        echo "| Log path    : $LOG_PATH"
    fi
    if [ -v LOADSQL_PATH ]; then
        echo "| sql path    : $LOADSQL_PATH"
    fi
    echo "+====================================================="
}


# Begin configuration before starting daemonized process
# and start generating host keys
function begin_config {
    echo "=> BEGIN POSTGRESQL CONFIGURATION"
    if [[ ! -d $DATA_PATH ]]; then
        echo "data directory $DATA_PATH not found"
        mkdir -p $DATA_PATH; chmod 0774 $DATA_PATH
        echo "data directory $DATA_PATH CREATED"
    else 
        echo "data directory $DATA_PATH EXIST"
    fi
    export PGDATA=$DATA_PATH
    if [[ ! -d $LOG_PATH ]]; then
        echo "log directory $LOG_PATH not found"
        mkdir -p $LOG_PATH; chmod 0774 $LOG_PATH
        echo "log directory $LOG_PATH CREATED"
    else 
        echo "log directory $LOG_PATH EXIST"
    fi
    chmod 0774 $DATA_PATH $LOG_PATH; 
    chown postgres:postgres $DATA_PATH $LOG_PATH; 
    if [[ -d $LOADSQL_PATH ]]; then
        echo "sql directory $LOADSQL_PATH EXIST"
        chmod 0774 $LOADSQL_PATH; 
        chown postgres:postgres $LOADSQL_PATH
    fi
    if [ "$(ls -A $DATA_PATH)" ]; then
        echo "postgresql directory is initialized"
        echo "Reusing PostgreSQL in $DATA_PATH ..."
   else
        echo "postgresql directory is empty or uninitialized"
        echo "Installing PostgreSQL in $DATA_PATH ..."
        su - postgres -c "initdb -D $DATA_PATH"
        export POSTGRESQL_CONF=$DATA_PATH/postgresql.conf
        echo "" >> $POSTGRESQL_CONF
        echo "log_directory = '$LOG_PATH'" >> $POSTGRESQL_CONF
        echo "log_connections = on" >> $POSTGRESQL_CONF
        echo "log_hostname = on" >> $POSTGRESQL_CONF
        echo "log_line_prefix = '<%u@%r%%%d>'" >> $POSTGRESQL_CONF
        { echo; echo "host all all 0.0.0.0/0 md5"; } >> "$DATA_PATH/pg_hba.conf"
        echo "Installing PostgreSQL in $DATA_PATH is DONE !"
        chown root:postgres $STARTUPLOG
        chmod ug+w $STARTUPLOG
        config_startserver
        config_createadmin
        config_createuser
        config_createdatabase
        config_importsql
        config_stopserver
    fi
}

function config_startserver {
    echo "start database for initial setup"
    su - postgres -c "pg_ctl -D $DATA_PATH -o \"-c listen_addresses=''\" --log=$STARTUPLOG -w start "
    sleep 6
}

function config_stopserver {
    echo "stop database after initial setup"
    su - postgres -c "pg_ctl -D $DATA_PATH -m fast -w stop" 
    set_listen_addresses '*'
    sleep 3
}


function config_createadmin {
    PASS=${POSTGRESQL_ROOT_PASSWORD:-$(pwgen -s 12 1)}
    _word=$( [ ${POSTGRESQL_ROOT_PASSWORD} ] && echo "preset" || echo "random" )
    echo "Creating PostgreSQL admin user with ${_word} password"
    su - postgres -c "psql -U postgres -c \"ALTER USER postgres WITH SUPERUSER PASSWORD '$PASS';\""
    echo ""
    echo " +------------------------------------------------------"
    echo " | SUPERADMIN USER CREATED ! "
    echo " | You can now connect to this server using:"
    echo " | "
    echo " | user     : postgres"
    echo " | password : $PASS"
    echo " | "
    echo " | shell    : psql --username=postgres --host=<host> --port=<port>"
    echo " +------------------------------------------------------"
    echo ""
}

function config_createuser {
    if [[ -n "$POSTGRESQL_USER" ]]; then
        echo "Creating PostgreSQL $POSTGRESQL_USER user with preset password"
        su - postgres -c 'psql -U postgres -c \"CREATE USER \"$POSTGRESQL_USER\" WITH PASSWORD \'$POSTGRESQL_PASSWORD\';\"'
        echo ""
        echo " +------------------------------------------------------"
        echo " | USER CREATED ! "
        echo " | You can now connect to this server using:"
        echo " | "
        echo " | user     : $POSTGRESQL_USER"
        echo " | password : $POSTGRESQL_PASSWORD"
        echo " | "
        echo " | shell    : psql --username=$POSTGRESQL_USER --host=<host> --port=<port>"
        echo " +------------------------------------------------------"
        echo ""
    fi
}

function config_createdatabase {
    if [[ -n "$POSTGRESQL_DATABASE" ]]; then
        echo "processing database " $POSTGRESQL_DATABASE
        if [[ ! -d $DIR_DB_DATA/$POSTGRESQL_DATABASE ]]; then
            echo "database " $POSTGRESQL_DATABASE " doesn't exist"
            su - postgres -c "psql -U postgres -c \"CREATE DATABASE $POSTGRESQL_DATABASE;\""
            echo ""
            echo " +------------------------------------------------------"
            echo " | DATABASE CREATED ! "
            echo " | You can connect to this database using :"
            echo " | "
            echo " | database : $POSTGRESQL_DATABASE"
            echo " | "
            echo " | shell    : psql --username=postgres --host=<host> --port=<port> $POSTGRESQL_DATABASE "
            echo " +------------------------------------------------------"
            echo ""
        else
            echo "database " $POSTGRESQL_DATABASE " already exist"
        fi
    else
        echo "no database to create"
    fi
}

function config_importsql {
    if [[ -n "$LOADSQL_PATH" ]]; then
        echo "import sql data into " $POSTGRESQL_DATABASE
        if [[ -d $LOADSQL_PATH ]]; then
            if [ "$(ls -A $LOADSQL_PATH/schema-*.sql)" ]; then
                echo "Importing schema from $LOADSQL_PATH "
                SCHEMALIST=$(find $LOADSQL_PATH/schema-*.sql -type f -printf "%f\n")
                for SCHEMAFILE in $SCHEMALIST; do 
                    echo -n "Creating schema " $SCHEMAFILE " ... "
                    su - postgres -c "psql -U postgres -d $POSTGRESQL_DATABASE < $LOADSQL_PATH/$SCHEMAFILE"
                    echo " DONE"
                done
            fi
            if [ "$(ls -A $LOADSQL_PATH/data-*.sql)" ]; then
                echo "Importing data from $LOADSQL_PATH "
                DATALIST=$(find $LOADSQL_PATH/data-*.sql -type f -printf "%f\n")
                for DATAFILE in $DATALIST; do 
                    echo -n "Creating data " $DATAFILE " ... "
                    su - postgres -c "psql -U postgres -d $POSTGRESQL_DATABASE < $LOADSQL_PATH/$DATAFILE"
                    echo " DONE"
                done
            fi
        fi
    else
        echo "no sql data to import into " $POSTGRESQL_DATABASE
    fi
}


# End configuration process just before starting daemon
function end_config {
    echo "=> END POSTGRESQL CONFIGURATION"
}

function stop_postgres_handler {
    su - postgres -c "pg_ctl -D $DATA_PATH -m fast -w stop" 
    set_listen_addresses '*'
    sleep 2
    echo "+=====================================================" | tee -a $STARTUPLOG
    echo "| Container $HOSTNAME is now STOPPED" | tee -a $STARTUPLOG
    echo "+=====================================================" | tee -a $STARTUPLOG
    exit 143; # 128 + 15 -- SIGTERM
}

# Start the postgresql server as a deamon and execute it inside 
# the running shell
function start_service_postgres {
    trap 'kill ${!}; stop_postgres_handler' SIGHUP SIGINT SIGQUIT SIGTERM SIGKILL SIGSTOP SIGCONT
    echo "+=====================================================" | tee -a $STARTUPLOG
    echo "| Container $HOSTNAME is now RUNNING" | tee -a $STARTUPLOG
    echo "+=====================================================" | tee -a $STARTUPLOG
    su - postgres -c "pg_ctl -D $DATA_PATH -w start " &
    while true
    do
      tail -f /dev/null & wait ${!}
    done
}


check_postgresql_environment | tee -a $STARTUPLOG
display_container_postgresql_header | tee -a $STARTUPLOG
begin_config | tee -a $STARTUPLOG
end_config | tee -a $STARTUPLOG
start_service_postgres