use testcontainers::{ContainerAsync, Image, TestcontainersError, core::WaitFor, runners::AsyncRunner};

/// Container port exposed by the local OCI registry used in tests.

pub const LYQUOR_REGISTRY_TEST_PORT: u16 = 5000;

const TEST_REGISTRY_IMAGE: &str = "registry";

const TEST_REGISTRY_TAG: &str = "3.0.0";

/// Running test OCI registry container.

#[derive(Debug)]

pub struct TestRegistry {

    _container: ContainerAsync<TestRegistryImage>,

    addr: String,

}

impl TestRegistry {

    /// Start a disposable OCI registry container.

    pub async fn start() -> Result<Self, TestcontainersError> {

        let container = TestRegistryImage.start().await
?0
;

        let host = container.get_host().await
?0
;

        let port = container.get_host_port_ipv4(LYQUOR_REGISTRY_TEST_PORT).await
?0
;

        let addr = format!("{host}:{port}");

        Ok(Self {

            _container: container,

            addr,

        })

    }

    /// Return the host address for clients to push and pull images.

    pub fn addr(&self) -> &str {

        &self.addr

    }

}

#[derive(Debug, Default)]

struct TestRegistryImage;

impl Image for TestRegistryImage {

    fn name(&self) -> &str {

        TEST_REGISTRY_IMAGE

    }

    fn tag(&self) -> &str {

        TEST_REGISTRY_TAG

    }

    fn ready_conditions(&self) -> Vec<WaitFor> {

        vec![WaitFor::message_on_stderr("listening on")]

    }

}