#!/bin/bash
set -e
version=4.4
date=20/01/2021
pixz_i=""
pigz_i=""
swap_was="0"
ftp_i=0
pv_i=0
retry_flag=0
daemon_on=0
dfolder="bootstrap_backup"
split_size=250

echo -e "\n\e[97mOkcash Blockchain Uploader v$version ($date)\e[0m"
echo -e "FTP version (backup)"
echo -e "wareck : wareck@gmail.com"
echo -e ""
echo -e "\e[93mArchive size \e[0m: $split_size Mo"
sleep 2

function init {
if ps -ef | grep -v grep | grep okcashd >/dev/null
then
daemon_on=1
echo -e "\n\e[93mStopping Okcashd :\e[0m"
okcashd stop 2>/dev/null || true && sudo /etc/init.d/cron stop 2>/dev/null || true
sleep 2
if  ps -ef | grep -v grep | grep okcashd >/dev/null ; then killall -9 okcashd; fi
sleep 2
if [ -f /usr/local/bin/okcashd ];then sudo mv /usr/local/bin/okcashd /usr/local/bin/okcashd_old |true ;fi
if  ps -ef | grep -v grep | grep okcashd >/dev/null ; then killall -9 okcashd; fi
echo "Done."
fi

if [ "`systemctl is-active watchdog.service`" = "active" ]
then
echo -e "\n\e[93mStopping watchdog :\e[0m"
sudo systemctl stop watchdog >/dev/null
echo "Done."
fi
if [ -d /home/$USER/scripts/ledstatus/ ];then sudo python3 /home/$USER/scripts/ledstatus/led_off.py; fi
if [ -f /var/swap ]
then
echo -e "\n\e[93mStopping Swap :\e[0m"
sudo dphys-swapfile swapoff /var/swap
sudo rm /var/swap
swap_was="1"
echo "Done."
fi
}

function check_software_ {
echo -e "\n\e[95mChecking Softwares :\e[0m"
echo -e -n "Check PIXZ  installed  : "
if ! [ -x "$(command -v pixz)" ]; then pixz_i="pixz" && echo -e "[\e[91m NO  \e[0m]"; else pixz_i="" && echo -e "[\e[92m YES \e[0m]"; fi
echo -e -n "Check PIGZ  installed  : "
if ! [ -x "$(command -v pigz)" ]; then pigz_i="pigz" && echo -e "[\e[91m NO  \e[0m]"; else pigz_i="" && echo -e "[\e[92m YES \e[0m]"; fi
echo -e -n "Check LFTP  installed  : "
if ! [ -x "$(command -v lftp)" ]; then ftp_i="lftp" && echo -e "[\e[91m NO  \e[0m]"; else ftp_i="" && echo -e "[\e[92m YES \e[0m]"; fi
echo -e -n "Check PV    installed  : "
if ! [ -x "$(command -v pv)" ]; then pv_i="pv" && echo -e "[\e[91m NO  \e[0m]"; else pv_i="" && echo -e "[\e[92m YES \e[0m]"; fi
if [[ ! $ftp_i = "" || ! $pixz_i = "" || ! $pv_i = "" || ! $pigz_i = "" ]]
then
echo -e "\n\e[95mSoftwares update & install :\e[0m"
sudo apt-get update
sudo apt install $pixz_i $ftp_i $pv_i $lrzip_i $pigz_i -y
sleep 1
check_software_
fi
echo -e "Done."
}

