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 {}