lemmy/server/src/apub/community.rs

120 lines
3.9 KiB
Rust
Raw Normal View History

use crate::apub::{create_apub_response, make_apub_endpoint, EndpointType};
use crate::convert_datetime;
use crate::db::community::Community;
use crate::db::community_view::CommunityFollowerView;
use crate::db::establish_unpooled_connection;
2020-03-14 00:05:42 +00:00
use crate::db::post_view::{PostQueryBuilder, PostView};
2020-03-19 01:16:17 +00:00
use activitystreams::collection::OrderedCollection;
use activitystreams::{
2020-03-19 01:16:17 +00:00
actor::{properties::ApActorProperties, Group},
collection::UnorderedCollection,
context,
ext::Extensible,
object::properties::ObjectProperties,
};
use actix_web::body::Body;
use actix_web::web::Path;
use actix_web::HttpResponse;
2020-03-16 17:30:25 +00:00
use actix_web::{web, Result};
use diesel::r2d2::{ConnectionManager, Pool};
use diesel::PgConnection;
use failure::Error;
use serde::Deserialize;
2020-03-16 17:30:25 +00:00
#[derive(Deserialize)]
pub struct CommunityQuery {
community_name: String,
}
2020-03-16 17:30:25 +00:00
pub async fn get_apub_community(
info: Path<CommunityQuery>,
db: web::Data<Pool<ConnectionManager<PgConnection>>>,
) -> Result<HttpResponse<Body>, Error> {
let community = Community::read_from_name(&&db.get()?, info.community_name.to_owned())?;
let base_url = make_apub_endpoint(EndpointType::Community, &community.name);
2020-03-16 17:30:25 +00:00
let mut group = Group::default();
let oprops: &mut ObjectProperties = group.as_mut();
2020-03-16 17:30:25 +00:00
oprops
.set_context_xsd_any_uri(context())?
.set_id(base_url.to_owned())?
.set_name_xsd_string(community.title.to_owned())?
.set_published(convert_datetime(community.published))?
.set_attributed_to_xsd_any_uri(make_apub_endpoint(
EndpointType::User,
&community.creator_id.to_string(),
))?;
2020-03-16 17:30:25 +00:00
if let Some(u) = community.updated.to_owned() {
oprops.set_updated(convert_datetime(u))?;
}
2020-03-16 17:30:25 +00:00
if let Some(d) = community.description {
oprops.set_summary_xsd_string(d)?;
}
2020-03-14 00:05:42 +00:00
2020-03-19 01:16:17 +00:00
let mut actor_props = ApActorProperties::default();
actor_props
2020-03-16 17:30:25 +00:00
.set_inbox(format!("{}/inbox", &base_url))?
.set_outbox(format!("{}/outbox", &base_url))?
.set_followers(format!("{}/followers", &base_url))?;
2020-03-14 00:05:42 +00:00
2020-03-19 01:16:17 +00:00
Ok(create_apub_response(&group.extend(actor_props)))
}
pub async fn get_apub_community_followers(
info: Path<CommunityQuery>,
2020-03-16 17:30:25 +00:00
db: web::Data<Pool<ConnectionManager<PgConnection>>>,
) -> Result<HttpResponse<Body>, Error> {
2020-03-16 17:30:25 +00:00
let community = Community::read_from_name(&&db.get()?, info.community_name.to_owned())?;
let base_url = make_apub_endpoint(EndpointType::Community, &community.name);
2020-03-16 17:30:25 +00:00
let connection = establish_unpooled_connection();
//As we are an object, we validated that the community id was valid
let community_followers =
CommunityFollowerView::for_community(&connection, community.id).unwrap();
let mut collection = UnorderedCollection::default();
let oprops: &mut ObjectProperties = collection.as_mut();
oprops
.set_context_xsd_any_uri(context())?
.set_id(base_url)?;
collection
.collection_props
.set_total_items(community_followers.len() as u64)?;
2020-03-19 01:16:17 +00:00
Ok(create_apub_response(&collection))
2020-03-14 00:05:42 +00:00
}
pub async fn get_apub_community_outbox(
info: Path<CommunityQuery>,
2020-03-16 17:30:25 +00:00
db: web::Data<Pool<ConnectionManager<PgConnection>>>,
2020-03-14 00:05:42 +00:00
) -> Result<HttpResponse<Body>, Error> {
2020-03-16 17:30:25 +00:00
let community = Community::read_from_name(&&db.get()?, info.community_name.to_owned())?;
let base_url = make_apub_endpoint(EndpointType::Community, &community.name);
2020-03-14 00:05:42 +00:00
2020-03-16 17:30:25 +00:00
let connection = establish_unpooled_connection();
//As we are an object, we validated that the community id was valid
let community_posts: Vec<PostView> = PostQueryBuilder::create(&connection)
.for_community_id(community.id)
.list()
.unwrap();
let mut collection = OrderedCollection::default();
let oprops: &mut ObjectProperties = collection.as_mut();
oprops
.set_context_xsd_any_uri(context())?
.set_id(base_url)?;
collection
.collection_props
.set_many_items_object_boxs(
community_posts
.iter()
.map(|c| c.as_page().unwrap())
.collect(),
)?
.set_total_items(community_posts.len() as u64)?;
2020-03-19 01:16:17 +00:00
Ok(create_apub_response(&collection))
}