Skip to content
Clone
main
122 lines 3.2 KB Rust
use sea_orm::JsonValue;
use sea_orm::entity::prelude::*;
use serde::{Deserialize, Serialize};
use time::OffsetDateTime;
use uuid::Uuid;
#[derive(
Debug,
Clone,
Copy,
PartialEq,
Eq,
Default,
EnumIter,
DeriveActiveEnum,
Serialize,
Deserialize,
utoipa::ToSchema,
)]
#[sea_orm(rs_type = "i16", db_type = "SmallInteger")]
pub enum AuthorType {
#[default]
#[sea_orm(num_value = 0)]
User,
#[sea_orm(num_value = 1)]
Bot,
#[sea_orm(num_value = 2)]
Webhook,
}
#[derive(
Debug,
Clone,
Copy,
PartialEq,
Eq,
Default,
EnumIter,
DeriveActiveEnum,
Serialize,
Deserialize,
utoipa::ToSchema,
)]
#[sea_orm(rs_type = "i16", db_type = "SmallInteger")]
pub enum ChatMessageType {
#[default]
#[sea_orm(num_value = 0)]
Default,
#[sea_orm(num_value = 1)]
Reply,
#[sea_orm(num_value = 2)]
ThreadCreated,
#[sea_orm(num_value = 3)]
ChannelJoin,
#[sea_orm(num_value = 4)]
ChannelLeave,
#[sea_orm(num_value = 5)]
ChannelTopic,
#[sea_orm(num_value = 6)]
ChannelPinnedItem,
#[sea_orm(num_value = 7)]
Reminder,
#[sea_orm(num_value = 9)]
System,
}
pub const FLAG_EPHEMERAL: i64 = 1;
pub const FLAG_URGENT: i64 = 1 << 4;
pub const FLAG_HAS_THREAD: i64 = 1 << 5;
pub const FLAG_SUPPRESS_EMBED: i64 = 1 << 2;
#[sea_orm::model]
#[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
#[sea_orm(table_name = "chat_message")]
pub struct Model {
#[sea_orm(primary_key, auto_increment = false)]
pub id: Uuid,
#[sea_orm(indexed, unique_key = "chat_message_room_seq")]
pub room_id: Uuid,
#[sea_orm(indexed)]
pub author_id: i64,
#[sea_orm(default_value = 0)]
pub author_type: AuthorType,
pub webhook_id: Option<Uuid>,
#[sea_orm(indexed)]
pub thread_id: Option<Uuid>,
#[sea_orm(default_value = "false")]
pub thread_broadcast: bool,
pub reply_to_id: Option<Uuid>,
pub forwarded_from_id: Option<Uuid>,
#[sea_orm(default_value = 0)]
pub message_type: ChatMessageType,
pub content: String,
#[sea_orm(column_type = "JsonBinary", default_value = "[]")]
pub mentions: JsonValue,
#[sea_orm(default_value = "false")]
pub mention_everyone: bool,
#[sea_orm(column_type = "JsonBinary", default_value = "[]")]
pub mention_group_ids: JsonValue,
#[sea_orm(column_type = "JsonBinary", default_value = "[]")]
pub attachments: JsonValue,
#[sea_orm(column_type = "JsonBinary", default_value = "[]")]
pub embeds: JsonValue,
#[sea_orm(default_value = 0)]
pub flags: i64,
pub nonce: Option<String>,
#[sea_orm(default_value = "false")]
pub pinned: bool,
pub pinned_by: Option<i64>,
pub pinned_at: Option<OffsetDateTime>,
pub edited_at: Option<OffsetDateTime>,
pub deleted_at: Option<OffsetDateTime>,
#[sea_orm(unique_key = "chat_message_room_seq")]
pub seq: i64,
pub created_at: OffsetDateTime,
#[sea_orm(belongs_to, from = "room_id", to = "id")]
pub room: BelongsTo<super::super::room::rooms::Entity>,
#[sea_orm(belongs_to, from = "author_id", to = "uid")]
pub author: BelongsTo<super::super::users::user::Entity>,
}
impl ActiveModelBehavior for ActiveModel {}