main
use sea_orm::JsonValue;
use sea_orm::entity::prelude::*;
use time::OffsetDateTime;
use uuid::Uuid;
#[derive(
Debug,
Clone,
Copy,
PartialEq,
Eq,
EnumIter,
DeriveActiveEnum,
serde::Serialize,
serde::Deserialize,
utoipa::ToSchema,
)]
#[serde(rename_all = "snake_case")]
#[sea_orm(rs_type = "String", db_type = "String(StringLen::N(16))")]
pub enum NotificationType {
#[sea_orm(string_value = "mention")]
Mention,
#[sea_orm(string_value = "reply")]
Reply,
#[sea_orm(string_value = "follow")]
Follow,
#[sea_orm(string_value = "invite")]
Invite,
#[sea_orm(string_value = "assignment")]
Assignment,
#[sea_orm(string_value = "system")]
System,
}
#[sea_orm::model]
#[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
#[sea_orm(table_name = "user_notifications")]
pub struct Model {
#[sea_orm(primary_key, auto_increment = false)]
pub id: Uuid,
pub user_id: i64,
#[sea_orm(belongs_to, from = "user_id", to = "uid")]
pub user: BelongsTo<super::user::Entity>,
pub notification_type: NotificationType,
pub title: String,
pub body: Option<String>,
#[sea_orm(column_name = "data")]
pub data_json: Option<JsonValue>,
pub actor_uid: Option<i64>,
#[sea_orm(belongs_to, relation_enum = "Actor", from = "actor_uid", to = "uid")]
pub actor: BelongsTo<Option<super::user::Entity>>,
pub read_at: Option<OffsetDateTime>,
pub created_at: OffsetDateTime,
}
impl ActiveModelBehavior for ActiveModel {}
impl Model {
/// Canonical payload for the WS `notification` user event; mirrors the
/// `NotificationDto` returned by `GET /me/notifications`.
pub fn to_event_value(&self) -> JsonValue {
#[derive(serde::Serialize)]
struct EventPayload<'a> {
id: Uuid,
notification_type: &'a NotificationType,
title: &'a str,
body: &'a Option<String>,
data: &'a Option<JsonValue>,
actor_id: &'a Option<i64>,
read_at: &'a Option<OffsetDateTime>,
created_at: &'a OffsetDateTime,
}
serde_json::to_value(EventPayload {
id: self.id,
notification_type: &self.notification_type,
title: &self.title,
body: &self.body,
data: &self.data_json,
actor_id: &self.actor_uid,
read_at: &self.read_at,
created_at: &self.created_at,
})
.unwrap_or(serde_json::Value::Null)
}
}
impl Related<super::user::Entity> for Entity {
fn to() -> RelationDef {
Relation::User.def()
}
}