barbacane-control 0.5.2

//! Artifacts repository for CRUD operations.

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

use super::models::Artifact;

/// Repository for artifact operations.
#[derive(Clone)]
pub struct ArtifactsRepository {
    pool: PgPool,
}

impl ArtifactsRepository {
    /// Create a new artifacts repository.
    pub fn new(pool: PgPool) -> Self {
        Self { pool }
    }

    /// Store a new artifact.
    pub async fn create(
        &self,
        project_id: Option<Uuid>,
        manifest: serde_json::Value,
        data: Vec<u8>,
        sha256: &str,
        compiler_version: &str,
    ) -> Result<Artifact, sqlx::Error> {
        let size_bytes = data.len() as i64;
        sqlx::query_as::<_, Artifact>(
            r#"
            INSERT INTO artifacts (project_id, manifest, data, sha256, size_bytes, compiler_version)
            VALUES ($1, $2, $3, $4, $5, $6)
            RETURNING *
            "#,
        )
        .bind(project_id)
        .bind(&manifest)
        .bind(&data)
        .bind(sha256)
        .bind(size_bytes)
        .bind(compiler_version)
        .fetch_one(&self.pool)
        .await
    }

    /// List all artifacts.
    pub async fn list(&self) -> Result<Vec<Artifact>, sqlx::Error> {
        sqlx::query_as::<_, Artifact>("SELECT * FROM artifacts ORDER BY compiled_at DESC")
            .fetch_all(&self.pool)
            .await
    }

    /// List artifacts for a project.
    pub async fn list_for_project(&self, project_id: Uuid) -> Result<Vec<Artifact>, sqlx::Error> {
        sqlx::query_as::<_, Artifact>(
            "SELECT * FROM artifacts WHERE project_id = $1 ORDER BY compiled_at DESC",
        )
        .bind(project_id)
        .fetch_all(&self.pool)
        .await
    }

    /// Get an artifact by ID (metadata only).
    pub async fn get(&self, id: Uuid) -> Result<Option<Artifact>, sqlx::Error> {
        sqlx::query_as::<_, Artifact>("SELECT * FROM artifacts WHERE id = $1")
            .bind(id)
            .fetch_optional(&self.pool)
            .await
    }

    /// Get the binary data of an artifact for download.
    pub async fn get_data(&self, id: Uuid) -> Result<Option<Vec<u8>>, sqlx::Error> {
        let row: Option<(Vec<u8>,)> = sqlx::query_as("SELECT data FROM artifacts WHERE id = $1")
            .bind(id)
            .fetch_optional(&self.pool)
            .await?;
        Ok(row.map(|(data,)| data))
    }

    /// Check if any artifact's manifest references a given plugin name+version.
    pub async fn plugin_is_referenced(
        &self,
        name: &str,
        version: &str,
    ) -> Result<bool, sqlx::Error> {
        // The artifact manifest is JSONB with a `plugins` array of objects
        // containing `name` and `version` fields. Use the @> containment operator.
        let exists: bool = sqlx::query_scalar(
            r#"
            SELECT EXISTS(
                SELECT 1 FROM artifacts
                WHERE manifest->'plugins' @> jsonb_build_array(
                    jsonb_build_object('name', $1::text, 'version', $2::text)
                )
            )
            "#,
        )
        .bind(name)
        .bind(version)
        .fetch_one(&self.pool)
        .await?;
        Ok(exists)
    }

    /// Delete an artifact.
    pub async fn delete(&self, id: Uuid) -> Result<bool, sqlx::Error> {
        let result = sqlx::query("DELETE FROM artifacts WHERE id = $1")
            .bind(id)
            .execute(&self.pool)
            .await?;
        Ok(result.rows_affected() > 0)
    }

    /// Link an artifact to a spec revision.
    pub async fn link_to_spec(
        &self,
        artifact_id: Uuid,
        spec_id: Uuid,
        spec_revision: i32,
    ) -> Result<(), sqlx::Error> {
        sqlx::query(
            r#"
            INSERT INTO artifact_specs (artifact_id, spec_id, spec_revision)
            VALUES ($1, $2, $3)
            ON CONFLICT (artifact_id, spec_id) DO NOTHING
            "#,
        )
        .bind(artifact_id)
        .bind(spec_id)
        .bind(spec_revision)
        .execute(&self.pool)
        .await?;
        Ok(())
    }
}