#!/bin/bash
: '
 @vendor    Chorke Academia, Inc.
 @web       https://cdn.chorke.org/docs/academia
 @version   1.0.00
 @since     1.0.00
'

:<<EOF
export ENV_LOCALE='staging'
export ENV_REMOTE='staging'
export SSH_REMOTE_USER='deploy'
export SSH_TUNNEL_USER='deploy'
export RDB_LOCALE_USER='postgres'
export SSH_TUNNEL_HOST='dev.xpresscover.io'
export SSH_REMOTE_HOST='dev.xpresscover.io'
bash <(curl -s 'https://cdn.chorke.org/exec/cli/psql/finology/commons/import/remote/db2db.sh.txt')
EOF


if [ -z "$CLI_PSQL" ];then CLI_PSQL='psql';fi
if [ -z "$CLI_DUMP" ];then CLI_DUMP='pg_dump';fi
if [ -z "$ENV_CLIENT" ];then ENV_CLIENT='demo';fi
if [ -z "$ENV_PREFIX" ];then ENV_PREFIX='xpresscover';fi


if [ -z "$RDB_LOCALE_USER" ];then RDB_LOCALE_USER='postgres';fi
if [ -z "$RDB_REMOTE_USER" ];then RDB_REMOTE_USER='postgres';fi


if [ -z "$ENV_LOCALE" ];then
    printf -v RDB_LOCALE_TDS    '%s_tds'       "$ENV_PREFIX"
    printf -v RDB_LOCALE_QUOTE  '%s_quote'     "$ENV_PREFIX"
    printf -v RDB_LOCALE_POLICY '%s_policy'    "$ENV_PREFIX"
else
    printf -v RDB_LOCALE_TDS    '%s_tds_%s'    "$ENV_PREFIX" "$ENV_LOCALE"
    printf -v RDB_LOCALE_QUOTE  '%s_quote_%s'  "$ENV_PREFIX" "$ENV_LOCALE"
    printf -v RDB_LOCALE_POLICY '%s_policy_%s' "$ENV_PREFIX" "$ENV_LOCALE"
fi


if [ -z "$ENV_REMOTE" ];then
    printf -v RDB_REMOTE_TDS    '%s_tds'       "$ENV_PREFIX"
    printf -v RDB_REMOTE_QUOTE  '%s_quote'     "$ENV_PREFIX"
    printf -v RDB_REMOTE_POLICY '%s_policy'    "$ENV_PREFIX"
else
    printf -v RDB_REMOTE_TDS    '%s_tds_%s'    "$ENV_PREFIX" "$ENV_REMOTE"
    printf -v RDB_REMOTE_QUOTE  '%s_quote_%s'  "$ENV_PREFIX" "$ENV_REMOTE"
    printf -v RDB_REMOTE_POLICY '%s_policy_%s' "$ENV_PREFIX" "$ENV_REMOTE"
fi


if [ -z "$SSH_REMOTE_HOST" ];then SSH_REMOTE_HOST='dev.xpresscover.io';fi
if [ -z "$SSH_TUNNEL_HOST" ];then SSH_TUNNEL_HOST="$SSH_REMOTE_HOST";fi
if [ -z "$SSH_TUNNEL_USER" ];then SSH_TUNNEL_USER="$SSH_REMOTE_USER";fi

if [ -z "$SSH_REMOTE_USER" ];then SSH_REMOTE_AUTH="$SSH_REMOTE_HOST"
else printf -v SSH_REMOTE_AUTH '%s@%s' "$SSH_REMOTE_USER" "$SSH_REMOTE_HOST";fi

if [ -z "$SSH_TUNNEL_USER" ];then SSH_TUNNEL_AUTH="$SSH_TUNNEL_HOST"
else printf -v SSH_TUNNEL_AUTH '%s@%s' "$SSH_TUNNEL_USER" "$SSH_TUNNEL_HOST";fi


function quit_pid(){
    printf -v RDB_WHERE_IN "'%s','%s','%s'"  "$RDB_LOCALE_TDS" "$RDB_LOCALE_QUOTE" "$RDB_LOCALE_POLICY"
    cat << EOF | psql -U "$RDB_LOCALE_USER"
SELECT pg_terminate_backend(pid)
FROM pg_stat_activity WHERE datname IN($RDB_WHERE_IN);
EOF
}


