composefs_oci/

oci_image.rs

//! OCI image and artifact storage for composefs.
//!
//! This module provides native OCI storage in composefs repositories. The key insight
//! is that OCI is a simple, extensible format that can represent any content - not just
//! container images. By standardizing on OCI, we get:
//!
//! - A well-defined manifest format with content-addressed blobs
//! - Built-in support for signatures (cosign, notation)
//! - Existing tooling (skopeo, crane, oras)
//! - A clear GC model: manifests are roots, everything else is garbage-collectable
//!
//! # Storage Model
//!
//! ```text
//! streams/
//!   oci-manifest-sha256:abc...  -> objects/XX/YYY  (manifest splitstream)
//!   oci-config-sha256:def...    -> objects/XX/YYY  (config splitstream)  
//!   oci-layer-sha256:ghi...     -> objects/XX/YYY  (layer splitstream)
//!   refs/
//!     oci/
//!       myimage:latest          -> ../../oci-manifest-sha256:abc...  (GC root!)
//!       myimage:v1.0            -> ../../oci-manifest-sha256:xyz...
//! ```
//!
//! Named references under `refs/oci/` act as GC roots. Manifests without references
//! will be garbage collected along with their unreferenced configs and layers.
//!
//! # Container Images vs Artifacts
//!
//! Container images have:
//! - Config with `application/vnd.oci.image.config.v1+json` mediaType
//! - Layers that are tar archives (gzip, zstd, or uncompressed)
//!
//! Artifacts can have:
//! - Any config mediaType (or empty config)
//! - Any blob types as "layers"
//!
//! This module handles both transparently. Use `is_container_image()` to check.

use std::collections::{HashMap, HashSet};
use std::sync::Arc;

use anyhow::{Context, Result, ensure};
use containers_image_proxy::oci_spec::image::{
    Descriptor, Digest as OciDigest, ImageConfiguration, ImageManifest, MediaType,
};
use rustix::fs::{AtFlags, Dir, Mode, OFlags, openat, readlinkat, unlinkat};
use rustix::io::Errno;
use serde::Serialize;

use composefs::{
    erofs::format::{FormatEpoch, FormatVersion},
    fsverity::FsVerityHashValue,
    repository::Repository,
};

use crate::ContentAndVerity;
use crate::layer::is_tar_media_type;
use crate::skopeo::{OCI_BLOB_CONTENT_TYPE, OCI_CONFIG_CONTENT_TYPE, OCI_MANIFEST_CONTENT_TYPE};

/// Error marker: an OCI reference (tag) does not exist.
#[derive(Debug, thiserror::Error)]
#[error("OCI reference not found: {name}")]
pub struct OciRefNotFound {
    /// The reference name that was not found.
    pub name: String,
}

/// Error marker: an OCI manifest/image does not exist in the repository.
#[derive(Debug, thiserror::Error)]
#[error("OCI image not found: {digest}")]
pub struct OciImageNotFound {
    /// The manifest digest that was not found.
    pub digest: String,
}

/// Data and named refs from a splitstream with external object storage.
type ExternalData<ObjectID> = (Vec<u8>, HashMap<Box<str>, ObjectID>);

/// Open a splitstream that stores its payload as a single external object.
///
/// Manifests, configs, and blobs are stored as external objects (not inline)
/// so that fsverity can be independently enabled on the raw content. This
/// function opens the splitstream, verifies it contains exactly one external
/// object reference, and returns that object's data along with the stream's
/// named refs (used for GC reachability to configs and layers).
pub(crate) fn read_external_splitstream<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    content_id: &str,
    verity: Option<&ObjectID>,
    expected_content_type: Option<u64>,
) -> Result<ExternalData<ObjectID>> {
    let mut stream = repo.open_stream(content_id, verity, expected_content_type)?;

    let mut object_refs = Vec::new();
    stream.get_object_refs(|id| object_refs.push(id.clone()))?;
    ensure!(
        object_refs.len() == 1,
        "Expected exactly 1 external object in splitstream, got {}",
        object_refs.len()
    );

    let data = repo.read_object(&object_refs[0])?;
    let named_refs = stream.into_named_refs();
    Ok((data, named_refs))
}

/// Prefix for OCI image references in the repository.
pub const OCI_REF_PREFIX: &str = "oci/";

/// An OCI image or artifact stored in a composefs repository.
///
/// This type provides access to the complete OCI structure including
/// manifest, config, and layer/blob references. All metadata is stored
/// locally, eliminating network access for queries.
#[derive(Debug)]
pub struct OciImage<ObjectID: FsVerityHashValue> {
    /// The manifest digest (sha256 content hash)
    manifest_digest: OciDigest,
    /// The parsed OCI manifest
    manifest: ImageManifest,
    /// The config digest (sha256 content hash)
    config_digest: OciDigest,
    /// The fs-verity ID of the config splitstream
    config_verity: ObjectID,
    /// The parsed OCI config (may be empty for artifacts)
    config: Option<ImageConfiguration>,
    /// Map from layer diff_id to its fs-verity object ID
    layer_refs: HashMap<Box<str>, ObjectID>,
    /// The V2 EROFS image ObjectID linked to this config, if any
    image_ref: Option<ObjectID>,
    /// The V1 EROFS image ObjectID linked to this config, if any
    image_ref_v1: Option<ObjectID>,
    /// The V2 boot EROFS image ObjectID linked to this config, if any
    boot_image_ref: Option<ObjectID>,
    /// The V1 boot EROFS image ObjectID linked to this config, if any
    boot_image_ref_v1: Option<ObjectID>,
    /// The fs-verity ID of the manifest splitstream
    manifest_verity: ObjectID,
}

impl<ObjectID: FsVerityHashValue> OciImage<ObjectID> {
    /// Opens an OCI image by its manifest digest.
    ///
    /// If `verity` is provided, it's used directly for fast lookup.
    /// Otherwise, the content is verified against the digest.
    pub fn open(
        repo: &Repository<ObjectID>,
        manifest_digest: &OciDigest,
        verity: Option<&ObjectID>,
    ) -> Result<Self> {
        let manifest_id = manifest_identifier(manifest_digest);
        let (data, named_refs) =
            read_external_splitstream(repo, &manifest_id, verity, Some(OCI_MANIFEST_CONTENT_TYPE))?;

        // Verify content hash when no verity was provided
        if verity.is_none() {
            let computed = hash_sha256(&data);
            ensure!(
                *manifest_digest == computed,
                "Manifest integrity failed: expected {manifest_digest}, got {computed}"
            );
        }

        let manifest = ImageManifest::from_reader(&data[..])?;

        let config_digest = manifest.config().digest().clone();
        let config_key = format!("config:{config_digest}");
        let config_verity = named_refs
            .get(config_key.as_str())
            .context("Manifest missing config reference")?
            .clone();

        let config_id = crate::config_identifier(&config_digest);
        let (config_data, config_named_refs) = read_external_splitstream(
            repo,
            &config_id,
            Some(&config_verity),
            Some(OCI_CONFIG_CONTENT_TYPE),
        )?;

        // Try to parse as ImageConfiguration, but don't fail for artifacts
        let (config, mut layer_refs) = match manifest.config().media_type() {
            MediaType::ImageConfig => {
                let config = ImageConfiguration::from_reader(&config_data[..])?;
                (Some(config), config_named_refs)
            }
            _ => {
                // Artifact - layer refs are in the manifest's named refs.
                // Filter to only include refs matching known layer digests
                // from the manifest, rather than removing the config key
                // and hoping nothing else leaks through.
                let layer_digests: HashSet<&str> = manifest
                    .layers()
                    .iter()
                    .map(|d| d.digest().as_ref())
                    .collect();
                let refs = named_refs
                    .into_iter()
                    .filter(|(k, _)| layer_digests.contains(k.as_ref()))
                    .collect();
                (None, refs)
            }
        };

        // Strip the EROFS image refs from layer_refs (they're not layers)
        let image_ref = layer_refs.remove(crate::IMAGE_REF_KEY);
        let image_ref_v1 = layer_refs.remove(crate::IMAGE_REF_KEY_V1);
        let boot_image_ref = layer_refs.remove(crate::BOOT_IMAGE_REF_KEY);
        let boot_image_ref_v1 = layer_refs.remove(crate::BOOT_IMAGE_REF_KEY_V1);

        let manifest_verity = if let Some(v) = verity {
            v.clone()
        } else {
            match repo.has_stream(&manifest_id)? {
                Some(v) => v,
                None => {
                    return Err(anyhow::Error::new(OciImageNotFound {
                        digest: manifest_digest.to_string(),
                    }));
                }
            }
        };

        Ok(Self {
            manifest_digest: manifest_digest.clone(),
            manifest,
            config_digest,
            config_verity,
            config,
            layer_refs,
            image_ref,
            image_ref_v1,
            boot_image_ref,
            boot_image_ref_v1,
            manifest_verity,
        })
    }

    /// Opens an OCI image by its tag/reference name.
    pub fn open_ref(repo: &Repository<ObjectID>, name: &str) -> Result<Self> {
        let (manifest_digest, verity) = resolve_ref(repo, name)?;
        Self::open(repo, &manifest_digest, Some(&verity))
    }

    /// Returns true if this is a container image (vs an artifact).
    pub fn is_container_image(&self) -> bool {
        matches!(self.manifest.config().media_type(), MediaType::ImageConfig)
    }

    /// Returns the manifest digest.
    pub fn manifest_digest(&self) -> &OciDigest {
        &self.manifest_digest
    }

    /// Returns the manifest fs-verity hash.
    pub fn manifest_verity(&self) -> &ObjectID {
        &self.manifest_verity
    }

    /// Returns the OCI manifest.
    pub fn manifest(&self) -> &ImageManifest {
        &self.manifest
    }

    /// Returns the config digest.
    pub fn config_digest(&self) -> &OciDigest {
        &self.config_digest
    }

    /// Returns the config fs-verity hash.
    pub fn config_verity(&self) -> &ObjectID {
        &self.config_verity
    }

    /// Returns the OCI config, if this is a container image.
    pub fn config(&self) -> Option<&ImageConfiguration> {
        self.config.as_ref()
    }

    /// Returns the layer refs map (diff_id → fs-verity ObjectID).
    pub fn layer_refs(&self) -> &HashMap<Box<str>, ObjectID> {
        &self.layer_refs
    }

    /// Returns the EROFS image ObjectID for `version`, if present.
    ///
    /// Maps `version` to its on-disk storage slot via [`FormatVersion::epoch`]:
    /// epoch1 (V0/V1) resolves the V1 ref; epoch2 (V2) resolves the V2 ref.
    /// No fallback — returns `None` if that specific format was not generated.
    pub fn image_ref(&self, version: FormatVersion) -> Option<&ObjectID> {
        match version.epoch() {
            FormatEpoch::Epoch1 => self.image_ref_v1.as_ref(),
            FormatEpoch::Epoch2 => self.image_ref.as_ref(),
        }
    }

    /// Returns the V2 EROFS image ObjectID linked to this config, if any.
    pub fn image_ref_v2(&self) -> Option<&ObjectID> {
        self.image_ref.as_ref()
    }

    /// Returns the V1 EROFS image ObjectID linked to this config, if any.
    pub fn image_ref_v1(&self) -> Option<&ObjectID> {
        self.image_ref_v1.as_ref()
    }

    /// Returns the boot EROFS image ObjectID for `version`, if present.
    ///
    /// Maps `version` to its on-disk storage slot via [`FormatVersion::epoch`].
    /// No fallback — returns `None` if that specific format was not generated.
    pub fn boot_image_ref(&self, version: FormatVersion) -> Option<&ObjectID> {
        match version.epoch() {
            FormatEpoch::Epoch1 => self.boot_image_ref_v1.as_ref(),
            FormatEpoch::Epoch2 => self.boot_image_ref.as_ref(),
        }
    }

    /// Returns the V2 boot EROFS image ObjectID linked to this config, if any.
    pub fn boot_image_ref_v2(&self) -> Option<&ObjectID> {
        self.boot_image_ref.as_ref()
    }

    /// Returns the V1 boot EROFS image ObjectID linked to this config, if any.
    pub fn boot_image_ref_v1(&self) -> Option<&ObjectID> {
        self.boot_image_ref_v1.as_ref()
    }

    /// Returns the image architecture (empty string for artifacts).
    pub fn architecture(&self) -> String {
        self.config
            .as_ref()
            .map(|c| c.architecture().to_string())
            .unwrap_or_default()
    }

    /// Returns the image OS (empty string for artifacts).
    pub fn os(&self) -> String {
        self.config
            .as_ref()
            .map(|c| c.os().to_string())
            .unwrap_or_default()
    }

    /// Returns the creation timestamp.
    pub fn created(&self) -> Option<&str> {
        self.config.as_ref().and_then(|c| c.created().as_deref())
    }

    /// Opens an artifact layer's backing object by index, returning a
    /// read-only file descriptor to the raw blob data.
    ///
    /// This only works for non-tar layers (OCI artifacts). Returns an
    /// error for tar layers — use the splitstream API for those.
    pub fn open_layer_fd(
        &self,
        repo: &Repository<ObjectID>,
        index: usize,
    ) -> Result<rustix::fd::OwnedFd> {
        let descriptor = self
            .manifest
            .layers()
            .get(index)
            .with_context(|| format!("Layer index {index} out of range"))?;

        ensure!(
            !is_tar_media_type(descriptor.media_type()),
            "open_layer_fd does not support tar layers (media type: {}); \
             use the splitstream API instead",
            descriptor.media_type()
        );

        let diff_id = descriptor.digest();
        let layer_verity = self
            .layer_verity(diff_id.as_ref())
            .with_context(|| format!("No verity for layer {diff_id}"))?;

        let content_id = crate::layer_identifier(diff_id);
        let mut stream = repo.open_stream(&content_id, Some(layer_verity), None)?;

        // Artifact layers are stored as a single object; the splitstream
        // exists only for GC tracking.
        let mut object_refs = vec![];
        stream.get_object_refs(|id| object_refs.push(id.clone()))?;
        ensure!(
            object_refs.len() == 1,
            "Expected exactly 1 external ref for artifact layer, got {}",
            object_refs.len()
        );
        repo.open_object(&object_refs[0])
    }

    /// Returns the layer diff_ids (for container images).
    pub fn layer_diff_ids(&self) -> Vec<&str> {
        self.config
            .as_ref()
            .map(|c| c.rootfs().diff_ids().iter().map(|s| s.as_str()).collect())
            .unwrap_or_default()
    }

    /// Returns the fs-verity ID for a layer.
    pub fn layer_verity(&self, diff_id: &str) -> Option<&ObjectID> {
        self.layer_refs.get(diff_id)
    }

    /// Returns layer descriptors from the manifest.
    pub fn layer_descriptors(&self) -> &[Descriptor] {
        self.manifest.layers()
    }

    /// Returns a label from the config.
    pub fn label(&self, key: &str) -> Option<&str> {
        self.config.as_ref().and_then(|c| {
            c.config()
                .as_ref()
                .and_then(|cfg| cfg.labels().as_ref())
                .and_then(|labels| labels.get(key).map(|s| s.as_str()))
        })
    }

    /// Returns all labels from the config.
    pub fn labels(&self) -> Option<&HashMap<String, String>> {
        self.config
            .as_ref()
            .and_then(|c| c.config().as_ref())
            .and_then(|cfg| cfg.labels().as_ref())
    }

