asimov-serpapi-module 0.1.2

ASIMOV module for data import powered by the SerpApi search data platform.
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

// This is free and unencumbered software released into the public domain.

mod bing;
pub use bing::*;

mod duckduckgo;
pub use duckduckgo::*;

mod google;
pub use google::*;

pub use asimov_module::secrecy::{ExposeSecret, SecretString};

use asimov_module::prelude::{Box, Result, String};
use core::error::Error;

#[derive(Debug)]
pub struct SerpApi {
    pub(crate) api_key: SecretString,
}

impl SerpApi {
    pub fn new(api_key: SecretString) -> Self {
        Self { api_key }
    }

    /// See: https://serpapi.com/bing-search-api
    pub fn search_bing(&self, request: &BingSearchRequest) -> Result<String, Box<dyn Error>> {
        let mut response = ureq::get("https://serpapi.com/search")
            .query("engine", "bing")
            .query("q", &request.q)
            .query("api_key", self.api_key.expose_secret())
            .call()?;
        let response_body = response.body_mut().read_to_string()?;
        Ok(response_body)
    }

    /// See: https://serpapi.com/duckduckgo-search-api
    pub fn search_duckduckgo(
        &self,
        request: &DuckDuckGoSearchRequest,
    ) -> Result<String, Box<dyn Error>> {
        let mut response = ureq::get("https://serpapi.com/search")
            .query("engine", "duckduckgo")
            .query("q", &request.q)
            .query("kl", request.kl.as_deref().unwrap_or_default())
            .query("api_key", self.api_key.expose_secret())
            .call()?;
        let response_body = response.body_mut().read_to_string()?;
        Ok(response_body)
    }

    /// See: https://serpapi.com/search-api
    pub fn search_google(&self, request: &GoogleSearchRequest) -> Result<String, Box<dyn Error>> {
        let mut response = ureq::get("https://serpapi.com/search")
            .query("engine", "google")
            .query("q", &request.q)
            .query("api_key", self.api_key.expose_secret())
            .call()?;
        let response_body = response.body_mut().read_to_string()?;
        Ok(response_body)
    }
}