libytdlp 0.1.0

Safe Rust wrapper around the libytdlp C ABI
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
98
99

use libytdlp::Client;

#[test]
fn network_extract() {
    let url = std::env::var("LIBYTDLP_TEST_URL").unwrap_or_default();

    if url.is_empty() {
        eprintln!("no LIBYTDLP_TEST_URL, skipping network test");
        return;
    }

    let client = Client::new("{}").expect("default options must be accepted");

    match client.extract(&url) {
        Ok(info) => {
            assert!(info.raw_json().len() > 100);
            assert!(info.value().is_object());
            assert!(info.title().is_some());
            assert!(info.extractor().is_some());
            assert!(info.webpage_url().is_some());

            let formats = info.formats();

            assert!(!formats.is_empty());
            assert!(formats.iter().all(|format| format.url().is_some()));
            assert!(formats.iter().any(|format| format.format_id().is_some()));
        }
        Err(error) => {
            eprintln!("network test failed (flaky is fine): {error}");
        }
    }
}

#[test]
fn network_search() {
    if std::env::var("LIBYTDLP_TEST_SEARCH").is_err() {
        eprintln!("no LIBYTDLP_TEST_SEARCH, skipping search test");
        return;
    }

    let client = Client::new("{\"extract_flat\": true}").expect("default options must be accepted");

    match client.extract("ytsearch5:never gonna give you up") {
        Ok(info) => {
            let entries = info
                .value()
                .get("entries")
                .and_then(|entries| entries.as_array());

            assert!(entries.is_some_and(|entries| !entries.is_empty()));
        }
        Err(error) => {
            eprintln!("search test failed (flaky is fine): {error}");
        }
    }
}

#[test]
fn network_download() {
    let url = std::env::var("LIBYTDLP_TEST_DOWNLOAD_URL").unwrap_or_default();

    if url.is_empty() {
        eprintln!("no LIBYTDLP_TEST_DOWNLOAD_URL, skipping download test");
        return;
    }

    let dir = std::env::temp_dir().join(format!("libytdlp-netdl-{}", std::process::id()));
    let _ = std::fs::remove_dir_all(&dir);
    std::fs::create_dir_all(&dir).expect("temp dir must be created");

    let target = dir.join("out.bin");
    let options = format!(
        "{{\"outtmpl\": \"{}\"}}",
        target.to_string_lossy().replace('\\', "\\\\")
    );

    let client = Client::new(&options).expect("options must be accepted");

    match client.download(&url, "{}") {
        Err(error) => {
            eprintln!("download test failed (flaky is fine): {error}");
        }
        Ok(job) => match job.wait() {
            Err(error) => {
                eprintln!("download wait failed (flaky is fine): {error}");
            }
            Ok(info) => {
                assert!(info.raw_json().len() > 100);
                assert!(info.title().is_some());

                let bytes = std::fs::read(&target).expect("output file must exist");

                assert!(!bytes.is_empty(), "downloaded file is empty");
            }
        },
    }

    let _ = std::fs::remove_dir_all(&dir);
}