function freeze_on {
NEW_FAN=5.0
if [ -f /home/$USER/scripts/run-fan.py ]
then
echo -e "\n\e[95mEnable \e[38;5;196mF\e[38;5;202mr\e[38;5;208me\e[38;5;214me\e[38;5;220mz\e[38;5;226mi\e[38;5;227mn\e[38;5;229mg\e[0m \e[95m:\e[0m"
sudo systemctl stop run-fan.service
if ! [ -f /home/$USER/freeze.txt ]
then
if  grep "# Author: Andreas Spiess" /home/$USER/scripts/run-fan.py >/dev/null
then
grep -i "desiredTemp = " /home/$USER/scripts/run-fan.py |awk 'NR==1 {print$3;exit}' >/home/$USER/freeze.txt
else
grep -i "self.startTemperature = " /home/$USER/scripts/run-fan.py |awk 'NR==1 {print$3;exit}' >/home/$USER/freeze.txt
fi
fi
OLD_FAN=`cat /home/$USER/freeze.txt| awk '{print $1}'`
echo "Fan PWM value : $OLD_FAN => $NEW_FAN"
if  grep "# Author: Andreas Spiess" ~/scripts/run-fan.py >/dev/null
then
sed -i -e "s/desiredTemp = "$OLD_FAN"/desiredTemp = "$NEW_FAN"/g" /home/$USER/scripts/run-fan.py
else
sed -i -e "s/self.startTemperature = "$OLD_FAN"/self.startTemperature = "$NEW_FAN"/g" /home/$USER/scripts/run-fan.py
fi
sudo systemctl restart run-fan.service
echo -e "Done."
sleep 1
fi
}

function freeze_off {
echo -e -n ""
if [ -f /home/$USER/scripts/run-fan.py ]
then
echo -e "\n\e[95mDisable Freezing :\e[0m"
OLD_FAN=`cat /home/$USER/freeze.txt| awk '{print $1}'`
if  grep "# Author: Andreas Spiess" ~/scripts/run-fan.py >/dev/null
then
echo "Fan PWM value : $NEW_FAN => $OLD_FAN"
else
echo "Fan value: $NEW_FAN =>$OLD_FAN"
fi
sudo systemctl stop run-fan.service
if  grep "# Author: Andreas Spiess" ~/scripts/run-fan.py >/dev/null
then
sed -i -e "s/desiredTemp = "$NEW_FAN"/desiredTemp = "$OLD_FAN"/g" /home/$USER/scripts/run-fan.py
else
sed -i -e "s/self.startTemperature = "$NEW_FAN"/self.startTemperature = "$OLD_FAN"/g" /home/$USER/scripts/run-fan.py
fi
sudo systemctl restart run-fan.service
if [ -f /home/$USER/freeze.txt ];then rm /home/$USER/freeze.txt;fi
echo -e "Done."
fi
}

