summaryrefslogtreecommitdiffstats
path: root/src/youtube_dl.rs
blob: a917c543bedada79947a7eacda9806c774780c4f (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
use std::process::{Command, Stdio};
use tokio_process::CommandExt;
use futures::compat::Future01CompatExt;

use serde::{Serialize, Deserialize};

use log::debug;

#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct AudioMetadata {
    pub url: String,
    pub title: Option<String>,
}

pub async fn get_audio_download_url(uri: String) -> Result<AudioMetadata, String> {
    let ytdl_args = [
        "--no-playlist",
        "-f",
        "bestaudio/best",
        "-j",
        &uri,
    ];

    let mut cmd = Command::new("youtube-dl");
    cmd.args(&ytdl_args);
    cmd.stdin(Stdio::null());

    debug!("yt-dl command: {:?}", cmd);

    let ytdl_output = cmd.output_async().compat().await.unwrap();

    if ytdl_output.status.success() == false {
        return Err(String::from_utf8(ytdl_output.stderr.clone()).unwrap());
    }

    let output_str = String::from_utf8(ytdl_output.stdout.clone()).unwrap();
    let output = serde_json::from_str(&output_str).map_err(|e| e.to_string())?;

    Ok(output)
}