2022-07-14 08:11:21 +00:00
|
|
|
use tokio_postgres::GenericClient;
|
|
|
|
use uuid::Uuid;
|
|
|
|
|
|
|
|
use crate::activitypub::{
|
|
|
|
activity::{create_activity, Activity},
|
2022-07-23 21:37:21 +00:00
|
|
|
actors::types::Actor,
|
2022-07-14 08:11:21 +00:00
|
|
|
constants::AP_PUBLIC,
|
|
|
|
deliverer::OutgoingActivity,
|
2022-07-15 17:31:02 +00:00
|
|
|
identifiers::local_object_id,
|
2022-07-14 08:11:21 +00:00
|
|
|
vocabulary::LIKE,
|
|
|
|
};
|
|
|
|
use crate::config::Instance;
|
|
|
|
use crate::errors::DatabaseError;
|
2022-07-20 21:53:40 +00:00
|
|
|
use crate::models::posts::types::{Post, Visibility};
|
2022-07-14 08:11:21 +00:00
|
|
|
use crate::models::profiles::types::DbActorProfile;
|
|
|
|
use crate::models::users::types::User;
|
|
|
|
|
2022-07-28 22:27:50 +00:00
|
|
|
pub fn get_like_note_audience(
|
|
|
|
note_author_id: &str,
|
|
|
|
note_visibility: &Visibility,
|
|
|
|
) -> (Vec<String>, Vec<String>) {
|
|
|
|
let mut primary_audience = vec![note_author_id.to_string()];
|
|
|
|
if matches!(note_visibility, Visibility::Public) {
|
|
|
|
primary_audience.push(AP_PUBLIC.to_string());
|
|
|
|
};
|
|
|
|
let secondary_audience = vec![];
|
|
|
|
(primary_audience, secondary_audience)
|
|
|
|
}
|
|
|
|
|
2022-07-14 08:11:21 +00:00
|
|
|
fn build_like_note(
|
|
|
|
instance_url: &str,
|
|
|
|
actor_profile: &DbActorProfile,
|
|
|
|
note_id: &str,
|
|
|
|
reaction_id: &Uuid,
|
2022-07-28 22:27:50 +00:00
|
|
|
note_author_id: &str,
|
|
|
|
note_visibility: &Visibility,
|
2022-07-14 08:11:21 +00:00
|
|
|
) -> Activity {
|
2022-07-15 17:31:02 +00:00
|
|
|
let activity_id = local_object_id(instance_url, reaction_id);
|
2022-07-28 22:27:50 +00:00
|
|
|
let (primary_audience, secondary_audience) =
|
|
|
|
get_like_note_audience(note_author_id, note_visibility);
|
2022-07-14 08:11:21 +00:00
|
|
|
let activity = create_activity(
|
|
|
|
instance_url,
|
|
|
|
&actor_profile.username,
|
|
|
|
LIKE,
|
|
|
|
activity_id,
|
|
|
|
note_id,
|
2022-07-20 21:53:40 +00:00
|
|
|
primary_audience,
|
2022-07-28 22:27:50 +00:00
|
|
|
secondary_audience,
|
2022-07-14 08:11:21 +00:00
|
|
|
);
|
|
|
|
activity
|
|
|
|
}
|
|
|
|
|
2022-07-14 12:02:01 +00:00
|
|
|
pub async fn get_like_note_recipients(
|
|
|
|
_db_client: &impl GenericClient,
|
2022-07-28 22:27:50 +00:00
|
|
|
_instance_url: &str,
|
2022-07-14 12:02:01 +00:00
|
|
|
post: &Post,
|
2022-07-28 22:27:50 +00:00
|
|
|
) -> Result<Vec<Actor>, DatabaseError> {
|
2022-07-14 12:02:01 +00:00
|
|
|
let mut recipients: Vec<Actor> = Vec::new();
|
|
|
|
if let Some(remote_actor) = post.author.actor_json.as_ref() {
|
|
|
|
recipients.push(remote_actor.clone());
|
|
|
|
};
|
2022-07-28 22:27:50 +00:00
|
|
|
Ok(recipients)
|
2022-07-14 12:02:01 +00:00
|
|
|
}
|
|
|
|
|
2022-07-14 08:11:21 +00:00
|
|
|
pub async fn prepare_like_note(
|
|
|
|
db_client: &impl GenericClient,
|
|
|
|
instance: Instance,
|
|
|
|
user: &User,
|
|
|
|
post: &Post,
|
|
|
|
reaction_id: &Uuid,
|
2022-07-13 23:40:33 +00:00
|
|
|
) -> Result<OutgoingActivity<Activity>, DatabaseError> {
|
2022-07-28 22:27:50 +00:00
|
|
|
let recipients = get_like_note_recipients(
|
2022-07-14 12:02:01 +00:00
|
|
|
db_client,
|
|
|
|
&instance.url(),
|
|
|
|
post,
|
|
|
|
).await?;
|
2022-07-14 08:11:21 +00:00
|
|
|
let note_id = post.get_object_id(&instance.url());
|
2022-07-28 22:27:50 +00:00
|
|
|
let note_author_id = post.author.actor_id(&instance.url());
|
2022-07-14 08:11:21 +00:00
|
|
|
let activity = build_like_note(
|
|
|
|
&instance.url(),
|
|
|
|
&user.profile,
|
|
|
|
¬e_id,
|
|
|
|
reaction_id,
|
2022-07-28 22:27:50 +00:00
|
|
|
¬e_author_id,
|
|
|
|
&post.visibility,
|
2022-07-14 08:11:21 +00:00
|
|
|
);
|
|
|
|
Ok(OutgoingActivity {
|
|
|
|
instance,
|
|
|
|
sender: user.clone(),
|
|
|
|
activity,
|
|
|
|
recipients,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use serde_json::json;
|
|
|
|
use crate::utils::id::new_uuid;
|
|
|
|
use super::*;
|
|
|
|
|
|
|
|
const INSTANCE_URL: &str = "https://example.com";
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_build_like_note() {
|
|
|
|
let author = DbActorProfile::default();
|
|
|
|
let note_id = "https://example.com/objects/123";
|
|
|
|
let note_author_id = "https://example.com/users/test";
|
|
|
|
let reaction_id = new_uuid();
|
|
|
|
let activity = build_like_note(
|
|
|
|
INSTANCE_URL,
|
|
|
|
&author,
|
|
|
|
note_id,
|
|
|
|
&reaction_id,
|
|
|
|
note_author_id,
|
2022-07-28 22:27:50 +00:00
|
|
|
&Visibility::Public,
|
2022-07-14 08:11:21 +00:00
|
|
|
);
|
|
|
|
assert_eq!(
|
|
|
|
activity.id,
|
|
|
|
format!("{}/objects/{}", INSTANCE_URL, reaction_id),
|
|
|
|
);
|
|
|
|
assert_eq!(activity.object, json!(note_id));
|
2022-07-20 21:53:40 +00:00
|
|
|
assert_eq!(activity.to.unwrap(), json!([note_author_id, AP_PUBLIC]));
|
2022-07-14 08:11:21 +00:00
|
|
|
}
|
|
|
|
}
|