1use crate::db::UserId;
 2use sea_orm::entity::prelude::*;
 3use serde::Serialize;
 4
 5#[derive(Clone, Debug, Default, PartialEq, Eq, DeriveEntityModel, Serialize)]
 6#[sea_orm(table_name = "users")]
 7pub struct Model {
 8    #[sea_orm(primary_key)]
 9    pub id: UserId,
10    pub github_login: String,
11    pub github_user_id: Option<i32>,
12    pub email_address: Option<String>,
13    pub admin: bool,
14    pub invite_code: Option<String>,
15    pub invite_count: i32,
16    pub inviter_id: Option<UserId>,
17    pub connected_once: bool,
18    pub metrics_id: Uuid,
19}
20
21#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
22pub enum Relation {
23    #[sea_orm(has_many = "super::access_token::Entity")]
24    AccessToken,
25    #[sea_orm(has_one = "super::room_participant::Entity")]
26    RoomParticipant,
27    #[sea_orm(has_many = "super::project::Entity")]
28    HostedProjects,
29    #[sea_orm(has_many = "super::channel_member::Entity")]
30    ChannelMemberships,
31    #[sea_orm(has_many = "super::user_feature::Entity")]
32    UserFeatures,
33}
34
35impl Related<super::access_token::Entity> for Entity {
36    fn to() -> RelationDef {
37        Relation::AccessToken.def()
38    }
39}
40
41impl Related<super::room_participant::Entity> for Entity {
42    fn to() -> RelationDef {
43        Relation::RoomParticipant.def()
44    }
45}
46
47impl Related<super::project::Entity> for Entity {
48    fn to() -> RelationDef {
49        Relation::HostedProjects.def()
50    }
51}
52
53impl Related<super::channel_member::Entity> for Entity {
54    fn to() -> RelationDef {
55        Relation::ChannelMemberships.def()
56    }
57}
58
59impl Related<super::user_feature::Entity> for Entity {
60    fn to() -> RelationDef {
61        Relation::UserFeatures.def()
62    }
63}
64
65impl ActiveModelBehavior for ActiveModel {}
66
67pub struct UserFlags;
68
69impl Linked for UserFlags {
70    type FromEntity = Entity;
71
72    type ToEntity = super::feature_flag::Entity;
73
74    fn link(&self) -> Vec<RelationDef> {
75        vec![
76            super::user_feature::Relation::User.def().rev(),
77            super::user_feature::Relation::Flag.def(),
78        ]
79    }
80}