diff options
Diffstat (limited to 'ar/.local')
| -rwxr-xr-x | ar/.local/bin/qndl | 129 |
1 files changed, 110 insertions, 19 deletions
diff --git a/ar/.local/bin/qndl b/ar/.local/bin/qndl index 33cdbf0..245a378 100755 --- a/ar/.local/bin/qndl +++ b/ar/.local/bin/qndl @@ -100,6 +100,44 @@ get_filename() { basename "$_fname" } +# HH:MM:SS / MM:SS / SS / "inf" β seconds (-1 for inf). awk handles +# leading zeros (POSIX sh's $((08)) treats them as octal and breaks). +parse_section_time() { + case "$1" in + inf) printf -- '-1' ;; + *) + printf '%s' "$1" | awk -F: '{ + if (NF == 1) print $1 + 0 + else if (NF == 2) print ($1 + 0) * 60 + ($2 + 0) + else if (NF == 3) print ($1 + 0) * 3600 + ($2 + 0) * 60 + ($3 + 0) + }' + ;; + esac +} + +# seconds β "HH-MM-SS" (filename-safe). -1 β "inf". +format_seconds() { + case "$1" in + -1 | inf) printf 'inf' ;; + *) printf '%02d-%02d-%02d' $(($1 / 3600)) $(($1 % 3600 / 60)) $(($1 % 60)) ;; + esac +} + +# Probe a URL for its multi_video parts. Returns one "<index>|<seconds>" +# line per part. Single-video URLs return one line with index "NA". +# Uses '|' rather than a tab because yt-dlp's --print template doesn't +# interpret backslash escapes (\t lands in output literally). +get_parts() { + _url="$1" + _cookies="$(get_cookies)" + if [ -n "$_cookies" ]; then + yt-dlp --no-warnings --print '%(playlist_index)s|%(duration)s' \ + --cookies-from-browser "$_cookies" "$_url" 2>/dev/null + else + yt-dlp --no-warnings --print '%(playlist_index)s|%(duration)s' "$_url" 2>/dev/null + fi +} + enqueue() { _dl_type="$1" _url="$2" @@ -257,31 +295,84 @@ download_video() { ;; esac - # Section downloads bypass playlist handling (sectioning a whole playlist - # makes no sense) and embed the range in the filename so multiple sections - # of the same VOD don't collide with --no-force-overwrites. - if [ -n "$_sections" ]; then - _pl_flag="--no-playlist" - _section_safe="$(printf '%s' "$_sections" | tr ':' '-')" - _fmt="${_output_dir}/%(title)s [%(id)s] [${_section_safe}].%(ext)s" - else + # Non-section paths use the regular playlist prompt + standard filename. + if [ -z "$_sections" ]; then _pl_result="$(handle_playlist "$_url" "video" "$_output_dir" "$_format")" _pl_flag="$(printf '%s' "$_pl_result" | head -n 1)" _fmt="$(printf '%s' "$_pl_result" | tail -n 1)" + _filename="$(get_filename "$_url")" + notify "π₯ Queuing video download:" "$_filename" fi - _filename="$(get_filename "$_url")" - notify "π₯ Queuing video download:" "$_filename" - + # Section downloads need special handling for multi_video URLs (e.g. Soop + # VODs split into multiple files): yt-dlp's --download-sections is applied + # per-part using each part's *local* timeline starting at 00:00:00. So a + # global "03:33:00-inf" must be split: each affected part gets its own + # locally-translated section, and the user gets one file per part. if [ -n "$_sections" ]; then - enqueue "video" "$_url" "$_filename" \ - "$_pl_flag" \ - --buffer-size 1M \ - --embed-thumbnail \ - --no-sponsorblock \ - --format "$_format_val" \ - --download-sections "*${_sections}" \ - --output "$_fmt" + _filename="$(get_filename "$_url")" + notify "π Probing VOD partsβ¦" "$_filename" + _parts="$(get_parts "$_url")" + _part_count="$(printf '%s\n' "$_parts" | awk 'NF{n++} END{print n+0}')" + + _sec_start_str="${_sections%-*}" + _sec_end_str="${_sections#*-}" + _sec_start="$(parse_section_time "$_sec_start_str")" + _sec_end="$(parse_section_time "$_sec_end_str")" + + if [ "$_part_count" -le 1 ]; then + # Single video β pass section straight through. + _section_safe="$(printf '%s' "$_sections" | tr ':' '-')" + _fmt="${_output_dir}/%(title)s [%(id)s] [${_section_safe}].%(ext)s" + notify "π₯ Queuing video download:" "$_filename" + enqueue "video" "$_url" "$_filename" \ + --no-playlist \ + --buffer-size 1M \ + --embed-thumbnail \ + --no-sponsorblock \ + --format "$_format_val" \ + --download-sections "*${_sections}" \ + --output "$_fmt" + else + # multi_video β translate global section into per-part local sections + # and dispatch one job per overlapping part. + notify "π₯ Queuing $_part_count parts:" "$_filename ($_sections)" + _offset=0 + printf '%s\n' "$_parts" | while IFS='|' read -r _idx _dur; do + case "$_dur" in '' | NA) continue ;; esac + + _part_start="$_offset" + _part_end=$((_offset + _dur)) + _offset="$_part_end" + + # Skip parts outside the requested global range. + [ "$_sec_end" != "-1" ] && [ "$_sec_end" -le "$_part_start" ] && continue + [ "$_sec_start" -ge "$_part_end" ] && continue + + _ls=$((_sec_start - _part_start)) + [ "$_ls" -lt 0 ] && _ls=0 + if [ "$_sec_end" = "-1" ]; then + _le=-1 + _yt_end="inf" + else + _le=$((_sec_end - _part_start)) + [ "$_le" -gt "$_dur" ] && _le="$_dur" + _yt_end="$_le" + fi + + _safe_section="$(format_seconds "$_ls")-$(format_seconds "$_le")" + _fmt_part="${_output_dir}/%(title)s [%(id)s] [part${_idx}_${_safe_section}].%(ext)s" + + enqueue "video" "$_url" "$_filename (part $_idx)" \ + --playlist-items "$_idx" \ + --buffer-size 1M \ + --embed-thumbnail \ + --no-sponsorblock \ + --format "$_format_val" \ + --download-sections "*${_ls}-${_yt_end}" \ + --output "$_fmt_part" + done + fi elif [ -n "$_recode_ext" ]; then enqueue "video" "$_url" "$_filename" \ "$_pl_flag" \ |