1use crate::db::{ProjectId, RoomId, RoomParticipantId, ServerId, UserId};
 2use rpc::ConnectionId;
 3use sea_orm::entity::prelude::*;
 4
 5#[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel)]
 6#[sea_orm(table_name = "room_participants")]
 7pub struct Model {
 8    #[sea_orm(primary_key)]
 9    pub id: RoomParticipantId,
10    pub room_id: RoomId,
11    pub user_id: UserId,
12    pub answering_connection_id: Option<i32>,
13    pub answering_connection_server_id: Option<ServerId>,
14    pub answering_connection_lost: bool,
15    pub location_kind: Option<i32>,
16    pub location_project_id: Option<ProjectId>,
17    pub initial_project_id: Option<ProjectId>,
18    pub calling_user_id: UserId,
19    pub calling_connection_id: i32,
20    pub calling_connection_server_id: Option<ServerId>,
21    pub participant_index: Option<i32>,
22}
23
24impl Model {
25    pub fn answering_connection(&self) -> Option<ConnectionId> {
26        Some(ConnectionId {
27            owner_id: self.answering_connection_server_id?.0 as u32,
28            id: self.answering_connection_id? as u32,
29        })
30    }
31}
32
33#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
34pub enum Relation {
35    #[sea_orm(
36        belongs_to = "super::user::Entity",
37        from = "Column::UserId",
38        to = "super::user::Column::Id"
39    )]
40    User,
41    #[sea_orm(
42        belongs_to = "super::room::Entity",
43        from = "Column::RoomId",
44        to = "super::room::Column::Id"
45    )]
46    Room,
47}
48
49impl Related<super::user::Entity> for Entity {
50    fn to() -> RelationDef {
51        Relation::User.def()
52    }
53}
54
55impl Related<super::room::Entity> for Entity {
56    fn to() -> RelationDef {
57        Relation::Room.def()
58    }
59}
60
61impl ActiveModelBehavior for ActiveModel {}