about summary refs log tree commit diff stats
path: root/hm/soispha/pkgs/scripts/wrappers/ytc
blob: f563f00df562c764d9955c9f8247ef417b5bea85 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#!/usr/bin/env dash

# shellcheck source=/dev/null
SHELL_LIBRARY_VERSION="1.10.0" . %SHELL_LIBRARY_PATH
CONCURRENT=4
OUTPUT_PATH="$XDG_RUNTIME_DIR/ytcc";

col() {
    echo "$1" | csvtool col "$2" -
}

play() {
    msg2 "Playing: $1"
    mpv "$1" --speed=2.7
    output="$?";

    if [ "$output" -eq 0 ]; then
        msg2 "Removing: $1"
        rm "$1"
        msg2 "Marking: " "$2"
        ytcc mark "$2"
    fi
    return "$output"
}

bases="$(ytcc --output json list --attributes url --ids "$@" | jq --raw-output 'map("\(.url),\(.id)") | join("\n")')";

yt_flags="$(mktmp)"
cat << EOF > "$yt_flags"
--format bestvideo[height<=?1080]+bestaudio/best
--embed-chapters
--progress
--write-comments
--extractor-args youtube:max_comments=150,all,100;comment_sort=top
--write-info-json
--sponsorblock-mark default
--sponsorblock-remove sponsor
--restrict-filenames
EOF

[ -d "$OUTPUT_PATH" ] || mkdir "$OUTPUT_PATH";
cd  "$OUTPUT_PATH" || die "(Bug): Was created"

filename_file="$(mktmp)";
files_to_play="$(mktmp)";
while read -r base; do
    url="$(col "$base" 1)";
    id="$(col "$base" 2)"

    if [ "$old_filename" ]; then
        echo "$old_filename,$old_id" >> "$files_to_play"

        # Check if the process (pid) exists
        dbg "PID is '$pid'"
        if ! kill -0 "$pid"; then
            saved_base="$(head -n 1 "$files_to_play")";
            sed -i '1d' "$files_to_play";
            saved_name="$(col "$saved_base" 1)";
            saved_id="$(col "$saved_base" 2)"

            dbg "Started play for '$saved_name'"
            play "$saved_name" "$saved_id" &
            pid=$!
        else
            dbg "Storing for later '$old_filename'"
        fi
    fi

    # The sub shell needs to be unquoted, as the arguments may not be treated as one.
    # shellcheck disable=2046
    yt-dlp $(cat "$yt_flags") --output "%(channel)s/%(title)s.%(ext)s" "$url" --print-to-file after_move:filepath "$filename_file"

    filename="$(cat "$filename_file")"
    printf "" > "$filename_file"

    if [ "$old_filename" ]; then
        while [ "$(wc -l < "$files_to_play")" -gt "$CONCURRENT" ]; do
            msg2 "Waiting for '$pid' to finish as we already have '$(wc -l < "$files_to_play")' files cached"
            wait "$pid"
        done
    fi

    old_filename="$filename";
    old_id="$id";
done < "$(tmp echo "$bases")"

wait "$pid"
echo "$old_filename,$old_id" >> "$files_to_play"

while read -r base; do
    name="$(col "$base" 1)";
    id="$(col "$base" 2)"

    dbg "Started play for '$name'"
    play "$name" "$id"
done < "$files_to_play"
# vim: ft=sh