From f7fc2ef0ebc980d59f7d9500c2ed73bfd6ef1c2b Mon Sep 17 00:00:00 2001
From: xypwn <darwin.schuppan1301@gmail.com>
Date: Sun, 13 Dec 2020 15:04:14 +0100
Subject: [PATCH] Add progress monitoring option

---
 southpark-downloader.sh | 25 ++++++++++++++++++++-----
 1 file changed, 20 insertions(+), 5 deletions(-)

diff --git a/southpark-downloader.sh b/southpark-downloader.sh
index 68c31a3..ba05bb3 100755
--- a/southpark-downloader.sh
+++ b/southpark-downloader.sh
@@ -20,18 +20,19 @@ p_error() {
 
 usage() {
     echo "Usage:"
-    echo "  $(basename $0) [-E|-D] -a                        -  Download all episodes"
-    echo "  $(basename $0) [-E|-D] -s <season>               -  Download all episodes in the specified season"
-    echo "  $(basename $0) [-E|-D] -s <season> -e <episode>  -  Download the specified episode"
+    echo "  $(basename $0) [OPTIONS] -a                        -  Download all episodes"
+    echo "  $(basename $0) [OPTIONS] -s <season>               -  Download all episodes in the specified season"
+    echo "  $(basename $0) [OPTIONS] -s <season> -e <episode>  -  Download the specified episode"
     echo "Options:"
     echo " -E                                        -  Download episodes in English (default)"
     echo " -D                                        -  Download episodes in German"
+    echo " -p                                        -  Show progress"
 }
 
-unset OPT_SEASON OPT_EPISODE OPT_ALL OPT_EN OPT_LANG
+unset OPT_SEASON OPT_EPISODE OPT_ALL OPT_EN OPT_LANG OPT_PROGRESS
 OPT_LANG="EN"
 
-while getopts "haEDs:e:" arg; do
+while getopts "haEDps:e:" arg; do
     case "$arg" in
 	h)
 	    usage
@@ -52,6 +53,9 @@ while getopts "haEDs:e:" arg; do
 	D)
 	    OPT_LANG="DE"
 	    ;;
+	p)
+	    OPT_PROGRESS=true
+	    ;;
 	?)
 	    echo "Invalid option: -$OPTARG"
 	    usage
@@ -119,6 +123,16 @@ tmp_cleanup() {
     rm -rf "$TMPDIR"
 }
 
+# Monitors size of downloaded video files; takes temp folder as arg
+monitor_progress() {
+    local TMP_DIR="$1"
+    while true; do
+	[ ! -e "$TMP_DIR" ] && break
+	printf " Downloaded: %s\r" $(du -bB M "$TMP_DIR" | cut -f1)
+	sleep 0.5
+    done
+}
+
 # Takes season and episode number as arguments
 download_episode() {
     local SEASON_NUMBER="$1"
@@ -129,6 +143,7 @@ download_episode() {
     [ -e "$OUTFILE" ] && echo "Already downloaded Season ${SEASON_NUMBER} Episode ${EPISODE_NUMBER}" && return
     p_info "Downloading Season $SEASON_NUMBER Episode $EPISODE_NUMBER ($URL)"
     TMPDIR=$(mktemp -d "/tmp/southparkdownloader.XXXXXXXXXX")
+    [ -n "OPT_PROGRESS" ] && monitor_progress "$TMPDIR"&
     pushd "$TMPDIR" > /dev/null
     if ! "$YOUTUBE_DL" "$URL" 2>/dev/null | grep --line-buffered "^\[download\]" | grep -v --line-buffered "^\[download\] Destination:"; then
 	p_info "possible youtube-dl \e[1;31mERROR\e[m"