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
//! Client for interacting with LTA API
use crate::utils::commons::Client;

/// A `Client` to make requests with
/// The `Client` holds a connection pool internally, so it is advised that you create one and reuse it
/// There are some instance where you might need to customise your client due to certain limitations.
///
/// The `Client` trait has a general constructor method and you should use the `reqwest` re-export
/// to build you own customised client from the ground up.
///
/// Take a look at the reqwest documentation on how to build your own client
///
/// ## Example
/// ```rust
/// use std::time::Duration;
/// use lta::reqwest::ClientBuilder;
/// use lta::lta_client::LTAClient;
/// use lta::utils::commons::Client;
///
/// fn my_custom_client() -> LTAClient {
///     let client = ClientBuilder::new()
///         .gzip(true)
///         .connect_timeout(Some(Duration::new(420,0)))
///         .build()
///         .unwrap();
///
///     LTAClient::new(Some("api_key".to_string()), client)     
/// }
/// ```
#[derive(Debug, Clone)]
pub struct LTAClient {
    api_key: Option<String>,
    client: reqwest::Client,
}

impl Client<reqwest::Client, reqwest::RequestBuilder> for LTAClient {
    type Output = LTAClient;

    fn new(api_key: Option<String>, client: reqwest::Client) -> LTAClient {
        LTAClient { api_key, client }
    }

    fn with_api_key<S>(api_key: S) -> LTAClient
    where
        S: Into<String>,
    {
        let api_key = api_key.into();

        let api_opt = if api_key.is_empty() {
            None
        } else {
            Some(api_key)
        };

        let client = reqwest::Client::new();

        LTAClient {
            api_key: api_opt,
            client,
        }
    }

    fn get_req_builder(&self, url: &str) -> reqwest::RequestBuilder {
        match &self.api_key {
            Some(s) => self.client.get(url).header("AccountKey", s.as_str()),
            None => panic!("API key not init!"),
        }
    }
}