lemmy/server/src/apub/user.rs

use crate::{
  apub::{
    activities::send_activity,
    create_apub_response,
    extensions::signatures::PublicKey,
    ActorType,
    FromApub,
    PersonExt,
    ToApub,
  },
  convert_datetime,
  db::{
    activity::insert_activity,
    user::{UserForm, User_},
  },
  naive_now,
  routes::DbPoolParam,
};
use activitystreams::{
  activity::{Follow, Undo},
  actor::{properties::ApActorProperties, Person},
  context,
  endpoint::EndpointProperties,
  object::{properties::ObjectProperties, AnyImage, Image, Tombstone},
};
use activitystreams_ext::Ext2;
use actix_web::{body::Body, web::Path, HttpResponse, Result};
use diesel::PgConnection;
use failure::Error;
use serde::Deserialize;

#[derive(Deserialize)]
pub struct UserQuery {
  user_name: String,
}

impl ToApub for User_ {
  type Response = PersonExt;

  // Turn a Lemmy Community into an ActivityPub group that can be sent out over the network.
  fn to_apub(&self, _conn: &PgConnection) -> Result<PersonExt, Error> {
    // TODO go through all these to_string and to_owned()
    let mut person = Person::default();
    let oprops: &mut ObjectProperties = person.as_mut();
    oprops
      .set_context_xsd_any_uri(context())?
      .set_id(self.actor_id.to_string())?
      .set_name_xsd_string(self.name.to_owned())?
      .set_published(convert_datetime(self.published))?;

    if let Some(u) = self.updated {
      oprops.set_updated(convert_datetime(u))?;
    }

    if let Some(i) = &self.preferred_username {
      oprops.set_name_xsd_string(i.to_owned())?;
    }

    if let Some(avatar_url) = &self.avatar {
      let mut image = Image::new();
      image
        .object_props
        .set_url_xsd_any_uri(avatar_url.to_owned())?;
      let any_image = AnyImage::from_concrete(image)?;
      oprops.set_icon_any_image(any_image)?;
    }

    let mut endpoint_props = EndpointProperties::default();

    endpoint_props.set_shared_inbox(self.get_shared_inbox_url())?;

    let mut actor_props = ApActorProperties::default();

    actor_props
      .set_inbox(self.get_inbox_url())?
      .set_outbox(self.get_outbox_url())?
      .set_endpoints(endpoint_props)?
      .set_followers(self.get_followers_url())?
      .set_following(self.get_following_url())?
      .set_liked(self.get_liked_url())?;

    Ok(Ext2::new(person, actor_props, self.get_public_key_ext()))
  }
  fn to_tombstone(&self) -> Result<Tombstone, Error> {
    unimplemented!()
  }
}

impl ActorType for User_ {
  fn actor_id(&self) -> String {
    self.actor_id.to_owned()
  }

  fn public_key(&self) -> String {
    self.public_key.to_owned().unwrap()
  }

  fn private_key(&self) -> String {
    self.private_key.to_owned().unwrap()
  }

  /// As a given local user, send out a follow request to a remote community.
  fn send_follow(&self, follow_actor_id: &str, conn: &PgConnection) -> Result<(), Error> {
    let mut follow = Follow::new();

    let id = format!("{}/follow/{}", self.actor_id, uuid::Uuid::new_v4());

    follow
      .object_props
      .set_context_xsd_any_uri(context())?
      .set_id(id)?;
    follow
      .follow_props
      .set_actor_xsd_any_uri(self.actor_id.to_owned())?
      .set_object_xsd_any_uri(follow_actor_id)?;
    let to = format!("{}/inbox", follow_actor_id);

    insert_activity(&conn, self.id, &follow, true)?;

    send_activity(&follow, self, vec![to])?;
    Ok(())
  }

