cf-modkit-db 0.7.2

#![allow(clippy::unwrap_used, clippy::expect_used)]
#![allow(dead_code)]
use anyhow::Result;
use std::path::PathBuf;
use std::time::Duration;

#[cfg(any(feature = "pg", feature = "mysql"))]
use testcontainers::{ImageExt, runners::AsyncRunner};

/// Returns a test data directory under target/test_data/modkit-db/
/// Creates the directory if it doesn't exist.
///
/// # Panics
/// Panics if the parent directories cannot be resolved or the directory cannot be created.
#[must_use]
pub fn test_data_dir() -> PathBuf {
    let project_root = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
        .parent()
        .unwrap()
        .parent()
        .unwrap()
        .to_path_buf();
    let test_dir = project_root
        .join("target")
        .join("test_data")
        .join("modkit-db");
    std::fs::create_dir_all(&test_dir).unwrap();
    test_dir
}

pub struct DbUnderTest {
    pub url: String,
    #[allow(dead_code, clippy::type_complexity)]
    _cleanup: Option<Box<dyn FnOnce() + Send + Sync>>,
}

#[cfg(feature = "sqlite")]
#[must_use]
pub fn bring_up_sqlite() -> DbUnderTest {
    DbUnderTest {
        url: "sqlite::memory:".into(),
        _cleanup: None,
    }
}

/// Bring up a `PostgreSQL` test container.
///
/// # Errors
/// Returns an error if the container fails to start or become ready.
#[cfg(feature = "pg")]
pub async fn bring_up_postgres() -> Result<DbUnderTest> {
    use testcontainers::ContainerRequest;
    use testcontainers_modules::postgres::Postgres;

    let postgres_image = Postgres::default();
    let container_request = ContainerRequest::from(postgres_image)
        .with_env_var("POSTGRES_PASSWORD", "pass")
        .with_env_var("POSTGRES_USER", "user")
        .with_env_var("POSTGRES_DB", "app");

    let container = container_request.start().await?;
    let port = container.get_host_port_ipv4(5432).await?;
    wait_for_tcp("127.0.0.1", port, Duration::from_secs(20)).await?;

    Ok(DbUnderTest {
        url: format!("postgres://user:pass@127.0.0.1:{port}/app"),
        _cleanup: Some(Box::new(move || drop(container))),
    })
}

/// Bring up a `MySQL` container for testing.
///
/// # Errors
/// Returns an error if the container fails to start or the port cannot be obtained.
#[cfg(feature = "mysql")]
pub async fn bring_up_mysql() -> Result<DbUnderTest> {
    use testcontainers::ContainerRequest;
    use testcontainers_modules::mysql::Mysql;
    let mysql_image = Mysql::default();
    let container_request = ContainerRequest::from(mysql_image)
        .with_env_var("MYSQL_ROOT_PASSWORD", "root")
        .with_env_var("MYSQL_USER", "user")
        .with_env_var("MYSQL_PASSWORD", "pass")
        .with_env_var("MYSQL_DATABASE", "app");

    let container = container_request.start().await?;
    let port = container.get_host_port_ipv4(3306).await?;
    wait_for_tcp("127.0.0.1", port, Duration::from_secs(30)).await?;

    Ok(DbUnderTest {
        url: format!("mysql://user:pass@127.0.0.1:{port}/app"),
        _cleanup: Some(Box::new(move || drop(container))),
    })
}

async fn wait_for_tcp(host: &str, port: u16, timeout: Duration) -> Result<()> {
    use tokio::{
        net::TcpStream,
        time::{Instant, sleep},
    };
    let deadline = Instant::now() + timeout;
    loop {
        if TcpStream::connect((host, port)).await.is_ok() {
            return Ok(());
        }
        if Instant::now() >= deadline {
            anyhow::bail!("Timeout waiting for {host}:{port}");
        }
        sleep(Duration::from_millis(200)).await;
    }
}