lisk_api_rust_client/http/
client.rs

1use api::models::{RequestError, Response};
2use api::Result;
3use reqwest::header::{HeaderMap, HeaderValue};
4use reqwest::{RequestBuilder, Url};
5use serde::de::DeserializeOwned;
6use serde::ser::Serialize;
7use serde_json::{from_str, from_value, to_string};
8use std::borrow::Borrow;
9use std::collections::HashMap;
10
11#[derive(Clone, Debug)]
12pub struct Client {
13    pub host: String,
14    client: ::reqwest::Client,
15    headers: HeaderMap,
16}
17
18impl Client {
19    pub fn new(host: &str) -> Client {
20        let mut headers = HeaderMap::new();
21        headers.insert("content-type", HeaderValue::from_static("application/json"));
22
23        Client {
24            host: host.to_owned(),
25            client: ::reqwest::Client::new(),
26            headers,
27        }
28    }
29
30    pub fn get<T: DeserializeOwned>(&self, endpoint: &str) -> Result<T> {
31        let url = Url::parse(&format!("{}{}", self.host, endpoint))?;
32        self.internal_get(&url)
33    }
34
35    pub fn get_with_params<T, I, K, V>(&self, endpoint: &str, parameters: I) -> Result<T>
36    where
37        T: DeserializeOwned,
38        I: IntoIterator,
39        I::Item: Borrow<(K, V)>,
40        K: AsRef<str>,
41        V: AsRef<str>,
42    {
43        let url = Url::parse_with_params(&format!("{}{}", self.host, endpoint), parameters)?;
44        self.internal_get(&url)
45    }
46
47    pub fn post<T, V>(&self, endpoint: &str, payload: Option<HashMap<&str, V>>) -> Result<T>
48    where
49        T: DeserializeOwned,
50        V: Serialize,
51    {
52        let url = Url::parse(&format!("{}{}", self.host, endpoint))?;
53        self.internal_post(&url, payload)
54    }
55
56    pub fn post_with_params<T, H, I, K, V>(
57        &self,
58        endpoint: &str,
59        payload: Option<HashMap<&str, H>>,
60        parameters: I,
61    ) -> Result<T>
62    where
63        T: DeserializeOwned,
64        H: Serialize,
65        I: IntoIterator,
66        I::Item: Borrow<(K, V)>,
67        K: AsRef<str>,
68        V: AsRef<str>,
69    {
70        let url = Url::parse_with_params(&format!("{}{}", self.host, endpoint), parameters)?;
71        self.internal_post(&url, payload)
72    }
73
74    fn internal_get<T: DeserializeOwned>(&self, url: &Url) -> Result<T> {
75        let builder = self.client.get(url.as_str());
76
77        self.send(builder)
78    }
79
80    fn internal_post<T, V>(&self, url: &Url, payload: Option<HashMap<&str, V>>) -> Result<T>
81    where
82        T: DeserializeOwned,
83        V: Serialize,
84    {
85        let builder = self.client.post(url.as_str());
86
87        let body = payload.map_or_else(|| Ok(String::default()), |v| to_string(&v));
88
89        self.send(builder.body(body?))
90    }
91
92    fn send<T: DeserializeOwned>(&self, builder: RequestBuilder) -> Result<T> {
93        let response = builder.headers(self.headers.clone()).send()?.text()?;
94        let value: serde_json::Value = from_str(&response)?;
95
96        let parsed = from_value::<Response<T>>(value.clone());
97
98        match parsed {
99            Ok(item) => Ok(item),
100            Err(_) => {
101                let request_error = from_value::<RequestError>(value)?;
102                Err(request_error.into())
103            }
104        }
105    }
106}