Finish porting remaining db methods to sea-orm

crates/collab/src/db/language_server.rs

@@ -0,0 +1,30 @@
+use super::ProjectId;
+use sea_orm::entity::prelude::*;
+
+#[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel)]
+#[sea_orm(table_name = "language_servers")]
+pub struct Model {
+    #[sea_orm(primary_key)]
+    pub project_id: ProjectId,
+    #[sea_orm(primary_key)]
+    pub id: i64,
+    pub name: String,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::project::Entity",
+        from = "Column::ProjectId",
+        to = "super::project::Column::Id"
+    )]
+    Project,
+}
+
+impl Related<super::project::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Project.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

crates/collab/src/db/project.rs

@@ -26,7 +26,11 @@ pub enum Relation {
     )]
     Room,
     #[sea_orm(has_many = "super::worktree::Entity")]
-    Worktree,
+    Worktrees,
+    #[sea_orm(has_many = "super::project_collaborator::Entity")]
+    Collaborators,
+    #[sea_orm(has_many = "super::language_server::Entity")]
+    LanguageServers,
 }
 
 impl Related<super::user::Entity> for Entity {
@@ -43,7 +47,19 @@ impl Related<super::room::Entity> for Entity {
 
 impl Related<super::worktree::Entity> for Entity {
     fn to() -> RelationDef {
-        Relation::Worktree.def()
+        Relation::Worktrees.def()
+    }
+}
+
+impl Related<super::project_collaborator::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Collaborators.def()
+    }
+}
+
+impl Related<super::language_server::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::LanguageServers.def()
     }
 }
 

crates/collab/src/db/project_collaborator.rs

@@ -14,6 +14,19 @@ pub struct Model {
 }
 
 #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
-pub enum Relation {}
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::project::Entity",
+        from = "Column::ProjectId",
+        to = "super::project::Column::Id"
+    )]
+    Project,
+}
+
+impl Related<super::project::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Project.def()
+    }
+}
 
 impl ActiveModelBehavior for ActiveModel {}

crates/collab/src/db/worktree.rs

@@ -1,17 +1,17 @@
-use super::{ProjectId, WorktreeId};
+use super::ProjectId;
 use sea_orm::entity::prelude::*;
 
 #[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel)]
 #[sea_orm(table_name = "worktrees")]
 pub struct Model {
     #[sea_orm(primary_key)]
-    pub id: WorktreeId,
+    pub id: i64,
     #[sea_orm(primary_key)]
     pub project_id: ProjectId,
     pub abs_path: String,
     pub root_name: String,
     pub visible: bool,
-    pub scan_id: u32,
+    pub scan_id: i64,
     pub is_complete: bool,
 }
 

crates/collab/src/db/worktree_diagnostic_summary.rs

@@ -0,0 +1,21 @@
+use super::ProjectId;
+use sea_orm::entity::prelude::*;
+
+#[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel)]
+#[sea_orm(table_name = "worktree_diagnostic_summaries")]
+pub struct Model {
+    #[sea_orm(primary_key)]
+    pub project_id: ProjectId,
+    #[sea_orm(primary_key)]
+    pub worktree_id: i64,
+    #[sea_orm(primary_key)]
+    pub path: String,
+    pub language_server_id: i64,
+    pub error_count: u32,
+    pub warning_count: u32,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {}
+
+impl ActiveModelBehavior for ActiveModel {}

crates/collab/src/db/worktree_entry.rs

@@ -1,23 +1,25 @@
-use super::{ProjectId, WorktreeEntryId, WorktreeId};
+use super::ProjectId;
 use sea_orm::entity::prelude::*;
 
 #[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel)]
 #[sea_orm(table_name = "worktree_entries")]
 pub struct Model {
     #[sea_orm(primary_key)]
-    project_id: ProjectId,
+    pub project_id: ProjectId,
     #[sea_orm(primary_key)]
-    worktree_id: WorktreeId,
+    pub worktree_id: i64,
     #[sea_orm(primary_key)]
-    id: WorktreeEntryId,
-    is_dir: bool,
-    path: String,
-    inode: u64,
-    mtime_seconds: u64,
-    mtime_nanos: u32,
-    is_symlink: bool,
-    is_ignored: bool,
+    pub id: i64,
+    pub is_dir: bool,
+    pub path: String,
+    pub inode: i64,
+    pub mtime_seconds: i64,
+    pub mtime_nanos: u32,
+    pub is_symlink: bool,
+    pub is_ignored: bool,
 }
 
 #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
 pub enum Relation {}
+
+impl ActiveModelBehavior for ActiveModel {}