function packit {
echo -e "\n\e[95mPacking & Compressing Files :\e[0m"
cd /home/$USER
SIZE1=`du -sk /home/$USER/.okcash/blk0001.dat | cut -f 1`
SIZE2=`du -sk /home/$USER/.okcash/txleveldb/ | cut -f 1`
SIZE=$(($SIZE2+$SIZE1))
#HSIZE=$(echo $SIZE | tail -1 | awk {'print $1'} | awk '{ total = $1 / 1024 ; print total "MB" }')
#echo -e "Folder size before compression : $HSIZE\n"

echo -e "\e[97mTar & PiXz :\e[0m"
tar cf - .okcash/blk0001.dat .okcash/txleveldb/* | pv -s ${SIZE}k | pixz  > bootstrap.tar.xz
#FSIZE=`du /home/$USER/bootstrap.tar.xz| tail -1 | awk {'print $1'} | awk '{ total = $1 / 1024 ; print total "MB" }'`
#echo -e "Folder size after compression : $FSIZE\n"
#echo -e "Gain ratio: ($FSIZE/$HSIZE)*100" | bc -l
sleep 1

echo -e "\n\e[95mSplitting Files (${split_size}M):\e[0m"
split --verbose --numeric-suffixes=1 -b ${split_size}M bootstrap.tar.xz "bootstrap.part"
rm bootstrap.tar.xz
sleep 1
echo "Done."

echo -e "\n\e[95mMake bootstrap_v.txt:\e[0m"
Raw_date="`date +%d%m%Y`"
YYYY=`ls -l bootstrap*.part* | wc -l`
cat <<'EOF'>> bootstrap_v.txt
Bootstrap : XXXX
Parts: YYYY
Compressed by :  tar / xz
EOF
sed -i -e "s/XXXX/$Raw_date/g" bootstrap_v.txt
sed -i -e "s/YYYY/$YYYY/g" bootstrap_v.txt
echo "Done."

echo -e "\n\e[95mMd5sum Files:\e[0m"
zap=$(ls -l bootstrap.part* | wc -l)
#zap=$((zap-1))
for i in `seq -w 01 $zap`
do
echo -e -n "md5sum Bootstrap.part$i: " && md5sum bootstrap.part$i > bootstrap$i.md5 && echo -e "\e[1mDone.\e[0m"
done
touch /home/$USER/bootstrap_uploadpass
}

function ftp_check {
FTP_SUCCESS_MSG="226 Transfer complete"
if fgrep "$FTP_SUCCESS_MSG" $Ftp_LOG ;then
   echo "ftp transfer : OK"
   echo ""
   rm $Ftp_LOG
else
   echo "ftp transfer Error: $ftp_file"
   rm $Ftp_LOG
   echo ""
   ftp_loop
fi
}

function ftp_loop {
lftp -u wareck,zorn692611 ftpperso.free.fr <<EOF
put $ftp_file -o /crypto/okcash/$dfolder/$ftp_file
EOF
echo -n $ftp_file && echo -e ": \e[1mDone.\e[0m";
}

function upload {
cd /home/$USER
echo -e "\n\e[95mUpload files to server :\e[0m"
ftp_file=bootstrap_v.txt && ftp_loop
zap=$(ls -l bootstrap.part* | wc -l)
#zap=$((zap-1))
for i in `seq -w 01 $zap`
do
ftp_file=bootstrap.part$i
ftp_loop
done
sleep 3

for j in `seq -w 01 $zap`
do
ftp_file=bootstrap$j.md5
ftp_loop
done

freeze_off
echo -e "\n\e[92mUpload Done ...\n\e[0m"
}

function end_restart {
if [ -f /usr/local/bin/okcashd_old ]
then
sudo mv /usr/local/bin/okcashd_old /usr/local/bin/okcashd | true
okcashd
fi
}

function retry {
if [ -f /home/$USER/bootstrap_uploadpass ]; then retry_flag=1 ;fi
if [ $retry_flag = "1" ]
then
echo -e "\n\e[95mRetry Upload to server :\e[0m"
echo -e "Old files are still present."
echo -e "If you want to force restart build/upload"
echo -e "remove files by using this command:"
echo -e ""
echo -e "rm -r -f /home/$USER/bootstrap.part* /home/$USER/bootstrap_v.txt /home/$USER/bootstrap*.md5 /home/$USER/bootstrap_uploadpass"
sleep 5
fi
}

if  ps -ef | grep -v grep | grep okcashd >/dev/null
then
echo -e "\n\e[38;5;166mOKcash daemon is working => shutdown and restart during this process !...\e[0m"
while true; do
read -p "Do you want to start processing (y/n) ? " yn
    case $yn in
        [Yy]* ) init ; check_software_; freeze_on ; retry ; packit ; upload ; end_restart ; break;;
        [Nn]* ) exit;;
        * ) echo "Please answer yes or no.";;
    esac
done
else
init
check_software_
freeze_on
retry
if [ $retry_flag = "1" ]
then
upload
end_restart
else
packit
upload
end_restart
fi
fi

if [ swap_was = 1 ]
then
echo -e "\n\e[95mRestart Swap:\e[0m"
sudo dphys-swapfile setup
sudo dphys-swapfile swapon
echo "Done."
fi
if [ -f /home/$USER/bootstrap_uploadpass ]; then rm /home/$USER/bootstrap_uploadpass;fi
if [ $daemon_on = 1 ]
then
echo -e "\n\e[38;5;166mOKcash daemon is restarting ...\e[0m"
if [ -f /usr/local/bin/okcashd_old ]
then
sudo mv /usr/local/bin/okcashd_old /usr/local/bin/okcashd | true
fi
fi
echo ""