    /// Reads the raw manifest JSON bytes from the repository.
    ///
    /// This retrieves the original manifest JSON as stored, which may differ
    /// slightly from re-serializing the parsed manifest (e.g., whitespace).
    pub fn read_manifest_json(&self, repo: &Repository<ObjectID>) -> Result<Vec<u8>> {
        let manifest_id = manifest_identifier(&self.manifest_digest);
        let (data, _) = read_external_splitstream(
            repo,
            &manifest_id,
            Some(&self.manifest_verity),
            Some(OCI_MANIFEST_CONTENT_TYPE),
        )?;
        Ok(data)
    }

    /// Reads the raw config JSON bytes from the repository.
    ///
    /// This retrieves the original config JSON as stored, which may differ
    /// slightly from re-serializing the parsed config (e.g., whitespace).
    pub fn read_config_json(&self, repo: &Repository<ObjectID>) -> Result<Vec<u8>> {
        let config_id = crate::config_identifier(&self.config_digest);

        let (data, _) = read_external_splitstream(
            repo,
            &config_id,
            Some(&self.config_verity),
            Some(OCI_CONFIG_CONTENT_TYPE),
        )?;
        Ok(data)
    }

    /// Returns the full inspect output as a JSON value.
    ///
    /// This includes the manifest, config, and referrers in a single JSON object.
    /// The manifest and config are included as their original JSON structure.
    pub fn inspect_json(&self, repo: &Repository<ObjectID>) -> Result<serde_json::Value> {
        let manifest_json = self.read_manifest_json(repo)?;
        let config_json = self.read_config_json(repo)?;
        let referrers = list_referrers(repo, &self.manifest_digest)?;

        let manifest_value: serde_json::Value = serde_json::from_slice(&manifest_json)?;
        let config_value: serde_json::Value = serde_json::from_slice(&config_json)?;

        let referrers_value: Vec<serde_json::Value> = referrers
            .iter()
            .map(|(digest, _verity)| serde_json::json!({ "digest": digest }))
            .collect();

        let mut result = serde_json::json!({
            "manifest": manifest_value,
            "config": config_value,
            "referrers": referrers_value,
        });

        if let Some(erofs_id) = self.image_ref(repo.erofs_version()) {
            result["composefs_erofs"] = serde_json::json!(erofs_id.to_hex());
        }

        if let Some(boot_id) = self.boot_image_ref(repo.erofs_version()) {
            result["composefs_boot_erofs"] = serde_json::json!(boot_id.to_hex());
        }

        Ok(result)
    }
}

// =============================================================================
// Reference Management (GC Roots)
// =============================================================================

/// Validate that a ref name doesn't start with `@`, which is reserved as
/// the digest prefix (e.g. `@sha256:abc...`).
fn validate_ref_name(name: &str) -> Result<()> {
    ensure!(
        !name.starts_with('@'),
        "Invalid ref name {name:?}: leading '@' is reserved for digest references"
    );
    Ok(())
}

/// Tags an image with a name, making it a GC root.
///
/// The name should be in the format `image:tag` or just `image` (implies `:latest`).
/// Names must not contain `@`, which is reserved for digest references.
pub fn tag_image<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    manifest_digest: &OciDigest,
    name: &str,
) -> Result<()> {
    validate_ref_name(name)?;
    let manifest_id = manifest_identifier(manifest_digest);
    let ref_name = oci_ref_path(name);
    repo.name_stream(&manifest_id, &ref_name)
}

/// Removes a tag from an image.
///
/// The image data is not deleted; it becomes eligible for garbage collection
/// if no other references point to it.
pub fn untag_image<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    name: &str,
) -> Result<()> {
    let ref_path = format!("streams/refs/{}", oci_ref_path(name));
    unlinkat(repo.repo_fd(), &ref_path, AtFlags::empty())
        .with_context(|| format!("Failed to remove tag {name}"))?;
    Ok(())
}

/// Resolves a reference name to (manifest_digest, verity).
pub fn resolve_ref<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    name: &str,
) -> Result<(OciDigest, ObjectID)> {
    let ref_path = format!("streams/refs/{}", oci_ref_path(name));

    // Read the symlink to get the manifest path
    let target = match readlinkat(repo.repo_fd(), &ref_path, vec![]) {
        Ok(t) => t,
        Err(Errno::NOENT) => {
            return Err(anyhow::Error::new(OciRefNotFound {
                name: name.to_string(),
            }));
        }
        Err(e) => {
            return Err(e).with_context(|| format!("Reference {name} not found"));
        }
    };

    let target_str = target
        .to_str()
        .context("Invalid UTF-8 in reference target")?;

    // Extract manifest digest from path like "../../oci-manifest-sha256:abc"
    let manifest_part = target_str
        .rsplit('/')
        .next()
        .context("Invalid reference target")?;

    let digest_str = manifest_part
        .strip_prefix("oci-manifest-")
        .with_context(|| format!("Invalid manifest reference: {manifest_part}"))?;

    let digest: OciDigest = digest_str
        .parse()
        .with_context(|| format!("Invalid OCI digest in reference: {digest_str}"))?;

    // Get the verity by looking up the manifest
    let verity = repo
        .has_stream(&manifest_identifier(&digest))?
        .with_context(|| format!("Manifest {digest} not found"))?;

    Ok((digest, verity))
}

/// Lists all tagged OCI images.
///
/// Returns (name, manifest_digest) pairs for each tag.
pub fn list_refs<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
) -> Result<Vec<(String, OciDigest)>> {
    let mut refs = Vec::new();

    // Use the repository's ref listing method
    for (name, target) in repo.list_stream_refs("oci")? {
        // Extract manifest digest from target path
        let manifest_part = target.rsplit('/').next().unwrap_or(&target);
        if let Some(digest_str) = manifest_part.strip_prefix("oci-manifest-")
            && let Ok(digest) = digest_str.parse()
        {
            // Decode the tag name from filesystem-safe encoding
            refs.push((decode_tag(&name), digest));
        }
    }

    Ok(refs)
}

/// Summary information about a stored OCI image.
/// FIXME change this to just have a struct of manifest+config JSON
/// plus a few helper methods. We shouldn't be re-parsing created timestamp here
/// callers should directly access that etc
#[derive(Debug, Clone, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct ImageInfo {
    /// The tag/name of the image
    pub name: String,
    /// The manifest digest
    pub manifest_digest: OciDigest,
    /// Whether this is a container image (vs artifact)
    pub is_container: bool,
    /// Architecture (empty for artifacts)
    pub architecture: String,
    /// OS (empty for artifacts)
    pub os: String,
    /// Creation timestamp
    pub created: Option<String>,
    /// Number of layers/blobs
    pub layer_count: usize,
    /// Number of OCI referrers (signatures, attestations, etc.)
    pub referrer_count: usize,
}

/// Lists all tagged images with their metadata.
pub fn list_images<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
) -> Result<Vec<ImageInfo>> {
    let mut images = Vec::new();

    for (name, digest) in list_refs(repo)? {
        match OciImage::open(repo, &digest, None) {
            Ok(img) => {
                let referrer_count = list_referrers(repo, &digest).map(|r| r.len()).unwrap_or(0);
                images.push(ImageInfo {
                    name,
                    manifest_digest: digest,
                    is_container: img.is_container_image(),
                    architecture: img.architecture(),
                    os: img.os(),
                    created: img.created().map(String::from),
                    layer_count: img.layer_descriptors().len(),
                    referrer_count,
                });
            }
            Err(e) => {
                tracing::warn!("skipping image {name}: {e:#}");
                continue;
            }
        }
    }

    Ok(images)
}

// =============================================================================
// Manifest Storage
// =============================================================================

/// Writes a manifest to the repository.
///
/// The manifest JSON is stored as an external object (not inline) so that
/// fsverity can be independently enabled on it. This is important for signing:
/// a signature can reference the fsverity digest of the manifest content directly.
///
/// The manifest becomes a GC root only if a `reference` name is provided.
/// The reference name must not contain `@`, which is reserved for digest
/// references.
pub fn write_manifest<ObjectID: FsVerityHashValue, S: AsRef<str>>(
    repo: &Arc<Repository<ObjectID>>,
    manifest: &ImageManifest,
    manifest_digest: &OciDigest,
    config_verity: &ObjectID,
    layer_verities: &[(S, ObjectID)],
    reference: Option<&str>,
) -> Result<ContentAndVerity<ObjectID>> {
    if let Some(name) = reference {
        validate_ref_name(name)?;
    }

    let content_id = manifest_identifier(manifest_digest);

    if let Some(verity) = repo.has_stream(&content_id)? {
        // Already exists - just add the reference if requested
        if let Some(name) = reference {
            tag_image(repo, manifest_digest, name)?;
        }
        return Ok((manifest_digest.clone(), verity));
    }

    let json = manifest.to_string()?;
    let json_bytes = json.as_bytes();

    let computed = hash_sha256(json_bytes);
    ensure!(
        *manifest_digest == computed,
        "Manifest digest mismatch: expected {manifest_digest}, got {computed}"
    );

    let mut stream = repo.create_stream(OCI_MANIFEST_CONTENT_TYPE)?;

    let config_key = format!("config:{}", manifest.config().digest());
    stream.add_named_stream_ref(&config_key, config_verity);

    for (diff_id, verity) in layer_verities {
        stream.add_named_stream_ref(diff_id.as_ref(), verity);
    }

    stream.write_external(json_bytes)?;

    let oci_ref = reference.map(oci_ref_path);
    let id = repo.write_stream(stream, &content_id, oci_ref.as_deref())?;

    Ok((computed, id))
}

/// Rewrites a manifest splitstream with updated named refs.
///
/// Unlike [`write_manifest`], this always writes the splitstream even if the
/// content identifier already exists. This is needed when the manifest JSON
/// hasn't changed but the config splitstream's verity has (e.g., because an
/// EROFS image ref was added to the config).
///
/// If `reference` is provided, the manifest is also tagged with that name.
pub(crate) fn rewrite_manifest<ObjectID: FsVerityHashValue, S: AsRef<str>>(
    repo: &Arc<Repository<ObjectID>>,
    manifest_json: &[u8],
    manifest_digest: &OciDigest,
    config_verity: &ObjectID,
    layer_verities: &[(S, ObjectID)],
    reference: Option<&str>,
) -> Result<(OciDigest, ObjectID)> {
    let content_id = manifest_identifier(manifest_digest);

    let config_digest = {
        let manifest = ImageManifest::from_reader(manifest_json)?;
        manifest.config().digest().to_string()
    };

    let mut stream = repo.create_stream(OCI_MANIFEST_CONTENT_TYPE)?;

    let config_key = format!("config:{config_digest}");
    stream.add_named_stream_ref(&config_key, config_verity);

    for (diff_id, verity) in layer_verities {
        stream.add_named_stream_ref(diff_id.as_ref(), verity);
    }

    stream.write_external(manifest_json)?;

    let oci_ref = reference.map(oci_ref_path);
    let id = repo.write_stream(stream, &content_id, oci_ref.as_deref())?;

    Ok((manifest_digest.clone(), id))
}

/// Checks if a manifest exists.
pub fn has_manifest<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    manifest_digest: &OciDigest,
) -> Result<Option<ObjectID>> {
    repo.has_stream(&manifest_identifier(manifest_digest))
}

/// Returns the content identifier for a manifest.
pub fn manifest_identifier(digest: &OciDigest) -> String {
    format!("oci-manifest-{digest}")
}

/// Returns the reference path for an OCI name.
fn oci_ref_path(name: &str) -> String {
    format!("{OCI_REF_PREFIX}{}", encode_tag(name))
}

/// Encode a tag name for safe filesystem storage.
///
/// Uses percent-encoding for characters that are problematic in paths:
/// - `/` becomes `%2F`
/// - `%` becomes `%25` (must be first to avoid double-encoding)
fn encode_tag(name: &str) -> String {
    name.replace('%', "%25").replace('/', "%2F")
}

/// Decode a tag name from filesystem storage.
///
/// Uses single-pass percent decoding to avoid order-dependent replacement bugs.
/// For example, `%252F` must decode to `%2F` (not `/`).
fn decode_tag(encoded: &str) -> String {
    let mut result = String::with_capacity(encoded.len());
    let mut chars = encoded.chars().peekable();
    while let Some(c) = chars.next() {
        if c == '%' {
            let hex: String = chars.by_ref().take(2).collect();
            match hex.as_str() {
                "2F" => result.push('/'),
                "25" => result.push('%'),
                _ => {
                    result.push('%');
                    result.push_str(&hex);
                }
            }
        } else {
            result.push(c);
        }
    }
    result
}

/// Computes sha256 content hash, returning an OCI `Digest`.
fn hash_sha256(bytes: &[u8]) -> OciDigest {
    crate::sha256_content_digest(bytes)
}

// =============================================================================
// Arbitrary Blob Storage (for OCI Artifacts)
// =============================================================================

/// Returns the content identifier for an arbitrary blob.
pub fn blob_identifier(digest: &OciDigest) -> String {
    format!("oci-blob-{digest}")
}

/// Writes an arbitrary blob to the repository.
///
/// This is used for OCI artifacts with non-tar media types. The blob is stored
/// as an external object so that fsverity can be independently enabled on the
/// raw content.
///
/// Returns (sha256 digest, fs-verity hash).
pub fn write_blob<ObjectID: FsVerityHashValue>(
    repo: &Arc<Repository<ObjectID>>,
    data: &[u8],
) -> Result<(OciDigest, ObjectID)> {
    let digest = hash_sha256(data);
    let content_id = blob_identifier(&digest);

    if let Some(verity) = repo.has_stream(&content_id)? {
        return Ok((digest, verity));
    }

    let mut stream = repo.create_stream(OCI_BLOB_CONTENT_TYPE)?;
    stream.write_external(data)?;
    let verity = repo.write_stream(stream, &content_id, None)?;

    Ok((digest, verity))
}

/// Opens an arbitrary blob from the repository.
///
/// Returns the blob data. If verity is provided, it's used for fast lookup;
/// otherwise, the content hash is verified against the digest.
pub fn open_blob<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    digest: &OciDigest,
    verity: Option<&ObjectID>,
) -> Result<Vec<u8>> {
    let content_id = blob_identifier(digest);
    let (data, _named_refs) =
        read_external_splitstream(repo, &content_id, verity, Some(OCI_BLOB_CONTENT_TYPE))?;

    if verity.is_none() {
        let computed = hash_sha256(&data);
        ensure!(
            *digest == computed,
            "Blob integrity failed: expected {digest}, got {computed}"
        );
    }

    Ok(data)
}

// =============================================================================
// Referrer Index (for OCI Artifacts with subject field)
// =============================================================================

/// Prefix for referrer index references.
const REFERRER_REF_PREFIX: &str = "oci-referrers/";

/// Records a referrer relationship: an artifact references a subject image.
///
/// Creates a symlink at `streams/refs/oci-referrers/{subject_digest}/{artifact_digest}`
/// pointing to the artifact's manifest stream. This enables discovery of all artifacts
/// that reference a given image (e.g. finding all signature artifacts for an image).
///
/// Both digests should be in the `sha256:...` format used by OCI.
pub fn add_referrer<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    subject_digest: &OciDigest,
    artifact_manifest_digest: &OciDigest,
) -> Result<()> {
    let subject_str: &str = subject_digest.as_ref();
    let artifact_str: &str = artifact_manifest_digest.as_ref();
    let ref_name = format!(
        "{REFERRER_REF_PREFIX}{}/{}",
        encode_tag(subject_str),
        encode_tag(artifact_str)
    );
    let manifest_id = manifest_identifier(artifact_manifest_digest);
    repo.name_stream(&manifest_id, &ref_name)
}

