stormchaser-api 1.4.2

use crate::UpdateStorageBackendRequest;
use serde_json::Value;
use sqlx::{PgPool, Postgres, Transaction};
use stormchaser_model::ConnectionId;

use stormchaser_model::connections;

/// Unsets the default Stormchaser File System.
/// Unset default sfs.
pub async fn unset_default_sfs(tx: &mut Transaction<'_, Postgres>) -> Result<(), sqlx::Error> {
    sqlx::query("UPDATE connections SET is_default_sfs = FALSE WHERE is_default_sfs = TRUE")
        .execute(&mut **tx)
        .await?;
    Ok(())
}

/// Creates a new storage backend.
/// Create storage backend.
#[allow(clippy::too_many_arguments)]
pub async fn create_connection(
    tx: &mut Transaction<'_, Postgres>,
    id: ConnectionId,
    name: &str,
    description: &Option<String>,
    connection_type: &connections::ConnectionType,
    config: &Value,
    aws_assume_role_arn: &Option<String>,
    encrypted_credentials: &Option<String>,
    is_default_sfs: bool,
) -> Result<(), sqlx::Error> {
    sqlx::query(
        r#"
        INSERT INTO connections (id, name, description, connection_type, config, aws_assume_role_arn, encrypted_credentials, is_default_sfs)
        VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
        "#,
    )
    .bind(id)
    .bind(name)
    .bind(description)
    .bind(connection_type)
    .bind(config)
    .bind(aws_assume_role_arn)
    .bind(encrypted_credentials)
    .bind(is_default_sfs)
    .execute(&mut **tx)
    .await?;
    Ok(())
}

/// Retrieves all storage backends.
/// List storage backends.
pub async fn list_connections(pool: &PgPool) -> Result<Vec<connections::Connection>, sqlx::Error> {
    sqlx::query_as("SELECT * FROM connections ORDER BY name ASC")
        .fetch_all(pool)
        .await
}

/// Retrieves a storage backend by ID.
/// Get storage backend.
pub async fn get_connection(
    pool: &PgPool,
    id: ConnectionId,
) -> Result<Option<connections::Connection>, sqlx::Error> {
    sqlx::query_as("SELECT * FROM connections WHERE id = $1")
        .bind(id)
        .fetch_optional(pool)
        .await
}

/// Retrieves a storage backend by its unique name.
///
/// Unlike [`get_connection`], which looks up by [`ConnectionId`], this helper
/// resolves a connection using the user-facing `name` field.
pub async fn get_connection_by_name(
    pool: &PgPool,
    name: &str,
) -> Result<Option<connections::Connection>, sqlx::Error> {
    sqlx::query_as("SELECT * FROM connections WHERE name = $1")
        .bind(name)
        .fetch_optional(pool)
        .await
}

/// Updates an existing storage backend.
/// Update storage backend.
pub async fn update_connection(
    tx: &mut Transaction<'_, Postgres>,
    id: ConnectionId,
    payload: &UpdateStorageBackendRequest,
) -> Result<(), sqlx::Error> {
    let mut query = sqlx::QueryBuilder::new("UPDATE connections SET ");
    let mut separated = query.separated(", ");

    if let Some(name) = &payload.name {
        separated.push("name = ").push_bind_unseparated(name);
    }
    if let Some(desc) = &payload.description {
        separated.push("description = ").push_bind_unseparated(desc);
    }
    if let Some(bt) = &payload.connection_type {
        separated
            .push("connection_type = ")
            .push_bind_unseparated(bt);
    }
    if let Some(cfg) = &payload.config {
        separated.push("config = ").push_bind_unseparated(cfg);
    }
    if let Some(role) = &payload.aws_assume_role_arn {
        // An empty string is treated as a request to clear the ARN (set to NULL).
        let value: Option<&str> = if role.is_empty() {
            None
        } else {
            Some(role.as_str())
        };
        separated
            .push("aws_assume_role_arn = ")
            .push_bind_unseparated(value);
    }
    if let Some(credentials) = &payload.encrypted_credentials {
        let value: Option<&str> = if credentials.is_empty() {
            None
        } else {
            Some(credentials.as_str())
        };
        separated
            .push("encrypted_credentials = ")
            .push_bind_unseparated(value);
    }
    if let Some(is_default) = payload.is_default_sfs {
        separated
            .push("is_default_sfs = ")
            .push_bind_unseparated(is_default);
    }

    query.push(" WHERE id = ").push_bind(id);

    query.build().execute(&mut **tx).await?;
    Ok(())
}

/// Deletes a storage backend from the database.
/// Delete storage backend.
pub async fn delete_connection(pool: &PgPool, id: ConnectionId) -> Result<(), sqlx::Error> {
    sqlx::query("DELETE FROM connections WHERE id = $1")
        .bind(id)
        .execute(pool)
        .await?;
    Ok(())
}