From: Steven Baltakatei Sandoval <baltakatei@gmail.com>
Date: Sun, 22 Sep 2024 11:17:47 +0000 (+0000)
Subject: feat(user/rand_media_pl.sh):Make robust and parallelize
X-Git-Url: https://zdv2.bktei.com/gitweb/BK-2020-03.git/commitdiff_plain/00fcff36572aa4cb92fe5363c52d6ed346a20120?ds=sidebyside;hp=b0f274d1f2b18d7d18ca86cbe612b3e37d2a7cdc

feat(user/rand_media_pl.sh):Make robust and parallelize
---

diff --git a/user/rand_media_pl.sh b/user/rand_media_pl.sh
index a9c981b..ecf26da 100755
--- a/user/rand_media_pl.sh
+++ b/user/rand_media_pl.sh
@@ -1,30 +1,31 @@
 #!/usr/bin/env bash
-# Desc
+# Description:
 #   - Finds audio and video files with specified extensions up to a max depth of 8.
 #   - Uses ffprobe to measure their durations.
 #   - Creates a playlist starting at a random location within the total runtime.
 #   - Stores durations and file paths in a dotfile for faster subsequent runs.
 #   - Prompts the user whether to use the cached data or regenerate it.
+#   - Uses an EDL file to specify the starting offset for the first file.
 # Usage: rand_media_pl.sh [DIR]
-# Version: 0.0.1
-# Dependencies: Bash 4, ffprobe, mpv
+# Version: 0.0.4
+# Dependencies: Bash 4, ffprobe, mpv, bc, shuf
 
-yell() { echo "$0: $*" >&2; } # print script path and all args to stderr
-die() { yell "$*"; exit 111; } # same as yell() but non-zero exit status
-must() { "$@" || die "cannot $*"; } # runs args as command, reports args if command fails
+yell() { echo "$0: $*" >&2; } # Print script path and all args to stderr
+die() { yell "$*"; exit 111; } # Same as yell() but exits with code 111
+must() { "$@" || die "cannot $*"; } # Runs args as command, reports args if command fails
 
 # Configurable variables
-SEARCH_DIR="$1";
-MAX_DEPTH=8;
-EXTENSIONS=("*.flac" "*.mp3" "*.opus" "*.m4a" "*.m4b" "*.mp4" "*.mkv" "*.webm");
-CACHE_FILE=".playlist_cache";
+SEARCH_DIR="${1:-.}"  # Default to current directory if no argument is provided
+MAX_DEPTH=8
+EXTENSIONS=("*.flac" "*.mp3" "*.opus" "*.m4a" "*.m4b" "*.mp4" "*.mkv" "*.webm")
+CACHE_FILE=".playlist_cache"
 
-declare -gA file_durations;
-declare total_duration;
+declare -gA file_durations
+declare total_duration
 
 # Function to prompt the user
 prompt_yes_no() {
-    yell "STATUS:User input required.";
+    yell "STATUS: User input required."
     local prompt_message="$1"
     local user_input
     while true; do
@@ -33,115 +34,163 @@ prompt_yes_no() {
             [Yy]*) return 0 ;;
             [Nn]*) return 1 ;;
             *) echo "Please answer yes or no." ;;
