Index bans

migrations/20260418000114_add_index_ban.down.sql

@@ -0,0 +1,2 @@
+-- Add down migration script here
+DROP TABLE IF EXISTS bans;

migrations/20260418000114_add_index_ban.up.sql

@@ -0,0 +1,9 @@
+-- Add up migration script here
+CREATE TABLE bans (
+	id SERIAL PRIMARY KEY NOT NULL,
+	developer_id INTEGER NOT NULL REFERENCES developers(id) ON DELETE CASCADE,
+	reason TEXT,
+	admin_id INTEGER REFERENCES developers(id) ON DELETE SET NULL,
+	created_at TIMESTAMPTZ DEFAULT CURRENT_TIMESTAMP NOT NULL,
+	revoked_at TIMESTAMPTZ
+);

src/database/repository/developers.rs

@@ -1,7 +1,8 @@
 use crate::database::DatabaseError;
 use crate::types::api::PaginatedData;
-use crate::types::models::developer::{Developer, ModDeveloper};
+use crate::types::models::developer::{Developer, DeveloperBan, ModDeveloper};
 use sqlx::PgConnection;
+use sqlx::types::chrono::{DateTime, Utc};
 use std::collections::HashMap;
 use uuid::Uuid;
 
@@ -427,3 +428,60 @@ pub async fn find_by_token(
     .inspect_err(|e| log::error!("{}", e))
     .map_err(|e| e.into())
 }
+
+pub async fn create_ban(
+    dev_id: i32,
+    admin_id: i32,
+    reason: Option<&str>,
+    revoked_at: Option<DateTime<Utc>>,
+    conn: &mut PgConnection,
+) -> Result<DeveloperBan, DatabaseError> {
+    sqlx::query_as!(DeveloperBan,
+        "INSERT INTO bans (developer_id, reason, admin_id, revoked_at)
+            VALUES ($1, $2, $3, $4)
+        RETURNING
+            id, developer_id, reason, admin_id, created_at, revoked_at",
+        dev_id, reason, admin_id, revoked_at
+    )
+    .fetch_one(&mut *conn)
+    .await
+    .inspect_err(|e| log::error!("Failed to insert create developer ban: {e}"))
+    .map_err(|e| e.into())
+}
+
+pub async fn check_ban(
+    dev_id: i32,
+    conn: &mut PgConnection,
+) -> Result<Option<DeveloperBan>, DatabaseError> {
+    sqlx::query_as!(DeveloperBan,
+        "SELECT developer_id, reason, admin_id, created_at, id, revoked_at FROM bans WHERE developer_id=$1 AND revoked_at > NOW() or revoked_at IS NULL ORDER BY revoked_at DESC NULLS FIRST LIMIT 1", dev_id
+    )
+    .fetch_optional(&mut *conn)
+    .await
+    .inspect_err(|e| log::error!("Failed to get developer ban: {e}"))
+    .map_err(|e| e.into())
+}
+
+pub async fn update_ban_revoke_time(ban_id: i32, revoked_at: Option<DateTime<Utc>>, conn: &mut PgConnection) -> Result<Option<DeveloperBan>, DatabaseError> {
+    sqlx::query_as!(DeveloperBan,
+        "UPDATE bans
+            SET revoked_at=$2 WHERE id=$1
+        RETURNING
+            id, developer_id, reason, admin_id, created_at, revoked_at",
+        ban_id, revoked_at)
+    .fetch_optional(&mut *conn)
+    .await
+    .inspect_err(|e| log::error!("Failed to update developer ban: {e}"))
+    .map_err(|e| e.into())
+}
+
+pub async fn delete_ban(dev_id: i32, conn: &mut PgConnection) -> Result<(), DatabaseError> {
+    if let Some(current_ban) = check_ban(dev_id, conn).await? {
+        sqlx::query!("UPDATE bans SET revoked_at=NOW() WHERE id=$1", current_ban.id)
+            .execute(conn)
+            .await
+        .inspect_err(|e| log::error!("Failed to revoke developer ban: {e}"))?;
+    }
+
+    Ok(())
+}