2 check_is_apub_id_valid,
4 objects::get_summary_from_string_or_source,
6 objects::person::{Person, UserTypes},
11 use activitystreams::{actor::Endpoints, object::kind::ImageType};
12 use chrono::NaiveDateTime;
13 use lemmy_api_common::blocking;
16 traits::{ActorType, ApubObject},
17 values::MediaTypeMarkdown,
18 verify::verify_domains_match,
20 use lemmy_db_schema::{
22 source::person::{Person as DbPerson, PersonForm},
25 utils::{check_slurs, check_slurs_opt, convert_datetime, markdown_to_html},
28 use lemmy_websocket::LemmyContext;
32 #[derive(Clone, Debug, PartialEq)]
33 pub struct ApubPerson(DbPerson);
35 impl Deref for ApubPerson {
36 type Target = DbPerson;
37 fn deref(&self) -> &Self::Target {
42 impl From<DbPerson> for ApubPerson {
43 fn from(p: DbPerson) -> Self {
48 #[async_trait::async_trait(?Send)]
49 impl ApubObject for ApubPerson {
50 type DataType = LemmyContext;
51 type ApubType = Person;
52 type TombstoneType = ();
54 fn last_refreshed_at(&self) -> Option<NaiveDateTime> {
55 Some(self.last_refreshed_at)
58 async fn read_from_apub_id(
60 context: &LemmyContext,
61 ) -> Result<Option<Self>, LemmyError> {
63 blocking(context.pool(), move |conn| {
64 DbPerson::read_from_apub_id(conn, object_id)
71 async fn delete(self, context: &LemmyContext) -> Result<(), LemmyError> {
72 blocking(context.pool(), move |conn| {
73 DbPerson::update_deleted(conn, self.id, true)
79 async fn to_apub(&self, _pool: &LemmyContext) -> Result<Person, LemmyError> {
80 let kind = if self.bot_account {
85 let source = self.bio.clone().map(|bio| Source {
87 media_type: MediaTypeMarkdown::Markdown,
89 let icon = self.avatar.clone().map(|url| ImageObject {
90 kind: ImageType::Image,
93 let image = self.banner.clone().map(|url| ImageObject {
94 kind: ImageType::Image,
100 id: ObjectId::new(self.actor_id.clone()),
101 preferred_username: self.name.clone(),
102 name: self.display_name.clone(),
103 summary: self.bio.as_ref().map(|b| markdown_to_html(b)),
107 matrix_user_id: self.matrix_user_id.clone(),
108 published: Some(convert_datetime(self.published)),
109 outbox: generate_outbox_url(&self.actor_id)?.into(),
110 endpoints: Endpoints {
111 shared_inbox: self.shared_inbox_url.clone().map(|s| s.into()),
114 public_key: self.get_public_key()?,
115 updated: self.updated.map(convert_datetime),
116 unparsed: Default::default(),
117 inbox: self.inbox_url.clone().into(),
122 fn to_tombstone(&self) -> Result<(), LemmyError> {
128 context: &LemmyContext,
129 expected_domain: &Url,
130 _request_counter: &mut i32,
131 ) -> Result<ApubPerson, LemmyError> {
132 verify_domains_match(person.id.inner(), expected_domain)?;
133 let actor_id = Some(person.id.clone().into());
134 let name = person.preferred_username.clone();
135 let display_name: Option<String> = person.name.clone();
136 let bio = get_summary_from_string_or_source(&person.summary, &person.source);
137 let shared_inbox = person.endpoints.shared_inbox.clone().map(|s| s.into());
138 let bot_account = match person.kind {
139 UserTypes::Person => false,
140 UserTypes::Service => true,
143 let slur_regex = &context.settings().slur_regex();
144 check_slurs(&name, slur_regex)?;
145 check_slurs_opt(&display_name, slur_regex)?;
146 check_slurs_opt(&bio, slur_regex)?;
148 check_is_apub_id_valid(person.id.inner(), false, &context.settings())?;
150 let person_form = PersonForm {
152 display_name: Some(display_name),
155 avatar: Some(person.icon.clone().map(|i| i.url.into())),
156 banner: Some(person.image.clone().map(|i| i.url.into())),
157 published: person.published.map(|u| u.clone().naive_local()),
158 updated: person.updated.map(|u| u.clone().naive_local()),
163 bot_account: Some(bot_account),
165 public_key: Some(Some(person.public_key.public_key_pem.clone())),
166 last_refreshed_at: Some(naive_now()),
167 inbox_url: Some(person.inbox.to_owned().into()),
168 shared_inbox_url: Some(shared_inbox),
169 matrix_user_id: Some(person.matrix_user_id.clone()),
171 let person = blocking(context.pool(), move |conn| {
172 DbPerson::upsert(conn, &person_form)
179 impl ActorType for ApubPerson {
180 fn is_local(&self) -> bool {
183 fn actor_id(&self) -> Url {
184 self.actor_id.to_owned().into()
186 fn name(&self) -> String {
190 fn public_key(&self) -> Option<String> {
191 self.public_key.to_owned()
194 fn private_key(&self) -> Option<String> {
195 self.private_key.to_owned()
198 fn inbox_url(&self) -> Url {
199 self.inbox_url.clone().into()
202 fn shared_inbox_url(&self) -> Option<Url> {
203 self.shared_inbox_url.clone().map(|s| s.into())
208 pub(crate) mod tests {
210 use crate::objects::tests::{file_to_json_object, init_context};
211 use lemmy_db_schema::traits::Crud;
212 use serial_test::serial;
214 pub(crate) async fn parse_lemmy_person(context: &LemmyContext) -> ApubPerson {
215 let json = file_to_json_object("assets/lemmy/objects/person.json");
216 let url = Url::parse("https://enterprise.lemmy.ml/u/picard").unwrap();
217 let mut request_counter = 0;
218 let person = ApubPerson::from_apub(&json, context, &url, &mut request_counter)
221 assert_eq!(request_counter, 0);
227 async fn test_parse_lemmy_person() {
228 let context = init_context();
229 let person = parse_lemmy_person(&context).await;
231 assert_eq!(person.display_name, Some("Jean-Luc Picard".to_string()));
232 assert!(person.public_key.is_some());
233 assert!(!person.local);
234 assert_eq!(person.bio.as_ref().unwrap().len(), 39);
236 DbPerson::delete(&*context.pool().get().unwrap(), person.id).unwrap();
241 async fn test_parse_pleroma_person() {
242 let context = init_context();
243 let json = file_to_json_object("assets/pleroma/objects/person.json");
244 let url = Url::parse("https://queer.hacktivis.me/users/lanodan").unwrap();
245 let mut request_counter = 0;
246 let person = ApubPerson::from_apub(&json, &context, &url, &mut request_counter)
250 assert_eq!(person.actor_id, url.into());
251 assert_eq!(person.name, "lanodan");
252 assert!(person.public_key.is_some());
253 assert!(!person.local);
254 assert_eq!(request_counter, 0);
255 assert_eq!(person.bio.as_ref().unwrap().len(), 873);
257 DbPerson::delete(&*context.pool().get().unwrap(), person.id).unwrap();