X-Git-Url: https://gitweb.arthurtaft.net/yt-dlp-wrapper.git/blobdiff_plain/795ebfda4485f98a96e00b549f2b6427acc3f03b..refs/heads/main:/linux/yt-dlp-wrapper.sh?ds=inline

diff --git a/linux/yt-dlp-wrapper.sh b/linux/yt-dlp-wrapper.sh
old mode 100644
new mode 100755
index ed83050..16dcfbf
--- a/linux/yt-dlp-wrapper.sh
+++ b/linux/yt-dlp-wrapper.sh
@@ -1,73 +1,182 @@
 #!/bin/bash
 
+# Cross-distro package checker and installer
+check_or_install() {
+    local pkg="$1"
+    local pm=""
+
+    if command -v pacman &>/dev/null; then
+        pm="pacman"
+        pacman -Q "$pkg" &>/dev/null
+
+    elif command -v dpkg &>/dev/null && command -v apt &>/dev/null; then
+        pm="apt"
+        dpkg -s "$pkg" &>/dev/null
+
+    elif command -v rpm &>/dev/null && command -v dnf &>/dev/null; then
+        pm="dnf"
+        rpm -q "$pkg" &>/dev/null
+
+    elif command -v rpm &>/dev/null && command -v yum &>/dev/null; then
+        pm="yum"
+        rpm -q "$pkg" &>/dev/null
+
+    elif command -v qlist &>/dev/null; then
+        pm="emerge"
+        qlist -I "$pkg" &>/dev/null
+
+    elif command -v zypper &>/dev/null; then
+        pm="zypper"
+        zypper se --installed-only "$pkg" | grep -q "$pkg"
+
+    else
+        echo "Unsupported package manager" >&2
+        return 2
+    fi
+
+    if [[ $? -eq 0 ]]; then
+        echo "$pkg is already installed."
+        return 0
+    fi
+
+    echo "Installing missing package: $pkg via $pm..."
+    case "$pm" in
+        pacman) sudo pacman -Sy --noconfirm "$pkg" ;;
+        apt) sudo apt update && sudo apt install -y "$pkg" ;;
+        dnf) sudo dnf install -y "$pkg" ;;
+        yum) sudo yum install -y "$pkg" ;;
+        emerge) sudo emerge "$pkg" ;;
+        zypper) sudo zypper install -y "$pkg" ;;
+        *) echo "Cannot install $pkg on unknown system." >&2; return 3 ;;
+    esac
+}
+
+# Ensure dependencies
+check_or_install wget
+check_or_install ffmpeg
+
 # Define variables
 current_dir="$PWD"
-exe_location="$current_dir"/bin/yt-dlp
-ffmpeg_location=/usr/bin/ffmpeg
-audio_dir="$current_dir"/audio
-video_dir="$current_dir"/video
+exe_location="$current_dir/bin/yt-dlp"
+ffmpeg_location="/usr/bin/ffmpeg"
+audio_dir="$current_dir/audio"
+video_dir="$current_dir/video"
+generated_signature="$current_dir/exe_sig.sha512"
+upstream_signature="$current_dir/sig/yt-dlp-linux.sha512"
 val=0
+dl_val=0
 url=""
+format=""
 type=""
+batch_dl_file="$current_dir/batch-dl.txt"
+
+# Ensure bin directory exists
+mkdir -p "$current_dir/bin"
 
-# Function to hold audio download command
+# Ensure sig directory exists
+mkdir -p "$current_dir/sig"
+
+# Function to download audio
 audio_dl () {
     "$exe_location" -x -f 'ba' --audio-format mp3 -o "$audio_dir/%(title)s.%(ext)s" "$url"
 }
 
-# Function to hold video download command
+audio_dl_batch () {
+    "$exe_location" -x -f 'ba' --audio-format mp3 -o "$audio_dir/%(title)s.%(ext)s" -a "$batch_dl_file"
+}
+
+# Function to download video
 video_dl () {
-    "$exe_location" -f 'bv*[ext=mp4]+ba[ext=m4a]/b[ext=mp4] / bv*+ba/b' -o "$video_dir/%(title)s.%(ext)s" "$url"
+    "$exe_location" -f "bv*[ext=mp4]+ba[ext=m4a]/b[ext=mp4]/bv*+ba/b" -o "$video_dir/%(title)s.%(ext)s" "$url"
 }
 
