logos-blockchain-testing/testing-framework/runners/compose/src/cleanup.rs

use std::{env, path::PathBuf, thread};

use testing_framework_core::scenario::CleanupGuard;

use crate::{
    cfgsync::CfgsyncServerHandle,
    compose::{ComposeCommandError, compose_down},
    workspace::ComposeWorkspace,
};

/// Cleans up a compose deployment and associated cfgsync container.
pub struct RunnerCleanup {
    pub compose_file: PathBuf,
    pub project_name: String,
    pub root: PathBuf,
    workspace: Option<ComposeWorkspace>,
    cfgsync: Option<CfgsyncServerHandle>,
}

impl RunnerCleanup {
    /// Construct a cleanup guard for the given compose deployment.
    pub fn new(
        compose_file: PathBuf,
        project_name: String,
        root: PathBuf,
        workspace: ComposeWorkspace,
        cfgsync: Option<CfgsyncServerHandle>,
    ) -> Self {
        debug_assert!(
            !compose_file.as_os_str().is_empty() && !project_name.is_empty(),
            "compose cleanup should receive valid identifiers"
        );
        Self {
            compose_file,
            project_name,
            root,
            workspace: Some(workspace),
            cfgsync,
        }
    }

    fn teardown_compose(&self) {
        if let Err(err) =
            run_compose_down_blocking(&self.compose_file, &self.project_name, &self.root)
        {
            eprintln!("[compose-runner] docker compose down failed: {err}");
        }
    }
}

fn run_compose_down_blocking(
    compose_file: &PathBuf,
    project_name: &str,
    root: &PathBuf,
) -> Result<(), ComposeCommandError> {
    let compose_file = compose_file.clone();
    let project_name = project_name.to_owned();
    let root = root.clone();

    let handle = thread::spawn(move || {
        tokio::runtime::Builder::new_current_thread()
            .enable_all()
            .build()
            .map_err(|err| ComposeCommandError::Spawn {
                command: "docker compose down".into(),
                source: std::io::Error::new(std::io::ErrorKind::Other, err),
            })?
            .block_on(compose_down(&compose_file, &project_name, &root))
    });

    handle.join().map_err(|_| ComposeCommandError::Spawn {
        command: "docker compose down".into(),
        source: std::io::Error::new(
            std::io::ErrorKind::Other,
            "join failure running compose down",
        ),
    })?
}
impl CleanupGuard for RunnerCleanup {
    fn cleanup(mut self: Box<Self>) {
        if self.should_preserve() {
            self.persist_workspace();
            return;
        }

        self.teardown_compose();

        if let Some(mut handle) = self.cfgsync.take() {
            handle.shutdown();
        }
    }
}

impl RunnerCleanup {
    fn should_preserve(&self) -> bool {
        env::var("COMPOSE_RUNNER_PRESERVE").is_ok() || env::var("TESTNET_RUNNER_PRESERVE").is_ok()
    }

    fn persist_workspace(&mut self) {
        if let Some(workspace) = self.workspace.take() {
            let keep = workspace.into_inner().keep();
            eprintln!(
                "[compose-runner] preserving docker state at {}",
                keep.display()
            );
        }

        eprintln!("[compose-runner] compose preserve flag set; skipping docker compose down");
    }
}