2022-12-09 20:15:24 +00:00
|
|
|
use std::path::Path;
|
|
|
|
|
|
|
|
use chrono::Utc;
|
2022-12-09 21:16:53 +00:00
|
|
|
use serde_json::Value;
|
2022-12-06 23:48:24 +00:00
|
|
|
use tokio_postgres::GenericClient;
|
|
|
|
|
|
|
|
use crate::activitypub::{
|
|
|
|
activity::{Activity, Object},
|
2022-12-09 20:15:24 +00:00
|
|
|
actors::types::Actor,
|
|
|
|
fetcher::fetchers::fetch_actor_images,
|
|
|
|
handlers::create::get_note_content,
|
2022-12-06 23:48:24 +00:00
|
|
|
vocabulary::{NOTE, PERSON},
|
|
|
|
};
|
2022-12-09 20:15:24 +00:00
|
|
|
use crate::config::{Config, Instance};
|
|
|
|
use crate::database::DatabaseError;
|
2022-12-06 23:48:24 +00:00
|
|
|
use crate::errors::ValidationError;
|
2022-12-09 20:15:24 +00:00
|
|
|
use crate::models::{
|
|
|
|
posts::queries::{
|
|
|
|
get_post_by_remote_object_id,
|
|
|
|
update_post,
|
|
|
|
},
|
|
|
|
posts::types::PostUpdateData,
|
|
|
|
profiles::queries::{
|
|
|
|
get_profile_by_remote_actor_id,
|
|
|
|
update_profile,
|
|
|
|
},
|
|
|
|
profiles::types::{DbActorProfile, ProfileUpdateData},
|
|
|
|
};
|
|
|
|
use super::{HandlerError, HandlerResult};
|
|
|
|
|
|
|
|
async fn handle_update_note(
|
|
|
|
db_client: &mut impl GenericClient,
|
|
|
|
object: Object,
|
|
|
|
) -> HandlerResult {
|
|
|
|
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))
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Updates remote actor's profile
|
|
|
|
pub async fn update_remote_profile(
|
|
|
|
db_client: &impl GenericClient,
|
|
|
|
instance: &Instance,
|
|
|
|
media_dir: &Path,
|
|
|
|
profile: DbActorProfile,
|
|
|
|
actor: Actor,
|
|
|
|
) -> Result<DbActorProfile, HandlerError> {
|
|
|
|
let actor_old = profile.actor_json.ok_or(HandlerError::LocalObject)?;
|
|
|
|
if actor_old.id != actor.id {
|
|
|
|
log::warn!(
|
|
|
|
"actor ID changed from {} to {}",
|
|
|
|
actor_old.id,
|
|
|
|
actor.id,
|
|
|
|
);
|
|
|
|
};
|
|
|
|
if actor_old.public_key.public_key_pem != actor.public_key.public_key_pem {
|
|
|
|
log::warn!(
|
|
|
|
"actor public key changed from {} to {}",
|
|
|
|
actor_old.public_key.public_key_pem,
|
|
|
|
actor.public_key.public_key_pem,
|
|
|
|
);
|
|
|
|
};
|
|
|
|
let (maybe_avatar, maybe_banner) = fetch_actor_images(
|
|
|
|
instance,
|
|
|
|
&actor,
|
|
|
|
media_dir,
|
|
|
|
profile.avatar_file_name,
|
|
|
|
profile.banner_file_name,
|
|
|
|
).await;
|
|
|
|
let (identity_proofs, payment_options, extra_fields) =
|
|
|
|
actor.parse_attachments();
|
|
|
|
let mut profile_data = ProfileUpdateData {
|
|
|
|
display_name: actor.name.clone(),
|
|
|
|
bio: actor.summary.clone(),
|
|
|
|
bio_source: actor.summary.clone(),
|
|
|
|
avatar: maybe_avatar,
|
|
|
|
banner: maybe_banner,
|
|
|
|
identity_proofs,
|
|
|
|
payment_options,
|
|
|
|
extra_fields,
|
|
|
|
actor_json: Some(actor),
|
|
|
|
};
|
|
|
|
profile_data.clean()?;
|
|
|
|
let profile = update_profile(db_client, &profile.id, profile_data).await?;
|
|
|
|
Ok(profile)
|
|
|
|
}
|
|
|
|
|
|
|
|
async fn handle_update_person(
|
|
|
|
config: &Config,
|
|
|
|
db_client: &impl GenericClient,
|
|
|
|
activity: Activity,
|
|
|
|
) -> HandlerResult {
|
|
|
|
let actor: Actor = serde_json::from_value(activity.object)
|
|
|
|
.map_err(|_| ValidationError("invalid actor data"))?;
|
|
|
|
if actor.id != activity.actor {
|
|
|
|
return Err(ValidationError("actor ID mismatch").into());
|
|
|
|
};
|
|
|
|
let profile = get_profile_by_remote_actor_id(
|
|
|
|
db_client,
|
|
|
|
&actor.id,
|
|
|
|
).await?;
|
|
|
|
update_remote_profile(
|
|
|
|
db_client,
|
|
|
|
&config.instance(),
|
|
|
|
&config.media_dir(),
|
|
|
|
profile,
|
|
|
|
actor,
|
|
|
|
).await?;
|
|
|
|
Ok(Some(PERSON))
|
|
|
|
}
|
2022-12-06 23:48:24 +00:00
|
|
|
|
|
|
|
pub async fn handle_update(
|
|
|
|
config: &Config,
|
|
|
|
db_client: &mut impl GenericClient,
|
2022-12-09 21:16:53 +00:00
|
|
|
activity: Value,
|
2022-12-06 23:48:24 +00:00
|
|
|
) -> HandlerResult {
|
2022-12-09 21:16:53 +00:00
|
|
|
let activity: Activity = serde_json::from_value(activity)
|
|
|
|
.map_err(|_| ValidationError("unexpected activity structure"))?;
|
2022-12-06 23:48:24 +00:00
|
|
|
let object_type = activity.object["type"].as_str()
|
|
|
|
.ok_or(ValidationError("unknown object type"))?;
|
|
|
|
match object_type {
|
|
|
|
NOTE => {
|
|
|
|
let object: Object = serde_json::from_value(activity.object)
|
|
|
|
.map_err(|_| ValidationError("invalid object"))?;
|
|
|
|
handle_update_note(db_client, object).await
|
|
|
|
},
|
|
|
|
PERSON => {
|
|
|
|
handle_update_person(config, db_client, activity).await
|
|
|
|
},
|
|
|
|
_ => {
|
|
|
|
log::warn!("unexpected object type {}", object_type);
|
|
|
|
Ok(None)
|
|
|
|
},
|
|
|
|
}
|
|
|
|
}
|