-# Check if ffmpeg is installed
-if [ ! "$ffmpeg_location" ]; then
-    echo "Missing dependency ffmpeg"
-    exit 1
-fi 
+video_dl_batch () {
+    "$exe_location" -f "bv*[ext=mp4]+ba[ext=m4a]/b[ext=mp4]/bv*+ba/b" --sponsorblock-remove sponsor,selfpromo,interaction -o "$video_dir/%(title)s.%(ext)s" -a "$batch_dl_file"
+}
 
-# Remove executable
-if [ "$exe_location" ]; then
-    rm "$exe_location"
-fi
+# Function to single download
+single_dl () {
+    local dl_val="$dl_val"
+    while [ "$dl_val" -ne 1 ]; do
+        read -p "Audio or Video?: " format
 
-# Download new executable
-wget -q -O "$exe_location" https://github.com/yt-dlp/yt-dlp/releases/latest/download/yt-dlp
+        case "$format" in
+            audio)
+                read -p "Enter URL Here: " url
+                audio_dl
+                dl_val=1
+                ;;
+            video)
+                read -p "Enter URL Here: " url
+                video_dl
+                dl_val=1
+                ;;
+            *)
+                echo "❗ Must be 'audio' or 'video'!"
+                ;;
+        esac
+    done
+}
+# Function to batch download
+batch_dl () {
+    local dl_val="$dl_val"
+    while [ "$dl_val" -ne 1 ]; do
+        read -p "Audio or Video?: " format
 
-# Make executable executable
-chmod +x "$exe_location"
+        case "$format" in
+            audio)
+                audio_dl_batch
+                dl_val=1
+                ;;
+            video)
+                video_dl_batch
+                dl_val=1
+                ;;
+            *)
+                echo "Must be 'audio' or 'video'!"
+                ;;
+        esac
+    done
+}
 
-# Check if audio directory exists, if not, create it
-if [ ! "$audio_dir" ]; then
-    mkdir audio
-fi
+# Pull yt-dlp cert
+wget -q -O "$current_dir/sig/yt-dlp-linux.sha512" https://github.com/arthur-taft/yt-dlp-signatures/releases/latest/download/yt-dlp-linux.sha512
 
-# Check if video directory exists, if not, create it
-if [ ! "$video_dir" ]; then
-    mkdir video
+# Generate signature for binary
+sha512sum "$exe_location" > "$current_dir/exe_sig.sha512" && sed 's/.\{8\}$//' "$current_dir/exe_sig.sha512"
+
+# Check if signatures match
+if ! grep -Ff "$current_dir/exe_sig.sha512" "$current_dir/sig/yt-dlp-linux.sha512"; then
+    echo "Old verion of yt-dlp detected downloading now"
+    # Remove existing executable (if it exists)
+    [ -f "$exe_location" ] && rm "$exe_location"
+
+    # Download yt-dlp
+    wget -q -O "$exe_location" https://github.com/yt-dlp/yt-dlp/releases/latest/download/yt-dlp
+    chmod +x "$exe_location"
+
+    rm "$current_dir/exe_sig.sha512"
 fi
 
-# While loop to contain download logic
+# Create directories if they don't exist
+mkdir -p "$audio_dir"
+mkdir -p "$video_dir"
+
+# Download loop
 while [ "$val" -ne 1 ]; do
-    # Ask user if they want to download audio or video
-    read -p "Audio or Video?: " type
-
-    # Download audio
-    if [ "$type" = "audio" ]; then
-        read -p "Enter URL Here: " url
-        audio_dl
-        val=1
-    # Download video
-    elif [ "$type" = "video" ]; then
-        read -p "Enter URL Here: " url
-        video_dl
-        val=1
-    # If invlaid response is given, start over
-    else
-        echo "Must be 'Audio' or 'Video'!"
-        continue
-    fi
+    read -p "Batch Download? (y/n): " type
+    case "$type" in
+        y)
+            batch_dl
+            val=1
+            ;;
+        n)
+            single_dl
+            val=1
+            ;;
+        *)
+            echo "Must be 'y' or 'n'!"
+            ;;
+    esac
 done
-
-# If you made it here you get a sweet exit code 0
 exit 0
\ No newline at end of file