99 lines
2.7 KiB
Rust
99 lines
2.7 KiB
Rust
use chrono::Utc;
|
|
use serde::Deserialize;
|
|
use serde_json::Value;
|
|
use tokio_postgres::GenericClient;
|
|
|
|
use crate::activitypub::{
|
|
actors::{
|
|
helpers::update_remote_profile,
|
|
types::Actor,
|
|
},
|
|
handlers::create::get_note_content,
|
|
types::Object,
|
|
vocabulary::{NOTE, PERSON},
|
|
};
|
|
use crate::config::Config;
|
|
use crate::database::DatabaseError;
|
|
use crate::errors::ValidationError;
|
|
use crate::models::{
|
|
posts::queries::{
|
|
get_post_by_remote_object_id,
|
|
update_post,
|
|
},
|
|
posts::types::PostUpdateData,
|
|
profiles::queries::get_profile_by_remote_actor_id,
|
|
};
|
|
use super::HandlerResult;
|
|
|
|
async fn handle_update_note(
|
|
db_client: &mut impl GenericClient,
|
|
activity: Value,
|
|
) -> HandlerResult {
|
|
let object: Object = serde_json::from_value(activity["object"].to_owned())
|
|
.map_err(|_| ValidationError("invalid object"))?;
|
|
let post_id = match get_post_by_remote_object_id(
|
|
db_client,
|
|
&object.id,
|
|
).await {
|
|
Ok(post) => post.id,
|
|
// Ignore Update if post is not found locally
|
|
Err(DatabaseError::NotFound(_)) => return Ok(None),
|
|
Err(other_error) => return Err(other_error.into()),
|
|
};
|
|
let content = get_note_content(&object)?;
|
|
let updated_at = object.updated.unwrap_or(Utc::now());
|
|
let post_data = PostUpdateData { content, 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: &impl GenericClient,
|
|
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(),
|
|
&config.media_dir(),
|
|
profile,
|
|
activity.object,
|
|
).await?;
|
|
Ok(Some(PERSON))
|
|
}
|
|
|
|
pub async fn handle_update(
|
|
config: &Config,
|
|
db_client: &mut impl GenericClient,
|
|
activity: Value,
|
|
) -> HandlerResult {
|
|
let object_type = activity["object"]["type"].as_str()
|
|
.ok_or(ValidationError("unknown object type"))?;
|
|
match object_type {
|
|
NOTE => {
|
|
handle_update_note(db_client, activity).await
|
|
},
|
|
PERSON => {
|
|
handle_update_person(config, db_client, activity).await
|
|
},
|
|
_ => {
|
|
log::warn!("unexpected object type {}", object_type);
|
|
Ok(None)
|
|
},
|
|
}
|
|
}
|