1use crate::db::{DevServerId, UserId};
 2use rpc::proto;
 3use sea_orm::entity::prelude::*;
 4
 5#[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel)]
 6#[sea_orm(table_name = "dev_servers")]
 7pub struct Model {
 8    #[sea_orm(primary_key)]
 9    pub id: DevServerId,
10    pub name: String,
11    pub user_id: UserId,
12    pub hashed_token: String,
13    pub ssh_connection_string: Option<String>,
14}
15
16impl ActiveModelBehavior for ActiveModel {}
17
18#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
19pub enum Relation {
20    #[sea_orm(has_many = "super::dev_server_project::Entity")]
21    RemoteProject,
22}
23
24impl Related<super::dev_server_project::Entity> for Entity {
25    fn to() -> RelationDef {
26        Relation::RemoteProject.def()
27    }
28}
29
30impl Model {
31    pub fn to_proto(&self, status: proto::DevServerStatus) -> proto::DevServer {
32        proto::DevServer {
33            dev_server_id: self.id.to_proto(),
34            name: self.name.clone(),
35            status: status as i32,
36            ssh_connection_string: self.ssh_connection_string.clone(),
37        }
38    }
39}