/// Lists all artifacts that reference the given subject manifest digest.
///
/// Returns `(artifact_manifest_digest, artifact_manifest_verity)` pairs for
/// each artifact that declared the subject as its referrer. The digests are
/// in `sha256:...` format.
pub fn list_referrers<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    subject_digest: &OciDigest,
) -> Result<Vec<(OciDigest, ObjectID)>> {
    let subject_str: &str = subject_digest.as_ref();
    let prefix = format!("{REFERRER_REF_PREFIX}{}", encode_tag(subject_str));

    let mut referrers = Vec::new();

    for (name, target) in repo.list_stream_refs(&prefix)? {
        // The name is the encoded artifact manifest digest
        let artifact_digest_str = decode_tag(&name);

        // Extract verity from the symlink target — it points to
        // a manifest stream path like "../../oci-manifest-sha256:abc..."
        let manifest_part = target.rsplit('/').next().unwrap_or(&target);
        if let Some(digest) = manifest_part.strip_prefix("oci-manifest-") {
            // Verify consistency: the ref name should match the target
            if digest != artifact_digest_str {
                continue;
            }
        }

        // Look up the verity for this manifest
        let artifact_digest: OciDigest = artifact_digest_str
            .parse()
            .with_context(|| format!("Parsing referrer digest '{artifact_digest_str}'"))?;
        match repo.has_stream(&manifest_identifier(&artifact_digest))? {
            Some(verity) => referrers.push((artifact_digest, verity)),
            None => {
                continue;
            }
        }
    }

    Ok(referrers)
}

/// Removes a specific referrer index entry.
///
/// Idempotent — returns Ok if the entry doesn't exist.
pub fn remove_referrer<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    subject_digest: &OciDigest,
    artifact_digest: &OciDigest,
) -> Result<()> {
    let subject_str: &str = subject_digest.as_ref();
    let artifact_str: &str = artifact_digest.as_ref();
    let ref_path = format!(
        "streams/refs/{REFERRER_REF_PREFIX}{}/{}",
        encode_tag(subject_str),
        encode_tag(artifact_str)
    );
    match unlinkat(repo.repo_fd(), &ref_path, AtFlags::empty()) {
        Ok(()) => Ok(()),
        Err(Errno::NOENT) => Ok(()),
        Err(e) => Err(e).with_context(|| format!("Failed to remove referrer {artifact_digest}")),
    }
}

/// Removes all referrer index entries for a subject.
///
/// Removes each referrer symlink and tries to remove the empty subject
/// directory afterwards. Idempotent — returns Ok if no entries exist.
pub fn remove_referrers_for_subject<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    subject_digest: &OciDigest,
) -> Result<()> {
    let referrers = list_referrers(repo, subject_digest)?;
    for (artifact_digest, _verity) in &referrers {
        remove_referrer(repo, subject_digest, artifact_digest)?;
    }
    // Try to remove the now-empty subject directory (ignore errors)
    let subject_str: &str = subject_digest.as_ref();
    let subject_dir = format!(
        "streams/refs/{REFERRER_REF_PREFIX}{}",
        encode_tag(subject_str)
    );
    let _ = unlinkat(repo.repo_fd(), &subject_dir, AtFlags::REMOVEDIR);
    Ok(())
}

/// Removes referrer index entries whose subject manifest no longer exists.
///
/// When a subject image is untagged and garbage collected, its referrer
/// artifacts become orphaned — their referrer symlinks under
/// `streams/refs/oci-referrers/{subject_digest}/` still act as GC roots,
/// preventing the artifact manifests from being collected.
///
/// Call this **before** running GC to ensure orphaned referrer artifacts
/// are also eligible for collection. The typical workflow is:
///
/// ```text
/// cleanup_dangling_referrers(&repo)?;
/// repo.gc(&[])?;
/// ```
///
/// Returns the number of referrer entries removed.
pub fn cleanup_dangling_referrers<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
) -> Result<u64> {
    let referrers_path = format!("streams/refs/{REFERRER_REF_PREFIX}");

    // Open the oci-referrers directory; if it doesn't exist, there's nothing to do
    let referrers_dir = match openat(
        repo.repo_fd(),
        &*referrers_path,
        OFlags::RDONLY | OFlags::DIRECTORY | OFlags::CLOEXEC,
        Mode::empty(),
    ) {
        Ok(fd) => fd,
        Err(Errno::NOENT) => return Ok(0),
        Err(e) => return Err(e).context("Opening oci-referrers directory")?,
    };

    let mut removed = 0u64;

    // Collect subject directory names first to avoid borrowing issues
    let mut subject_dirs = Vec::new();
    for item in Dir::read_from(&referrers_dir).context("Reading oci-referrers directory")? {
        let entry = item.context("Reading oci-referrers entry")?;
        let name = entry.file_name();
        if name == c"." || name == c".." {
            continue;
        }
        if let Ok(s) = std::str::from_utf8(name.to_bytes()) {
            subject_dirs.push(s.to_string());
        }
    }

    for encoded_subject in &subject_dirs {
        let subject_digest_str = decode_tag(encoded_subject);
        let subject_digest: OciDigest = subject_digest_str
            .parse()
            .with_context(|| format!("Parsing subject digest '{subject_digest_str}'"))?;

        // Check if the subject manifest still exists in the repository
        if has_manifest(repo, &subject_digest)?.is_some() {
            continue;
        }

        // Subject is gone — remove all referrer entries in this directory
        let subject_dir_fd = match openat(
            &referrers_dir,
            encoded_subject.as_str(),
            OFlags::RDONLY | OFlags::DIRECTORY | OFlags::CLOEXEC,
            Mode::empty(),
        ) {
            Ok(fd) => fd,
            Err(Errno::NOENT) => continue,
            Err(e) => {
                return Err(e)
                    .context(format!("Opening referrer subject dir {encoded_subject}"))?;
            }
        };

        for item in Dir::read_from(&subject_dir_fd).context("Reading referrer subject directory")? {
            let entry = item.context("Reading referrer entry")?;
            let name = entry.file_name();
            if name == c"." || name == c".." {
                continue;
            }
            unlinkat(&subject_dir_fd, name, AtFlags::empty())
                .with_context(|| format!("Removing referrer entry {name:?}"))?;
            removed += 1;
        }

        // Remove the now-empty subject directory
        unlinkat(&referrers_dir, encoded_subject.as_str(), AtFlags::REMOVEDIR)
            .with_context(|| format!("Removing empty referrer subject dir {encoded_subject}"))?;
    }

    Ok(removed)
}

// =============================================================================
// Filesystem Consistency Checks (fsck)
// =============================================================================

/// A structured error found during an OCI-level consistency check.
///
/// Each variant corresponds to a specific kind of OCI metadata integrity
/// problem. The `Display` implementation produces a kebab-case error type
/// prefix followed by the image name/context and any relevant details.
#[derive(Debug, Clone, serde::Serialize, thiserror::Error)]
#[serde(tag = "type", rename_all = "kebab-case")]
#[non_exhaustive]
#[allow(missing_docs)]
pub enum OciFsckError {
    #[error("fsck: manifest-read-failed: {name}: {detail}")]
    ManifestReadFailed { name: String, detail: String },

    #[error("fsck: manifest-digest-mismatch: {name}: expected {expected}, got {actual}")]
    ManifestDigestMismatch {
        name: String,
        expected: String,
        actual: String,
    },

    #[error("fsck: manifest-parse-failed: {name}: {detail}")]
    ManifestParseFailed { name: String, detail: String },

    #[error("fsck: config-ref-missing: {name}: {digest}")]
    ConfigRefMissing { name: String, digest: String },

    #[error("fsck: config-read-failed: {name}: {detail}")]
    ConfigReadFailed { name: String, detail: String },

    #[error("fsck: config-digest-mismatch: {name}: expected {expected}, got {actual}")]
    ConfigDigestMismatch {
        name: String,
        expected: String,
        actual: String,
    },

    #[error("fsck: config-parse-failed: {name}: {detail}")]
    ConfigParseFailed { name: String, detail: String },

    #[error("fsck: layer-ref-missing: {name}: {diff_id}")]
    #[serde(rename_all = "camelCase")]
    LayerRefMissing { name: String, diff_id: String },

    #[error("fsck: layer-stream-missing: {name}: {diff_id}")]
    #[serde(rename_all = "camelCase")]
    LayerStreamMissing { name: String, diff_id: String },

    #[error("fsck: layer-check-failed: {name}: {diff_id}: {detail}")]
    #[serde(rename_all = "camelCase")]
    LayerCheckFailed {
        name: String,
        diff_id: String,
        detail: String,
    },

    #[error("fsck: layer-object-missing: {name}: {diff_id}: {detail}")]
    #[serde(rename_all = "camelCase")]
    LayerObjectMissing {
        name: String,
        diff_id: String,
        detail: String,
    },

    #[error("fsck: seal-image-missing: {name}: {digest}: {detail}")]
    SealImageMissing {
        name: String,
        digest: String,
        detail: String,
    },

    #[error("fsck: artifact-layer-ref-missing: {name}: {digest}")]
    ArtifactLayerRefMissing { name: String, digest: String },

    #[error("fsck: artifact-layer-object-missing: {name}: {digest}: {detail}")]
    ArtifactLayerObjectMissing {
        name: String,
        digest: String,
        detail: String,
    },

    #[error("fsck: ref-resolve-failed: {name}: {detail}")]
    RefResolveFailed { name: String, detail: String },

    #[error("fsck: invalid-ref-name: {name}: leading '@' is reserved for digest references")]
    InvalidRefName { name: String },
}

/// Results from an OCI-level filesystem consistency check.
///
/// Returned by [`oci_fsck`] and [`oci_fsck_image`] to report integrity status
/// of OCI images stored in the repository. This includes checks at both the
/// OCI metadata level (manifest/config digests, layer references) and the
/// underlying repository level (object integrity, splitstream validity).
#[derive(Debug, Clone, Default, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct OciFsckResult {
    pub(crate) repo_result: composefs::repository::FsckResult,
    pub(crate) images_checked: u64,
    pub(crate) images_corrupted: u64,
    pub(crate) errors: Vec<OciFsckError>,
}

impl OciFsckResult {
    /// Returns true if no corruption or errors were found at any level.
    pub fn is_ok(&self) -> bool {
        debug_assert!(
            self.images_corrupted == 0 || !self.errors.is_empty(),
            "images_corrupted is non-zero but no OCI error messages recorded"
        );
        self.repo_result.is_ok() && self.errors.is_empty()
    }

    /// Results from the underlying repository fsck.
    pub fn repo_result(&self) -> &composefs::repository::FsckResult {
        &self.repo_result
    }

    /// Number of OCI images checked.
    pub fn images_checked(&self) -> u64 {
        self.images_checked
    }

    /// Number of OCI images with issues.
    pub fn images_corrupted(&self) -> u64 {
        self.images_corrupted
    }

    /// OCI-level errors found during the check.
    pub fn errors(&self) -> &[OciFsckError] {
        &self.errors
    }
}

impl std::fmt::Display for OciFsckResult {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.repo_result)?;
        writeln!(
            f,
            "oci images: {}/{} ok",
            self.images_checked.saturating_sub(self.images_corrupted),
            self.images_checked
        )?;
        if !self.errors.is_empty() {
            writeln!(f, "oci errors: {}", self.errors.len())?;
            for err in &self.errors {
                writeln!(f, "  - {err}")?;
            }
        }
        Ok(())
    }
}

/// Run a full OCI-aware consistency check on the repository.
///
/// This performs the underlying repository fsck (object integrity, splitstream
/// validation, symlink checks) and then additionally validates all tagged OCI
/// images: manifest digest verification, config digest verification, layer
/// reference existence, and seal consistency.
pub async fn oci_fsck<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
) -> Result<OciFsckResult> {
    let repo_result = repo.fsck().await?;
    let mut result = OciFsckResult {
        repo_result,
        ..Default::default()
    };

    // Check all tagged OCI images
    let refs = list_refs(repo).context("listing OCI refs")?;
    for (name, manifest_digest) in refs {
        if name.starts_with('@') {
            result.images_checked += 1;
            result.images_corrupted += 1;
            result
                .errors
                .push(OciFsckError::InvalidRefName { name: name.clone() });
            continue;
        }
        fsck_single_image(repo, &name, &manifest_digest, &mut result);
    }

    Ok(result)
}

/// Run an OCI-aware consistency check on a single image by tag name.
///
/// Performs the underlying repository fsck, then validates the specified image.
pub async fn oci_fsck_image<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    name: &str,
) -> Result<OciFsckResult> {
    let repo_result = repo.fsck().await?;
    let mut result = OciFsckResult {
        repo_result,
        ..Default::default()
    };

    let (manifest_digest, _verity) = match resolve_ref(repo, name) {
        Ok(v) => v,
        Err(e) => {
            result.images_corrupted += 1;
            result.images_checked += 1;
            result.errors.push(OciFsckError::RefResolveFailed {
                name: name.to_string(),
                detail: e.to_string(),
            });
            return Ok(result);
        }
    };

    fsck_single_image(repo, name, &manifest_digest, &mut result);
    Ok(result)
}

