potd 0.1.0

Retrieve Picture of the Day feed from Wikimedia Commons
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


/// Error type for HTTP client
#[derive(Debug)]
pub enum HttpError {
    NetworkError(reqwest::Error),
    StatusCode(reqwest::StatusCode),
}

impl From<reqwest::Error> for HttpError {
    fn from(e: reqwest::Error) -> Self {
        HttpError::NetworkError(e)
    }
}

impl From<reqwest::StatusCode> for HttpError {
    fn from(e: reqwest::StatusCode) -> Self {
        HttpError::StatusCode(e)
    }
}

impl std::fmt::Display for HttpError {
    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        match self {
            HttpError::NetworkError(e) => write!(f, "NetworkError: {}", e),
            HttpError::StatusCode(e) => write!(f, "StatusCode: {}", e),
        }
    }
}

/// HTTP client for POTD engine
#[derive(Debug, Clone)]
pub struct HttpClient {
    client: reqwest::Client,
}

impl HttpClient {
    pub fn new() -> HttpClient {
        let version = env!("CARGO_PKG_VERSION");
        let user_agent = format!("Mozilla/5.0 (compatible; potd-rs/{}; +https://github.com/menhera-org/potd-rs)", version);
        HttpClient {
            client: reqwest::Client::builder()
                .user_agent(user_agent)
                .build()
                .unwrap(),
        }
    }

    pub async fn fetch(&self, url: &str, expect_ok: bool) -> Result<String, HttpError> {
        let res = self.client.get(url).send().await?;
        if expect_ok && res.status() != 200 {
            return Err(res.status().into());
        }
        let body = res.text().await?;
        Ok(body)
    }

    pub async fn fetch_bytes(&self, url: &str, expect_ok: bool) -> Result<Vec<u8>, HttpError> {
        let res = self.client.get(url).send().await?;
        if expect_ok && res.status() != 200 {
            return Err(res.status().into());
        }
        let body = res.bytes().await?;
        Ok(body.to_vec())
    }
}