use sea_orm::entity::prelude::*; use super::Audience; #[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)] #[sea_orm(table_name = "objects")] pub struct Model { #[sea_orm(primary_key)] pub id: String, pub object_type: apb::ObjectType, pub attributed_to: Option, pub name: Option, pub summary: Option, pub content: Option, pub likes: i64, pub shares: i64, pub comments: i64, pub context: Option, pub in_reply_to: Option, pub cc: Audience, pub bcc: Audience, pub to: Audience, pub bto: Audience, pub published: ChronoDateTimeUtc, } impl Model { pub fn new(object: &impl apb::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(), name: object.name().map(|x| x.to_string()), summary: object.summary().map(|x| x.to_string()), content: object.content().map(|x| x.to_string()), context: object.context().id(), in_reply_to: object.in_reply_to().id(), published: object.published().ok_or(super::FieldError("published"))?, comments: 0, likes: 0, shares: 0, to: object.to().into(), bto: object.bto().into(), cc: object.cc().into(), bcc: object.bcc().into(), }) } } #[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, #[sea_orm(has_many = "super::addressing::Entity")] Addressing, } impl Related for Entity { fn to() -> RelationDef { Relation::Activity.def() } } impl Related for Entity { fn to() -> RelationDef { Relation::User.def() } } impl Related for Entity { fn to() -> RelationDef { Relation::Addressing.def() } } impl ActiveModelBehavior for ActiveModel {}