function drop_dbs(){
    quit_pid
    cat << EOF | psql -U "$RDB_LOCALE_USER"
DROP DATABASE IF EXISTS $RDB_LOCALE_TDS;
DROP DATABASE IF EXISTS $RDB_LOCALE_QUOTE;
DROP DATABASE IF EXISTS $RDB_LOCALE_POLICY;
EOF
}


function create_dbs(){
    cat << EOF | psql -U "$RDB_LOCALE_USER"
CREATE DATABASE $RDB_LOCALE_TDS;
CREATE DATABASE $RDB_LOCALE_QUOTE;
CREATE DATABASE $RDB_LOCALE_POLICY;
EOF
}


function tunnel_import_dbs(){
    printf -v SSH_TUNNEL_TDS    "ssh %s '%s -h localhost -U %s %s'" "$SSH_REMOTE_AUTH" "$CLI_DUMP" "$RDB_REMOTE_USER" "$RDB_REMOTE_TDS"
    printf -v SSH_TUNNEL_QUOTE  "ssh %s '%s -h localhost -U %s %s'" "$SSH_REMOTE_AUTH" "$CLI_DUMP" "$RDB_REMOTE_USER" "$RDB_REMOTE_QUOTE"
    printf -v SSH_TUNNEL_POLICY "ssh %s '%s -h localhost -U %s %s'" "$SSH_REMOTE_AUTH" "$CLI_DUMP" "$RDB_REMOTE_USER" "$RDB_REMOTE_POLICY"

    ssh "$SSH_TUNNEL_AUTH" "$SSH_TUNNEL_TDS"    | "$CLI_PSQL" -U "$RDB_LOCALE_USER" "$RDB_LOCALE_TDS"
    ssh "$SSH_TUNNEL_AUTH" "$SSH_TUNNEL_QUOTE"  | "$CLI_PSQL" -U "$RDB_LOCALE_USER" "$RDB_LOCALE_QUOTE"
    ssh "$SSH_TUNNEL_AUTH" "$SSH_TUNNEL_POLICY" | "$CLI_PSQL" -U "$RDB_LOCALE_USER" "$RDB_LOCALE_POLICY"
}


function remote_import_dbs(){
    ssh "$SSH_REMOTE_AUTH" "$CLI_DUMP -h localhost -U $RDB_REMOTE_USER $RDB_REMOTE_TDS"    | "$CLI_PSQL" -U "$RDB_LOCALE_USER" "$RDB_LOCALE_TDS"
    ssh "$SSH_REMOTE_AUTH" "$CLI_DUMP -h localhost -U $RDB_REMOTE_USER $RDB_REMOTE_QUOTE"  | "$CLI_PSQL" -U "$RDB_LOCALE_USER" "$RDB_LOCALE_QUOTE"
    ssh "$SSH_REMOTE_AUTH" "$CLI_DUMP -h localhost -U $RDB_REMOTE_USER $RDB_REMOTE_POLICY" | "$CLI_PSQL" -U "$RDB_LOCALE_USER" "$RDB_LOCALE_POLICY"
}


function import_dbs(){
    if [ "$SSH_TUNNEL_AUTH" == "$SSH_REMOTE_AUTH" ];then SSH_TUNNEL='OFF';else SSH_TUNNEL='ON';fi
    if [ "$SSH_TUNNEL" == "ON" ];then tunnel_import_dbs;else remote_import_dbs;fi
    echo "$ENV_CLIENT" > "$HOME/.finology/xpresscover/var/bkp/.last_env_client"
    echo "$ENV_REMOTE" > "$HOME/.finology/xpresscover/var/bkp/.last_env_locale"
}


function init(){
    mkdir -p "$HOME/.finology/xpresscover/var/bkp"
    if [ -x "$(command -v psql)" ];then drop_dbs; create_dbs; import_dbs
    else printf '\npsql \033[0;31mcommand not found! \033[0m:(\n';fi
}


init