use sea_orm::entity::prelude::*; use crate::{activitypub::jsonld::LD, activitystream::{object::{ObjectMut, ObjectType}, BaseMut, BaseType, Node}}; #[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)] #[sea_orm(table_name = "objects")] pub struct Model { #[sea_orm(primary_key)] /// must be full uri!!! maybe not great? pub id: String, pub object_type: ObjectType, pub attributed_to: Option, pub name: Option, pub summary: Option, pub content: Option, pub published: ChronoDateTimeUtc, } #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)] pub enum Relation { #[sea_orm(has_many = "super::activity::Entity")] Activity, #[sea_orm( belongs_to = "super::user::Entity", from = "Column::AttributedTo", to = "super::user::Column::Id", )] User, } impl Related for Entity { fn to() -> RelationDef { Relation::Activity.def() } } impl Related for Entity { fn to() -> RelationDef { Relation::User.def() } } impl ActiveModelBehavior for ActiveModel {} impl crate::activitystream::Base for Model { fn id(&self) -> Option<&str> { Some(&self.id) } fn base_type(&self) -> Option { Some(BaseType::Object(self.object_type)) } fn underlying_json_object(self) -> serde_json::Value { serde_json::Value::new_object() .set_id(Some(&self.id)) .set_object_type(Some(self.object_type)) .set_attributed_to(Node::maybe_link(self.attributed_to)) .set_name(self.name.as_deref()) .set_summary(self.summary.as_deref()) .set_content(self.content.as_deref()) .set_published(Some(self.published)) } } impl crate::activitystream::object::Object for Model { fn object_type(&self) -> Option { Some(self.object_type) } fn attributed_to(&self) -> Node { Node::::from(self.attributed_to.as_deref()) } fn name(&self) -> Option<&str> { self.name.as_deref() } fn summary(&self) -> Option<&str> { self.summary.as_deref() } fn content(&self) -> Option<&str> { self.content.as_deref() } fn published (&self) -> Option> { Some(self.published) } } impl Model { pub fn new(object: &impl crate::activitystream::object::Object) -> Result { Ok(Model { id: object.id().ok_or(super::FieldError("id"))?.to_string(), object_type: object.object_type().ok_or(super::FieldError("type"))?, attributed_to: object.attributed_to().id().map(|x| x.to_string()), name: object.name().map(|x| x.to_string()), summary: object.summary().map(|x| x.to_string()), content: object.content().map(|x| x.to_string()), published: object.published().ok_or(super::FieldError("published"))?, }) } }