  fn send_unfollow(&self, follow_actor_id: &str, conn: &PgConnection) -> Result<(), Error> {
    let mut follow = Follow::new();

    let id = format!("{}/follow/{}", self.actor_id, uuid::Uuid::new_v4());

    follow
      .object_props
      .set_context_xsd_any_uri(context())?
      .set_id(id)?;
    follow
      .follow_props
      .set_actor_xsd_any_uri(self.actor_id.to_owned())?
      .set_object_xsd_any_uri(follow_actor_id)?;
    let to = format!("{}/inbox", follow_actor_id);

    // TODO
    // Undo that fake activity
    let undo_id = format!("{}/undo/follow/{}", self.actor_id, uuid::Uuid::new_v4());
    let mut undo = Undo::default();

    undo
      .object_props
      .set_context_xsd_any_uri(context())?
      .set_id(undo_id)?;

    undo
      .undo_props
      .set_actor_xsd_any_uri(self.actor_id.to_owned())?
      .set_object_base_box(follow)?;

    insert_activity(&conn, self.id, &undo, true)?;

    send_activity(&undo, self, vec![to])?;
    Ok(())
  }

  fn send_delete(&self, _creator: &User_, _conn: &PgConnection) -> Result<(), Error> {
    unimplemented!()
  }

  fn send_undo_delete(&self, _creator: &User_, _conn: &PgConnection) -> Result<(), Error> {
    unimplemented!()
  }

  fn send_remove(&self, _creator: &User_, _conn: &PgConnection) -> Result<(), Error> {
    unimplemented!()
  }

  fn send_undo_remove(&self, _creator: &User_, _conn: &PgConnection) -> Result<(), Error> {
    unimplemented!()
  }

  fn send_accept_follow(&self, _follow: &Follow, _conn: &PgConnection) -> Result<(), Error> {
    unimplemented!()
  }

  fn get_follower_inboxes(&self, _conn: &PgConnection) -> Result<Vec<String>, Error> {
    unimplemented!()
  }
}

impl FromApub for UserForm {
  type ApubType = PersonExt;
  /// Parse an ActivityPub person received from another instance into a Lemmy user.
  fn from_apub(person: &PersonExt, _conn: &PgConnection) -> Result<Self, Error> {
    let oprops = &person.inner.object_props;
    let aprops = &person.ext_one;
    let public_key: &PublicKey = &person.ext_two.public_key;

    let avatar = match oprops.get_icon_any_image() {
      Some(any_image) => any_image
        .to_owned()
        .into_concrete::<Image>()?
        .object_props
        .get_url_xsd_any_uri()
        .map(|u| u.to_string()),
      None => None,
    };

    Ok(UserForm {
      name: oprops.get_name_xsd_string().unwrap().to_string(),
      preferred_username: aprops.get_preferred_username().map(|u| u.to_string()),
      password_encrypted: "".to_string(),
      admin: false,
      banned: false,
      email: None,
      avatar,
      updated: oprops
        .get_updated()
        .map(|u| u.as_ref().to_owned().naive_local()),
      show_nsfw: false,
      theme: "".to_string(),
      default_sort_type: 0,
      default_listing_type: 0,
      lang: "".to_string(),
      show_avatars: false,
      send_notifications_to_email: false,
      matrix_user_id: None,
      actor_id: oprops.get_id().unwrap().to_string(),
      bio: oprops.get_summary_xsd_string().map(|s| s.to_string()),
      local: false,
      private_key: None,
      public_key: Some(public_key.to_owned().public_key_pem),
      last_refreshed_at: Some(naive_now()),
    })
  }
}

/// Return the user json over HTTP.
pub async fn get_apub_user_http(
  info: Path<UserQuery>,
  db: DbPoolParam,
) -> Result<HttpResponse<Body>, Error> {
  let user = User_::find_by_email_or_username(&&db.get()?, &info.user_name)?;
  let u = user.to_apub(&db.get().unwrap())?;
  Ok(create_apub_response(&u))
}