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
//! # Search With Google
//! ## Usage
//!
//! ```rust
//! extern crate search_with_google;
//!
//! fn main() {
//!     use search_with_google::search;
//!     
//!     let results = search("rust", 3, None);
//!     if let Ok(result_list) = results{
//!         println!("Title: {}", result_list[0].title);
//!     }
//! }
//! ```
extern crate reqwest;
extern crate scraper;
extern crate select;

mod search;
use search::*;

/// # Examples
/// ```rust
/// use search_with_google::search;
/// let results = search("rust", 3, None);
/// if let Ok(result_list) = results {
///     println!("Title : {}\nLink : {}", result_list[0].title, result_list[0].link);
/// }
/// ```
pub fn search<T: Into<Option<u32>>, U: Into<Option<String>>>(
    query: &str,
    limit: T,
    agent: U,
) -> Result<Vec<SearchResult>, Error> {
    let limit: Option<u32> = limit.into();
    let agent: Option<String> = agent.into();
    google(
        query,
        limit.unwrap_or(10),
        agent.unwrap_or(
            "Mozilla/5.0 (X11; Linux x86_64; rv:77.0) Gecko/20100101 Firefox/77.0".to_string(),
        ),
    )
}

#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn search_test() {
        let results = search("minecraft", None, None);
        if let Ok(result_list) = results {
            assert!(result_list[0].title.starts_with("Minecraft"));
            assert!(result_list[0].link.len() > 0);
            assert!(result_list[0].description.len() > 0);
        }
    }
    #[test]
    fn limit_test() {
        let results = search("minecraft", 4, None);
        if let Ok(result_list) = results {
            assert!(result_list.len() <= 4);
        }
    }
}