server/src/apub/user.rs

   1 use crate::apub::signatures::PublicKey;
   2 use crate::apub::{create_apub_response, PersonExt};
   3 use crate::db::user::{UserForm, User_};
   4 use crate::{convert_datetime, naive_now};
   5 use activitystreams::{
   6   actor::{properties::ApActorProperties, Person},
   7   context,
   8   ext::Extensible,
   9   object::properties::ObjectProperties,
  10 };
  11 use actix_web::body::Body;
  12 use actix_web::web::Path;
  13 use actix_web::HttpResponse;
  14 use actix_web::{web, Result};
  15 use diesel::r2d2::{ConnectionManager, Pool};
  16 use diesel::PgConnection;
  17 use failure::Error;
  18 use serde::Deserialize;
  19
  20 #[derive(Deserialize)]
  21 pub struct UserQuery {
  22   user_name: String,
  23 }
  24
  25 pub async fn get_apub_user(
  26   info: Path<UserQuery>,
  27   db: web::Data<Pool<ConnectionManager<PgConnection>>>,
  28 ) -> Result<HttpResponse<Body>, Error> {
  29   let user = User_::find_by_email_or_username(&&db.get()?, &info.user_name)?;
  30
  31   let mut person = Person::default();
  32   let oprops: &mut ObjectProperties = person.as_mut();
  33   oprops
  34     .set_context_xsd_any_uri(context())?
  35     .set_id(user.actor_id.to_string())?
  36     .set_name_xsd_string(user.name.to_owned())?
  37     .set_published(convert_datetime(user.published))?;
  38
  39   if let Some(u) = user.updated {
  40     oprops.set_updated(convert_datetime(u))?;
  41   }
  42
  43   if let Some(i) = &user.preferred_username {
  44     oprops.set_name_xsd_string(i.to_owned())?;
  45   }
  46
  47   let mut actor_props = ApActorProperties::default();
  48
  49   actor_props
  50     .set_inbox(format!("{}/inbox", &user.actor_id))?
  51     .set_outbox(format!("{}/outbox", &user.actor_id))?
  52     .set_following(format!("{}/following", &user.actor_id))?
  53     .set_liked(format!("{}/liked", &user.actor_id))?;
  54
  55   let public_key = PublicKey {
  56     id: format!("{}#main-key", user.actor_id),
  57     owner: user.actor_id.to_owned(),
  58     public_key_pem: user.public_key.unwrap(),
  59   };
  60
  61   Ok(create_apub_response(
  62     &person.extend(actor_props).extend(public_key.to_ext()),
  63   ))
  64 }
  65
  66 impl UserForm {
  67   pub fn from_person(person: &PersonExt) -> Result<Self, Error> {
  68     let oprops = &person.base.base.object_props;
  69     let aprops = &person.base.extension;
  70     let public_key: &PublicKey = &person.extension.public_key;
  71
  72     Ok(UserForm {
  73       name: oprops.get_name_xsd_string().unwrap().to_string(),
  74       preferred_username: aprops.get_preferred_username().map(|u| u.to_string()),
  75       password_encrypted: "".to_string(),
  76       admin: false,
  77       banned: false,
  78       email: None,
  79       avatar: None, // -> icon, image
  80       updated: oprops
  81         .get_updated()
  82         .map(|u| u.as_ref().to_owned().naive_local()),
  83       show_nsfw: false,
  84       theme: "".to_string(),
  85       default_sort_type: 0,
  86       default_listing_type: 0,
  87       lang: "".to_string(),
  88       show_avatars: false,
  89       send_notifications_to_email: false,
  90       matrix_user_id: None,
  91       actor_id: oprops.get_id().unwrap().to_string(),
  92       bio: oprops.get_summary_xsd_string().map(|s| s.to_string()),
  93       local: false,
  94       private_key: None,
  95       public_key: Some(public_key.to_owned().public_key_pem),
  96       last_refreshed_at: Some(naive_now()),
  97     })
  98   }
  99 }