2023-01-01 19:38:14 +00:00
|
|
|
use std::collections::BTreeMap;
|
2022-07-24 21:19:17 +00:00
|
|
|
|
2021-11-17 21:35:17 +00:00
|
|
|
use actix_web::http::Method;
|
2023-03-06 16:55:21 +00:00
|
|
|
use reqwest::Client;
|
2021-10-30 22:35:18 +00:00
|
|
|
use rsa::RsaPrivateKey;
|
2022-12-11 18:41:08 +00:00
|
|
|
use serde::{Deserialize, Serialize};
|
2022-12-04 22:31:02 +00:00
|
|
|
use serde_json::Value;
|
2021-10-30 22:35:18 +00:00
|
|
|
|
2023-02-18 23:52:48 +00:00
|
|
|
use mitra_config::Instance;
|
2023-02-18 22:25:49 +00:00
|
|
|
use mitra_utils::crypto_rsa::deserialize_private_key;
|
|
|
|
|
2023-01-17 23:14:18 +00:00
|
|
|
use crate::database::{
|
|
|
|
DatabaseClient,
|
|
|
|
DatabaseError,
|
|
|
|
};
|
2022-10-23 16:04:02 +00:00
|
|
|
use crate::http_signatures::create::{
|
|
|
|
create_http_signature,
|
|
|
|
HttpSignatureError,
|
|
|
|
};
|
2022-10-20 23:18:45 +00:00
|
|
|
use crate::json_signatures::create::{
|
2022-11-01 00:02:25 +00:00
|
|
|
is_object_signed,
|
2022-10-20 23:18:45 +00:00
|
|
|
sign_object,
|
|
|
|
JsonSignatureError,
|
|
|
|
};
|
2023-02-24 20:17:32 +00:00
|
|
|
use crate::models::users::types::User;
|
2023-03-06 16:55:21 +00:00
|
|
|
use super::{
|
|
|
|
actors::types::Actor,
|
|
|
|
constants::AP_MEDIA_TYPE,
|
|
|
|
http_client::build_federation_client,
|
|
|
|
identifiers::{local_actor_id, local_actor_key_id},
|
|
|
|
queues::OutgoingActivityJobData,
|
|
|
|
};
|
2021-04-09 00:22:17 +00:00
|
|
|
|
2023-02-22 21:54:08 +00:00
|
|
|
const DELIVERER_TIMEOUT: u64 = 30;
|
|
|
|
|
2021-04-09 00:22:17 +00:00
|
|
|
#[derive(thiserror::Error, Debug)]
|
|
|
|
pub enum DelivererError {
|
|
|
|
#[error("key error")]
|
|
|
|
KeyDeserializationError(#[from] rsa::pkcs8::Error),
|
|
|
|
|
|
|
|
#[error(transparent)]
|
2022-10-23 16:04:02 +00:00
|
|
|
HttpSignatureError(#[from] HttpSignatureError),
|
2021-04-09 00:22:17 +00:00
|
|
|
|
2022-10-20 23:18:45 +00:00
|
|
|
#[error(transparent)]
|
|
|
|
JsonSignatureError(#[from] JsonSignatureError),
|
|
|
|
|
2021-04-09 00:22:17 +00:00
|
|
|
#[error("activity serialization error")]
|
|
|
|
SerializationError(#[from] serde_json::Error),
|
|
|
|
|
|
|
|
#[error(transparent)]
|
|
|
|
RequestError(#[from] reqwest::Error),
|
|
|
|
|
|
|
|
#[error("http error {0:?}")]
|
|
|
|
HttpError(reqwest::StatusCode),
|
2022-12-03 21:06:15 +00:00
|
|
|
|
|
|
|
#[error(transparent)]
|
|
|
|
DatabaseError(#[from] DatabaseError),
|
2021-04-09 00:22:17 +00:00
|
|
|
}
|
|
|
|
|
2022-10-19 18:39:47 +00:00
|
|
|
fn build_client(instance: &Instance) -> reqwest::Result<Client> {
|
2023-03-06 16:55:21 +00:00
|
|
|
build_federation_client(instance, DELIVERER_TIMEOUT)
|
2022-10-19 18:39:47 +00:00
|
|
|
}
|
|
|
|
|
2021-04-09 00:22:17 +00:00
|
|
|
async fn send_activity(
|
2021-11-18 14:56:52 +00:00
|
|
|
instance: &Instance,
|
2021-10-30 22:35:18 +00:00
|
|
|
actor_key: &RsaPrivateKey,
|
|
|
|
actor_key_id: &str,
|
|
|
|
activity_json: &str,
|
2021-04-09 00:22:17 +00:00
|
|
|
inbox_url: &str,
|
|
|
|
) -> Result<(), DelivererError> {
|
|
|
|
let headers = create_http_signature(
|
2021-11-17 21:35:17 +00:00
|
|
|
Method::POST,
|
2021-10-30 22:35:18 +00:00
|
|
|
inbox_url,
|
|
|
|
activity_json,
|
2021-04-09 00:22:17 +00:00
|
|
|
actor_key,
|
|
|
|
actor_key_id,
|
|
|
|
)?;
|
|
|
|
|
2022-10-19 18:39:47 +00:00
|
|
|
let client = build_client(instance)?;
|
2021-11-17 21:35:17 +00:00
|
|
|
let request = client.post(inbox_url)
|
|
|
|
.header("Host", headers.host)
|
|
|
|
.header("Date", headers.date)
|
|
|
|
.header("Digest", headers.digest.unwrap())
|
|
|
|
.header("Signature", headers.signature)
|
2022-10-01 16:56:57 +00:00
|
|
|
.header(reqwest::header::CONTENT_TYPE, AP_MEDIA_TYPE)
|
2022-02-08 19:51:40 +00:00
|
|
|
.header(reqwest::header::USER_AGENT, instance.agent())
|
2021-11-17 21:35:17 +00:00
|
|
|
.body(activity_json.to_owned());
|
|
|
|
|
2021-11-18 14:56:52 +00:00
|
|
|
if instance.is_private {
|
2021-12-31 15:29:44 +00:00
|
|
|
log::info!(
|
2021-11-18 14:56:52 +00:00
|
|
|
"private mode: not sending activity to {}",
|
|
|
|
inbox_url,
|
|
|
|
);
|
|
|
|
} else {
|
|
|
|
let response = request.send().await?;
|
|
|
|
let response_status = response.status();
|
2022-09-04 11:57:31 +00:00
|
|
|
let response_text: String = response.text().await?
|
2023-01-24 21:36:36 +00:00
|
|
|
.chars().filter(|chr| *chr != '\n' && *chr != '\r').take(75)
|
|
|
|
.collect();
|
2021-11-18 14:56:52 +00:00
|
|
|
log::info!(
|
2023-01-24 21:36:36 +00:00
|
|
|
"response from {}: [{}] {}",
|
2021-12-24 00:44:01 +00:00
|
|
|
inbox_url,
|
2023-01-24 21:36:36 +00:00
|
|
|
response_status.as_str(),
|
2021-11-18 14:56:52 +00:00
|
|
|
response_text,
|
|
|
|
);
|
|
|
|
if response_status.is_client_error() || response_status.is_server_error() {
|
|
|
|
return Err(DelivererError::HttpError(response_status));
|
|
|
|
};
|
2021-04-09 00:22:17 +00:00
|
|
|
};
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
2022-12-11 18:41:08 +00:00
|
|
|
#[derive(Deserialize, Serialize)]
|
|
|
|
pub struct Recipient {
|
2023-02-24 20:17:32 +00:00
|
|
|
pub id: String,
|
2022-12-31 17:14:36 +00:00
|
|
|
inbox: String,
|
2023-02-25 00:13:07 +00:00
|
|
|
#[serde(default)]
|
2023-02-24 20:17:32 +00:00
|
|
|
pub is_delivered: bool, // default to false if serialized data contains no value
|
2022-12-31 17:14:36 +00:00
|
|
|
}
|
|
|
|
|
2021-10-30 22:35:18 +00:00
|
|
|
async fn deliver_activity_worker(
|
2021-11-18 14:56:52 +00:00
|
|
|
instance: Instance,
|
2021-10-30 22:35:18 +00:00
|
|
|
sender: User,
|
2022-12-04 22:31:02 +00:00
|
|
|
activity: Value,
|
2023-02-24 20:17:32 +00:00
|
|
|
recipients: &mut [Recipient],
|
2021-10-30 22:35:18 +00:00
|
|
|
) -> Result<(), DelivererError> {
|
|
|
|
let actor_key = deserialize_private_key(&sender.private_key)?;
|
2023-02-07 18:28:19 +00:00
|
|
|
let actor_id = local_actor_id(
|
|
|
|
&instance.url(),
|
|
|
|
&sender.profile.username,
|
2021-10-30 22:35:18 +00:00
|
|
|
);
|
2023-02-07 18:28:19 +00:00
|
|
|
let actor_key_id = local_actor_key_id(&actor_id);
|
2022-12-04 22:31:02 +00:00
|
|
|
let activity_signed = if is_object_signed(&activity) {
|
2022-11-01 00:02:25 +00:00
|
|
|
log::warn!("activity is already signed");
|
2022-12-04 22:31:02 +00:00
|
|
|
activity
|
2022-11-01 00:02:25 +00:00
|
|
|
} else {
|
2022-12-04 22:31:02 +00:00
|
|
|
sign_object(&activity, &actor_key, &actor_key_id)?
|
2022-11-01 00:02:25 +00:00
|
|
|
};
|
2022-10-20 23:18:45 +00:00
|
|
|
let activity_json = serde_json::to_string(&activity_signed)?;
|
2022-12-31 17:14:36 +00:00
|
|
|
|
2023-02-24 20:17:32 +00:00
|
|
|
for recipient in recipients.iter_mut() {
|
|
|
|
if recipient.is_delivered {
|
|
|
|
continue;
|
2022-07-24 21:19:17 +00:00
|
|
|
};
|
2023-02-24 20:17:32 +00:00
|
|
|
if let Err(error) = send_activity(
|
|
|
|
&instance,
|
|
|
|
&actor_key,
|
|
|
|
&actor_key_id,
|
|
|
|
&activity_json,
|
|
|
|
&recipient.inbox,
|
|
|
|
).await {
|
|
|
|
log::warn!(
|
|
|
|
"failed to deliver activity to {}: {}",
|
|
|
|
recipient.inbox,
|
|
|
|
error,
|
|
|
|
);
|
|
|
|
} else {
|
|
|
|
recipient.is_delivered = true;
|
2022-12-03 21:06:15 +00:00
|
|
|
};
|
|
|
|
};
|
2021-10-30 22:35:18 +00:00
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
|
2022-12-04 22:31:02 +00:00
|
|
|
pub struct OutgoingActivity {
|
2022-12-11 18:41:08 +00:00
|
|
|
pub instance: Instance,
|
|
|
|
pub sender: User,
|
2022-12-04 22:31:02 +00:00
|
|
|
pub activity: Value,
|
2022-12-11 18:41:08 +00:00
|
|
|
pub recipients: Vec<Recipient>,
|
2022-06-01 00:53:35 +00:00
|
|
|
}
|
|
|
|
|
2022-12-04 22:31:02 +00:00
|
|
|
impl OutgoingActivity {
|
|
|
|
pub fn new(
|
|
|
|
instance: &Instance,
|
|
|
|
sender: &User,
|
|
|
|
activity: impl Serialize,
|
|
|
|
recipients: Vec<Actor>,
|
|
|
|
) -> Self {
|
2022-12-31 17:14:36 +00:00
|
|
|
// Sort and de-duplicate recipients
|
|
|
|
let mut recipient_map = BTreeMap::new();
|
|
|
|
for actor in recipients {
|
|
|
|
if !recipient_map.contains_key(&actor.id) {
|
|
|
|
let recipient = Recipient {
|
|
|
|
id: actor.id.clone(),
|
|
|
|
inbox: actor.inbox,
|
2023-02-25 00:13:07 +00:00
|
|
|
is_delivered: false,
|
2022-12-31 17:14:36 +00:00
|
|
|
};
|
|
|
|
recipient_map.insert(actor.id, recipient);
|
|
|
|
};
|
|
|
|
};
|
2022-12-04 22:31:02 +00:00
|
|
|
Self {
|
|
|
|
instance: instance.clone(),
|
|
|
|
sender: sender.clone(),
|
|
|
|
activity: serde_json::to_value(activity)
|
|
|
|
.expect("activity should be serializable"),
|
2022-12-31 17:14:36 +00:00
|
|
|
recipients: recipient_map.into_values().collect(),
|
2022-12-04 22:31:02 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-02-25 16:37:24 +00:00
|
|
|
pub(super) async fn deliver(
|
2023-02-24 20:17:32 +00:00
|
|
|
mut self,
|
|
|
|
) -> Result<Vec<Recipient>, DelivererError> {
|
2022-06-01 00:53:35 +00:00
|
|
|
deliver_activity_worker(
|
|
|
|
self.instance,
|
|
|
|
self.sender,
|
|
|
|
self.activity,
|
2023-02-24 20:17:32 +00:00
|
|
|
&mut self.recipients,
|
|
|
|
).await?;
|
|
|
|
Ok(self.recipients)
|
2022-12-03 21:06:15 +00:00
|
|
|
}
|
|
|
|
|
2022-12-11 18:41:08 +00:00
|
|
|
pub async fn enqueue(
|
|
|
|
self,
|
2023-01-17 23:14:18 +00:00
|
|
|
db_client: &impl DatabaseClient,
|
2022-12-11 18:41:08 +00:00
|
|
|
) -> Result<(), DatabaseError> {
|
2023-02-25 00:13:07 +00:00
|
|
|
if self.recipients.is_empty() {
|
|
|
|
return Ok(());
|
|
|
|
};
|
|
|
|
log::info!(
|
|
|
|
"delivering activity to {} inboxes: {}",
|
|
|
|
self.recipients.len(),
|
|
|
|
self.activity,
|
|
|
|
);
|
2022-12-11 18:41:08 +00:00
|
|
|
let job_data = OutgoingActivityJobData {
|
|
|
|
activity: self.activity,
|
|
|
|
sender_id: self.sender.id,
|
|
|
|
recipients: self.recipients,
|
2023-02-24 20:17:32 +00:00
|
|
|
failure_count: 0,
|
2022-12-11 18:41:08 +00:00
|
|
|
};
|
2023-02-24 20:17:32 +00:00
|
|
|
job_data.into_job(db_client, 0).await
|
2022-08-03 14:57:46 +00:00
|
|
|
}
|
|
|
|
}
|