music: reject live streams on the yt-dlp path

2 files changed, 9 insertions, 0 deletions

diff --git a/src/music.rs b/src/music.rs
index acec278..0b907d3 100644
--- a/src/music.rs
+++ b/src/music.rs
@@ -547,6 +547,13 @@ pub(crate) fn sanitize_yt_url(url: &str) -> String {
 }
 
 async fn build_input(request: &TrackRequest, http: &reqwest::Client) -> Result<Input> {
+    // yt-dlp would block forever collecting a live stream into memory (and
+    // `--match-filter !is_live` rejects the remaining cases).
+    if request.kind == SourceKind::YtDlp && is_live_stream_hint(&request.url) {
+        return Err(Error::Custom(
+            "live streams are not supported through yt-dlp".into(),
+        ));
+    }
     if request.compressed {
         let audio_bytes = match request.kind {
             SourceKind::YtDlp => {
diff --git a/src/voice/mod.rs b/src/voice/mod.rs
index 2866f54..8f73b7a 100644
--- a/src/voice/mod.rs
+++ b/src/voice/mod.rs
@@ -253,6 +253,8 @@ pub async fn ytdlp_download(url: &str) -> Result<Vec<u8>> {
                 "--no-playlist",
                 "--max-filesize",
                 "128M",
+                "--match-filter",
+                "!is_live",
                 "--",
                 url,
             ])