Allow toggling repository active status

TabbyML · Apr 9, 2024 · 6ff67ac · 6ff67ac
1 parent 8c4480c
commit 6ff67ac
Show file tree

Hide file tree

Showing 4 changed files with 43 additions and 0 deletions.
diff --git a/ee/tabby-db/src/github_repository_provider.rs b/ee/tabby-db/src/github_repository_provider.rs
@@ -153,4 +153,26 @@ impl DbConn {
         .await?;
         Ok(repos)
     }
+
+    pub async fn update_github_provided_repository_active(
+        &self,
+        id: i64,
+        active: bool,
+    ) -> Result<()> {
+        let not_active = !active;
+        let res = query!(
+            "UPDATE github_provided_repositories SET active = ? WHERE id = ? AND active = ?",
+            active,
+            id,
+            not_active
+        )
+        .execute(&self.pool)
+        .await?;
+
+        if res.rows_affected() != 1 {
+            return Err(anyhow!("Repository active status was not changed"));
+        }
+
+        Ok(())
+    }
 }
diff --git a/ee/tabby-webserver/src/schema/github_repository_provider.rs b/ee/tabby-webserver/src/schema/github_repository_provider.rs
@@ -86,4 +86,6 @@ pub trait GithubRepositoryProviderService: Send + Sync {
         first: Option<usize>,
         last: Option<usize>,
     ) -> Result<Vec<GithubProvidedRepository>>;
+
+    async fn update_github_provided_repository_active(&self, id: ID, active: bool) -> Result<()>;
 }
diff --git a/ee/tabby-webserver/src/schema/mod.rs b/ee/tabby-webserver/src/schema/mod.rs
@@ -716,6 +716,18 @@ impl Mutation {
             .await?;
         Ok(true)
     }
+
+    async fn update_github_provided_repository_active(
+        ctx: &Context,
+        id: ID,
+        active: bool,
+    ) -> Result<bool> {
+        ctx.locator
+            .github_repository_provider()
+            .update_github_provided_repository_active(id, active)
+            .await?;
+        Ok(true)
+    }
 }
 
 async fn check_analytic_access(ctx: &Context, users: &[ID]) -> Result<(), CoreError> {

diff --git a/ee/tabby-webserver/src/service/github_repository_provider.rs b/ee/tabby-webserver/src/service/github_repository_provider.rs
@@ -99,4 +99,11 @@ impl GithubRepositoryProviderService for GithubRepositoryProviderServiceImpl {
             .map(GithubProvidedRepository::from)
             .collect())
     }
+
+    async fn update_github_provided_repository_active(&self, id: ID, active: bool) -> Result<()> {
+        self.db
+            .update_github_provided_repository_active(id.as_rowid()? as i64, active)
+            .await?;
+        Ok(())
+    }
 }