Skip to main content

fedimint_connectors/
http.rs

1use std::sync::Arc;
2
3use anyhow::anyhow;
4use fedimint_core::util::SafeUrl;
5use fedimint_core::{apply, async_trait_maybe_send};
6use reqwest::{Method, StatusCode};
7use serde_json::Value;
8
9use crate::error::ServerError;
10use crate::{
11    Connectivity, DynGatewayConnection, DynGuaridianConnection, IConnection, IGatewayConnection,
12    ServerResult,
13};
14
15#[derive(Clone, Debug, Default)]
16pub(crate) struct HttpConnector {
17    client: Arc<reqwest::Client>,
18}
19
20#[async_trait::async_trait]
21impl crate::Connector for HttpConnector {
22    async fn connect_guardian(
23        &self,
24        _url: &SafeUrl,
25        _api_secret: Option<&str>,
26    ) -> ServerResult<DynGuaridianConnection> {
27        Err(ServerError::InternalClientError(anyhow!(
28            "Unsupported transport mechanism"
29        )))
30    }
31
32    async fn connect_gateway(&self, url: &SafeUrl) -> anyhow::Result<DynGatewayConnection> {
33        let http_connection = HttpConnection {
34            client: self.client.clone(),
35            base_url: url.clone(),
36        };
37
38        Ok(IGatewayConnection::into_dyn(http_connection))
39    }
40
41    fn connectivity(&self, _url: &SafeUrl) -> Connectivity {
42        Connectivity::Direct
43    }
44}
45
46#[derive(Debug)]
47pub(crate) struct HttpConnection {
48    client: Arc<reqwest::Client>,
49    base_url: SafeUrl,
50}
51
52#[apply(async_trait_maybe_send!)]
53impl IConnection for HttpConnection {
54    async fn await_disconnection(&self) {}
55
56    fn is_connected(&self) -> bool {
57        // `reqwest::Client` already implemented connection pooling. So instead of
58        // keeping this `HttpConnection` alive, we always return false here and
59        // force the `ConnectionRegistry` to re-create the connection. `HttpConnector`
60        // manages the `reqwest::Client` lifetime, so the same underlying TCP
61        // connection will be used for subsequent requests.
62        false
63    }
64}
65
66#[apply(async_trait_maybe_send!)]
67impl IGatewayConnection for HttpConnection {
68    async fn request(
69        &self,
70        password: Option<String>,
71        method: Method,
72        route: &str,
73        payload: Option<Value>,
74    ) -> ServerResult<Value> {
75        let url = self.base_url.join(route).expect("Invalid base url");
76        let mut builder = self.client.request(method, url.clone().to_unsafe());
77        if let Some(password) = password.clone() {
78            builder = builder.bearer_auth(password);
79        }
80        if let Some(payload) = payload {
81            builder = builder.json(&payload);
82        }
83
84        let response = builder
85            .send()
86            .await
87            .map_err(|e| ServerError::ServerError(e.into()))?;
88
89        match response.status() {
90            StatusCode::OK => Ok(response
91                .json::<Value>()
92                .await
93                .map_err(|e| ServerError::InvalidResponse(e.into()))?),
94            status => Err(ServerError::InvalidRequest(anyhow::anyhow!(
95                "HTTP request returned unexpected status: {status}"
96            ))),
97        }
98    }
99}