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
70
71
72
73
74
75
76
77
78
79
80

// Copyright (C) 2019 Boyu Yang
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

use reqwest::{
    header::HeaderMap, Client as RawClient, ClientBuilder as RawClientBuilder, IntoUrl,
    RequestBuilder as RawRequestBuilder, Response as RawResponse,
};

use jsonrpc_sdk_prelude::{jsonrpc_core::Response, CommonPart, JsonRpcRequest, Result};

pub struct Client(RawClient);
pub struct ClientBuilder(RawClientBuilder);
pub struct RequestBuilder(RawRequestBuilder);

impl Client {
    pub fn new() -> Self {
        Client(RawClient::new())
    }

    pub fn builder() -> ClientBuilder {
        ClientBuilder(RawClient::builder())
    }

    pub fn post<U>(&self, url: U) -> RequestBuilder
    where
        U: IntoUrl,
    {
        RequestBuilder(self.0.post(url))
    }
}

impl Default for Client {
    fn default() -> Self {
        Self::new()
    }
}

impl ClientBuilder {
    pub fn build(self) -> Result<Client> {
        Ok(Client(self.0.build()?))
    }

    pub fn tcp_nodelay(self) -> Self {
        ClientBuilder(self.0.tcp_nodelay())
    }

    pub fn default_headers(self, headers: HeaderMap) -> Self {
        ClientBuilder(self.0.default_headers(headers))
    }

    pub fn gzip(self, enable: bool) -> Self {
        ClientBuilder(self.0.gzip(enable))
    }

    pub fn connect_timeout(self, timeout: ::std::time::Duration) -> Self {
        ClientBuilder(self.0.connect_timeout(timeout))
    }
}

impl RequestBuilder {
    pub fn send<T>(self, content: T, common: CommonPart) -> Result<T::Output>
    where
        T: JsonRpcRequest,
    {
        content.to_single_request(common).and_then(|request| {
            self.0
                .json(&request)
                .send()
                .and_then(RawResponse::error_for_status)
                .and_then(|mut r| r.json::<Response>())
                .map_err(std::convert::Into::into)
                .and_then(T::parse_single_response)
        })
    }
}