pub mod activities; pub mod comment; pub mod community; pub mod extensions; pub mod fetcher; pub mod inbox; pub mod post; pub mod private_message; pub mod user; use crate::{ apub::extensions::{ group_extensions::GroupExtension, page_extension::PageExtension, signatures::{PublicKey, PublicKeyExtension}, }, blocking, request::{retry, RecvError}, routes::webfinger::WebFingerResponse, DbPool, LemmyError, }; use activitystreams_ext::{Ext1, Ext2}; use activitystreams_new::{ activity::Follow, actor::{ApActor, Group, Person}, object::{Page, Tombstone}, prelude::*, }; use actix_web::{body::Body, client::Client, HttpResponse}; use chrono::NaiveDateTime; use failure::_core::fmt::Debug; use lemmy_db::{activity::do_insert_activity, user::User_}; use lemmy_utils::{convert_datetime, get_apub_protocol_string, settings::Settings, MentionData}; use log::debug; use serde::Serialize; use url::{ParseError, Url}; type GroupExt = Ext2, GroupExtension, PublicKeyExtension>; type PersonExt = Ext1, PublicKeyExtension>; type PageExt = Ext1; pub static APUB_JSON_CONTENT_TYPE: &str = "application/activity+json"; /// Convert the data to json and turn it into an HTTP Response with the correct ActivityPub /// headers. fn create_apub_response(data: &T) -> HttpResponse where T: Serialize, { HttpResponse::Ok() .content_type(APUB_JSON_CONTENT_TYPE) .json(data) } fn create_apub_tombstone_response(data: &T) -> HttpResponse where T: Serialize, { HttpResponse::Gone() .content_type(APUB_JSON_CONTENT_TYPE) .json(data) } // Checks if the ID has a valid format, correct scheme, and is in the allowed instance list. fn is_apub_id_valid(apub_id: &Url) -> bool { debug!("Checking {}", apub_id); if apub_id.scheme() != get_apub_protocol_string() { debug!("invalid scheme: {:?}", apub_id.scheme()); return false; } let allowed_instances: Vec = Settings::get() .federation .allowed_instances .split(',') .map(|d| d.to_string()) .collect(); match apub_id.domain() { Some(d) => { let contains = allowed_instances.contains(&d.to_owned()); if !contains { debug!("{} not in {:?}", d, allowed_instances); } contains } None => { debug!("missing domain"); false } } } #[async_trait::async_trait(?Send)] pub trait ToApub { type Response; async fn to_apub(&self, pool: &DbPool) -> Result; fn to_tombstone(&self) -> Result; } /// Updated is actually the deletion time fn create_tombstone( deleted: bool, object_id: &str, updated: Option, former_type: T, ) -> Result where T: ToString, { if deleted { if let Some(updated) = updated { let mut tombstone = Tombstone::new(); tombstone.set_id(object_id.parse()?); tombstone.set_former_type(former_type.to_string()); tombstone.set_deleted(convert_datetime(updated)); Ok(tombstone) } else { Err(format_err!("Cant convert to tombstone because updated time was None.").into()) } } else { Err(format_err!("Cant convert object to tombstone if it wasnt deleted").into()) } } #[async_trait::async_trait(?Send)] pub trait FromApub { type ApubType; async fn from_apub( apub: &Self::ApubType, client: &Client, pool: &DbPool, ) -> Result where Self: Sized; } #[async_trait::async_trait(?Send)] pub trait ApubObjectType { async fn send_create( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_update( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_delete( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_undo_delete( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_remove( &self, mod_: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_undo_remove( &self, mod_: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; } #[async_trait::async_trait(?Send)] pub trait ApubLikeableType { async fn send_like( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_dislike( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_undo_like( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; } pub fn get_shared_inbox(actor_id: &Url) -> String { format!( "{}://{}{}/inbox", &actor_id.scheme(), &actor_id.host_str().unwrap(), if let Some(port) = actor_id.port() { format!(":{}", port) } else { "".to_string() }, ) } #[async_trait::async_trait(?Send)] pub trait ActorType { fn actor_id_str(&self) -> String; fn public_key(&self) -> String; fn private_key(&self) -> String; /// numeric id in the database, used for insert_activity fn user_id(&self) -> i32; // These two have default impls, since currently a community can't follow anything, // and a user can't be followed (yet) #[allow(unused_variables)] async fn send_follow( &self, follow_actor_id: &str, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_unfollow( &self, follow_actor_id: &str, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; #[allow(unused_variables)] async fn send_accept_follow( &self, follow: Follow, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_delete( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_undo_delete( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_remove( &self, mod_: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; async fn send_undo_remove( &self, mod_: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError>; /// For a given community, returns the inboxes of all followers. async fn get_follower_inboxes(&self, pool: &DbPool) -> Result, LemmyError>; fn actor_id(&self) -> Result { Url::parse(&self.actor_id_str()) } // TODO move these to the db rows fn get_inbox_url(&self) -> String { format!("{}/inbox", &self.actor_id_str()) } // TODO: make this return `Result fn get_shared_inbox_url(&self) -> String { get_shared_inbox(&self.actor_id().unwrap()) } fn get_outbox_url(&self) -> String { format!("{}/outbox", &self.actor_id_str()) } fn get_followers_url(&self) -> String { format!("{}/followers", &self.actor_id_str()) } fn get_following_url(&self) -> String { format!("{}/following", &self.actor_id_str()) } fn get_liked_url(&self) -> String { format!("{}/liked", &self.actor_id_str()) } fn get_public_key_ext(&self) -> PublicKeyExtension { PublicKey { id: format!("{}#main-key", self.actor_id_str()), owner: self.actor_id_str(), public_key_pem: self.public_key(), } .to_ext() } } pub async fn fetch_webfinger_url( mention: &MentionData, client: &Client, ) -> Result { let fetch_url = format!( "{}://{}/.well-known/webfinger?resource=acct:{}@{}", get_apub_protocol_string(), mention.domain, mention.name, mention.domain ); debug!("Fetching webfinger url: {}", &fetch_url); let mut response = retry(|| client.get(&fetch_url).send()).await?; let res: WebFingerResponse = response .json() .await .map_err(|e| RecvError(e.to_string()))?; let link = res .links .iter() .find(|l| l.type_.eq(&Some("application/activity+json".to_string()))) .ok_or_else(|| format_err!("No application/activity+json link found."))?; link .href .to_owned() .map(|u| Url::parse(&u)) .transpose()? .ok_or_else(|| format_err!("No href found.").into()) } pub async fn insert_activity( user_id: i32, data: T, local: bool, pool: &DbPool, ) -> Result<(), LemmyError> where T: Serialize + Debug + Send + 'static, { blocking(pool, move |conn| { do_insert_activity(conn, user_id, &data, local) }) .await??; Ok(()) }