2022-05-30 23:08:49 +00:00
|
|
|
use tokio_postgres::GenericClient;
|
|
|
|
|
|
|
|
use crate::activitypub::{
|
|
|
|
activity::Activity,
|
2022-07-15 17:31:02 +00:00
|
|
|
receiver::find_object_id,
|
2022-05-30 23:08:49 +00:00
|
|
|
vocabulary::{ANNOUNCE, LIKE},
|
|
|
|
};
|
|
|
|
use crate::errors::{DatabaseError, ValidationError};
|
|
|
|
use crate::models::posts::queries::{
|
|
|
|
delete_post,
|
|
|
|
get_post_by_object_id,
|
|
|
|
};
|
|
|
|
use crate::models::profiles::queries::get_profile_by_actor_id;
|
|
|
|
use crate::models::reactions::queries::{
|
|
|
|
delete_reaction,
|
|
|
|
get_reaction_by_activity_id,
|
|
|
|
};
|
|
|
|
use super::HandlerResult;
|
|
|
|
|
|
|
|
pub async fn handle_undo(
|
|
|
|
db_client: &mut impl GenericClient,
|
|
|
|
activity: Activity,
|
|
|
|
) -> HandlerResult {
|
|
|
|
let actor_profile = get_profile_by_actor_id(db_client, &activity.actor).await?;
|
2022-07-15 17:31:02 +00:00
|
|
|
let object_id = find_object_id(&activity.object)?;
|
2022-05-30 23:08:49 +00:00
|
|
|
match get_reaction_by_activity_id(db_client, &object_id).await {
|
|
|
|
Ok(reaction) => {
|
|
|
|
// Undo(Like)
|
|
|
|
if reaction.author_id != actor_profile.id {
|
|
|
|
return Err(ValidationError("actor is not an author").into());
|
|
|
|
};
|
|
|
|
delete_reaction(
|
|
|
|
db_client,
|
|
|
|
&reaction.author_id,
|
|
|
|
&reaction.post_id,
|
|
|
|
).await?;
|
|
|
|
Ok(Some(LIKE))
|
|
|
|
},
|
|
|
|
Err(DatabaseError::NotFound(_)) => {
|
|
|
|
// Undo(Announce)
|
|
|
|
let post = match get_post_by_object_id(db_client, &object_id).await {
|
|
|
|
Ok(post) => post,
|
|
|
|
// Ignore undo if neither reaction nor repost is found
|
|
|
|
Err(DatabaseError::NotFound(_)) => return Ok(None),
|
|
|
|
Err(other_error) => return Err(other_error.into()),
|
|
|
|
};
|
|
|
|
if post.author.id != actor_profile.id {
|
|
|
|
return Err(ValidationError("actor is not an author").into());
|
|
|
|
};
|
|
|
|
match post.repost_of_id {
|
|
|
|
// Ignore returned data because reposts don't have attached files
|
|
|
|
Some(_) => delete_post(db_client, &post.id).await?,
|
|
|
|
// Can't undo regular post
|
|
|
|
None => return Err(ValidationError("object is not a repost").into()),
|
|
|
|
};
|
|
|
|
Ok(Some(ANNOUNCE))
|
|
|
|
},
|
2022-05-30 23:58:11 +00:00
|
|
|
Err(other_error) => Err(other_error.into()),
|
2022-05-30 23:08:49 +00:00
|
|
|
}
|
|
|
|
}
|