-        esac;
-    done;
-};
+        esac
+    done
+}
 
 # Function to find files with specified extensions
 find_media_files() {
-    yell "STATUS:Finding media files.";
-    local find_cmd=("find" "-L" "$SEARCH_DIR" "-maxdepth" "$MAX_DEPTH" "-type" "f" "(");
-    #declare -p find_cmd 1>&2; # debug
+    yell "STATUS: Finding media files."
+    local find_cmd=("find" "-L" "$SEARCH_DIR" "-maxdepth" "$MAX_DEPTH" "-type" "f" "(")
     for ext in "${EXTENSIONS[@]}"; do
-        find_cmd+=("-iname" "$ext" "-o");
-        #declare -p find_cmd 1>&2; # debug
-    done;
+        find_cmd+=("-iname" "$ext" "-o")
+    done
     # Remove the last "-o" and add ")"
-    unset 'find_cmd[-1]';
-    find_cmd+=(")");
-    "${find_cmd[@]}";
-};
+    unset 'find_cmd[-1]'
+    find_cmd+=(")")
+    "${find_cmd[@]}"
+}
 
 # Function to generate the playlist cache
 generate_playlist_cache() {
     yell "Generating playlist cache. This may take a while..."
     # Initialize or empty the cache file
-    if [[ ! -f "$CACHE_FILE" ]]; then
-        touch "$CACHE_FILE";
-    else
-        rm "$CACHE_FILE";
-    fi;
-    
+    : > "$CACHE_FILE"  # Truncate or create the cache file
+
     total_duration=0
     while IFS= read -r file; do
         # Get the duration using ffprobe
-        duration=$(ffprobe -v error -show_entries format=duration -of default=noprint_wrappers=1:nokey=1 "$file")
-        if [[ -n "$duration" ]]; then
+        duration=$(ffprobe -v error -show_entries format=duration \
+            -of default=noprint_wrappers=1:nokey=1 "$file")
+        declare -p file duration 1>&2  # Debugging statement
+
+        # Validate the duration
+        if [[ -n "$duration" && "$duration" =~ ^[0-9]+(\.[0-9]+)?$ ]]; then
             # Append the file path and duration to the cache file
-            printf "%s|%s\n" "$file" "$duration" >> "$CACHE_FILE";
-            total_duration=$(echo "$total_duration + $duration" | bc);
-        fi;
-    done < <(find_media_files | sort);
-    total_duration_s="$(echo "scale=1; ${total_duration} / 1" | bc -l)";
-    total_duration_h="$(echo "scale=1; ${total_duration} / 3600" | bc -l)";
-    
-    yell "Total duration of playlist: ${total_duration_s} seconds. (${total_duration_h} hours.)";
-};
+            printf "%s|%s\n" "$file" "$duration" >> "$CACHE_FILE"
+            total_duration=$(echo "$total_duration + $duration" | bc)
+        else
+            yell "WARNING: Invalid duration '$duration' for file '$file', skipping." 1>&2
+        fi
+    done < <(find_media_files | sort)
+
+    total_duration_s=$(printf "%.1f" "$total_duration")
+    total_duration_h=$(echo "scale=1; $total_duration / 3600" | bc -l)
+
+    yell "Total duration of playlist: ${total_duration_s} seconds (${total_duration_h} hours)."
+}
 
 # Function to read the playlist cache
 read_playlist_cache() {
-    # Input: file_durations  assoc. array
-    #        total_duration  array
-    yell "STATUS:Reading playlist cache.";
-    
+    # Input: file_durations associative array
+    #        total_duration scalar
+    yell "STATUS: Reading playlist cache."
+
     total_duration=0
     while IFS='|' read -r file duration; do
-        file_durations["$file"]="$duration"
-        total_duration=$(echo "$total_duration + $duration" | bc)
+        declare -p file duration 1>&2  # Debugging statement
+
+        # Validate the duration
+        if [[ -n "$duration" && "$duration" =~ ^[0-9]+(\.[0-9]+)?$ ]]; then
+            file_durations["$file"]="$duration"
+            total_duration=$(echo "$total_duration + $duration" | bc)
+        else
+            yell "WARNING: Invalid duration '$duration' for file '$file', skipping." 1>&2
+        fi
     done < "$CACHE_FILE"
-};
+}
 
 # Main script
 if [[ -f "$CACHE_FILE" ]]; then
     if prompt_yes_no "Playlist cache detected. Do you want to use it? (Faster)"; then
-        read_playlist_cache;
+        read_playlist_cache
     else
-        generate_playlist_cache;
-        read_playlist_cache;
-    fi;
+        generate_playlist_cache
+        read_playlist_cache
+    fi
 else
