feat: moved object creation out of entity

This commit is contained in:
əlemi 2024-03-23 06:10:53 +01:00
parent 05245c2776
commit c361528b2f
Signed by: alemi
GPG key ID: A4895B84D311642C
2 changed files with 160 additions and 102 deletions

View file

@ -12,7 +12,7 @@ use std::sync::Arc;
use axum::{extract::{Path, State}, http::StatusCode}; use axum::{extract::{Path, State}, http::StatusCode};
use sea_orm::{DatabaseConnection, EntityTrait}; use sea_orm::{DatabaseConnection, EntityTrait};
use crate::{activitystream::Base, model::user, server::Context}; use crate::{activitystream::{object::{actor::ActorMut, collection::{CollectionMut, CollectionType}, document::{DocumentMut, DocumentType}, ObjectMut}, BaseMut, Node}, model::{self, user}, server::Context, url};
use super::{jsonld::LD, JsonLD}; use super::{jsonld::LD, JsonLD};
@ -21,8 +21,54 @@ pub async fn list(State(_db) : State<Arc<DatabaseConnection>>) -> Result<JsonLD<
} }
pub async fn view(State(ctx) : State<Context>, Path(id): Path<String>) -> Result<JsonLD<serde_json::Value>, StatusCode> { pub async fn view(State(ctx) : State<Context>, Path(id): Path<String>) -> Result<JsonLD<serde_json::Value>, StatusCode> {
match user::Entity::find_by_id(ctx.uid(id)).one(ctx.db()).await { match user::Entity::find_by_id(ctx.uid(id.clone()))
Ok(Some(user)) => Ok(JsonLD(user.underlying_json_object().ld_context())), .find_also_related(model::config::Entity)
.one(ctx.db()).await
{
// local user
Ok(Some((user, Some(cfg)))) => {
Ok(JsonLD(serde_json::Value::new_object()
.set_id(Some(&user.id))
.set_actor_type(Some(user.actor_type))
.set_name(user.name.as_deref())
.set_summary(user.summary.as_deref())
.set_icon(Node::maybe_object(user.icon.map(|i|
serde_json::Value::new_object()
.set_document_type(Some(DocumentType::Image))
.set_url(Node::link(i.clone()))
)))
.set_image(Node::maybe_object(user.image.map(|i|
serde_json::Value::new_object()
.set_document_type(Some(DocumentType::Image))
.set_url(Node::link(i.clone()))
)))
.set_published(Some(user.created))
.set_preferred_username(Some(&user.preferred_username))
.set_inbox(Node::link(url!(ctx, "/users/{id}/inbox")))
.set_outbox(Node::link(url!(ctx, "/users/{id}/outbox")))
.set_following(Node::object(
serde_json::Value::new_object()
.set_id(Some(&url!(ctx, "/users/{id}/following")))
.set_collection_type(Some(CollectionType::OrderedCollection))
.set_total_items(if cfg.show_following_count { Some(0 /* user.following_count TODO */) } else { None })
.set_first(Node::link(url!(ctx, "/users/{id}/following?page=true")))
))
.set_followers(Node::object(
serde_json::Value::new_object()
.set_id(Some(&url!(ctx, "/users/{id}/followers")))
.set_collection_type(Some(CollectionType::OrderedCollection))
.set_total_items(if cfg.show_followers_count { Some(0 /* user.followers_count TODO */) } else { None })
.set_first(Node::link(url!(ctx, "/users/{id}/followers?page=true")))
))
// .set_public_key(user.public_key) // TODO
.set_discoverable(Some(true))
.set_endpoints(None)
))
},
// remote user
Ok(Some((_user, None))) => {
Err(StatusCode::NOT_IMPLEMENTED)
},
Ok(None) => Err(StatusCode::NOT_FOUND), Ok(None) => Err(StatusCode::NOT_FOUND),
Err(e) => { Err(e) => {
tracing::error!("error querying for user: {e}"); tracing::error!("error querying for user: {e}");

View file

@ -69,107 +69,119 @@ impl Related<super::object::Entity> for Entity {
} }
} }
impl Related<super::config::Entity> for Entity {
fn to() -> RelationDef {
Relation::Config.def()
}
}
impl Related<super::credential::Entity> for Entity {
fn to() -> RelationDef {
Relation::Credential.def()
}
}
impl ActiveModelBehavior for ActiveModel {} impl ActiveModelBehavior for ActiveModel {}
impl crate::activitystream::Base for Model { // impl crate::activitystream::Base for Model {
fn id(&self) -> Option<&str> { // fn id(&self) -> Option<&str> {
Some(&self.id) // Some(&self.id)
} // }
//
fn base_type(&self) -> Option<BaseType> { // fn base_type(&self) -> Option<BaseType> {
Some(BaseType::Object(ObjectType::Actor(self.actor_type))) // Some(BaseType::Object(ObjectType::Actor(self.actor_type)))
} // }
//
fn underlying_json_object(self) -> serde_json::Value { // fn underlying_json_object(self) -> serde_json::Value {
serde_json::Value::new_object() // serde_json::Value::new_object()
.set_id(Some(&self.id)) // .set_id(Some(&self.id))
.set_actor_type(Some(self.actor_type)) // .set_actor_type(Some(self.actor_type))
.set_name(self.name.as_deref()) // .set_name(self.name.as_deref())
.set_summary(self.summary.as_deref()) // .set_summary(self.summary.as_deref())
.set_icon(self.icon()) // .set_icon(self.icon())
.set_image(self.image()) // .set_image(self.image())
.set_published(Some(self.created)) // .set_published(Some(self.created))
.set_preferred_username(Some(&self.preferred_username)) // .set_preferred_username(Some(&self.preferred_username))
.set_inbox(self.inbox()) // .set_inbox(self.inbox())
.set_outbox(self.outbox()) // .set_outbox(self.outbox())
.set_following(self.following()) // .set_following(self.following())
.set_followers(self.followers()) // .set_followers(self.followers())
.set_public_key(self.public_key()) // .set_public_key(self.public_key())
.set_discoverable(Some(true)) // .set_discoverable(Some(true))
.set_endpoints(None) // TODO dirty fix to put an empty object // .set_endpoints(None) // TODO dirty fix to put an empty object
} // }
} // }
//
impl crate::activitystream::object::Object for Model { // impl crate::activitystream::object::Object for Model {
fn name(&self) -> Option<&str> { // fn name(&self) -> Option<&str> {
self.name.as_deref() // self.name.as_deref()
} // }
//
fn summary(&self) -> Option<&str> { // fn summary(&self) -> Option<&str> {
self.summary.as_deref() // self.summary.as_deref()
} // }
//
fn icon(&self) -> Node<impl Image> { // fn icon(&self) -> Node<impl Image> {
match &self.icon { // match &self.icon {
Some(x) => Node::object( // Some(x) => Node::object(
serde_json::Value::new_object() // serde_json::Value::new_object()
.set_document_type(Some(DocumentType::Image)) // .set_document_type(Some(DocumentType::Image))
.set_url(Node::link(x.clone())) // .set_url(Node::link(x.clone()))
), // ),
None => Node::Empty, // None => Node::Empty,
} // }
} // }
//
fn image(&self) -> Node<impl Image> { // fn image(&self) -> Node<impl Image> {
match &self.image { // match &self.image {
Some(x) => Node::object( // Some(x) => Node::object(
serde_json::Value::new_object() // serde_json::Value::new_object()
.set_document_type(Some(DocumentType::Image)) // .set_document_type(Some(DocumentType::Image))
.set_url(Node::link(x.clone())) // .set_url(Node::link(x.clone()))
), // ),
None => Node::Empty, // None => Node::Empty,
} // }
} // }
//
fn published(&self) -> Option<chrono::DateTime<chrono::Utc>> { // fn published(&self) -> Option<chrono::DateTime<chrono::Utc>> {
Some(self.created) // Some(self.created)
} // }
} // }
//
impl crate::activitystream::object::actor::Actor for Model { // impl crate::activitystream::object::actor::Actor for Model {
fn actor_type(&self) -> Option<ActorType> { // fn actor_type(&self) -> Option<ActorType> {
Some(self.actor_type) // Some(self.actor_type)
} // }
//
fn preferred_username(&self) -> Option<&str> { // fn preferred_username(&self) -> Option<&str> {
Some(&self.preferred_username) // Some(&self.preferred_username)
} // }
//
fn inbox(&self) -> Node<impl Collection> { // fn inbox(&self) -> Node<impl Collection> {
Node::link(self.inbox.clone().unwrap_or(format!("https://{}/users/{}/inbox", self.domain, self.preferred_username))) // Node::link(self.inbox.clone().unwrap_or(format!("https://{}/users/{}/inbox", self.domain, self.preferred_username)))
} // }
//
fn outbox(&self) -> Node<impl Collection> { // fn outbox(&self) -> Node<impl Collection> {
Node::link(self.outbox.clone().unwrap_or(format!("https://{}/users/{}/outbox", self.domain, self.preferred_username))) // Node::link(self.outbox.clone().unwrap_or(format!("https://{}/users/{}/outbox", self.domain, self.preferred_username)))
} // }
//
fn following(&self) -> Node<impl Collection> { // fn following(&self) -> Node<impl Collection> {
Node::link(self.following.clone().unwrap_or(format!("https://{}/users/{}/following", self.domain, self.preferred_username))) // Node::link(self.following.clone().unwrap_or(format!("https://{}/users/{}/following", self.domain, self.preferred_username)))
} // }
//
fn followers(&self) -> Node<impl Collection> { // fn followers(&self) -> Node<impl Collection> {
Node::link(self.following.clone().unwrap_or(format!("https://{}/users/{}/followers", self.domain, self.preferred_username))) // Node::link(self.following.clone().unwrap_or(format!("https://{}/users/{}/followers", self.domain, self.preferred_username)))
} // }
//
fn public_key(&self) -> Node<impl crate::activitystream::key::PublicKey> { // fn public_key(&self) -> Node<impl crate::activitystream::key::PublicKey> {
Node::object( // Node::object(
serde_json::Value::new_object() // serde_json::Value::new_object()
.set_id(Some(&format!("{}#main-key", self.id))) // TODO is this some standard?? // .set_id(Some(&format!("{}#main-key", self.id))) // TODO is this some standard??
.set_public_key_pem(&self.public_key) // .set_public_key_pem(&self.public_key)
.set_owner(Some(&self.id)) // .set_owner(Some(&self.id))
) // )
} // }
} // }
impl Model { impl Model {
pub fn new(object: &impl Actor) -> Result<Self, super::FieldError> { pub fn new(object: &impl Actor) -> Result<Self, super::FieldError> {