circus/crates/common/src/repo/remote_builders.rs

use sqlx::PgPool;
use uuid::Uuid;

use crate::{
  config::DeclarativeRemoteBuilder,
  error::{CiError, Result},
  models::{CreateRemoteBuilder, RemoteBuilder},
};

pub async fn create(
  pool: &PgPool,
  input: CreateRemoteBuilder,
) -> Result<RemoteBuilder> {
  sqlx::query_as::<_, RemoteBuilder>(
    "INSERT INTO remote_builders (name, ssh_uri, systems, max_jobs, \
     speed_factor, supported_features, mandatory_features, public_host_key, \
     ssh_key_file) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9) RETURNING *",
  )
  .bind(&input.name)
  .bind(&input.ssh_uri)
  .bind(&input.systems)
  .bind(input.max_jobs.unwrap_or(1))
  .bind(input.speed_factor.unwrap_or(1))
  .bind(input.supported_features.as_deref().unwrap_or(&[]))
  .bind(input.mandatory_features.as_deref().unwrap_or(&[]))
  .bind(&input.public_host_key)
  .bind(&input.ssh_key_file)
  .fetch_one(pool)
  .await
  .map_err(|e| {
    match &e {
      sqlx::Error::Database(db_err) if db_err.is_unique_violation() => {
        CiError::Conflict(format!(
          "Remote builder '{}' already exists",
          input.name
        ))
      },
      _ => CiError::Database(e),
    }
  })
}

pub async fn get(pool: &PgPool, id: Uuid) -> Result<RemoteBuilder> {
  sqlx::query_as::<_, RemoteBuilder>(
    "SELECT * FROM remote_builders WHERE id = $1",
  )
  .bind(id)
  .fetch_optional(pool)
  .await?
  .ok_or_else(|| CiError::NotFound(format!("Remote builder {id} not found")))
}

pub async fn list(pool: &PgPool) -> Result<Vec<RemoteBuilder>> {
  sqlx::query_as::<_, RemoteBuilder>(
    "SELECT * FROM remote_builders ORDER BY speed_factor DESC, name",
  )
  .fetch_all(pool)
  .await
  .map_err(CiError::Database)
}

pub async fn list_enabled(pool: &PgPool) -> Result<Vec<RemoteBuilder>> {
  sqlx::query_as::<_, RemoteBuilder>(
    "SELECT * FROM remote_builders WHERE enabled = true ORDER BY speed_factor \
     DESC, name",
  )
  .fetch_all(pool)
  .await
  .map_err(CiError::Database)
}

/// Find a suitable builder for the given system.
pub async fn find_for_system(
  pool: &PgPool,
  system: &str,
) -> Result<Vec<RemoteBuilder>> {
  sqlx::query_as::<_, RemoteBuilder>(
    "SELECT * FROM remote_builders WHERE enabled = true AND $1 = ANY(systems) \
     ORDER BY speed_factor DESC",
  )
  .bind(system)
  .fetch_all(pool)
  .await
  .map_err(CiError::Database)
}

pub async fn update(
  pool: &PgPool,
  id: Uuid,
  input: crate::models::UpdateRemoteBuilder,
) -> Result<RemoteBuilder> {
  // Build dynamic update — use COALESCE pattern
  sqlx::query_as::<_, RemoteBuilder>(
    "UPDATE remote_builders SET name = COALESCE($1, name), ssh_uri = \
     COALESCE($2, ssh_uri), systems = COALESCE($3, systems), max_jobs = \
     COALESCE($4, max_jobs), speed_factor = COALESCE($5, speed_factor), \
     supported_features = COALESCE($6, supported_features), \
     mandatory_features = COALESCE($7, mandatory_features), enabled = \
     COALESCE($8, enabled), public_host_key = COALESCE($9, public_host_key), \
     ssh_key_file = COALESCE($10, ssh_key_file) WHERE id = $11 RETURNING *",
  )
  .bind(&input.name)
  .bind(&input.ssh_uri)
  .bind(&input.systems)
  .bind(input.max_jobs)
  .bind(input.speed_factor)
  .bind(&input.supported_features)
  .bind(&input.mandatory_features)
  .bind(input.enabled)
  .bind(&input.public_host_key)
  .bind(&input.ssh_key_file)
  .bind(id)
  .fetch_optional(pool)
  .await?
  .ok_or_else(|| CiError::NotFound(format!("Remote builder {id} not found")))
}

