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
81
82
83
84
85
86
87
88
89
90
91
92
93

// 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 futures::{future, Future};
use reqwest::{
    header::HeaderMap,
    r#async::{
        Client as RawClient, ClientBuilder as RawClientBuilder,
        RequestBuilder as RawRequestBuilder, Response as RawResponse,
    },
    IntoUrl,
};

use jsonrpc_sdk_prelude::{jsonrpc_core::Response, CommonPart, Error, 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,
    ) -> impl Future<Item = T::Output, Error = Error>
    where
        T: JsonRpcRequest,
    {
        match content.to_single_request(common) {
            Ok(request) => future::ok(request),
            Err(error) => future::err(error),
        }
        .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)
        })
    }
}