2020-07-29 08:56:18 +00:00
|
|
|
// Copyright 2020 The Matrix.org Foundation C.I.C.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
//
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
|
2020-08-04 09:23:24 +00:00
|
|
|
use std::{convert::TryFrom, sync::Arc};
|
2020-07-29 08:56:18 +00:00
|
|
|
|
|
|
|
use http::{Method as HttpMethod, Response as HttpResponse};
|
2020-07-31 18:35:27 +00:00
|
|
|
use reqwest::{Client, Response};
|
2020-07-29 08:56:18 +00:00
|
|
|
use tracing::trace;
|
|
|
|
use url::Url;
|
|
|
|
|
2020-07-31 18:35:27 +00:00
|
|
|
use matrix_sdk_common::{locks::RwLock, FromHttpRequestError, FromHttpResponseError, Outgoing};
|
2020-07-29 11:50:01 +00:00
|
|
|
|
2020-07-31 18:35:27 +00:00
|
|
|
use crate::{Endpoint, Error, Result, Session};
|
2020-07-29 08:56:18 +00:00
|
|
|
|
|
|
|
#[derive(Clone, Debug)]
|
|
|
|
pub(crate) struct HttpClient {
|
|
|
|
pub(crate) inner: Client,
|
|
|
|
pub(crate) homeserver: Arc<Url>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl HttpClient {
|
2020-07-31 18:35:27 +00:00
|
|
|
async fn send_request<Request>(
|
2020-07-29 08:56:18 +00:00
|
|
|
&self,
|
|
|
|
requires_auth: bool,
|
|
|
|
method: HttpMethod,
|
2020-07-31 18:35:27 +00:00
|
|
|
request: Request,
|
2020-07-29 11:50:01 +00:00
|
|
|
session: Arc<RwLock<Option<Session>>>,
|
2020-07-31 18:35:27 +00:00
|
|
|
) -> Result<Response>
|
|
|
|
where
|
|
|
|
Request: Endpoint,
|
|
|
|
<Request as Outgoing>::Incoming:
|
|
|
|
TryFrom<http::Request<Vec<u8>>, Error = FromHttpRequestError>,
|
|
|
|
<Request::Response as Outgoing>::Incoming: TryFrom<
|
|
|
|
http::Response<Vec<u8>>,
|
|
|
|
Error = FromHttpResponseError<<Request as Endpoint>::ResponseError>,
|
|
|
|
>,
|
|
|
|
{
|
|
|
|
let request = {
|
|
|
|
let read_guard;
|
|
|
|
let access_token = if requires_auth {
|
|
|
|
read_guard = session.read().await;
|
|
|
|
|
|
|
|
if let Some(session) = read_guard.as_ref() {
|
|
|
|
Some(session.access_token.as_str())
|
|
|
|
} else {
|
|
|
|
return Err(Error::AuthenticationRequired);
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
None
|
|
|
|
};
|
|
|
|
|
|
|
|
request.try_into_http_request(&self.homeserver.to_string(), access_token)?
|
|
|
|
};
|
2020-07-29 08:56:18 +00:00
|
|
|
|
2020-07-31 18:35:27 +00:00
|
|
|
let url = &request.uri().to_string();
|
2020-07-29 08:56:18 +00:00
|
|
|
|
|
|
|
let request_builder = match method {
|
|
|
|
HttpMethod::GET => self.inner.get(url),
|
|
|
|
HttpMethod::POST => {
|
|
|
|
let body = request.body().clone();
|
|
|
|
self.inner
|
|
|
|
.post(url)
|
|
|
|
.body(body)
|
|
|
|
.header(reqwest::header::CONTENT_TYPE, "application/json")
|
|
|
|
}
|
|
|
|
HttpMethod::PUT => {
|
|
|
|
let body = request.body().clone();
|
|
|
|
self.inner
|
|
|
|
.put(url)
|
|
|
|
.body(body)
|
|
|
|
.header(reqwest::header::CONTENT_TYPE, "application/json")
|
|
|
|
}
|
|
|
|
HttpMethod::DELETE => {
|
|
|
|
let body = request.body().clone();
|
|
|
|
self.inner
|
|
|
|
.delete(url)
|
|
|
|
.body(body)
|
|
|
|
.header(reqwest::header::CONTENT_TYPE, "application/json")
|
|
|
|
}
|
|
|
|
method => panic!("Unsupported method {}", method),
|
|
|
|
};
|
|
|
|
|
|
|
|
Ok(request_builder.send().await?)
|
|
|
|
}
|
|
|
|
|
|
|
|
async fn response_to_http_response(
|
|
|
|
&self,
|
|
|
|
mut response: Response,
|
|
|
|
) -> Result<http::Response<Vec<u8>>> {
|
|
|
|
let status = response.status();
|
|
|
|
let mut http_builder = HttpResponse::builder().status(status);
|
|
|
|
let headers = http_builder.headers_mut().unwrap();
|
|
|
|
|
|
|
|
for (k, v) in response.headers_mut().drain() {
|
|
|
|
if let Some(key) = k {
|
|
|
|
headers.insert(key, v);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
let body = response.bytes().await?.as_ref().to_owned();
|
|
|
|
Ok(http_builder.body(body).unwrap())
|
|
|
|
}
|
|
|
|
|
2020-07-31 18:35:27 +00:00
|
|
|
pub async fn send<Request>(
|
2020-07-29 08:56:18 +00:00
|
|
|
&self,
|
|
|
|
request: Request,
|
2020-07-29 11:50:01 +00:00
|
|
|
session: Arc<RwLock<Option<Session>>>,
|
2020-07-31 18:35:27 +00:00
|
|
|
) -> Result<<Request::Response as Outgoing>::Incoming>
|
|
|
|
where
|
|
|
|
Request: Endpoint,
|
|
|
|
<Request as Outgoing>::Incoming:
|
|
|
|
TryFrom<http::Request<Vec<u8>>, Error = FromHttpRequestError>,
|
|
|
|
<Request::Response as Outgoing>::Incoming: TryFrom<
|
|
|
|
http::Response<Vec<u8>>,
|
|
|
|
Error = FromHttpResponseError<<Request as Endpoint>::ResponseError>,
|
|
|
|
>,
|
|
|
|
Error: From<FromHttpResponseError<<Request as Endpoint>::ResponseError>>,
|
|
|
|
{
|
2020-07-29 08:56:18 +00:00
|
|
|
let response = self
|
|
|
|
.send_request(
|
|
|
|
Request::METADATA.requires_authentication,
|
|
|
|
Request::METADATA.method,
|
|
|
|
request,
|
|
|
|
session,
|
|
|
|
)
|
|
|
|
.await?;
|
|
|
|
|
|
|
|
trace!("Got response: {:?}", response);
|
|
|
|
|
|
|
|
let response = self.response_to_http_response(response).await?;
|
|
|
|
|
2020-07-31 18:35:27 +00:00
|
|
|
Ok(<Request::Response as Outgoing>::Incoming::try_from(
|
|
|
|
response,
|
|
|
|
)?)
|
2020-07-29 08:56:18 +00:00
|
|
|
}
|
|
|
|
}
|