145 lines
3.8 KiB
Rust
145 lines
3.8 KiB
Rust
use std::collections::HashMap;
|
|
|
|
use chrono::Utc;
|
|
use serde::Deserialize;
|
|
use serde_json::Value;
|
|
|
|
use mitra_config::Config;
|
|
use mitra_models::{
|
|
database::{DatabaseClient, DatabaseError},
|
|
posts::queries::{
|
|
get_post_by_remote_object_id,
|
|
update_post,
|
|
},
|
|
posts::types::PostUpdateData,
|
|
profiles::queries::get_profile_by_remote_actor_id,
|
|
};
|
|
|
|
use crate::activitypub::{
|
|
actors::{
|
|
helpers::update_remote_profile,
|
|
types::Actor,
|
|
},
|
|
handlers::create::{
|
|
create_content_link,
|
|
get_object_attachments,
|
|
get_object_content,
|
|
get_object_tags,
|
|
get_object_url,
|
|
},
|
|
types::Object,
|
|
vocabulary::{NOTE, PERSON},
|
|
};
|
|
use crate::errors::ValidationError;
|
|
use crate::media::MediaStorage;
|
|
|
|
use super::HandlerResult;
|
|
|
|
async fn handle_update_note(
|
|
config: &Config,
|
|
db_client: &mut impl DatabaseClient,
|
|
activity: Value,
|
|
) -> HandlerResult {
|
|
let object: Object = serde_json::from_value(activity["object"].to_owned())
|
|
.map_err(|_| ValidationError("invalid object"))?;
|
|
let post = match get_post_by_remote_object_id(
|
|
db_client,
|
|
&object.id,
|
|
).await {
|
|
Ok(post) => post,
|
|
// Ignore Update if post is not found locally
|
|
Err(DatabaseError::NotFound(_)) => return Ok(None),
|
|
Err(other_error) => return Err(other_error.into()),
|
|
};
|
|
let mut content = get_object_content(&object)?;
|
|
if object.object_type != NOTE {
|
|
// Append link to object
|
|
let object_url = get_object_url(&object)?;
|
|
content += &create_content_link(object_url);
|
|
};
|
|
let instance = config.instance();
|
|
let storage = MediaStorage::from(config);
|
|
let (attachments, unprocessed) = get_object_attachments(
|
|
db_client,
|
|
&instance,
|
|
&storage,
|
|
&object,
|
|
&post.author,
|
|
).await?;
|
|
for attachment_url in unprocessed {
|
|
content += &create_content_link(attachment_url);
|
|
};
|
|
if content.is_empty() && attachments.is_empty() {
|
|
return Err(ValidationError("post is empty").into());
|
|
};
|
|
let (mentions, hashtags, links, emojis) = get_object_tags(
|
|
db_client,
|
|
&instance,
|
|
&storage,
|
|
&object,
|
|
&HashMap::new(),
|
|
).await?;
|
|
let updated_at = object.updated.unwrap_or(Utc::now());
|
|
let post_data = PostUpdateData {
|
|
content,
|
|
attachments,
|
|
mentions,
|
|
tags: hashtags,
|
|
links,
|
|
emojis,
|
|
updated_at,
|
|
};
|
|
update_post(db_client, &post.id, post_data).await?;
|
|
Ok(Some(NOTE))
|
|
}
|
|
|
|
#[derive(Deserialize)]
|
|
struct UpdatePerson {
|
|
actor: String,
|
|
object: Actor,
|
|
}
|
|
|
|
async fn handle_update_person(
|
|
config: &Config,
|
|
db_client: &mut impl DatabaseClient,
|
|
activity: Value,
|
|
) -> HandlerResult {
|
|
let activity: UpdatePerson = serde_json::from_value(activity)
|
|
.map_err(|_| ValidationError("invalid actor data"))?;
|
|
if activity.object.id != activity.actor {
|
|
return Err(ValidationError("actor ID mismatch").into());
|
|
};
|
|
let profile = get_profile_by_remote_actor_id(
|
|
db_client,
|
|
&activity.object.id,
|
|
).await?;
|
|
update_remote_profile(
|
|
db_client,
|
|
&config.instance(),
|
|
&MediaStorage::from(config),
|
|
profile,
|
|
activity.object,
|
|
).await?;
|
|
Ok(Some(PERSON))
|
|
}
|
|
|
|
pub async fn handle_update(
|
|
config: &Config,
|
|
db_client: &mut impl DatabaseClient,
|
|
activity: Value,
|
|
) -> HandlerResult {
|
|
let object_type = activity["object"]["type"].as_str()
|
|
.ok_or(ValidationError("unknown object type"))?;
|
|
match object_type {
|
|
NOTE => {
|
|
handle_update_note(config, db_client, activity).await
|
|
},
|
|
PERSON => {
|
|
handle_update_person(config, db_client, activity).await
|
|
},
|
|
_ => {
|
|
log::warn!("unexpected object type {}", object_type);
|
|
Ok(None)
|
|
},
|
|
}
|
|
}
|