#!/usr/bin/env bash

## - !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
## -
## - assuming your remote user is called "back", you have
## - to give sudoer rights to him as follow:
## -
## -    visudo (edits /etc/sudoers file)
## -
## - add lines:
## -    back ALL = (postgres) NOPASSWD: /usr/bin/psql
## -    back ALL = (postgres) NOPASSWD: /usr/bin/pg_dump
## -    back ALL = (postgres) NOPASSWD: /usr/bin/pg_dumpall
## -
## - !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

err_Log=${LOCK_DIR}/pgsql.err.log
> $err_Log


## - load functions
## -
. $rcopy_functions_file

# --------------------------------------------------- #
# -------------------- Variable --------------------- #
#

_backupDestArchiveDir="${script_backup_dir}/PostgreSQL"

if [ ! -d $_backupDestArchiveDir ]; then
   mkdir -p $_backupDestArchiveDir
fi


if [ $srcHost != "localhost" ] || $_via_ssh_tunnel ;then
   psql=`$ssh ${ssh_user}@$srcHost which psql`
   pg_dump=`$ssh ${ssh_user}@$srcHost which pg_dump`
   pg_dumpall=`$ssh ${ssh_user}@$srcHost which pg_dumpall`
   sudo=`$ssh ${ssh_user}@$srcHost which sudo`
   su=`$ssh ${ssh_user}@$srcHost which su`
   ssh_options="-o BatchMode=yes -o ConnectTimeout=360"
else
   psql=`which psql`
   pg_dump=`which pg_dump`
   pg_dumpall=`which pg_dumpall`
fi

if [ -z "$psql" ]; then
   psql="/usr/local/pgsql/bin/psql"
fi

if [ -z "$pg_dump" ]; then
   pg_dump="/usr/local/pgsql/bin/pg_dump"
fi

if [ $srcHost != "localhost" ] || $_via_ssh_tunnel ;then
   DATABASES=`$ssh ${ssh_user}@$srcHost "$sudo -u $pgsql_user $psql  -lt" \
      | $grep -v -e"^$" | $awk '{print$1}'` 
else
   DATABASES=`$su - $pgsql_user -c "$psql  -lt" | $grep -v -e"^$" | $awk '{print$1}'`
fi


# ------------------ Ende Variable ------------------ #
# --------------------------------------------------- #

