use nethsm::Url;
use rustainers::{
    ExposedPort,
    ImageName,
    RunnableContainer,
    RunnableContainerBuilder,
    ToRunnableContainer,
    WaitStrategy,
};
use testresult::TestResult;
use uuid::{timestamp::Timestamp, NoContext, Uuid};

/// The NetHSM container image and specific tag
///
/// We are currently pinning to "c16fe4ed" due to https://gitlab.archlinux.org/archlinux/signstar/-/issues/32
/// In the future we will probably want to stick to a specific release tag (representing an actual
/// upstream release) and not "testing"
const IMAGE_NAME: &str = "docker.io/nitrokey/nethsm:c16fe4ed";
const DEFAULT_PORT: u16 = 8443;
const DEFAULT_PATH: &str = "/api/v1";

#[derive(Debug)]
pub struct NetHsmImage {
    pub image: ImageName,
    pub port: ExposedPort,
}

impl NetHsmImage {
    pub async fn url(&self) -> TestResult<Url> {
        Ok(Url::new(&format!(
            "https://localhost:{}{}",
            self.port.host_port().await?,
            DEFAULT_PATH
        ))?)
    }
}

impl Default for NetHsmImage {
    fn default() -> Self {
        Self {
            image: ImageName::new(IMAGE_NAME),
            port: ExposedPort::new(DEFAULT_PORT),
        }
    }
}

impl ToRunnableContainer for NetHsmImage {
    fn to_runnable(&self, builder: RunnableContainerBuilder) -> RunnableContainer {
        builder
            .with_image(self.image.clone())
            .with_container_name(Some(format!(
                "nethsm-test-{}",
                Uuid::new_v7(Timestamp::now(NoContext))
            )))
            .with_wait_strategy(WaitStrategy::stderr_contains(
                "listening on 8443/TCP for HTTPS",
            ))
            .with_port_mappings([self.port.clone()])
            .build()
    }
}