anidl 3.0.6

A simple anime downloader that uses GoGoAnime as a source
Documentation 
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

use select::document::Document;
use select::node::Node;
use select::predicate::{Attr, Name, Predicate};

/// Extracts the URLs of anime from the provided HTML body.
/// This function searches for 'a' tags within elements with class 'name'.
///
/// # Arguments
/// * `body` - HTML content as a string.
///
/// # Returns
/// A `Vec<String>` containing the URL endings of each anime.
pub fn get_anime_url(body: String) -> Result<Vec<String>, &'static str> {
    let document = parse_document(&body);
    let nodes = find_nodes(&document, "name");

    let anime_list: Vec<String> = nodes
        .iter()
        .filter_map(|node| {
            node.attr("href")
                .and_then(|href| href.split('/').last())
                .map(|url_ending| url_ending.to_string())
        })
        .collect();

    Ok(anime_list)
}

/// Extracts the names of anime from the provided HTML body.
///
/// # Arguments
///
/// * `body` - A string containing the HTML content.
///
/// # Returns
///
/// A vector of strings, each representing the name of an anime.
pub fn get_anime_name(body: String) -> Vec<String> {
    let document = parse_document(&body);
    let nodes = find_nodes(&document, "name");

    nodes.iter().map(|node| node.text()).collect()
}

/// Extracts the video URLs from the provided HTML body.
///
/// # Arguments
///
/// * `body` - A string containing the HTML content.
///
/// # Returns
///
/// A vector of strings, each representing a video URL.
pub fn get_video_url(body: String) -> Vec<String> {
    let mut anime_list: Vec<_> = vec![];

    let document = Document::from(body.as_str());
    for node in document.find(Attr("class", "cf-download").descendant(Name("a"))) {
        anime_list.push(node.attr("href").unwrap().to_string());
    }

    anime_list
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::utils;
    use crate::URL;
    use tokio;

    /// Tests the `get_anime_url` function.
    #[tokio::test]
    async fn test_get_anime_url() {
        let body = utils::get_html(format!("{}{}", URL, "search.html?keyword=Kenka%20Dokugaku"))
            .await
            .unwrap();

        let urls = get_anime_url(body).unwrap();
        assert_eq!(
            urls,
            vec![
                "kenka-dokugaku".to_string(),
                "kenka-dokugaku-dub".to_string()
            ]
        );
    }

    /// Tests the `get_anime_name` function.
    #[tokio::test]
    async fn test_get_anime_name() {
        let body = utils::get_html(format!("{}{}", URL, "search.html?keyword=kenka%20dokugaku"))
            .await
            .unwrap();

        let names = get_anime_name(body);
        println!("{:?}", names);
        assert_eq!(names, vec!["Kenka Dokugaku", "Kenka Dokugaku (Dub)"]);
    }

    #[tokio::test]
    async fn test_get_anime_url_with_empty_body() {
        let body = "".to_string();
        let urls = get_anime_url(body).unwrap();
        assert!(urls.is_empty());
    }
}

fn parse_document(body: &str) -> Document {
    Document::from(body)
}

fn find_nodes<'a>(document: &'a Document, selector: &'a str) -> Vec<Node<'a>> {
    document
        .find(Attr("class", selector).descendant(Name("a")))
        .collect()
}