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 into_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 name = person.preferred_username;
134 let display_name: Option<String> = person.name;
135 let bio = get_summary_from_string_or_source(&person.summary, &person.source);
136 let shared_inbox = person.endpoints.shared_inbox.map(|s| s.into());
137 let bot_account = match person.kind {
138 UserTypes::Person => false,
139 UserTypes::Service => true,
142 let slur_regex = &context.settings().slur_regex();
143 check_slurs(&name, slur_regex)?;
144 check_slurs_opt(&display_name, slur_regex)?;
145 check_slurs_opt(&bio, slur_regex)?;
147 check_is_apub_id_valid(person.id.inner(), false, &context.settings())?;
149 let person_form = PersonForm {
151 display_name: Some(display_name),
154 avatar: Some(person.icon.map(|i| i.url.into())),
155 banner: Some(person.image.map(|i| i.url.into())),
156 published: person.published.map(|u| u.naive_local()),
157 updated: person.updated.map(|u| u.naive_local()),
158 actor_id: Some(person.id.into()),
162 bot_account: Some(bot_account),
164 public_key: Some(Some(person.public_key.public_key_pem)),
165 last_refreshed_at: Some(naive_now()),
166 inbox_url: Some(person.inbox.into()),
167 shared_inbox_url: Some(shared_inbox),
168 matrix_user_id: Some(person.matrix_user_id),
170 let person = blocking(context.pool(), move |conn| {
171 DbPerson::upsert(conn, &person_form)
178 impl ActorType for ApubPerson {
179 fn is_local(&self) -> bool {
182 fn actor_id(&self) -> Url {
183 self.actor_id.to_owned().into()
185 fn name(&self) -> String {
189 fn public_key(&self) -> Option<String> {
190 self.public_key.to_owned()
193 fn private_key(&self) -> Option<String> {
194 self.private_key.to_owned()
197 fn inbox_url(&self) -> Url {
198 self.inbox_url.clone().into()
201 fn shared_inbox_url(&self) -> Option<Url> {
202 self.shared_inbox_url.clone().map(|s| s.into())
207 pub(crate) mod tests {
209 use crate::objects::tests::{file_to_json_object, init_context};
210 use lemmy_db_schema::traits::Crud;
211 use serial_test::serial;
213 pub(crate) async fn parse_lemmy_person(context: &LemmyContext) -> ApubPerson {
214 let json = file_to_json_object("assets/lemmy/objects/person.json");
215 let url = Url::parse("https://enterprise.lemmy.ml/u/picard").unwrap();
216 let mut request_counter = 0;
217 let person = ApubPerson::from_apub(json, context, &url, &mut request_counter)
220 assert_eq!(request_counter, 0);
226 async fn test_parse_lemmy_person() {
227 let context = init_context();
228 let person = parse_lemmy_person(&context).await;
230 assert_eq!(person.display_name, Some("Jean-Luc Picard".to_string()));
231 assert!(person.public_key.is_some());
232 assert!(!person.local);
233 assert_eq!(person.bio.as_ref().unwrap().len(), 39);
235 DbPerson::delete(&*context.pool().get().unwrap(), person.id).unwrap();
240 async fn test_parse_pleroma_person() {
241 let context = init_context();
242 let json = file_to_json_object("assets/pleroma/objects/person.json");
243 let url = Url::parse("https://queer.hacktivis.me/users/lanodan").unwrap();
244 let mut request_counter = 0;
245 let person = ApubPerson::from_apub(json, &context, &url, &mut request_counter)
249 assert_eq!(person.actor_id, url.into());
250 assert_eq!(person.name, "lanodan");
251 assert!(person.public_key.is_some());
252 assert!(!person.local);
253 assert_eq!(request_counter, 0);
254 assert_eq!(person.bio.as_ref().unwrap().len(), 873);
256 DbPerson::delete(&*context.pool().get().unwrap(), person.id).unwrap();