/// Internal: validate a single OCI image's metadata integrity.
fn fsck_single_image<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    name: &str,
    manifest_digest: &OciDigest,
    result: &mut OciFsckResult,
) {
    result.images_checked += 1;
    let error_count_before = result.errors.len();

    // 1. Verify manifest content hash
    let manifest_id = manifest_identifier(manifest_digest);
    let (manifest_data, manifest_named_refs) = match read_external_splitstream(
        repo,
        &manifest_id,
        None,
        Some(OCI_MANIFEST_CONTENT_TYPE),
    ) {
        Ok(v) => v,
        Err(e) => {
            result.images_corrupted += 1;
            result.errors.push(OciFsckError::ManifestReadFailed {
                name: name.to_string(),
                detail: e.to_string(),
            });
            return;
        }
    };

    let computed_digest = hash_sha256(&manifest_data);
    if *manifest_digest != computed_digest {
        result.images_corrupted += 1;
        result.errors.push(OciFsckError::ManifestDigestMismatch {
            name: name.to_string(),
            expected: manifest_digest.to_string(),
            actual: computed_digest.to_string(),
        });
        return;
    }

    // 2. Parse manifest
    let manifest = match ImageManifest::from_reader(&manifest_data[..]) {
        Ok(m) => m,
        Err(e) => {
            result.images_corrupted += 1;
            result.errors.push(OciFsckError::ManifestParseFailed {
                name: name.to_string(),
                detail: e.to_string(),
            });
            return;
        }
    };

    // 3. Verify config reference exists in manifest's named refs
    let config_digest = manifest.config().digest().clone();
    let config_key = format!("config:{config_digest}");
    let config_verity = match manifest_named_refs.get(config_key.as_str()) {
        Some(v) => v.clone(),
        None => {
            result.images_corrupted += 1;
            result.errors.push(OciFsckError::ConfigRefMissing {
                name: name.to_string(),
                digest: config_digest.to_string(),
            });
            return;
        }
    };

    // 4. Verify config content hash
    let config_id = crate::config_identifier(&config_digest);
    let (config_data, config_named_refs) = match read_external_splitstream(
        repo,
        &config_id,
        Some(&config_verity),
        Some(OCI_CONFIG_CONTENT_TYPE),
    ) {
        Ok(v) => v,
        Err(e) => {
            result.images_corrupted += 1;
            result.errors.push(OciFsckError::ConfigReadFailed {
                name: name.to_string(),
                detail: e.to_string(),
            });
            return;
        }
    };

    let computed_config = hash_sha256(&config_data);
    if config_digest != computed_config {
        result.images_corrupted += 1;
        result.errors.push(OciFsckError::ConfigDigestMismatch {
            name: name.to_string(),
            expected: config_digest.to_string(),
            actual: computed_config.to_string(),
        });
        return;
    }

    // 5. Parse config and verify layer references
    let is_container = matches!(manifest.config().media_type(), MediaType::ImageConfig);

    if is_container {
        let config = match ImageConfiguration::from_reader(&config_data[..]) {
            Ok(c) => c,
            Err(e) => {
                result.images_corrupted += 1;
                result.errors.push(OciFsckError::ConfigParseFailed {
                    name: name.to_string(),
                    detail: e.to_string(),
                });
                return;
            }
        };

        // Verify each layer diff_id has a corresponding named ref and stream
        for diff_id_str in config.rootfs().diff_ids() {
            let layer_verity = match config_named_refs.get(diff_id_str.as_str()) {
                Some(v) => v,
                None => {
                    result.errors.push(OciFsckError::LayerRefMissing {
                        name: name.to_string(),
                        diff_id: diff_id_str.to_string(),
                    });
                    continue;
                }
            };

            let diff_id: OciDigest = match diff_id_str.parse() {
                Ok(d) => d,
                Err(e) => {
                    result.errors.push(OciFsckError::LayerCheckFailed {
                        name: name.to_string(),
                        diff_id: diff_id_str.to_string(),
                        detail: format!("Invalid diff_id: {e}"),
                    });
                    continue;
                }
            };

            // Check the layer stream exists
            let layer_id = crate::layer_identifier(&diff_id);
            match repo.has_stream(&layer_id) {
                Ok(Some(_)) => {}
                Ok(None) => {
                    result.errors.push(OciFsckError::LayerStreamMissing {
                        name: name.to_string(),
                        diff_id: diff_id.to_string(),
                    });
                }
                Err(e) => {
                    result.errors.push(OciFsckError::LayerCheckFailed {
                        name: name.to_string(),
                        diff_id: diff_id.to_string(),
                        detail: e.to_string(),
                    });
                }
            }

            // Verify the layer's object exists
            match repo.open_object(layer_verity) {
                Ok(_) => {}
                Err(e) => {
                    result.errors.push(OciFsckError::LayerObjectMissing {
                        name: name.to_string(),
                        diff_id: diff_id.to_string(),
                        detail: e.to_string(),
                    });
                }
            }
        }

        // 6. If sealed, verify the seal image exists
        if let Some(seal_digest) = config.get_config_annotation("containers.composefs.fsverity") {
            match repo.open_image(seal_digest) {
                Ok(_) => {}
                Err(e) => {
                    result.errors.push(OciFsckError::SealImageMissing {
                        name: name.to_string(),
                        digest: seal_digest.to_string(),
                        detail: e.to_string(),
                    });
                }
            }
        }
    } else {
        // Artifact: verify layer references from manifest named refs
        for layer_desc in manifest.layers() {
            let layer_digest = layer_desc.digest().to_string();
            match manifest_named_refs.get(layer_digest.as_str()) {
                Some(verity) => {
                    // Verify the layer object exists
                    match repo.open_object(verity) {
                        Ok(_) => {}
                        Err(e) => {
                            result
                                .errors
                                .push(OciFsckError::ArtifactLayerObjectMissing {
                                    name: name.to_string(),
                                    digest: layer_digest,
                                    detail: e.to_string(),
                                });
                        }
                    }
                }
                None => {
                    result.errors.push(OciFsckError::ArtifactLayerRefMissing {
                        name: name.to_string(),
                        digest: layer_digest,
                    });
                }
            }
        }
    }

    // Count at most once per image
    if result.errors.len() > error_count_before {
        result.images_corrupted += 1;
    }
}

// =============================================================================
// Layer Inspection
// =============================================================================

/// Metadata about a layer stored in the repository.
#[derive(Debug, Clone, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct LayerInfo {
    /// The layer diff_id (sha256 hash of uncompressed content)
    pub diff_id: String,
    /// The fs-verity hash of the layer splitstream
    pub verity: String,
    /// Size of the uncompressed tar layer in bytes
    pub size: u64,
    /// Number of files/entries in the layer
    pub entry_count: usize,
    /// Splitstream metadata
    pub splitstream: SplitstreamInfo,
}

/// Metadata about the splitstream representation of a layer.
#[derive(Debug, Clone, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct SplitstreamInfo {
    /// Number of external object references (large files stored separately)
    pub external_objects: usize,
    /// Total size of external objects in bytes
    pub external_size: u64,
    /// Size of inline data in bytes (small files + tar headers)
    pub inline_size: u64,
}

/// Opens a layer by its diff_id and returns metadata about it.
///
/// The diff_id should be in the `sha256:...` format used by OCI.
pub fn layer_info<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    diff_id: &OciDigest,
) -> Result<LayerInfo> {
    let content_id = crate::layer_identifier(diff_id);
    let verity = repo
        .has_stream(&content_id)?
        .with_context(|| format!("Layer {diff_id} not found"))?;

    let mut stream = repo.open_stream(
        &content_id,
        Some(&verity),
        Some(crate::skopeo::TAR_LAYER_CONTENT_TYPE),
    )?;

    // Get the total size from the splitstream header (this is the merged/tar size)
    let size = stream.total_size;

    // Count external object references (this doesn't consume the stream)
    let mut external_objects = 0usize;
    stream.get_object_refs(|_| external_objects += 1)?;

    // Iterate entries and gather sizes
    let mut entry_count = 0usize;
    let mut external_size = 0u64;

    while let Some(entry) = crate::tar::get_entry(&mut stream)? {
        entry_count += 1;
        if let crate::tar::TarItem::Leaf(composefs::tree::LeafContent::Regular(
            composefs::tree::RegularFile::External(_, file_size),
        )) = entry.item
        {
            external_size += file_size;
        }
    }

    // inline_size includes tar headers, small files, and other metadata
    let inline_size = size.saturating_sub(external_size);

    Ok(LayerInfo {
        diff_id: diff_id.to_string(),
        verity: verity.to_hex(),
        size,
        entry_count,
        splitstream: SplitstreamInfo {
            external_objects,
            external_size,
            inline_size,
        },
    })
}

/// Writes the layer contents in composefs dumpfile format.
///
/// Each entry is written on its own line in the composefs dumpfile format,
/// which includes path, size, mode, ownership, timestamps, and content references.
pub fn layer_dumpfile<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    diff_id: &OciDigest,
    output: &mut impl std::io::Write,
) -> Result<()> {
    let content_id = crate::layer_identifier(diff_id);
    let verity = repo
        .has_stream(&content_id)?
        .with_context(|| format!("Layer {diff_id} not found"))?;

    let mut stream = repo.open_stream(
        &content_id,
        Some(&verity),
        Some(crate::skopeo::TAR_LAYER_CONTENT_TYPE),
    )?;

    while let Some(entry) = crate::tar::get_entry(&mut stream)? {
        writeln!(output, "{entry}")?;
    }

    Ok(())
}

/// Reconstitutes and writes the original tar layer.
///
/// This merges the splitstream back into the original tar format by
/// combining inline data with external object references.
pub fn layer_tar<ObjectID: FsVerityHashValue>(
    repo: &Repository<ObjectID>,
    diff_id: &OciDigest,
    output: &mut impl std::io::Write,
) -> Result<()> {
    let content_id = crate::layer_identifier(diff_id);
    let verity = repo
        .has_stream(&content_id)?
        .with_context(|| format!("Layer {diff_id} not found"))?;

    repo.merge_splitstream(
        &content_id,
        Some(&verity),
        Some(crate::skopeo::TAR_LAYER_CONTENT_TYPE),
        output,
    )
}

#[cfg(test)]
mod test {
    use super::*;
    use composefs::fsverity::Sha256HashValue;
    use composefs::test::TestRepo;
    use containers_image_proxy::oci_spec::image::{
        ConfigBuilder, DescriptorBuilder, ImageConfigurationBuilder, ImageManifestBuilder,
        RootFsBuilder,
    };
    use std::fs::File;
    use std::io::Read;

    /// Helper to create a synthetic container image in the repository.
    ///
    /// Creates a minimal but valid container image with:
    /// - A single "layer" (stored as an external object)
    /// - Proper OCI manifest and config structure
    /// - Optional tag
    ///
    /// Returns (manifest_digest, manifest_verity, config_digest).
    fn create_test_image(
        repo: &Arc<Repository<Sha256HashValue>>,
        tag: Option<&str>,
        arch: &str,
    ) -> (OciDigest, Sha256HashValue, OciDigest) {
        // Create a fake layer - in real usage this would be a tar splitstream
        // For testing the manifest/config storage, we just need valid references
        let layer_data = format!("fake-layer-{arch}").into_bytes();
        let layer_digest = hash_sha256(&layer_data);

        let mut layer_stream = repo
            .create_stream(crate::skopeo::TAR_LAYER_CONTENT_TYPE)
            .unwrap();
        layer_stream.write_external(&layer_data).unwrap();
        let layer_verity = repo
            .write_stream(layer_stream, &crate::layer_identifier(&layer_digest), None)
            .unwrap();

        let rootfs = RootFsBuilder::default()
            .typ("layers")
            .diff_ids(vec![layer_digest.to_string()])
            .build()
            .unwrap();

        let cfg = ConfigBuilder::default().build().unwrap();

        let config = ImageConfigurationBuilder::default()
            .architecture(arch)
            .os("linux")
            .rootfs(rootfs)
            .config(cfg)
            .build()
            .unwrap();

        let config_json = config.to_string().unwrap();
        let config_digest = hash_sha256(config_json.as_bytes());

        let mut config_stream = repo.create_stream(OCI_CONFIG_CONTENT_TYPE).unwrap();
        config_stream.add_named_stream_ref(layer_digest.as_ref(), &layer_verity);
        config_stream
            .write_external(config_json.as_bytes())
            .unwrap();
        let config_verity = repo
            .write_stream(
                config_stream,
                &crate::config_identifier(&config_digest),
                None,
            )
            .unwrap();

        let config_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::ImageConfig)
            .digest(config_digest.clone())
            .size(config_json.len() as u64)
            .build()
            .unwrap();

