yts 0.1.0

lib and cli tool for crawling movies out of yts.mx
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

use clap::ArgMatches;
use std::error::Error;
use yts::request::ListRequest;

pub fn validate_natural_one_plus(msg: String) -> impl Fn(String) -> Result<(), String> {
    move |val| match val.trim().parse::<u16>() {
        Ok(x) if x > 0 => Ok(()),
        _ => Err(msg.clone()),
    }
}

pub fn validate_min_rating(msg: String) -> impl Fn(String) -> Result<(), String> {
    move |val| match val.trim().parse::<u8>() {
        Ok(x) if (1..=9).contains(&x) => Ok(()),
        _ => Err(msg.clone()),
    }
}

pub fn list_req_from_clap<'a>(matches: &'a ArgMatches) -> Result<ListRequest<'a>, Box<dyn Error>> {
    let mut r = ListRequest::new();

    if let Some(val) = matches.value_of("limit") {
        r.limit(val.parse()?);
    }
    if let Some(val) = matches.value_of("page") {
        r.page(val.parse()?);
    }
    if let Some(val) = matches.value_of("quality") {
        r.quality(val.parse()?);
    }
    if let Some(val) = matches.value_of("rating") {
        r.rating(val.parse()?);
    }
    if let Some(vals) = matches.values_of("search") {
        let query = vals
            .into_iter()
            .map(|x| x.to_string())
            .collect::<Vec<String>>()
            .join(" ");
        r.query_term(query);
    }
    if let Some(val) = matches.value_of("genre") {
        r.genre(val);
    }
    if let Some(val) = matches.value_of("sort") {
        r.sort_by(val.parse()?);
    }
    if let Some(val) = matches.value_of("order") {
        r.order_by(val.parse()?);
    }
    if let Some(val) = matches.value_of("mirror") {
        r.mirror(val)?;
    }
    r.wirth_rt_ratings(matches.is_present("with_rotten_tomatoes"));

    Ok(r)
}

#[cfg(test)]
mod tests {
    use crate::cli::app::clap_app;
    use crate::list_req_from_clap;

    fn test_url(vec: Vec<&str>) -> String {
        let actual_vec = &mut vec!["yts", "list"];
        actual_vec.extend(vec);
        let matches = clap_app().get_matches_from(actual_vec.to_vec());
        let request = list_req_from_clap(matches.subcommand_matches("list").unwrap())
            .expect("expected a request");
        request.url().to_string()
    }

    #[test]
    fn limit() {
        assert_eq!(
            test_url(vec!["-l", "14", "-p", "13"]),
            "https://yts.mx/api/v2/list_movies.json?limit=14&page=13"
        );
    }
    #[test]
    fn search() {
        assert_eq!(
            test_url(vec!["mama", "are", "mere"]),
            "https://yts.mx/api/v2/list_movies.json?limit=50&query_term=mama+are+mere"
        );
    }
    #[test]
    fn rotten() {
        assert_eq!(
            test_url(vec!["--rt"]),
            "https://yts.mx/api/v2/list_movies.json?limit=50&with_rt_ratings"
        );
    }
    #[test]
    fn sort() {
        assert_eq!(
            test_url(vec!["--sort", "title"]),
            "https://yts.mx/api/v2/list_movies.json?limit=50&sort_by=title"
        );
    }
    #[test]
    fn mirror() {
        assert_eq!(
            test_url(vec!["--mirror", "yts.ag"]),
            "https://yts.ag/api/v2/list_movies.json?limit=50"
        );
    }
    #[test]
    fn quality() {
        assert_eq!(
            test_url(vec!["--quality", "720p"]),
            "https://yts.mx/api/v2/list_movies.json?limit=50&quality=720p"
        );
    }
}