mirror of
https://github.com/astro/buzzrelay.git
synced 2024-05-23 11:48:05 +00:00
62 lines
1.9 KiB
Rust
62 lines
1.9 KiB
Rust
use std::{
|
|
sync::Arc,
|
|
time::{Instant, SystemTime},
|
|
};
|
|
use http::StatusCode;
|
|
use metrics::histogram;
|
|
use serde::Serialize;
|
|
use sigh::{PrivateKey, SigningConfig, alg::RsaSha256};
|
|
use crate::{digest, error::Error};
|
|
|
|
pub async fn send<T: Serialize>(
|
|
client: &reqwest::Client,
|
|
uri: &str,
|
|
key_id: &str,
|
|
private_key: &PrivateKey,
|
|
body: &T,
|
|
) -> Result<(), Error> {
|
|
let body = Arc::new(
|
|
serde_json::to_vec(body)?
|
|
);
|
|
send_raw(client, uri, key_id, private_key, body).await
|
|
}
|
|
|
|
pub async fn send_raw(
|
|
client: &reqwest::Client,
|
|
uri: &str,
|
|
key_id: &str,
|
|
private_key: &PrivateKey,
|
|
body: Arc<Vec<u8>>,
|
|
) -> Result<(), Error> {
|
|
let url = reqwest::Url::parse(uri)
|
|
.map_err(|_| Error::InvalidUri)?;
|
|
let host = format!("{}", url.host().ok_or(Error::InvalidUri)?);
|
|
let digest_header = digest::generate_header(&body)
|
|
.map_err(|()| Error::Digest)?;
|
|
let mut req = http::Request::builder()
|
|
.method("POST")
|
|
.uri(uri)
|
|
.header("host", &host)
|
|
.header("content-type", "application/activity+json")
|
|
.header("date", httpdate::fmt_http_date(SystemTime::now()))
|
|
.header("digest", digest_header)
|
|
.body(body.as_ref().clone())?;
|
|
let t1 = Instant::now();
|
|
SigningConfig::new(RsaSha256, private_key, key_id)
|
|
.sign(&mut req)?;
|
|
let t2 = Instant::now();
|
|
let req: reqwest::Request = req.try_into()?;
|
|
let res = client.execute(req)
|
|
.await?;
|
|
let t3 = Instant::now();
|
|
histogram!("relay_http_request_duration", t2 - t1);
|
|
if res.status() >= StatusCode::OK && res.status() < StatusCode::MULTIPLE_CHOICES {
|
|
histogram!("relay_http_response_duration", t3 - t2, "res" => "ok");
|
|
Ok(())
|
|
} else {
|
|
histogram!("relay_http_response_duration", t3 - t2, "res" => "err");
|
|
let response = res.text().await?;
|
|
Err(Error::Response(response))
|
|
}
|
|
}
|