From 963f39d6ed3cc871adfdd7632fb2416984dd34c8 Mon Sep 17 00:00:00 2001
From: rootzoll <christian@geektank.de>
Date: Thu, 6 Sep 2018 17:06:59 +0200
Subject: [PATCH] using screen instead of lftp background jobs

---
 home.admin/50torrentHDD.sh | 180 ++++++++++++++++++++++++++++---------
 1 file changed, 140 insertions(+), 40 deletions(-)

diff --git a/home.admin/50torrentHDD.sh b/home.admin/50torrentHDD.sh
index cd24cc7..1ae3cc3 100755
--- a/home.admin/50torrentHDD.sh
+++ b/home.admin/50torrentHDD.sh
@@ -3,71 +3,171 @@ echo ""
 
 # *** BITCOIN Torrent ***
 bitcoinTorrent="raspiblitz-bitcoin-2018-07-16"
+bitcoinTorrentsize=231230512
                    
 # *** LITECOIN Torrent ***
 litecoinTorrent="raspiblitz-litecoin-2018-07-29"
+litecoinTorrentsize=10240000
 
 # load network
 network=`cat .network`
 
-targetDir="/mnt/hdd/torrent/"
-
 # settings based on network
 torrent=$bitcoinTorrent
+size=$bitcoinTorrentsize
 if [ "$network" = "litecoin" ]; then
   torrent=$litecoinTorrent
+  size=$litecoinTorrentsize
 fi
 
-sudo apt-get install lftp -y
-echo ""
+# screen background monitoring settings
+name="Torrent"
+targetDir="/mnt/hdd/torrent/"
+targetSize=$size
+maxTimeoutLoops=100000
+command="sudo lftp -c 'torrent -O ${targetDir} /home/admin/assets/${torrent}.torrent; bye'"
 