        let layer_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::ImageLayerGzip)
            .digest(layer_digest.clone())
            .size(layer_data.len() as u64)
            .build()
            .unwrap();

        let manifest = ImageManifestBuilder::default()
            .schema_version(2u32)
            .media_type(MediaType::ImageManifest)
            .config(config_descriptor)
            .layers(vec![layer_descriptor])
            .build()
            .unwrap();

        let layer_verities = [(layer_digest, layer_verity)];

        let manifest_json = manifest.to_string().unwrap();
        let manifest_digest = hash_sha256(manifest_json.as_bytes());

        let (_stored_digest, manifest_verity) = write_manifest(
            repo,
            &manifest,
            &manifest_digest,
            &config_verity,
            &layer_verities,
            tag,
        )
        .unwrap();

        (manifest_digest, manifest_verity, config_digest)
    }

    #[test]
    fn test_manifest_identifier() {
        let digest: OciDigest =
            "sha256:abc1230000000000000000000000000000000000000000000000000000000000"
                .parse()
                .unwrap();
        assert_eq!(
            manifest_identifier(&digest),
            "oci-manifest-sha256:abc1230000000000000000000000000000000000000000000000000000000000"
        );
    }

    #[test]
    fn test_oci_ref_path() {
        assert_eq!(oci_ref_path("myimage:latest"), "oci/myimage:latest");
        // Slashes get encoded
        assert_eq!(oci_ref_path("library/nginx"), "oci/library%2Fnginx");
        assert_eq!(oci_ref_path("docker://busybox"), "oci/docker:%2F%2Fbusybox");
    }

    #[test]
    fn test_encode_decode_tag() {
        // Simple names pass through
        assert_eq!(encode_tag("myimage:latest"), "myimage:latest");
        assert_eq!(decode_tag("myimage:latest"), "myimage:latest");

        // Slashes get encoded
        assert_eq!(encode_tag("library/nginx"), "library%2Fnginx");
        assert_eq!(decode_tag("library%2Fnginx"), "library/nginx");

        // Double slashes
        assert_eq!(encode_tag("docker://busybox"), "docker:%2F%2Fbusybox");
        assert_eq!(decode_tag("docker:%2F%2Fbusybox"), "docker://busybox");

        // Percent signs get encoded first to avoid conflicts
        assert_eq!(encode_tag("test%2F"), "test%252F");
        assert_eq!(decode_tag("test%252F"), "test%2F");

        // Round-trip including tricky inputs where order-dependent
        // replacement would produce wrong results
        let names = [
            "simple",
            "with:tag",
            "registry.io/image:v1",
            "docker://busybox:latest",
            "containers-storage:myimage",
            "weird%name/with/slashes",
            "%2F",
            "a/b%c",
            "100%",
            "normal:tag",
            "%25already-encoded",
            "double%%percent",
        ];
        for name in names {
            assert_eq!(
                decode_tag(&encode_tag(name)),
                name,
                "round-trip failed for {name}"
            );
        }
    }

    #[test]
    fn test_hash_sha256() {
        assert_eq!(
            hash_sha256(b"hello world").as_ref(),
            "sha256:b94d27b9934d3e08a52e52d7da7dabfac484efe37a5380ee9088f7ace2efcde9"
        );
    }

    #[test]
    fn test_blob_identifier() {
        let digest: OciDigest =
            "sha256:abc1230000000000000000000000000000000000000000000000000000000000"
                .parse()
                .unwrap();
        assert_eq!(
            blob_identifier(&digest),
            "oci-blob-sha256:abc1230000000000000000000000000000000000000000000000000000000000"
        );
    }

    #[test]
    fn test_write_and_read_blob() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let data = b"This is some arbitrary blob data for an OCI artifact.";
        let (digest, verity) = write_blob(repo, data).unwrap();

        assert!(digest.as_ref().starts_with("sha256:"));

        // Read back with verity (fast path)
        let read_data = open_blob(&repo, &digest, Some(&verity)).unwrap();
        assert_eq!(read_data, data);

        // Read back without verity (verifies content hash)
        let read_data2 = open_blob(&repo, &digest, None).unwrap();
        assert_eq!(read_data2, data);
    }

    #[test]
    fn test_write_blob_deduplication() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let data = b"duplicate blob content";

        let (digest1, verity1) = write_blob(repo, data).unwrap();
        let (digest2, verity2) = write_blob(repo, data).unwrap();

        assert_eq!(digest1, digest2);
        assert_eq!(verity1, verity2);
    }

    #[test]
    fn test_open_blob_bad_digest() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let data = b"some blob data";
        let (_digest, _verity) = write_blob(repo, data).unwrap();

        let bad_digest: OciDigest =
            "sha256:0000000000000000000000000000000000000000000000000000000000000000"
                .parse()
                .unwrap();
        let result = open_blob::<Sha256HashValue>(&repo, &bad_digest, None);
        assert!(result.is_err());
    }

    /// Verify that manifest JSON is stored as an external object, not inline.
    ///
    /// External storage gives each manifest its own file in objects/, allowing
    /// fsverity to be independently enabled on the raw content. This is a
    /// prerequisite for signing: a signature can reference the fsverity digest
    /// of the manifest bytes directly.
    #[test]
    fn test_manifest_stored_as_external_object() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (manifest_digest, manifest_verity, _) =
            create_test_image(repo, Some("ext-test"), "amd64");

        let manifest_id = manifest_identifier(&manifest_digest);
        let mut stream = repo
            .open_stream(&manifest_id, Some(&manifest_verity), None)
            .unwrap();

        let mut object_refs = Vec::new();
        stream
            .get_object_refs(|id| object_refs.push(id.clone()))
            .unwrap();

        // Should have at least one external object (the manifest JSON itself)
        assert!(
            !object_refs.is_empty(),
            "Manifest splitstream should contain external object references"
        );

        let img = OciImage::open(&repo, &manifest_digest, Some(&manifest_verity)).unwrap();
        let manifest_json = img.manifest().to_string().unwrap();
        let expected_verity: Sha256HashValue =
            composefs::fsverity::compute_verity(manifest_json.as_bytes());

        assert!(
            object_refs.contains(&expected_verity),
            "Manifest JSON fsverity digest should appear in splitstream object refs"
        );
    }

    /// Verify that blob content is stored as an external object.
    #[test]
    fn test_blob_stored_as_external_object() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let data = b"artifact blob content for external storage test";
        let (digest, verity) = write_blob(repo, data).unwrap();

        let content_id = blob_identifier(&digest);
        let mut stream = repo.open_stream(&content_id, Some(&verity), None).unwrap();

        let mut object_refs = Vec::new();
        stream
            .get_object_refs(|id| object_refs.push(id.clone()))
            .unwrap();

        assert_eq!(
            object_refs.len(),
            1,
            "Blob should be stored as exactly one external object"
        );

        let expected_verity: Sha256HashValue = composefs::fsverity::compute_verity(data);
        assert_eq!(
            object_refs[0], expected_verity,
            "External object verity should match independently computed verity of blob data"
        );
    }

    /// Test storing and retrieving an OCI artifact with non-tar media type.
    ///
    /// This simulates what would happen when storing something like a
    /// Helm chart, WASM module, or other non-container artifact.
    #[test]
    fn test_oci_artifact_roundtrip() {
        use containers_image_proxy::oci_spec::image::{DescriptorBuilder, ImageManifestBuilder};

        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        // Create an artifact with a custom media type (simulating a WASM module)
        let wasm_bytes = b"\x00asm\x01\x00\x00\x00"; // WASM magic header
        let (blob_digest, blob_verity) = write_blob(repo, wasm_bytes).unwrap();

        // Create an empty config (common for artifacts)
        let empty_config = b"{}";
        let config_digest = hash_sha256(empty_config);

        let mut config_stream = repo.create_stream(OCI_CONFIG_CONTENT_TYPE).unwrap();
        config_stream.write_external(empty_config).unwrap();
        let config_verity = repo
            .write_stream(
                config_stream,
                &crate::config_identifier(&config_digest),
                None,
            )
            .unwrap();

        let config_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::Other(
                "application/vnd.wasm.config.v1+json".to_string(),
            ))
            .digest(config_digest.clone())
            .size(empty_config.len() as u64)
            .build()
            .unwrap();

        let blob_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::Other("application/wasm".to_string()))
            .digest(blob_digest.clone())
            .size(wasm_bytes.len() as u64)
            .build()
            .unwrap();

        let manifest = ImageManifestBuilder::default()
            .schema_version(2u32)
            .media_type(MediaType::ImageManifest)
            .config(config_descriptor)
            .layers(vec![blob_descriptor])
            .build()
            .unwrap();

        // For artifacts, we use the blob digest as the "diff_id" equivalent
        let layer_verities = [(blob_digest.clone(), blob_verity.clone())];

        let manifest_json = manifest.to_string().unwrap();
        let manifest_digest = hash_sha256(manifest_json.as_bytes());

        let (stored_digest, manifest_verity) = write_manifest(
            &repo,
            &manifest,
            &manifest_digest,
            &config_verity,
            &layer_verities,
            Some("my-wasm-artifact:v1"),
        )
        .unwrap();

        assert_eq!(stored_digest, manifest_digest);

        let opened = OciImage::open(&repo, &manifest_digest, Some(&manifest_verity)).unwrap();

        assert!(!opened.is_container_image()); // Not a container image
        assert_eq!(opened.manifest_digest(), &manifest_digest);
        assert_eq!(opened.config_digest(), &config_digest);
        assert_eq!(opened.layer_descriptors().len(), 1);
        assert_eq!(
            opened.layer_descriptors()[0].media_type(),
            &MediaType::Other("application/wasm".to_string())
        );

        let by_tag = OciImage::open_ref(&repo, "my-wasm-artifact:v1").unwrap();
        assert_eq!(by_tag.manifest_digest(), &manifest_digest);

        let images = list_images(&repo).unwrap();
        assert_eq!(images.len(), 1);
        assert_eq!(images[0].name, "my-wasm-artifact:v1");
        assert!(!images[0].is_container);

        let read_wasm = open_blob(&repo, &blob_digest, Some(&blob_verity)).unwrap();
        assert_eq!(read_wasm, wasm_bytes);
    }

    /// Test the OCI 1.1 empty config artifact pattern from the spec:
    /// config is `application/vnd.oci.empty.v1+json`, layers use custom
    /// media types, and layer digests are used as diff_ids.
    /// See: https://github.com/opencontainers/image-spec/blob/main/artifacts-guidance.md
    #[test]
    fn test_oci_artifact_empty_config() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let sbom_data = br#"{"spdxVersion":"SPDX-2.3","name":"example"}"#;
        let layer_digest = hash_sha256(sbom_data);

        // Store the raw layer as an object with external ref splitstream
        let blob_object_id = repo.ensure_object(sbom_data).unwrap();
        let layer_content_id = crate::layer_identifier(&layer_digest);
        let mut layer_stream = repo
            .create_stream(crate::skopeo::OCI_BLOB_CONTENT_TYPE)
            .unwrap();
        layer_stream.add_external_size(sbom_data.len() as u64);
        layer_stream
            .write_reference(blob_object_id.clone())
            .unwrap();
        let layer_verity = repo
            .write_stream(layer_stream, &layer_content_id, None)
            .unwrap();

        // The OCI 1.1 empty config: `{}` with the well-known digest
        let empty_config = b"{}";
        let config_digest = hash_sha256(empty_config);
        assert_eq!(
            config_digest.as_ref(),
            "sha256:44136fa355b3678a1146ad16f7e8649e94fb4fc21fe77e8310c060f61caaff8a"
        );

        // Store the config — for artifacts we still write it as a config
        // splitstream, but it contains no diff_ids-derived named refs.
        // Instead, the layer refs come from the manifest layer digests.
        let mut config_stream = repo.create_stream(OCI_CONFIG_CONTENT_TYPE).unwrap();
        config_stream.write_external(empty_config).unwrap();
        let config_verity = repo
            .write_stream(
                config_stream,
                &crate::config_identifier(&config_digest),
                None,
            )
            .unwrap();

        let config_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::EmptyJSON)
            .digest(config_digest.clone())
            .size(empty_config.len() as u64)
            .build()
            .unwrap();

        let layer_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::Other("text/spdx+json".to_string()))
            .digest(layer_digest.clone())
            .size(sbom_data.len() as u64)
            .build()
            .unwrap();

        let manifest = ImageManifestBuilder::default()
            .schema_version(2u32)
            .media_type(MediaType::ImageManifest)
            .config(config_descriptor.clone())
            .layers(vec![layer_descriptor])
            .build()
            .unwrap();

        assert_ne!(*config_descriptor.media_type(), MediaType::ImageConfig);

        // Store manifest — layer_verities uses the layer digest as key
        // (same logic as ensure_config_with_layers when !is_image_config)
        let layer_verities = [(layer_digest.clone(), layer_verity.clone())];

        let manifest_json = manifest.to_string().unwrap();
        let manifest_digest = hash_sha256(manifest_json.as_bytes());

        let (_stored_digest, manifest_verity) = write_manifest(
            &repo,
            &manifest,
            &manifest_digest,
            &config_verity,
            &layer_verities,
            Some("my-sbom:v1"),
        )
        .unwrap();

        let opened = OciImage::open(&repo, &manifest_digest, Some(&manifest_verity)).unwrap();
        assert!(!opened.is_container_image());
        assert_eq!(opened.layer_descriptors().len(), 1);
        assert_eq!(
            opened.layer_descriptors()[0].media_type(),
            &MediaType::Other("text/spdx+json".to_string())
        );

        let fd = opened.open_layer_fd(&repo, 0).unwrap();
        let mut recovered = vec![];
        File::from(fd).read_to_end(&mut recovered).unwrap();
        assert_eq!(recovered, sbom_data);

        assert!(opened.open_layer_fd(&repo, 1).is_err());

        let gc = repo.gc(&[]).unwrap();
        assert_eq!(gc.objects_removed, 0);

        untag_image(&repo, "my-sbom:v1").unwrap();
        let gc = repo.gc(&[]).unwrap();
        assert!(gc.objects_removed > 0);
    }

    /// Test that open_layer_fd rejects tar layers.
    #[test]
    fn test_open_layer_fd_rejects_tar() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (digest, verity, _) = create_test_image(repo, Some("myimage:v1"), "amd64");
        let img = OciImage::open(&repo, &digest, Some(&verity)).unwrap();
        assert!(img.is_container_image());

        // Tar layer should be rejected
        let err = img.open_layer_fd(&repo, 0).unwrap_err();
        let msg = format!("{err}");
        assert!(msg.contains("does not support tar layers"), "got: {msg}");
    }

    /// Test storing a non-tar layer as a splitstream with a single
    /// external reference, simulating how `ensure_layer` handles
    /// non-tar media types. The raw bytes go into objects/ and a
    /// tiny splitstream holds the reference for GC tracking.
    #[test]
    fn test_non_tar_layer_storage() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let sbom_data = br#"{"spdxVersion":"SPDX-2.3","name":"example"}"#;
        let diff_id = hash_sha256(sbom_data);

        let object_id = repo.ensure_object(sbom_data).unwrap();

        let content_id = crate::layer_identifier(&diff_id);
        let mut stream = repo
            .create_stream(crate::skopeo::OCI_BLOB_CONTENT_TYPE)
            .unwrap();
        stream.add_external_size(sbom_data.len() as u64);
        stream.write_reference(object_id.clone()).unwrap();
        let stream_verity = repo.write_stream(stream, &content_id, None).unwrap();

        let found = repo.has_stream(&content_id).unwrap();
        assert!(found.is_some());
        assert_eq!(found.unwrap(), stream_verity);

        let mut reader = repo
            .open_stream(
                &content_id,
                Some(&stream_verity),
                Some(crate::skopeo::OCI_BLOB_CONTENT_TYPE),
            )
            .unwrap();
        let mut refs = vec![];
        reader.get_object_refs(|id| refs.push(id.clone())).unwrap();
        assert_eq!(refs.len(), 1);
        assert_eq!(refs[0], object_id);

        let mut recovered = vec![];
        File::from(repo.open_object(&object_id).unwrap())
            .read_to_end(&mut recovered)
            .unwrap();
        assert_eq!(recovered, sbom_data);
    }

    /// Test that a non-tar artifact layer (stored as an external ref)
    /// is preserved by GC when referenced from a tagged manifest.
    #[test]
    fn test_non_tar_artifact_gc() {
        use containers_image_proxy::oci_spec::image::{DescriptorBuilder, ImageManifestBuilder};

        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let sbom_data = br#"{"spdxVersion":"SPDX-2.3","name":"example"}"#;
        let diff_id = hash_sha256(sbom_data);
        let blob_object_id = repo.ensure_object(sbom_data).unwrap();

        let layer_content_id = crate::layer_identifier(&diff_id);
        let mut layer_stream = repo
            .create_stream(crate::skopeo::OCI_BLOB_CONTENT_TYPE)
            .unwrap();
        layer_stream.add_external_size(sbom_data.len() as u64);
        layer_stream
            .write_reference(blob_object_id.clone())
            .unwrap();
        let layer_verity = repo
            .write_stream(layer_stream, &layer_content_id, None)
            .unwrap();

        let config_bytes = b"{}";
        let config_digest = hash_sha256(config_bytes);
        let mut config_stream = repo.create_stream(OCI_CONFIG_CONTENT_TYPE).unwrap();
        config_stream.write_external(config_bytes).unwrap();
        let config_verity = repo
            .write_stream(
                config_stream,
                &crate::config_identifier(&config_digest),
                None,
            )
            .unwrap();

        let config_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::ImageConfig)
            .digest(config_digest.clone())
            .size(config_bytes.len() as u64)
            .build()
            .unwrap();
        let layer_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::Other("text/spdx+json".to_string()))
            .digest(diff_id.clone())
            .size(sbom_data.len() as u64)
            .build()
            .unwrap();
        let manifest = ImageManifestBuilder::default()
            .schema_version(2u32)
            .media_type(MediaType::ImageManifest)
            .config(config_descriptor)
            .layers(vec![layer_descriptor])
            .build()
            .unwrap();

        let layer_verities = [(diff_id.clone(), layer_verity)];

        let manifest_json = manifest.to_string().unwrap();
        let manifest_digest = hash_sha256(manifest_json.as_bytes());

        let (_stored_digest, _manifest_verity) = write_manifest(
            &repo,
            &manifest,
            &manifest_digest,
            &config_verity,
            &layer_verities,
            Some("my-sbom:v1"),
        )
        .unwrap();

        // GC should preserve everything — the blob object is reachable via
        // manifest → config named ref → layer splitstream → external ref
        let gc = repo.gc(&[]).unwrap();
        assert_eq!(gc.objects_removed, 0, "tagged artifact should be preserved");

        let mut recovered = vec![];
        File::from(repo.open_object(&blob_object_id).unwrap())
            .read_to_end(&mut recovered)
            .unwrap();
        assert_eq!(recovered, sbom_data);
    }

    /// Test storing and listing multiple container images.
    #[test]
    fn test_multiple_images() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (digest1, verity1, _) = create_test_image(repo, Some("app:v1"), "amd64");
        let (digest2, verity2, _) = create_test_image(repo, Some("app:v2"), "amd64");
        let (digest3, verity3, _) = create_test_image(repo, Some("other:latest"), "arm64");

        let images = list_images(repo).unwrap();
        assert_eq!(images.len(), 3);

        let names: Vec<_> = images.iter().map(|i| i.name.as_str()).collect();
        assert!(names.contains(&"app:v1"));
        assert!(names.contains(&"app:v2"));
        assert!(names.contains(&"other:latest"));

        for img in &images {
            if img.name == "other:latest" {
                assert_eq!(img.architecture, "arm64");
            } else {
                assert_eq!(img.architecture, "amd64");
            }
            assert!(img.is_container);
        }

        let img1 = OciImage::open_ref(repo, "app:v1").unwrap();
        assert_eq!(img1.manifest_digest(), &digest1);
        assert_eq!(img1.manifest_verity(), &verity1);

        let img2 = OciImage::open_ref(repo, "app:v2").unwrap();
        assert_eq!(img2.manifest_digest(), &digest2);
        assert_eq!(img2.manifest_verity(), &verity2);

        let img3 = OciImage::open_ref(repo, "other:latest").unwrap();
        assert_eq!(img3.manifest_digest(), &digest3);
        assert_eq!(img3.manifest_verity(), &verity3);
    }

    /// Test that untagging removes the image from listing but preserves data.
    #[test]
    fn test_untag_image() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (digest1, verity1, _) = create_test_image(repo, Some("myapp:v1"), "amd64");
        let (digest2, _verity2, _) = create_test_image(repo, Some("myapp:v2"), "amd64");

        let images = list_images(repo).unwrap();
        assert_eq!(images.len(), 2);

        untag_image(repo, "myapp:v1").unwrap();

        let images = list_images(repo).unwrap();
        assert_eq!(images.len(), 1);
        assert_eq!(images[0].name, "myapp:v2");
        assert_eq!(images[0].manifest_digest, digest2);

        let img = OciImage::open(repo, &digest1, Some(&verity1)).unwrap();
        assert_eq!(img.manifest_digest(), &digest1);

        let result = OciImage::open_ref(repo, "myapp:v1");
        assert!(result.is_err());
    }

    /// Test resolving refs and listing refs.
    #[test]
    fn test_refs() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (digest, verity, _) = create_test_image(repo, Some("test:latest"), "amd64");

        let refs = list_refs(repo).unwrap();
        assert_eq!(refs.len(), 1);
        assert_eq!(refs[0].0, "test:latest");
        assert_eq!(refs[0].1, digest);

        let (resolved_digest, resolved_verity) = resolve_ref(repo, "test:latest").unwrap();
        assert_eq!(resolved_digest, digest);
        assert_eq!(resolved_verity, verity);

        let result = resolve_ref::<Sha256HashValue>(repo, "nonexistent:tag");
        assert!(result.is_err());
    }

    /// Test that tag_image rejects names containing `@`.
    #[test]
    fn test_tag_rejects_leading_at_sign() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (digest, _, _) = create_test_image(repo, Some("valid:v1"), "amd64");

        // Leading @ is rejected
        let result = tag_image(repo, &digest, "@sha256:bad");
        assert!(result.is_err());
        let err = result.unwrap_err().to_string();
        assert!(err.contains("'@' is reserved"), "unexpected error: {err}");

        // @ in the middle is fine
        let result = tag_image(repo, &digest, "name@digest");
        assert!(result.is_ok());
    }

    /// Test that fsck catches refs starting with `@`.
    #[tokio::test]
    async fn test_oci_fsck_detects_invalid_ref_name() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (digest, _, _) = create_test_image(repo, Some("good:v1"), "amd64");

        // Bypass validate_ref_name by creating the ref symlink directly
        let bad_name = "@badref";
        let ref_path = format!("streams/refs/{}", oci_ref_path(bad_name));
        let manifest_id = manifest_identifier(&digest);
        let target = format!("../../{manifest_id}");
        repo.symlink(&ref_path, &target)
            .expect("create bad ref symlink");

        let result = oci_fsck(repo).await.unwrap();
        assert!(
            result.images_corrupted > 0,
            "fsck should report corruption for @ in ref name"
        );
        assert!(
            result
                .errors
                .iter()
                .any(|e| matches!(e, OciFsckError::InvalidRefName { name } if name == bad_name)),
            "fsck should report InvalidRefName error"
        );
        // The bad ref should be counted exactly once
        let invalid_count = result
            .errors
            .iter()
            .filter(|e| matches!(e, OciFsckError::InvalidRefName { .. }))
            .count();
        assert_eq!(invalid_count, 1, "should report exactly one InvalidRefName");
    }

    /// Test that tagging an existing manifest with a new name works.
    #[test]
    fn test_tag_existing_manifest() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (digest, verity, _) = create_test_image(repo, Some("original:v1"), "amd64");

        tag_image(repo, &digest, "alias:latest").unwrap();

        let (d1, v1) = resolve_ref(repo, "original:v1").unwrap();
        let (d2, v2) = resolve_ref(repo, "alias:latest").unwrap();
        assert_eq!(d1, d2);
        assert_eq!(v1, v2);
        assert_eq!(d1, digest);
        assert_eq!(v1, verity);

        let images = list_images(repo).unwrap();
        assert_eq!(images.len(), 2);

        untag_image(repo, "original:v1").unwrap();
        let (d3, _) = resolve_ref(repo, "alias:latest").unwrap();
        assert_eq!(d3, digest);

        let images = list_images(repo).unwrap();
        assert_eq!(images.len(), 1);
        assert_eq!(images[0].name, "alias:latest");
    }

    /// Test opening image by manifest digest (no tag required).
    #[test]
    fn test_open_by_digest() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (digest, verity, config_digest) = create_test_image(repo, None, "amd64");

        let images = list_images(repo).unwrap();
        assert!(images.is_empty());

        let img = OciImage::open(repo, &digest, Some(&verity)).unwrap();
        assert_eq!(img.manifest_digest(), &digest);
        assert_eq!(img.config_digest(), &config_digest);
        assert!(img.is_container_image());
        assert_eq!(img.architecture(), "amd64");

        let img2 = OciImage::open(repo, &digest, None).unwrap();
        assert_eq!(img2.manifest_digest(), &digest);
    }

    /// Test fetching manifest and config from stored image.
    #[test]
    fn test_fetch_manifest_config() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (digest, verity, config_digest) =
            create_test_image(repo, Some("fetchtest:v1"), "amd64");

        let img = OciImage::open_ref(repo, "fetchtest:v1").unwrap();

        assert_eq!(img.manifest_digest(), &digest);
        assert_eq!(img.manifest_verity(), &verity);
        let manifest = img.manifest();
        assert_eq!(manifest.schema_version(), 2u32);
        assert_eq!(manifest.layers().len(), 1);

        assert_eq!(img.config_digest(), &config_digest);
        let config = img.config().expect("should have config");
        assert_eq!(config.architecture().to_string(), "amd64");
        assert_eq!(config.os().to_string(), "linux");
        assert_eq!(config.rootfs().diff_ids().len(), 1);

        let diff_ids = img.layer_diff_ids();
        assert_eq!(diff_ids.len(), 1);
        let layer_verity = img.layer_verity(diff_ids[0]);
        assert!(layer_verity.is_some());
    }

    /// Test that has_manifest correctly detects existing manifests.
    #[test]
    fn test_has_manifest() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let nonexistent: OciDigest =
            "sha256:0000000000000000000000000000000000000000000000000000000000000000"
                .parse()
                .unwrap();
        assert!(has_manifest(repo, &nonexistent).unwrap().is_none());

        let (digest, verity, _) = create_test_image(repo, None, "amd64");

        let found = has_manifest(repo, &digest).unwrap();
        assert!(found.is_some());
        assert_eq!(found.unwrap(), verity);

        assert!(has_manifest(repo, &nonexistent).unwrap().is_none());
    }

    /// Test empty repository behavior.
    #[test]
    fn test_empty_repo() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        // List should return empty vec, not error
        let images = list_images(repo).unwrap();
        assert!(images.is_empty());

        let refs = list_refs(repo).unwrap();
        assert!(refs.is_empty());
    }

    /// Test untagging non-existent tag.
    #[test]
    fn test_untag_nonexistent() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let result = untag_image(repo, "nonexistent:tag");
        assert!(result.is_err());
    }

    // ==================== GC Integration Tests ====================
    //
    // These tests verify that garbage collection correctly handles OCI images:
    // - Tagged images are preserved (tags act as GC roots)
    // - Untagged images can be collected
    // - Shared layers between images are handled correctly

    /// Test that GC preserves a tagged OCI image and all its components.
    #[test]
    fn test_gc_preserves_tagged_oci_image() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (manifest_digest, manifest_verity, config_digest) =
            create_test_image(repo, Some("myapp:v1"), "amd64");

        let gc_result = repo.gc(&[]).unwrap();

        assert_eq!(gc_result.objects_removed, 0);
        assert_eq!(gc_result.streams_pruned, 0);

        let img = OciImage::open_ref(repo, "myapp:v1").unwrap();
        assert_eq!(img.manifest_digest(), &manifest_digest);
        assert_eq!(img.manifest_verity(), &manifest_verity);
        assert_eq!(img.config_digest(), &config_digest);

        let diff_ids = img.layer_diff_ids();
        assert_eq!(diff_ids.len(), 1);
        assert!(img.layer_verity(diff_ids[0]).is_some());
    }

    /// Test that GC removes an untagged OCI image.
    #[test]
    fn test_gc_removes_untagged_oci_image() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (manifest_digest, manifest_verity, _config_digest) =
            create_test_image(repo, None, "amd64");

        let img = OciImage::open(repo, &manifest_digest, Some(&manifest_verity)).unwrap();
        let diff_ids = img.layer_diff_ids();
        assert_eq!(diff_ids.len(), 1);
        drop(img);

        let gc_result = repo.gc(&[]).unwrap();

        assert!(gc_result.objects_removed > 0);

        let result = has_manifest(repo, &manifest_digest);
        assert!(
            result.unwrap().is_none(),
            "manifest should be gone after GC"
        );
    }

    /// Test that untagging an image makes it eligible for GC.
    #[test]
    fn test_gc_after_untag_removes_image() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (manifest_digest, manifest_verity, _) =
            create_test_image(repo, Some("temporary:v1"), "amd64");

        let gc_result = repo.gc(&[]).unwrap();
        assert_eq!(gc_result.objects_removed, 0);

        untag_image(repo, "temporary:v1").unwrap();

        assert!(OciImage::open_ref(repo, "temporary:v1").is_err());

        assert!(OciImage::open(repo, &manifest_digest, Some(&manifest_verity)).is_ok());

        let gc_result = repo.gc(&[]).unwrap();
        assert!(gc_result.objects_removed > 0);

        assert!(has_manifest(repo, &manifest_digest).unwrap().is_none());
    }

    /// Test GC with two images sharing layers - removing one preserves shared layers.
    #[test]
    fn test_gc_with_shared_layers() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let shared_layer_data = b"shared-base-layer-content";
        let shared_layer_digest = hash_sha256(shared_layer_data);

        let mut shared_layer_stream = repo
            .create_stream(crate::skopeo::TAR_LAYER_CONTENT_TYPE)
            .unwrap();
        shared_layer_stream
            .write_external(shared_layer_data)
            .unwrap();
        let shared_layer_verity = repo
            .write_stream(
                shared_layer_stream,
                &crate::layer_identifier(&shared_layer_digest),
                None,
            )
            .unwrap();

        // Helper to create an image using the shared layer
        let create_image_with_shared_layer = |repo: &Arc<Repository<Sha256HashValue>>,
                                              tag: Option<&str>,
                                              extra_data: &[u8]|
         -> (OciDigest, Sha256HashValue) {
            let rootfs = RootFsBuilder::default()
                .typ("layers")
                .diff_ids(vec![shared_layer_digest.to_string()])
                .build()
                .unwrap();

            let cfg = ConfigBuilder::default().build().unwrap();

            // Add unique data to make configs different
            let config = ImageConfigurationBuilder::default()
                .architecture("amd64")
                .os("linux")
                .rootfs(rootfs)
                .config(cfg)
                .created(String::from_utf8_lossy(extra_data).to_string())
                .build()
                .unwrap();

            let config_json = config.to_string().unwrap();
            let config_digest = hash_sha256(config_json.as_bytes());

            let mut config_stream = repo.create_stream(OCI_CONFIG_CONTENT_TYPE).unwrap();
            config_stream.add_named_stream_ref(shared_layer_digest.as_ref(), &shared_layer_verity);
            config_stream
                .write_external(config_json.as_bytes())
                .unwrap();
            let config_verity = repo
                .write_stream(
                    config_stream,
                    &crate::config_identifier(&config_digest),
                    None,
                )
                .unwrap();

            let config_descriptor = DescriptorBuilder::default()
                .media_type(MediaType::ImageConfig)
                .digest(config_digest.clone())
                .size(config_json.len() as u64)
                .build()
                .unwrap();

            let layer_descriptor = DescriptorBuilder::default()
                .media_type(MediaType::ImageLayerGzip)
                .digest(shared_layer_digest.clone())
                .size(shared_layer_data.len() as u64)
                .build()
                .unwrap();

            let manifest = ImageManifestBuilder::default()
                .schema_version(2u32)
                .media_type(MediaType::ImageManifest)
                .config(config_descriptor)
                .layers(vec![layer_descriptor])
                .build()
                .unwrap();

            let layer_verities = [(shared_layer_digest.clone(), shared_layer_verity.clone())];

            let manifest_json = manifest.to_string().unwrap();
            let manifest_digest = hash_sha256(manifest_json.as_bytes());

            let (_stored_digest, manifest_verity) = write_manifest(
                repo,
                &manifest,
                &manifest_digest,
                &config_verity,
                &layer_verities,
                tag,
            )
            .unwrap();

            (manifest_digest, manifest_verity)
        };

        let (digest1, verity1) = create_image_with_shared_layer(repo, Some("tagged:v1"), b"image1");
        let (digest2, _verity2) = create_image_with_shared_layer(repo, None, b"image2");

        assert!(has_manifest(repo, &digest1).unwrap().is_some());
        assert!(has_manifest(repo, &digest2).unwrap().is_some());

        let gc_result = repo.gc(&[]).unwrap();

        assert!(gc_result.objects_removed > 0);

        let img1 = OciImage::open(repo, &digest1, Some(&verity1)).unwrap();
        assert_eq!(img1.layer_diff_ids().len(), 1);
        assert!(img1.layer_verity(shared_layer_digest.as_ref()).is_some());

        assert!(has_manifest(repo, &digest2).unwrap().is_none());

        // Shared layer still exists because the tagged image references it
        assert!(
            repo.has_stream(&crate::layer_identifier(&shared_layer_digest))
                .unwrap()
                .is_some()
        );
    }

    /// Test that multiple tags on the same manifest are handled correctly.
    #[test]
    fn test_gc_with_multiple_tags_same_manifest() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        // Create an image with one tag
        let (manifest_digest, manifest_verity, _) =
            create_test_image(repo, Some("original:v1"), "amd64");

        tag_image(repo, &manifest_digest, "alias:latest").unwrap();

        assert_eq!(list_images(repo).unwrap().len(), 2);

        untag_image(repo, "original:v1").unwrap();

        let gc_result = repo.gc(&[]).unwrap();

        assert_eq!(gc_result.objects_removed, 0);

        let img = OciImage::open_ref(repo, "alias:latest").unwrap();
        assert_eq!(img.manifest_digest(), &manifest_digest);
        assert_eq!(img.manifest_verity(), &manifest_verity);

        let diff_ids = img.layer_diff_ids();
        assert!(img.layer_verity(diff_ids[0]).is_some());

        untag_image(repo, "alias:latest").unwrap();

        let gc_result = repo.gc(&[]).unwrap();

        assert!(gc_result.objects_removed > 0);
        assert!(has_manifest(repo, &manifest_digest).unwrap().is_none());
    }

    /// Test gc_dry_run with OCI images.
    #[test]
    fn test_gc_dry_run_oci_image() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        // Create one tagged and one untagged image with DIFFERENT architectures
        // to ensure they have unique layer content (create_test_image uses arch in layer data)
        let (tagged_digest, tagged_verity, _) = create_test_image(repo, Some("keep:v1"), "amd64");
        let (untagged_digest, _untagged_verity, _) = create_test_image(repo, None, "arm64");

        assert!(has_manifest(repo, &tagged_digest).unwrap().is_some());
        assert!(has_manifest(repo, &untagged_digest).unwrap().is_some());

        let dry_run_result = repo.gc_dry_run(&[]).unwrap();
        assert!(
            dry_run_result.objects_removed > 0,
            "dry-run should report objects to remove, got {:?}",
            dry_run_result
        );

        // But nothing should actually be removed
        assert!(has_manifest(repo, &tagged_digest).unwrap().is_some());
        assert!(has_manifest(repo, &untagged_digest).unwrap().is_some());

        let img = OciImage::open(repo, &tagged_digest, Some(&tagged_verity)).unwrap();
        assert!(img.layer_verity(img.layer_diff_ids()[0]).is_some());

        let real_result = repo.gc(&[]).unwrap();

        assert_eq!(real_result.objects_removed, dry_run_result.objects_removed);

        assert!(has_manifest(repo, &untagged_digest).unwrap().is_none());
        assert!(has_manifest(repo, &tagged_digest).unwrap().is_some());
    }

    /// Test referrer index: store an artifact, add a referrer entry,
    /// then discover it via list_referrers.
    #[test]
    fn test_referrer_index_roundtrip() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (subject_digest, _, _) = create_test_image(repo, Some("subject:v1"), "amd64");

        let empty_config = b"{}";
        let config_digest = hash_sha256(empty_config);
        let mut config_stream = repo.create_stream(OCI_CONFIG_CONTENT_TYPE).unwrap();
        config_stream.write_external(empty_config).unwrap();
        let config_verity = repo
            .write_stream(
                config_stream,
                &crate::config_identifier(&config_digest),
                None,
            )
            .unwrap();

        let mut artifact_digests = Vec::new();
        for i in 0..2u8 {
            let blob_data = format!("artifact-blob-{i}").into_bytes();
            let (blob_digest, blob_verity) = write_blob(repo, &blob_data).unwrap();

            let config_descriptor = DescriptorBuilder::default()
                .media_type(MediaType::EmptyJSON)
                .digest(config_digest.clone())
                .size(empty_config.len() as u64)
                .build()
                .unwrap();

            let layer_descriptor = DescriptorBuilder::default()
                .media_type(MediaType::Other("application/octet-stream".to_string()))
                .digest(blob_digest.clone())
                .size(blob_data.len() as u64)
                .build()
                .unwrap();

            let manifest = ImageManifestBuilder::default()
                .schema_version(2u32)
                .media_type(MediaType::ImageManifest)
                .config(config_descriptor)
                .layers(vec![layer_descriptor])
                .build()
                .unwrap();

            let layer_verities = [(blob_digest, blob_verity)];

            let manifest_json = manifest.to_string().unwrap();
            let manifest_digest = hash_sha256(manifest_json.as_bytes());

            write_manifest(
                repo,
                &manifest,
                &manifest_digest,
                &config_verity,
                &layer_verities,
                None,
            )
            .unwrap();

            add_referrer(repo, &subject_digest, &manifest_digest).unwrap();
            artifact_digests.push(manifest_digest);
        }

        let referrers = list_referrers(repo, &subject_digest).unwrap();
        assert_eq!(referrers.len(), 2);

        let found_digests: Vec<&OciDigest> = referrers.iter().map(|(d, _)| d).collect();
        for expected in &artifact_digests {
            assert!(
                found_digests.contains(&expected),
                "Missing artifact {expected} in referrers"
            );
        }
    }

    /// Helper to create a minimal OCI artifact manifest in the repository.
    ///
    /// Returns (manifest_digest, manifest_verity).
    fn create_test_artifact(
        repo: &Arc<Repository<Sha256HashValue>>,
        blob_data: &[u8],
    ) -> (OciDigest, Sha256HashValue) {
        let (blob_digest, blob_verity) = write_blob(repo, blob_data).unwrap();

        let empty_config = b"{}";
        let config_digest = hash_sha256(empty_config);

        let mut config_stream = repo.create_stream(OCI_CONFIG_CONTENT_TYPE).unwrap();
        config_stream.write_external(empty_config).unwrap();
        let config_verity = repo
            .write_stream(
                config_stream,
                &crate::config_identifier(&config_digest),
                None,
            )
            .unwrap();

        let config_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::EmptyJSON)
            .digest(config_digest.clone())
            .size(empty_config.len() as u64)
            .build()
            .unwrap();

        let layer_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::Other("application/octet-stream".to_string()))
            .digest(blob_digest.clone())
            .size(blob_data.len() as u64)
            .build()
            .unwrap();

        let manifest = ImageManifestBuilder::default()
            .schema_version(2u32)
            .media_type(MediaType::ImageManifest)
            .config(config_descriptor)
            .layers(vec![layer_descriptor])
            .build()
            .unwrap();

        let layer_verities = [(blob_digest, blob_verity)];

        let manifest_json = manifest.to_string().unwrap();
        let manifest_digest = hash_sha256(manifest_json.as_bytes());

        let (_stored_digest, manifest_verity) = write_manifest(
            repo,
            &manifest,
            &manifest_digest,
            &config_verity,
            &layer_verities,
            None,
        )
        .unwrap();

        (manifest_digest, manifest_verity)
    }

    /// Test that GC collects referrer artifacts when their subject is untagged.
    ///
    /// Referrer symlinks under `streams/refs/oci-referrers/` act as GC roots,
    /// so orphaned referrer entries must be cleaned up before GC to allow
    /// the artifact manifests and their objects to be collected.
    #[test]
    fn test_gc_cleans_referrer_artifacts() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        // 1. Create a subject image with a tag
        let (subject_digest, _subject_verity, _) =
            create_test_image(repo, Some("subject:v1"), "amd64");

        // 2. Create an artifact referencing the subject
        let (artifact_digest, _artifact_verity) =
            create_test_artifact(repo, b"fake-signature-data");

        // 3. Register the referrer relationship
        add_referrer(repo, &subject_digest, &artifact_digest).unwrap();

        // 4. Verify the referrer is discoverable
        let referrers = list_referrers(repo, &subject_digest).unwrap();
        assert_eq!(referrers.len(), 1);
        assert_eq!(referrers[0].0, artifact_digest);

        // Verify GC preserves everything while subject is tagged
        let gc = repo.gc(&[]).unwrap();
        assert_eq!(gc.objects_removed, 0, "nothing should be collected yet");

        // Artifact should still be accessible
        assert!(
            has_manifest(repo, &artifact_digest).unwrap().is_some(),
            "artifact manifest should exist"
        );

        // 5. Untag the subject image
        untag_image(repo, "subject:v1").unwrap();

        // 6. First GC pass: collects the subject's objects and cleans up
        //    its broken stream symlink. The artifact survives because the
        //    referrer symlink still acts as a GC root.
        let gc1 = repo.gc(&[]).unwrap();
        assert!(gc1.objects_removed > 0, "should collect subject objects");
        assert!(
            has_manifest(repo, &subject_digest).unwrap().is_none(),
            "subject manifest should be gone after first GC"
        );
        // Artifact is still alive — rooted by referrer symlink
        assert!(
            has_manifest(repo, &artifact_digest).unwrap().is_some(),
            "artifact should survive first GC (referrer symlink roots it)"
        );

        // 7. Clean up dangling referrers (subject no longer exists)
        let cleaned = cleanup_dangling_referrers(repo).unwrap();
        assert_eq!(cleaned, 1, "should remove 1 dangling referrer entry");

        // 8. Second GC pass: now collects the artifact (no longer rooted)
        let gc2 = repo.gc(&[]).unwrap();
        assert!(gc2.objects_removed > 0, "should collect artifact objects");

        // 9. Verify the artifact manifest is gone
        assert!(
            has_manifest(repo, &artifact_digest).unwrap().is_none(),
            "artifact manifest should be collected"
        );

        // 10. Verify list_referrers returns empty
        let referrers = list_referrers(repo, &subject_digest).unwrap();
        assert!(referrers.is_empty(), "no referrers should remain after GC");

        // Also verify the subject manifest is gone
        assert!(
            has_manifest(repo, &subject_digest).unwrap().is_none(),
            "subject manifest should be collected"
        );
    }

    /// Test that cleanup_dangling_referrers preserves referrers for tagged subjects.
    #[test]
    fn test_cleanup_referrers_preserves_tagged_subjects() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        // Create a tagged subject
        let (subject_digest, _, _) = create_test_image(repo, Some("subject:v1"), "amd64");

        // Create an artifact and register it as a referrer
        let (artifact_digest, _) = create_test_artifact(repo, b"sig-data");
        add_referrer(repo, &subject_digest, &artifact_digest).unwrap();

        // Cleanup should not remove anything — subject is still tagged
        let cleaned = cleanup_dangling_referrers(repo).unwrap();
        assert_eq!(cleaned, 0, "should not remove referrers for tagged subject");

        // Referrer should still be discoverable
        let referrers = list_referrers(repo, &subject_digest).unwrap();
        assert_eq!(referrers.len(), 1);
    }

    /// Test that cleanup handles multiple subjects, only removing dangling ones.
    #[test]
    fn test_cleanup_referrers_mixed_subjects() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        // Create two subjects
        let (subject1_digest, _, _) = create_test_image(repo, Some("subject1:v1"), "amd64");
        let (subject2_digest, _, _) = create_test_image(repo, Some("subject2:v1"), "arm64");

        // Create artifacts for both
        let (artifact1_digest, _) = create_test_artifact(repo, b"sig-for-subject1");
        let (artifact2_digest, _) = create_test_artifact(repo, b"sig-for-subject2");

        add_referrer(repo, &subject1_digest, &artifact1_digest).unwrap();
        add_referrer(repo, &subject2_digest, &artifact2_digest).unwrap();

        // Untag only subject1
        untag_image(repo, "subject1:v1").unwrap();

        // First GC pass to actually remove subject1's manifest stream
        // (cleanup_dangling_referrers checks has_manifest, which checks the
        // stream symlink; GC removes the broken symlink after object deletion)
        repo.gc(&[]).unwrap();

        // Now cleanup should only remove referrers for subject1
        let cleaned = cleanup_dangling_referrers(repo).unwrap();
        assert_eq!(cleaned, 1, "should remove 1 referrer for untagged subject");

        // Run GC again to collect the now-unrooted artifact1
        let gc = repo.gc(&[]).unwrap();
        assert!(gc.objects_removed > 0);

        // subject2's referrer should still exist
        let referrers2 = list_referrers(repo, &subject2_digest).unwrap();
        assert_eq!(referrers2.len(), 1);
        assert_eq!(referrers2[0].0, artifact2_digest);

        // subject1's artifact should be gone
        assert!(has_manifest(repo, &artifact1_digest).unwrap().is_none());
        // subject2's artifact should still exist
        assert!(has_manifest(repo, &artifact2_digest).unwrap().is_some());
    }

    /// Test that cleanup_dangling_referrers is a no-op on an empty repository.
    #[test]
    fn test_cleanup_referrers_empty_repo() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let cleaned = cleanup_dangling_referrers(repo).unwrap();
        assert_eq!(cleaned, 0);
    }

    /// Test removing a single referrer: add, remove, verify gone, and
    /// confirm that a second remove is idempotent (no error).
    #[test]
    fn test_remove_referrer() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (subject_digest, _, _) = create_test_image(repo, Some("subject:v1"), "amd64");
        let (artifact_digest, _) = create_test_artifact(repo, b"sig-remove-test");

        add_referrer(repo, &subject_digest, &artifact_digest).unwrap();
        assert_eq!(list_referrers(repo, &subject_digest).unwrap().len(), 1);

        // Remove the referrer
        remove_referrer(repo, &subject_digest, &artifact_digest).unwrap();
        assert!(list_referrers(repo, &subject_digest).unwrap().is_empty());

        // Second remove is idempotent
        remove_referrer(repo, &subject_digest, &artifact_digest).unwrap();
    }

    // ==================== Property Tests ====================

    mod proptests {
        use super::*;
        use proptest::prelude::*;

        proptest! {
            #[test]
            fn encode_decode_tag_roundtrip(s in "\\PC*") {
                prop_assert_eq!(decode_tag(&encode_tag(&s)), s);
            }

            #[test]
            fn encode_tag_no_slashes(s in "\\PC*") {
                prop_assert!(!encode_tag(&s).contains('/'));
            }

            #[test]
            fn hash_deterministic_and_prefixed(data in proptest::collection::vec(any::<u8>(), 0..4096)) {
                let h1 = hash_sha256(&data);
                let h2 = hash_sha256(&data);
                prop_assert_eq!(&h1, &h2);
                prop_assert!(AsRef::<str>::as_ref(&h1).starts_with("sha256:"));
            }

            #[test]
            fn manifest_identifier_format(hex in "[0-9a-f]{64}") {
                let digest_str = format!("sha256:{hex}");
                let digest: OciDigest = digest_str.parse().unwrap();
                let id = manifest_identifier(&digest);
                prop_assert!(id.starts_with("oci-manifest-"));
                prop_assert!(id.ends_with(&digest_str));
            }

            #[test]
            fn blob_identifier_format(hex in "[0-9a-f]{64}") {
                let digest_str = format!("sha256:{hex}");
                let digest: OciDigest = digest_str.parse().unwrap();
                let id = blob_identifier(&digest);
                prop_assert!(id.starts_with("oci-blob-"));
                prop_assert!(id.ends_with(&digest_str));
            }

            #[test]
            fn write_read_blob_roundtrip(data in proptest::collection::vec(any::<u8>(), 1..4096)) {
                let test_repo = TestRepo::<Sha256HashValue>::new();
                let repo = &test_repo.repo;

                let (digest, verity) = write_blob(repo, &data).unwrap();
                let read_back = open_blob(repo, &digest, Some(&verity)).unwrap();
                prop_assert_eq!(read_back, data);
            }
        }
    }

    /// Test removing all referrers for a subject at once.
    #[test]
    fn test_remove_referrers_for_subject() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (subject_digest, _, _) = create_test_image(repo, Some("subject:v1"), "amd64");
        let (artifact1_digest, _) = create_test_artifact(repo, b"sig-bulk-1");
        let (artifact2_digest, _) = create_test_artifact(repo, b"sig-bulk-2");

        add_referrer(repo, &subject_digest, &artifact1_digest).unwrap();
        add_referrer(repo, &subject_digest, &artifact2_digest).unwrap();
        assert_eq!(list_referrers(repo, &subject_digest).unwrap().len(), 2);

        // Remove all referrers for this subject
        remove_referrers_for_subject(repo, &subject_digest).unwrap();
        assert!(list_referrers(repo, &subject_digest).unwrap().is_empty());

        // Idempotent: calling again on an already-empty subject is fine
        remove_referrers_for_subject(repo, &subject_digest).unwrap();
    }

    // ==================== OCI Fsck Tests ====================

    #[tokio::test]
    async fn test_oci_fsck_healthy_image() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        create_test_image(repo, Some("healthy:v1"), "amd64");

        let result = oci_fsck(repo).await.unwrap();

        assert!(
            result.is_ok(),
            "oci_fsck should pass on healthy repo: {result}"
        );
        assert_eq!(result.images_checked, 1);
        assert_eq!(result.images_corrupted, 0);
        assert!(result.repo_result.is_ok());
        assert!(result.errors.is_empty());
    }

    #[tokio::test]
    async fn test_oci_fsck_detects_corrupt_manifest() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (manifest_digest, manifest_verity, _) =
            create_test_image(repo, Some("corrupt:v1"), "amd64");

        // The manifest is stored as an external object in a splitstream.
        // Find the object file that holds the manifest JSON and corrupt it.
        let manifest_id = manifest_identifier(&manifest_digest);
        let mut stream = repo
            .open_stream(&manifest_id, Some(&manifest_verity), None)
            .unwrap();

        let mut object_refs: Vec<Sha256HashValue> = Vec::new();
        stream
            .get_object_refs(|id| object_refs.push(id.clone()))
            .unwrap();
        assert!(
            !object_refs.is_empty(),
            "manifest should have an external object ref"
        );

        // Corrupt the first (manifest JSON) object on disk.
        // Objects may be immutable due to fs-verity, so delete and recreate.
        let obj = &object_refs[0];
        let hex = obj.to_hex();
        let (dir, file) = hex.split_at(2);
        let obj_path = test_repo.path().join(format!("objects/{dir}/{file}"));
        std::fs::remove_file(&obj_path).unwrap();
        std::fs::write(&obj_path, b"not valid manifest json").unwrap();

        let result = oci_fsck(repo).await.unwrap();

        // The underlying repo fsck should detect the corrupted object
        assert!(
            !result.is_ok(),
            "oci_fsck should fail with corrupted manifest object: {result}"
        );
        assert!(
            result.repo_result().objects_corrupted() > 0,
            "repo fsck should detect corrupted object"
        );
    }

    #[tokio::test]
    async fn test_oci_fsck_detects_missing_layer() {
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (manifest_digest, manifest_verity, _) =
            create_test_image(repo, Some("missing-layer:v1"), "amd64");

        // Open the image to find the layer diff_id
        let img = OciImage::open(repo, &manifest_digest, Some(&manifest_verity)).unwrap();
        let diff_ids = img.layer_diff_ids();
        assert_eq!(diff_ids.len(), 1);

        // Find the layer stream and its backing splitstream object, then
        // delete the stream symlink so the layer appears missing.
        let diff_id_parsed: OciDigest = diff_ids[0].parse().unwrap();
        let layer_id = crate::layer_identifier(&diff_id_parsed);
        let stream_symlink = test_repo.path().join(format!("streams/{layer_id}"));
        std::fs::remove_file(&stream_symlink).unwrap();

        let result = oci_fsck(repo).await.unwrap();

        assert!(
            !result.is_ok(),
            "oci_fsck should detect missing layer: {result}"
        );
        assert!(
            result.images_corrupted > 0,
            "should report corrupted OCI image"
        );
        assert!(
            result
                .errors
                .iter()
                .any(|e| e.to_string().contains("layer-stream-missing")),
            "errors should mention missing layer stream: {:?}",
            result.errors
        );
    }

    // ==================== Additional OCI Fsck Gap Tests ====================

    #[tokio::test]
    async fn test_oci_fsck_detects_config_digest_mismatch() {
        // Exercises fsck_single_image config digest mismatch (line ~1109).
        // Corrupts the config JSON object so its sha256 hash no longer
        // matches the digest recorded in the manifest.
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let (manifest_digest, manifest_verity, config_digest) =
            create_test_image(repo, Some("config-corrupt:v1"), "amd64");

        // Open image to get config verity, then find and corrupt the config object
        let img = OciImage::open(repo, &manifest_digest, Some(&manifest_verity)).unwrap();
        let config_verity = img.config_verity.clone();
        drop(img);

        let config_id = crate::config_identifier(&config_digest);
        let mut stream = repo
            .open_stream(&config_id, Some(&config_verity), None)
            .unwrap();
        let mut config_obj_refs: Vec<Sha256HashValue> = Vec::new();
        stream
            .get_object_refs(|id| config_obj_refs.push(id.clone()))
            .unwrap();
        assert!(!config_obj_refs.is_empty());

        // Corrupt the config object — replace with valid JSON that has
        // a different hash
        let obj = &config_obj_refs[0];
        let hex = obj.to_hex();
        let (prefix, rest) = hex.split_at(2);
        let dir =
            cap_std::fs::Dir::open_ambient_dir(test_repo.path(), cap_std::ambient_authority())
                .unwrap();
        let obj_rel = format!("objects/{prefix}/{rest}");
        dir.remove_file(&obj_rel).unwrap();
        // Write valid JSON config but with modified content
        dir.write(
            &obj_rel,
            br#"{"architecture":"arm64","os":"linux","rootfs":{"type":"layers","diff_ids":[]}}"#,
        )
        .unwrap();

        let result = oci_fsck(repo).await.unwrap();

        // The repo-level fsck will flag the object digest mismatch,
        // which makes the overall result not ok.
        assert!(
            !result.is_ok(),
            "oci_fsck should detect config corruption: {result}"
        );
    }

    #[tokio::test]
    async fn test_oci_fsck_detects_missing_config_named_ref() {
        // Exercises the "manifest missing config reference" branch (line ~1079).
        // Deletes the config named ref from the manifest splitstream by
        // rewriting the manifest splitstream without the config named ref.
        //
        // Approach: create a manifest splitstream that stores the manifest
        // JSON externally but has NO named ref for the config, then point
        // the oci ref to it.
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        // Build a valid manifest JSON
        let layer_data = b"fake-layer-data";
        let layer_digest = hash_sha256(layer_data);

        let mut layer_stream = repo
            .create_stream(crate::skopeo::TAR_LAYER_CONTENT_TYPE)
            .unwrap();
        layer_stream.write_external(layer_data).unwrap();
        let layer_verity = repo
            .write_stream(layer_stream, &crate::layer_identifier(&layer_digest), None)
            .unwrap();

        let rootfs = RootFsBuilder::default()
            .typ("layers")
            .diff_ids(vec![layer_digest.to_string()])
            .build()
            .unwrap();
        let cfg = ConfigBuilder::default().build().unwrap();
        let config = ImageConfigurationBuilder::default()
            .architecture("amd64")
            .os("linux")
            .rootfs(rootfs)
            .config(cfg)
            .build()
            .unwrap();
        let config_json = config.to_string().unwrap();
        let config_digest = hash_sha256(config_json.as_bytes());

        // Store config normally
        let mut config_stream = repo.create_stream(OCI_CONFIG_CONTENT_TYPE).unwrap();
        config_stream.add_named_stream_ref(layer_digest.as_ref(), &layer_verity);
        config_stream
            .write_external(config_json.as_bytes())
            .unwrap();
        let _config_verity = repo
            .write_stream(
                config_stream,
                &crate::config_identifier(&config_digest),
                None,
            )
            .unwrap();

        let config_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::ImageConfig)
            .digest(config_digest.clone())
            .size(config_json.len() as u64)
            .build()
            .unwrap();
        let layer_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::ImageLayerGzip)
            .digest(layer_digest.clone())
            .size(layer_data.len() as u64)
            .build()
            .unwrap();
        let manifest = ImageManifestBuilder::default()
            .schema_version(2u32)
            .media_type(MediaType::ImageManifest)
            .config(config_descriptor)
            .layers(vec![layer_descriptor])
            .build()
            .unwrap();

        let manifest_json = manifest.to_string().unwrap();
        let manifest_digest = hash_sha256(manifest_json.as_bytes());

        // Store manifest WITHOUT config named ref — this is the bug we test
        let manifest_id = manifest_identifier(&manifest_digest);
        let mut manifest_stream = repo.create_stream(OCI_MANIFEST_CONTENT_TYPE).unwrap();
        // Deliberately omit: manifest_stream.add_named_stream_ref(...)
        manifest_stream
            .write_external(manifest_json.as_bytes())
            .unwrap();
        let _manifest_verity = repo
            .write_stream(manifest_stream, &manifest_id, None)
            .unwrap();

        // Create the OCI ref pointing to this manifest
        let ref_path = oci_ref_path("no-config-ref:v1");
        let stream_path = format!("streams/{manifest_id}");
        repo.symlink(&format!("streams/refs/{ref_path}"), &stream_path)
            .unwrap();

        let result = oci_fsck_image(repo, "no-config-ref:v1").await.unwrap();

        assert!(
            !result.is_ok(),
            "oci_fsck should detect missing config ref: {result}"
        );
        assert!(
            result
                .errors
                .iter()
                .any(|e| e.to_string().contains("config-ref-missing")),
            "errors should mention missing config reference: {:?}",
            result.errors
        );
    }

    #[tokio::test]
    async fn test_oci_fsck_healthy_artifact() {
        // Exercises the artifact validation path (line ~1183).
        // Creates a non-container artifact and verifies oci_fsck passes.
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        // Create an artifact with non-ImageConfig media type
        let blob_data = b"artifact-content-for-fsck-test";
        let (blob_digest, blob_verity) = write_blob(repo, blob_data).unwrap();

        let empty_config = b"{}";
        let config_digest = hash_sha256(empty_config);
        let mut config_stream = repo.create_stream(OCI_CONFIG_CONTENT_TYPE).unwrap();
        config_stream.write_external(empty_config).unwrap();
        let config_verity = repo
            .write_stream(
                config_stream,
                &crate::config_identifier(&config_digest),
                None,
            )
            .unwrap();

        let config_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::EmptyJSON) // NOT ImageConfig
            .digest(config_digest.clone())
            .size(empty_config.len() as u64)
            .build()
            .unwrap();
        let layer_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::Other("application/octet-stream".to_string()))
            .digest(blob_digest.clone())
            .size(blob_data.len() as u64)
            .build()
            .unwrap();
        let manifest = ImageManifestBuilder::default()
            .schema_version(2u32)
            .media_type(MediaType::ImageManifest)
            .config(config_descriptor)
            .layers(vec![layer_descriptor])
            .build()
            .unwrap();

        let layer_verities = [(blob_digest.to_string(), blob_verity)];

        let manifest_json = manifest.to_string().unwrap();
        let manifest_digest = hash_sha256(manifest_json.as_bytes());

        write_manifest(
            repo,
            &manifest,
            &manifest_digest,
            &config_verity,
            &layer_verities,
            Some("artifact-fsck:v1"),
        )
        .unwrap();

        let result = oci_fsck(repo).await.unwrap();
        assert!(
            result.is_ok(),
            "oci_fsck should pass for healthy artifact: {result}"
        );
        assert_eq!(result.images_checked, 1);
        assert_eq!(result.images_corrupted, 0);
    }

    #[tokio::test]
    async fn test_oci_fsck_detects_missing_artifact_layer_ref() {
        // Exercises the artifact "manifest missing layer reference" branch
        // (line ~1198). Creates an artifact where the manifest named refs
        // don't include the layer digest.
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let blob_data = b"artifact-blob-missing-ref";
        let (blob_digest, _blob_verity) = write_blob(repo, blob_data).unwrap();

        let empty_config = b"{}";
        let config_digest = hash_sha256(empty_config);
        let mut config_stream = repo.create_stream(OCI_CONFIG_CONTENT_TYPE).unwrap();
        config_stream.write_external(empty_config).unwrap();
        let config_verity = repo
            .write_stream(
                config_stream,
                &crate::config_identifier(&config_digest),
                None,
            )
            .unwrap();

        let config_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::EmptyJSON)
            .digest(config_digest.clone())
            .size(empty_config.len() as u64)
            .build()
            .unwrap();
        let layer_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::Other("application/wasm".to_string()))
            .digest(blob_digest.clone())
            .size(blob_data.len() as u64)
            .build()
            .unwrap();
        let manifest = ImageManifestBuilder::default()
            .schema_version(2u32)
            .media_type(MediaType::ImageManifest)
            .config(config_descriptor)
            .layers(vec![layer_descriptor])
            .build()
            .unwrap();

        // Deliberately pass empty layer_verities — no layer refs in manifest
        let layer_verities: Vec<(String, Sha256HashValue)> = Vec::new();

        let manifest_json = manifest.to_string().unwrap();
        let manifest_digest = hash_sha256(manifest_json.as_bytes());

        write_manifest(
            repo,
            &manifest,
            &manifest_digest,
            &config_verity,
            &layer_verities,
            Some("artifact-no-layer-ref:v1"),
        )
        .unwrap();

        let result = oci_fsck(repo).await.unwrap();

        assert!(
            !result.is_ok(),
            "oci_fsck should detect missing artifact layer ref: {result}"
        );
        assert!(
            result
                .errors
                .iter()
                .any(|e| e.to_string().contains("artifact-layer-ref-missing")),
            "errors should mention missing layer reference: {:?}",
            result.errors
        );
    }

    #[tokio::test]
    async fn test_oci_fsck_image_unresolvable_ref() {
        // Exercises oci_fsck_image with an unresolvable ref (line ~1011).
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let result = oci_fsck_image(repo, "nonexistent:tag").await.unwrap();

        assert!(!result.is_ok(), "should fail for nonexistent ref");
        assert_eq!(result.images_checked, 1);
        assert_eq!(result.images_corrupted, 1);
        assert!(
            result
                .errors
                .iter()
                .any(|e| e.to_string().contains("ref-resolve-failed")),
            "errors should mention cannot resolve ref: {:?}",
            result.errors
        );
    }

    #[tokio::test]
    async fn test_oci_fsck_multiple_images_partial_corruption() {
        // Verifies that oci_fsck checks ALL images and correctly counts
        // corrupted vs healthy ones when there's a mix.
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        // Create two healthy images
        create_test_image(repo, Some("healthy1:v1"), "amd64");
        let (manifest_digest2, manifest_verity2, _) =
            create_test_image(repo, Some("corrupt1:v1"), "arm64");

        // Corrupt the second image's layer
        let img = OciImage::open(repo, &manifest_digest2, Some(&manifest_verity2)).unwrap();
        let diff_ids = img.layer_diff_ids();
        let diff_id_parsed: OciDigest = diff_ids[0].parse().unwrap();
        let layer_id = crate::layer_identifier(&diff_id_parsed);
        let dir =
            cap_std::fs::Dir::open_ambient_dir(test_repo.path(), cap_std::ambient_authority())
                .unwrap();
        dir.remove_file(format!("streams/{layer_id}")).unwrap();

        let result = oci_fsck(repo).await.unwrap();

        assert!(!result.is_ok(), "should detect corruption: {result}");
        assert_eq!(result.images_checked, 2);
        assert_eq!(
            result.images_corrupted, 1,
            "only one image should be corrupt"
        );
    }

    #[tokio::test]
    async fn test_oci_fsck_detects_missing_layer_named_ref_in_config() {
        // Exercises the "config missing layer reference" branch (line ~1134).
        // Creates a container image where the config splitstream is missing
        // the named ref for a layer diff_id.
        let test_repo = TestRepo::<Sha256HashValue>::new();
        let repo = &test_repo.repo;

        let layer_data = b"layer-for-missing-ref-test";
        let layer_digest = hash_sha256(layer_data);

        let mut layer_stream = repo
            .create_stream(crate::skopeo::TAR_LAYER_CONTENT_TYPE)
            .unwrap();
        layer_stream.write_external(layer_data).unwrap();
        let layer_verity = repo
            .write_stream(layer_stream, &crate::layer_identifier(&layer_digest), None)
            .unwrap();

        let rootfs = RootFsBuilder::default()
            .typ("layers")
            .diff_ids(vec![layer_digest.to_string()])
            .build()
            .unwrap();
        let cfg = ConfigBuilder::default().build().unwrap();
        let config = ImageConfigurationBuilder::default()
            .architecture("amd64")
            .os("linux")
            .rootfs(rootfs)
            .config(cfg)
            .build()
            .unwrap();
        let config_json = config.to_string().unwrap();
        let config_digest = hash_sha256(config_json.as_bytes());

        // Store config WITHOUT the layer named ref — this is the bug
        let mut config_stream = repo.create_stream(OCI_CONFIG_CONTENT_TYPE).unwrap();
        // Deliberately omit: config_stream.add_named_stream_ref(&layer_digest, &layer_verity);
        config_stream
            .write_external(config_json.as_bytes())
            .unwrap();
        let config_verity = repo
            .write_stream(
                config_stream,
                &crate::config_identifier(&config_digest),
                None,
            )
            .unwrap();

        let config_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::ImageConfig)
            .digest(config_digest.clone())
            .size(config_json.len() as u64)
            .build()
            .unwrap();
        let layer_descriptor = DescriptorBuilder::default()
            .media_type(MediaType::ImageLayerGzip)
            .digest(layer_digest.clone())
            .size(layer_data.len() as u64)
            .build()
            .unwrap();
        let manifest = ImageManifestBuilder::default()
            .schema_version(2u32)
            .media_type(MediaType::ImageManifest)
            .config(config_descriptor)
            .layers(vec![layer_descriptor])
            .build()
            .unwrap();

        let layer_verities = [(layer_digest.to_string(), layer_verity)];
        let manifest_json = manifest.to_string().unwrap();
        let manifest_digest = hash_sha256(manifest_json.as_bytes());

        write_manifest(
            repo,
            &manifest,
            &manifest_digest,
            &config_verity,
            &layer_verities,
            Some("missing-layer-ref:v1"),
        )
        .unwrap();

        let result = oci_fsck(repo).await.unwrap();

        assert!(
            !result.is_ok(),
            "oci_fsck should detect missing layer ref in config: {result}"
        );
        assert!(
            result
                .errors
                .iter()
                .any(|e| e.to_string().contains("layer-ref-missing")),
            "errors should mention config missing layer reference: {:?}",
            result.errors
        );
    }
}