echolog ""
if $ARCHIVE ;then

   if [ "$_TEST" = "1" ];then
      progArgs="-n $progArgs --stats"
   elif [ "$_DEBUG" -gt 0 ];then
      progArgs="$progArgs --stats"
      if [ "$_DEBUG" -gt 1 ];then
         formats="%t -- %o %f %b Bytes[%l]"
      fi
   fi

   ## ---------------------------
   ## - Backup PostgreSQL's Roles
   ## -
   info_msg="save PostgreSQL's Roles"
   echononl "\t$info_msg"

   ## - begin timestamp
   ## -
   b_timestamp=`$date +"%s"`

   filedate=`$date +"%Y-%m-%d-%H%M"`

   if [ $srcHost != "localhost" ] || $_via_ssh_tunnel ;then
      $(
         $ssh $ssh_options ${ssh_user}@$srcHost "$sudo -u $pgsql_user $pg_dumpall -g -c " \
            > ${_backupDestArchiveDir}/PostreSQL_Roles-${filedate}.sql 2> $err_Log
         exit $?
      )
      retval=$?
   else
      $(
         $su $pgsql_user -c "$pg_dumpall -g -c" \
            > ${_backupDestArchiveDir}/PostreSQL_Roles-${filedate}.sql 2> $err_Log
         exit $?
      )
      retval=$?
   fi
   err_msg="Cannot save PostgreSQL's Roles"
   keep_backup_on_error=false

   if [ "$retval" = 0 ];then
      [ -z $pgsql_gzip ] && pgsql_gzip=false
      if $pgsql_gzip ; then
         $gzip ${_backupDestArchiveDir}/PostreSQL_Roles-${filedate}.sql 2> $err_Log
         retval=$?
         err_msg="Cannot gzip PostreSQL_Roles-${filedate}.sql"
         keep_backup_on_error=true
      fi
   fi

   ## - end timestamp
   ## -
   e_timestamp=`$date +"%s"`

   ## - determin duration
   ## -
   _time=`expr $e_timestamp - $b_timestamp`
   t_h=`expr $_time / 60 / 60`
   t_rest_h=`expr $_time - $t_h \\* 60 \\* 60`
   t_m=`expr $t_rest_h / 60`
   t_s=`expr $t_rest_h - $t_m \\* 60`
   duration=""
   if [ $t_h -gt 0 ]; then
      duration="$t_h h : $t_m min : $t_s sec"
   elif [ $t_m -gt 0 ];then
      duration="$t_m min : $t_s sec"
   else
      duration="$t_s sec"
   fi

   ## - look about errors..
   ## -
   if [ "$retval" != "0" ]; then
      echolog ""
      echolog "\t[ERROR] ${err_msg}  [ $duration ]\n\t`$cat $err_Log`\n"
      if ! $keep_backup_on_error ; then
         rm -f ${_backupDestArchiveDir}/PostreSQL_Roles-${filedate}.sql
      fi
   else

      ## - print durations right-aligned
      ## -
      [ -z $right_tabstop ] && right_tabstop=65
      _tmp_string="${info_msg}${duration}"
      _strlen=${#_tmp_string}
      _count_blank=`expr $right_tabstop - $_strlen`
      _str_blanks=""
      while [ $_count_blank -gt 1 ]; do
         _str_blanks="$_str_blanks "
         _count_blank=`expr $_count_blank - 1`
      done
      echononl "$_str_blanks"

      echolog "  [ $duration ]"
   fi

   ## - END: Backup PostgreSQL's Roles
   ## ---------------------------------


   ## -----------------------------------
   ## - Create Database Creation SQL file
   ## -
   info_msg="create Database Creation SQL file"
   echononl "\t$info_msg"

   ## - begin timestamp
   ## -
   b_timestamp=`$date +"%s"`

   filedate=`$date +"%Y-%m-%d-%H%M"`

   if [ $srcHost != "localhost" ] || $_via_ssh_tunnel ;then
      $(
         $ssh $ssh_options ${ssh_user}@$srcHost "$sudo -u $pgsql_user $pg_dumpall -s " | \
            grep -e "^CREATE DATABASE" \
            > ${_backupDestArchiveDir}/PostreSQL_Create_Databases-${filedate}.sql 2> $err_Log
         exit $?
      )
      retval=$?
   else
      $(
         $su - $pgsql_user -c "$pg_dumpall -s" | \
            grep -e "^CREATE DATABASE" \
            > ${_backupDestArchiveDir}/PostreSQL_Create_Databases-${filedate}.sql 2> $err_Log
         exit $?
      )
      retval=$?
   fi
   err_msg="Cannot create Database Creation SQL file"
   keep_backup_on_error=false

   if [ "$retval" = 0 ];then
      [ -z $pgsql_gzip ] && pgsql_gzip=false
      if $pgsql_gzip ; then
         $gzip ${_backupDestArchiveDir}/PostreSQL_Create_Databases-${filedate}.sql 2> $err_Log
         retval=$?
         err_msg="Cannot gzip PostreSQL_Databases-${filedate}.sql"
         keep_backup_on_error=true
      fi
   fi

   ## - end timestamp
   ## -
   e_timestamp=`$date +"%s"`

   ## - determin duration
   ## -
   _time=`expr $e_timestamp - $b_timestamp`
   t_h=`expr $_time / 60 / 60`
   t_rest_h=`expr $_time - $t_h \\* 60 \\* 60`
   t_m=`expr $t_rest_h / 60`
   t_s=`expr $t_rest_h - $t_m \\* 60`
   duration=""
   if [ $t_h -gt 0 ]; then
      duration="$t_h h : $t_m min : $t_s sec"
   elif [ $t_m -gt 0 ];then
      duration="$t_m min : $t_s sec"
   else
      duration="$t_s sec"
   fi

   ## - look about errors..
   ## -
   if [ "$retval" != "0" ]; then
      echolog ""
      echolog "\t[ERROR] ${err_msg}  [ $duration ]\n\t`$cat $err_Log`\n"
      if ! $keep_backup_on_error ; then
         rm -f ${_backupDestArchiveDir}/PostreSQL_Create_Databases-${filedate}.sql
      fi
   else

      ## - print durations right-aligned
      ## -
      [ -z $right_tabstop ] && right_tabstop=65
      _tmp_string="${info_msg}${duration}"
      _strlen=${#_tmp_string}
      _count_blank=`expr $right_tabstop - $_strlen`
      _str_blanks=""
      while [ $_count_blank -gt 1 ]; do
         _str_blanks="$_str_blanks "
         _count_blank=`expr $_count_blank - 1`
      done
      echononl "$_str_blanks"

      echolog "  [ $duration ]"
   fi

   ## - End: Create Database Creation SQL file
   ## ----------------------------------------


   ## - backup all databases..
   ## -
   for i in $DATABASES ; do
      if [ "$i" = "template0" ]; then
         continue 
      fi
      if [ "$i" = ":" -o "$i" = "|" ]; then
         continue
      fi
      filedate=`$date +"%Y-%m-%d-%H%M"`

      ## - begin timestamp
      ## -
      b_timestamp=`$date +"%s"`

      if [ "$destHost" = "localhost" ]; then
         if [ ! -d ${_backupDestArchiveDir}/${i} ] ; then 
            $mkdir -p ${_backupDestArchiveDir}/${i}
         fi
      fi

      info_msg="save database $i.."
      echononl "\t$info_msg"
      if [ $srcHost != "localhost" ] || $_via_ssh_tunnel ;then
         $(
            $ssh $ssh_options ${ssh_user}@$srcHost "$sudo -u $pgsql_user $pg_dump  -c  $i" \
               > ${_backupDestArchiveDir}/${i}/${i}-${filedate}.sql 2> $err_Log
            exit $?
         )
         retval=$?
      else
         $(
            $su - $pgsql_user -c "$pg_dump  -c  $i" \
               > ${_backupDestArchiveDir}/${i}/${i}-${filedate}.sql 2> $err_Log
            exit $?
         )
         retval=$?
      fi
      err_msg="Cannot save database \"$i\""
      keep_backup_on_error=false

      if [ "$retval" = 0 ];then
         [ -z $pgsql_gzip ] && pgsql_gzip=false
         if $pgsql_gzip ; then
            $gzip ${_backupDestArchiveDir}/${i}/${i}-${filedate}.sql 2> $err_Log
            retval=$?
            err_msg="Cannot gzip database \"$i\""
            keep_backup_on_error=true
         fi
      fi

      ## - end timestamp
      ## -
      e_timestamp=`$date +"%s"`

      ## - determin duration
      ## -
      _time=`expr $e_timestamp - $b_timestamp`
      t_h=`expr $_time / 60 / 60`
      t_rest_h=`expr $_time - $t_h \\* 60 \\* 60`
      t_m=`expr $t_rest_h / 60`
      t_s=`expr $t_rest_h - $t_m \\* 60`
      duration=""
      if [ $t_h -gt 0 ]; then
         duration="$t_h h : $t_m min : $t_s sec"
      elif [ $t_m -gt 0 ];then
         duration="$t_m min : $t_s sec"
      else
         duration="$t_s sec"
      fi

      ## - look about errors..
      ## -
      if [ "$retval" != "0" ]; then
         echolog ""
         echolog "\t[ERROR] ${err_msg}  [ $duration ]\n\t`$cat $err_Log`\n"
         if ! $keep_backup_on_error ; then
            rm -f ${_backupDestArchiveDir}/${i}/${i}-${filedate}.sql
         fi
         continue
      else


         ## - print durations right-aligned
         ## -
         [ -z $right_tabstop ] && right_tabstop=65
         _tmp_string="${info_msg}${duration}"
         _strlen=${#_tmp_string}
         _count_blank=`expr $right_tabstop - $_strlen`
         _str_blanks=""
         while [ $_count_blank -gt 1 ]; do
            _str_blanks="$_str_blanks "
            _count_blank=`expr $_count_blank - 1`
         done
         echononl "$_str_blanks"

         echolog "  [ $duration ]"
      fi

   done
   
   if [ $srcHost != "localhost" ] || $_via_ssh_tunnel ;then
      $ssh ${ssh_user}@$srcHost "$sudo -u $pgsql_user $psql  -lt" > ${_backupDestArchiveDir}/database-list_${filedate}.txt
   else
      $su - $pgsql_user -c "$psql  -lt" > ${_backupDestArchiveDir}/database-list_${filedate}.txt
   fi

else
   echolog "\tARCHIVE is set to $ARCHIVE. So nothing to do."   
fi
echolog ""

## - remove all files older than $days days..
## - 
$find $_backupDestArchiveDir -type f -mtime +${days} -exec rm {} \;