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