-    generate_playlist_cache;
-    read_playlist_cache;
-fi;
+    generate_playlist_cache
+    read_playlist_cache
+fi
 
 # Check if any files were found
 if [[ ${#file_durations[@]} -eq 0 ]]; then
-    declare -p file_durations 1>&2;
-    die "FATAL:No media files found.";    
+    declare -p file_durations 1>&2
+    die "FATAL: No media files found."
+fi
+
+# Ensure total_duration is not empty
+if [[ -z "$total_duration" ]]; then
+    die "FATAL: total_duration is empty."
 fi
 
+# Convert total_duration to an integer
+total_duration_int=$(printf "%.0f" "$total_duration")
+
+# Check if total_duration_int is a valid integer
+if ! [[ "$total_duration_int" =~ ^[0-9]+$ ]]; then
+    die "FATAL: total_duration_int is not a valid integer."
+fi
+
+# Generate a random integer between 0 and total_duration_int - 1
+random_point=$(shuf -i 0-$((total_duration_int - 1)) -n1)
+yell "DEBUG: total_duration=$total_duration, total_duration_int=$total_duration_int, random_point=$random_point" 1>&2
+
 # Get sorted list of files
 mapfile -t sorted_files < <(printf '%s\n' "${!file_durations[@]}" | sort)
 
-# Generate a random starting point within the total duration
-random_point=$(awk -v max="$total_duration" 'BEGIN{srand(); print rand()*max}')
-
 # Find the file and offset corresponding to the random starting point
 accumulated_duration=0
-for file in "${sorted_files[@]}"; do
+accumulated_duration_int=0
+
+for idx in "${!sorted_files[@]}"; do
+    file="${sorted_files[$idx]}"
     duration="${file_durations["$file"]}"
-    new_accumulated_duration=$(echo "$accumulated_duration + $duration" | bc)
-    if (( $(echo "$random_point < $new_accumulated_duration" | bc -l) )); then
-        offset=$(echo "$random_point - $accumulated_duration" | bc)
+    declare -p idx file duration accumulated_duration 1>&2  # Debugging statement
+
+    # Validate the duration
+    if [[ -n "$duration" && "$duration" =~ ^[0-9]+(\.[0-9]+)?$ ]]; then
+        new_accumulated_duration=$(echo "$accumulated_duration + $duration" | bc)
+        # Convert accumulated durations to integers for comparison
+        accumulated_duration_int=$(printf "%.0f" "$accumulated_duration")
+        new_accumulated_duration_int=$(printf "%.0f" "$new_accumulated_duration")
+    else
+        yell "WARNING: Invalid duration '$duration' for file '$file', skipping." 1>&2
+        continue
+    fi
+
+    if (( random_point < new_accumulated_duration_int )); then
+        offset=$(echo "$random_point - $accumulated_duration_int" | bc)
         yell "Starting playback from $offset seconds into file: $file"
 
-        # Create a playlist file
-        playlist_file=$(mktemp)
-        printf '%s\n' "${sorted_files[@]}" > "$playlist_file"
+        # Create an EDL file
+        edl_file=$(mktemp)
+        echo "# mpv EDL v0" > "$edl_file"
+
+        # First line: file with offset
+        printf '%s,%s\n' "$file" "$offset" >> "$edl_file"
+
+        # Append the rest of the files
+        declare -p file offset idx sorted_files edl_file 1>&2; # debug
+        for (( i=idx+1; i<${#sorted_files[@]}; i++ )); do
+            next_file="${sorted_files[$i]}"
+            yell "STATUS:Adding:$next_file";
+            printf '%s\n' "$next_file" >> "$edl_file"
+        done
 
         # Start playback using mpv
-        mpv --playlist="$playlist_file" --start="$offset" \
-            --playlist-start=$(($(printf '%s\n' "${sorted_files[@]}" | grep -n "^$file$" | cut -d: -f1)-1))
+        mpv "$edl_file"
 
-        # Securely delete the playlist file
-        if [[ -n "$playlist_file" && -f "$playlist_file" ]]; then
-            rm "$playlist_file"; exit 0;
+        # Securely delete the EDL file
+        if [[ -n "$edl_file" && -f "$edl_file" ]]; then
+            rm "$edl_file"
+            exit 0
         else
-            die "FATAL: playlist_file is either unset or not a valid file, skipping deletion.";
-        fi;
+            die "FATAL: edl_file is either unset or not a valid file, skipping deletion."
+        fi
     fi
-    accumulated_duration="$new_accumulated_duration";
+    accumulated_duration="$new_accumulated_duration"
+    accumulated_duration_int="$new_accumulated_duration_int"
 done
 
-die "FATAL: Could not find file corresponding to random point.";
-
+die "FATAL: Could not find file corresponding to random point."