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

:<<EOF
export ENV_REMOTE='feature';\
export RDB_LOCALE_USER='shahed';\
export SSH_REMOTE_USER='deploy';\
export SSH_REMOTE_HOST='fincon-dev.xpresscover.io';\
export SSH_PUBKEY_PATH="$HOME/.ssh/ci_finology_com_my_rsa";\
bash <(curl -s 'https://cdn.chorke.org/exec/cli/psql/finology/xpresscover/xc/feature.model.sh.txt')
EOF


:<<EOF
unset SSH_REMOTE_USER;\
export ENV_REMOTE='feature';\
export RDB_LOCALE_USER='shahed';\
export SSH_REMOTE_HOST='fincon-dev.xpresscover.io';\
export SSH_PUBKEY_PATH="$HOME/.ssh/ci_finology_com_my_rsa";\
bash <(curl -s 'https://cdn.chorke.org/exec/cli/psql/finology/xpresscover/xc/feature.model.sh.txt')
EOF


function common_env(){
    if [ -z "$ENV_LOCALE" ];then export ENV_LOCALE='staging';fi
    if [ -z "$ENV_REMOTE" ];then export ENV_REMOTE='staging';fi


    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_MODEL  '%s_model'     "$ENV_PREFIX"
    else
        printf -v RDB_LOCALE_MODEL  '%s_model_%s'  "$ENV_PREFIX" "$ENV_LOCALE"
    fi


    if [ -z "$ENV_REMOTE" ];then
        printf -v RDB_REMOTE_MODEL  '%s_model'     "$ENV_PREFIX"
    else
        printf -v RDB_REMOTE_MODEL  '%s_model_%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'" "$RDB_LOCALE_MODEL"
    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_MODEL;
EOF
}


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


function tunnel_import_dbs(){
    printf -v SSH_TUNNEL_POLICY "ssh %s '%s -U %s %s'" "$SSH_REMOTE_AUTH" "$CLI_DUMP" "$RDB_REMOTE_USER" "$RDB_REMOTE_MODEL"
    ssh "$SSH_TUNNEL_AUTH" "$SSH_TUNNEL_POLICY" | "$CLI_PSQL" -U "$RDB_LOCALE_USER" "$RDB_LOCALE_MODEL"
}


function remote_import_dbs(){
    ssh "$SSH_REMOTE_AUTH" "$CLI_DUMP -U $RDB_REMOTE_USER $RDB_REMOTE_MODEL"  | "$CLI_PSQL" -U "$RDB_LOCALE_USER" "$RDB_LOCALE_MODEL"
}


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 common_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
}

function init(){
    unset SSH_TUNNEL_USER
    unset SSH_TUNNEL_HOST

    export CLI_PSQL='psql'
    export ENV_CLIENT='xc'
    export CLI_DUMP='pg_dump'

    if [ -z "$ENV_LOCALE" ];then export ENV_LOCALE='staging';fi
    if [ -z "$ENV_REMOTE" ];then export ENV_REMOTE='staging';fi

    if [ -z "$PGPASSWORD" ];then printf 'PSQL  \033[0;32mPassword\033[0m: ';read -s PGPASSWORD;if [ ! -z "$PGPASSWORD" ];then export PGPASSWORD;fi;fi
    if [ ! -z "$SSH_PUBKEY_PATH" ]&&[ -f "$SSH_PUBKEY_PATH" ];then eval "$(ssh-agent -s)";if [[ "$OSTYPE" == "darwin"* ]];then ssh-add -K $SSH_PUBKEY_PATH; else ssh-add -k $SSH_PUBKEY_PATH;fi;fi

    common_env
    common_init
    if [ ! -z "$PGPASSWORD" ];then unset PGPASSWORD;fi
}


init