-# check if lftp is running in background
-pid=$(pgrep lftp | head -n 1)
-echo "${pid}"
-if [ ${#pid} -eq 0 ]; then
-  echo "Starting lftp"
+# starting session if needed
+echo "checking if ${name} has a running screen session"
+screen -wipe 1>/dev/null
+isRunning=$( screen -S ${name} -ls | grep "${name}" -c )
+echo "isRunning(${isRunning})"
+if [ ${isRunning} -eq 0 ]; then
+  echo "Starting screen session"
   sudo mkdir ${targetDir} 2>/dev/null
-  sudo lftp -c "torrent -O ${targetDir} /home/admin/assets/${torrent}.torrent; bye"
+  screen -S ${name} -dm ${command}
 else
-  echo "Reattaching lftp (${pid})"
-  sudo lftp -c "attach ${pid}"
+  echo "Continue screen session"
 fi
+sleep 3
+
+# monitor session
+screenDump="... started ..."
+actualSize=0
+timeout=1
+timeoutInfo="-"
+while :
+  do
+
+    # check if session is still running
+    screen -wipe 1>/dev/null
+    isRunning=$( screen -S ${name} -ls | grep "${name}" -c )
+    if [ ${isRunning} -eq 0 ]; then
+      timeout=0
+      echo "OK - session finished"
+      break
+    fi
+
+    # calculate progress and write it to file for LCD to read
+    freshSize=$( du -s ${targetDir} | head -n1 | awk '{print $1;}' )
+    if [ ${#actualSize} -eq 0 ]; then
+      freshSize=0
+    fi
+    progress=$(echo "scale=2; $freshSize*100/$targetSize" | bc)
+    echo $progress > '.${name}.progress'
+
+    # detect if since last loop any progress occured
+    if [ ${actualSize} -eq ${freshSize} ]; then
+      timeoutInfo="${timeout}/${maxTimeoutLoops}"
+      timeout=$(( $timeout + 1 ))
+    else
+      timeout=1
+      timeoutInfo="no timeout detected"
+    fi
+    actualSize=$freshSize
+
+    # detect if mx timeout loop limit is reached
+    if [ ${timeout} -gt ${maxTimeoutLoops} ]; then
+      echo "FAIL - download hit timeout"
+      break
+    fi
+
+    # display info screen
+    clear
+    echo "****************************************************"
+    echo "Monitoring Screen Session: ${name}"
+    echo "Progress: ${progress}% (${actualSize} of ${targetSize})"
+    echo "Timeout: ${timeoutInfo}"
+    echo "If needed press key x to stop ${name}"
+    echo "NOTICE: This can take multiple hours or days !!"
+    echo "Its OK to close terminal now and SSH back in later."
+    echo "****************************************************"
+    screen -S ${name} -X hardcopy .${name}.out
+    newScreenDump=$(cat .Download.out | grep . | tail -8)
+    if [ ${#newScreenDump} -gt 0 ]; then
+      screenDump=$newScreenDump
+    fi
+    echo "$screenDump"
 
-exit 1
+    # wait 2 seconds for key input
+    read -n 1 -t 2 keyPressed
 
-# TODO check success by size
+    # check if user wants to abort session
+    if [ "${keyPressed}" = "x" ]; then
+      echo ""
+      echo "Aborting ${name}"
+      break
+    fi
 
-# the path the actual data will be in
-#targetPath="${targetDir}${torrent}"
-#echo "path to downloaded data is ${targetPath}"
+  done
+
+# clean up
+rm -f .${name}.out
+rm -f .${name}.progress
+
+# quit session if still running
+if [ ${isRunning} -eq 1 ]; then
+  # get the PID of screen session
+  sessionPID=$(screen -ls | grep "${name}" | cut -d "." -f1 | xargs)
+  echo "killing screen session PID(${sessionPID})"
+  # kill all child processes of screen sceesion
+  pkill -P ${sessionPID}
+  echo "proccesses klilled"
+  sleep 3
+ # tell the screen session to quit and wait a bit
+  screen -S ${name} -X quit 1>/dev/null
+  sleep 3
+  echo "cleaning screen"
+  screen -wipe 1>/dev/null
+  sleep 3
+fi
+
+# the path torrent will download to
+targetPath="${targetDir}${torrent}"
+echo "path to downloaded data is ${targetPath}"
 
 # calculate progress and write it to file for LCD to read
-#finalSize=$( du -s ${targetDir} 2>/dev/null | head -n1 | awk '{print $1;}' )
-#if [ ${#finalSize} -eq 0 ]; then
-#  finalSize=0
-#fi
-#echo "final size is ${finalSize} of targeted size ${targetSize}"
+finalSize=$( du -s ${targetDir} 2>/dev/null | head -n1 | awk '{print $1;}' )
+if [ ${#finalSize} -eq 0 ]; then
+  finalSize=0
+fi
+echo "final size is ${finalSize} of targeted size ${targetSize}"
+
+echo "PRESS KEY TO CONTINUE"
+read key
 
 # check result
-#if [ ${finalSize} -lt ${targetSize} ]; then
+if [ ${finalSize} -lt ${targetSize} ]; then
  
  # Download failed
-#  sleep 3
-#  echo -ne '\007'
-#  dialog --title " WARNING " --yesno "The download failed or is not complete. Maybe try again (later). Do you want keep already downloaded data for next try?" 8 57
-#  response=$?
-#  case $response in
-#    1) sudo rm -rf ${targetDir} ;;
-#  esac
-#  ./00mainMenu.sh
-#  exit 1;
-#  
-#else
-
-#  # Download worked
-#  echo "*** Moving Files ***"
-#  sudo mv ${targetDir}${targetPath} /mnt/hdd/${network}
-#  echo "OK"
+  sleep 3
+  echo -ne '\007'
+  dialog --title " WARNING " --yesno "The download failed or is not complete. Maybe try again (later). Do you want keep already downloaded data for next try?" 8 57
+  response=$?
+  case $response in
+    1) sudo rm -rf ${targetDir} ;;
+  esac
+  ./00mainMenu.sh
+  exit 1;
+  
+else
+
+  # Download worked
+  echo "*** Moving Files ***"
+  sudo mv ${targetPath} /mnt/hdd/${network}
+  echo "OK"
 
   # continue setup
-#  ./60finishHDD.sh
\ No newline at end of file
+  ./60finishHDD.sh
+
+fi
\ No newline at end of file