From 5e463e2cc9335b308b024da9e9f84e2945d00175 Mon Sep 17 00:00:00 2001 From: Kitaiti Makoto Date: Tue, 22 Mar 2022 00:20:53 +0900 Subject: [PATCH] Implement FromId07 for Post --- plume-models/src/posts.rs | 193 +++++++++++++++++++++++++++++++++++++- 1 file changed, 189 insertions(+), 4 deletions(-) diff --git a/plume-models/src/posts.rs b/plume-models/src/posts.rs index 1e29f243..8e374614 100644 --- a/plume-models/src/posts.rs +++ b/plume-models/src/posts.rs @@ -13,7 +13,7 @@ use activitystreams::{ activity::{Create as Create07, Delete as Delete07, Update as Update07}, base::{AnyBase, Base}, iri_string::types::IriString, - link as link07, + link::{self as link07, kind::MentionType}, object::{ApObject, Article as Article07, Image as Image07, Tombstone as Tombstone07}, prelude::*, time::OffsetDateTime, @@ -23,10 +23,11 @@ use diesel::{self, BelongingToDsl, ExpressionMethods, QueryDsl, RunQueryDsl}; use once_cell::sync::Lazy; use plume_common::{ activity_pub::{ - inbox::{AsActor, AsObject, FromId}, + inbox::{AsActor, AsObject, AsObject07, FromId, FromId07}, sign::Signer, - Hashtag, Hashtag07, Id, IntoId, Licensed, Licensed07, LicensedArticle as LicensedArticle07, - Source, SourceProperty, PUBLIC_VISIBILITY, + Hashtag, Hashtag07, HashtagType07, Id, IntoId, Licensed, Licensed07, + LicensedArticle as LicensedArticle07, Source, SourceProperty, ToAsString, ToAsUri, + PUBLIC_VISIBILITY, }, utils::{iri_percent_encode_seg, md_to_html}, }; @@ -1013,6 +1014,190 @@ impl FromId for Post { } } +impl FromId07 for Post { + type Error = Error; + type Object = LicensedArticle07; + + fn from_db07(conn: &DbConn, id: &str) -> Result { + Self::find_by_ap_url(conn, id) + } + + fn from_activity07(conn: &DbConn, article: LicensedArticle07) -> Result { + let license = article.ext_one.license; + let source = article.ext_two.source.content; + let article = article.inner; + + let (blog, authors) = article + .attributed_to() + .ok_or(Error::MissingApProperty) + .iter() + .fold((None, vec![]), |(blog, mut authors), link| { + let url = link.to_as_uri().expect("Exists"); + match User::from_id(conn, &url, None, CONFIG.proxy()) { + Ok(u) => { + authors.push(u); + (blog, authors) + } + Err(_) => ( + blog.or_else(|| Blog::from_id(conn, &url, None, CONFIG.proxy()).ok()), + authors, + ), + } + }); + + let cover = article.icon().and_then(|icon| { + icon.iter().next().and_then(|img| { + let image: Image07 = img.extend().ok()??; + Media::from_activity07(conn, &image).ok().map(|m| m.id) + }) + }); + + let title = article + .name() + .and_then(|name| name.to_as_string()) + .ok_or(Error::MissingApProperty)?; + let ap_url = article + .url() + .and_then(|url| { + url.to_as_uri().or_else(|| { + AnyBase::from_extended(article) + .ok()? + .id() + .map(|id| id.to_string()) + }) + }) + .ok_or(Error::MissingApProperty)?; + let post = Post::from_db07(conn, &ap_url) + .and_then(|mut post| { + let mut updated = false; + + let slug = Self::slug(&title); + let content = SafeString::new( + &article + .content() + .and_then(|content| content.to_as_string()) + .ok_or(Error::MissingApProperty)?, + ); + let subtitle = article + .summary() + .and_then(|summary| summary.to_as_string()) + .ok_or(Error::MissingApProperty)?; + + if post.slug != slug { + post.slug = slug.to_string(); + updated = true; + } + if post.title != title { + post.title = title.clone(); + updated = true; + } + if post.content != content { + post.content = content; + updated = true; + } + if post.license != license { + post.license = license.clone(); + updated = true; + } + if post.subtitle != subtitle { + post.subtitle = subtitle; + updated = true; + } + if post.source != source { + post.source = source; + updated = true; + } + if post.cover_id != cover { + post.cover_id = cover; + updated = true; + } + + if updated { + post.update(conn)?; + } + + Ok(post) + }) + .or_else(|_| { + Post::insert( + conn, + NewPost { + blog_id: blog.ok_or(Error::NotFound)?.id, + slug: Self::slug(&title).to_string(), + title, + content: SafeString::new( + &article + .content() + .and_then(|content| content.to_as_string()) + .ok_or(Error::MissingApProperty)?, + ), + published: true, + license, + // FIXME: This is wrong: with this logic, we may use the display URL as the AP ID. We need two different fields + ap_url, + creation_date: article.published().map(|published| { + let timestamp_secs = published.unix_timestamp(); + let timestamp_nanos = published.unix_timestamp_nanos() + - (timestamp_secs as i128) * 1000i128 * 1000i128 * 1000i128; + NaiveDateTime::from_timestamp(timestamp_secs, timestamp_nanos as u32) + }), + subtitle: article + .summary() + .and_then(|summary| summary.to_as_string()) + .ok_or(Error::MissingApProperty)?, + source: source, + cover_id: cover, + }, + ) + .and_then(|post| { + for author in authors { + PostAuthor::insert( + conn, + NewPostAuthor { + post_id: post.id, + author_id: author.id, + }, + )?; + } + + Ok(post) + }) + })?; + + // save mentions and tags + let mut hashtags = md_to_html(&post.source, None, false, None) + .2 + .into_iter() + .collect::>(); + if let Some(tags) = article.tag() { + for tag in tags.iter() { + tag.extend::() + .map(|mention| { + mention.map(|m| Mention::from_activity07(conn, &m, post.id, true, true)) + }) + .ok(); + + tag.extend::() + .and_then(|hashtag| { + Ok(hashtag.and_then(|t| { + let tag_name = t.name?.as_str(); + Tag::from_activity07(conn, &t, post.id, hashtags.remove(tag_name)).ok() + })) + }) + .ok(); + } + } + + Timeline::add_to_all_timelines(conn, &post, Kind::Original)?; + + Ok(post) + } + + fn get_sender07() -> &'static dyn Signer { + Instance::get_local_instance_user().expect("Failed to get local instance user") + } +} + impl AsObject for Post { type Error = Error; type Output = Post;