pub async fn delete(pool: &PgPool, id: Uuid) -> Result<()> {
  let result = sqlx::query("DELETE FROM remote_builders WHERE id = $1")
    .bind(id)
    .execute(pool)
    .await
    .map_err(CiError::Database)?;
  if result.rows_affected() == 0 {
    return Err(CiError::NotFound(format!("Remote builder {id} not found")));
  }
  Ok(())
}

pub async fn count(pool: &PgPool) -> Result<i64> {
  let row: (i64,) = sqlx::query_as("SELECT COUNT(*) FROM remote_builders")
    .fetch_one(pool)
    .await
    .map_err(CiError::Database)?;
  Ok(row.0)
}

/// Upsert a remote builder (insert or update on conflict by name).
pub async fn upsert(
  pool: &PgPool,
  name: &str,
  ssh_uri: &str,
  systems: &[String],
  max_jobs: i32,
  speed_factor: i32,
  supported_features: &[String],
  mandatory_features: &[String],
  enabled: bool,
  public_host_key: Option<&str>,
  ssh_key_file: Option<&str>,
) -> Result<RemoteBuilder> {
  sqlx::query_as::<_, RemoteBuilder>(
    "INSERT INTO remote_builders (name, ssh_uri, systems, max_jobs, \
     speed_factor, supported_features, mandatory_features, enabled, \
     public_host_key, ssh_key_file) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, \
     $10) ON CONFLICT (name) DO UPDATE SET ssh_uri = EXCLUDED.ssh_uri, systems = \
     EXCLUDED.systems, max_jobs = EXCLUDED.max_jobs, speed_factor = \
     EXCLUDED.speed_factor, supported_features = EXCLUDED.supported_features, \
     mandatory_features = EXCLUDED.mandatory_features, enabled = \
     EXCLUDED.enabled, public_host_key = COALESCE(EXCLUDED.public_host_key, \
     remote_builders.public_host_key), ssh_key_file = \
     COALESCE(EXCLUDED.ssh_key_file, remote_builders.ssh_key_file) RETURNING *",
  )
  .bind(name)
  .bind(ssh_uri)
  .bind(systems)
  .bind(max_jobs)
  .bind(speed_factor)
  .bind(supported_features)
  .bind(mandatory_features)
  .bind(enabled)
  .bind(public_host_key)
  .bind(ssh_key_file)
  .fetch_one(pool)
  .await
  .map_err(CiError::Database)
}

/// Sync remote builders from declarative config.
/// Deletes builders not in the declarative list and upserts those that are.
pub async fn sync_all(
  pool: &PgPool,
  builders: &[DeclarativeRemoteBuilder],
) -> Result<()> {
  // Get builder names from declarative config
  let names: Vec<&str> = builders.iter().map(|b| b.name.as_str()).collect();

  // Delete builders not in declarative config
  sqlx::query("DELETE FROM remote_builders WHERE name != ALL($1::text[])")
    .bind(&names)
    .execute(pool)
    .await
    .map_err(CiError::Database)?;

  // Upsert each builder
  for builder in builders {
    upsert(
      pool,
      &builder.name,
      &builder.ssh_uri,
      &builder.systems,
      builder.max_jobs,
      builder.speed_factor,
      &builder.supported_features,
      &builder.mandatory_features,
      builder.enabled,
      builder.public_host_key.as_deref(),
      builder.ssh_key_file.as_deref(),
    )
    .await?;
  }

  Ok(())
}