archy/core/container/src/runtime.rs

use crate::manifest::{AppManifest, BuildConfig};
use crate::podman_client::{ContainerState, ContainerStatus, PodmanClient};
use anyhow::{Context, Result};
use async_trait::async_trait;
use std::process::Command;
use tokio::process::Command as TokioCommand;

#[async_trait]
pub trait ContainerRuntime: Send + Sync {
    async fn pull_image(&self, image: &str, signature: Option<&str>) -> Result<()>;
    async fn create_container(
        &self,
        manifest: &AppManifest,
        name: &str,
        port_offset: u16,
    ) -> Result<String>;
    async fn start_container(&self, name: &str) -> Result<()>;
    async fn stop_container(&self, name: &str) -> Result<()>;
    async fn remove_container(&self, name: &str) -> Result<()>;
    async fn get_container_status(&self, name: &str) -> Result<ContainerStatus>;
    async fn get_container_logs(&self, name: &str, lines: u32) -> Result<Vec<String>>;
    async fn list_containers(&self) -> Result<Vec<ContainerStatus>>;

    /// Check whether an image reference exists in local storage.
    ///
    /// The reconciler calls this before deciding to build. `true` means
    /// `image inspect <image_ref>` succeeded (or equivalent); `false` means
    /// the image is not present. Registry/network state is explicitly NOT
    /// consulted — this is a local-storage check only.
    async fn image_exists(&self, image_ref: &str) -> Result<bool>;

    /// Build a local image from a `BuildConfig`.
    ///
    /// Equivalent to `podman build -t <tag> -f <dockerfile> [--build-arg K=V ...] <context>`.
    /// The resulting image is referenceable by `config.tag` for subsequent
    /// `create_container` / `image_exists` calls. Stdout/stderr are collected
    /// and included in the error on failure; on success they are discarded.
    async fn build_image(&self, config: &BuildConfig) -> Result<()>;
}

pub struct PodmanRuntime {
    client: PodmanClient,
}

impl PodmanRuntime {
    pub fn new(user: String) -> Self {
        Self {
            client: PodmanClient::new(user),
        }
    }

    /// Run `podman <args>`, returning an error with captured stderr on non-zero
    /// exit. Used for operations (build, image inspect) that are awkward over the
    /// HTTP API. The daemon runs as the target user already, so no sudo hop.
    async fn podman_cli(&self, args: &[&str]) -> Result<std::process::Output> {
        let mut cmd = TokioCommand::new("podman");
        cmd.args(args);
        cmd.output()
            .await
            .with_context(|| format!("failed to execute podman {}", args.join(" ")))
    }
}

#[async_trait]
impl ContainerRuntime for PodmanRuntime {
    async fn pull_image(&self, image: &str, signature: Option<&str>) -> Result<()> {
        self.client.pull_image(image, signature).await
    }

    async fn create_container(
        &self,
        manifest: &AppManifest,
        name: &str,
        port_offset: u16,
    ) -> Result<String> {
        // Apply port offset to manifest ports
        let mut dev_manifest = manifest.clone();
        for port in &mut dev_manifest.app.ports {
            port.host += port_offset;
        }

        // PodmanClient doesn't take port_offset, so we use the modified manifest
        self.client.create_container(&dev_manifest, name).await
    }

    async fn start_container(&self, name: &str) -> Result<()> {
        self.client.start_container(name).await
    }

    async fn stop_container(&self, name: &str) -> Result<()> {
        self.client.stop_container(name).await
    }

    async fn remove_container(&self, name: &str) -> Result<()> {
        self.client.remove_container(name).await
    }

    async fn get_container_status(&self, name: &str) -> Result<ContainerStatus> {
        self.client.get_container_status(name).await
    }

    async fn get_container_logs(&self, name: &str, lines: u32) -> Result<Vec<String>> {
        self.client.get_container_logs(name, lines).await
    }

    async fn list_containers(&self) -> Result<Vec<ContainerStatus>> {
        self.client.list_containers().await
    }

    async fn image_exists(&self, image_ref: &str) -> Result<bool> {
        // `podman image exists` returns 0 if present, 1 if absent. Any other
        // exit code is an environment failure we should surface.
        let output = self.podman_cli(&["image", "exists", image_ref]).await?;
        match output.status.code() {
            Some(0) => Ok(true),
            Some(1) => Ok(false),
            Some(code) => {
                let stderr = String::from_utf8_lossy(&output.stderr);
                Err(anyhow::anyhow!(
                    "podman image exists {image_ref} exited with {code}: {stderr}"
                ))
            }
            None => Err(anyhow::anyhow!(
                "podman image exists {image_ref} terminated by signal"
            )),
        }
    }

    async fn build_image(&self, config: &BuildConfig) -> Result<()> {
        let args = build_args_for_podman(config);
        let borrowed: Vec<&str> = args.iter().map(|s| s.as_str()).collect();
        let output = self.podman_cli(&borrowed).await?;
        if !output.status.success() {
            let stderr = String::from_utf8_lossy(&output.stderr);
            let stdout = String::from_utf8_lossy(&output.stdout);
            return Err(anyhow::anyhow!(
                "podman build -t {} failed: {stderr}{}{stdout}",
                config.tag,
                if stderr.is_empty() || stdout.is_empty() {
                    ""
                } else {
                    "\n---stdout---\n"
                }
            ));
        }
        Ok(())
    }
}

/// Build the argv for `podman build` from a BuildConfig.
///
/// Extracted so it can be unit-tested without actually invoking podman.
/// Order is fixed for deterministic tests: subcommand, -t, -f, build-args
/// (sorted by key), context.
fn build_args_for_podman(config: &BuildConfig) -> Vec<String> {
    let mut args: Vec<String> = vec![
        "build".to_string(),
        "-t".to_string(),
        config.tag.clone(),
        "-f".to_string(),
        config.dockerfile.clone(),
    ];
    let mut kv: Vec<(&String, &String)> = config.build_args.iter().collect();
    kv.sort_by(|a, b| a.0.cmp(b.0));
    for (k, v) in kv {
        args.push("--build-arg".to_string());
        args.push(format!("{k}={v}"));
    }
    args.push(config.context.clone());
    args
}

pub struct DockerRuntime {
    _user: String,
}

impl DockerRuntime {
    pub fn new(user: String) -> Self {
        Self { _user: user }
    }

    fn docker_async(&self) -> TokioCommand {
        let mut cmd = TokioCommand::new("docker");
        // Use actual HOME environment variable instead of hardcoded /home
        if let Ok(home) = std::env::var("HOME") {
            cmd.env("HOME", home);
        }
        cmd
    }
}

#[async_trait]
impl ContainerRuntime for DockerRuntime {
    async fn pull_image(&self, image: &str, _signature: Option<&str>) -> Result<()> {
        let mut cmd = self.docker_async();
        cmd.arg("pull").arg(image);

        let output = cmd
            .output()
            .await
            .context("Failed to execute docker pull")?;

        if !output.status.success() {
            let stderr = String::from_utf8_lossy(&output.stderr);
            return Err(anyhow::anyhow!("Failed to pull image: {}", stderr));
        }

        Ok(())
    }

    async fn create_container(
        &self,
        manifest: &AppManifest,
        name: &str,
        port_offset: u16,
    ) -> Result<String> {
        let mut cmd = self.docker_async();
        cmd.arg("create");

        cmd.arg("--name").arg(name);

        if manifest.app.security.readonly_root {
            cmd.arg("--read-only");
        }

        match manifest.app.security.network_policy.as_str() {
            "host" => {
                cmd.arg("--network").arg("host");
            }
            "isolated" => {
                // Docker uses bridge network by default
            }
            _ => {
                cmd.arg("--network")
                    .arg(&manifest.app.security.network_policy);
            }
        }

        // Port mappings with offset
        for port in &manifest.app.ports {
            let host_port = port.host + port_offset;
            cmd.arg("-p")
                .arg(format!("{}:{}", host_port, port.container));
        }

        // Volumes
        for volume in &manifest.app.volumes {
            let mut mount = format!("{}:{}", volume.source, volume.target);
            if !volume.options.is_empty() {
                mount.push_str(&format!(":{}", volume.options.join(",")));
            }
            cmd.arg("-v").arg(mount);
        }

        // Devices
        for device in &manifest.app.devices {
            cmd.arg("--device").arg(device);
        }

        // Environment variables
        for env in &manifest.app.environment {
            cmd.arg("-e").arg(env);
        }

        // Resource limits
        if let Some(cpu) = manifest.app.resources.cpu_limit {
            cmd.arg("--cpus").arg(cpu.to_string());
        }

        if let Some(memory) = &manifest.app.resources.memory_limit {
            cmd.arg("--memory").arg(memory);
        }

        // Capabilities
        cmd.arg("--cap-drop").arg("ALL");
        for cap in &manifest.app.security.capabilities {
            cmd.arg("--cap-add").arg(cap);
        }

        let image_ref = manifest.app.container.image_ref().ok_or_else(|| {
            anyhow::anyhow!(
                "container config for {} has neither a valid image nor build source",
                manifest.app.id
            )
        })?;
        cmd.arg(&image_ref);

        let output = cmd.output().await.context("Failed to create container")?;

        if !output.status.success() {
            let stderr = String::from_utf8_lossy(&output.stderr);
            return Err(anyhow::anyhow!("Failed to create container: {}", stderr));
        }

        let container_id = String::from_utf8_lossy(&output.stdout).trim().to_string();

        Ok(container_id)
    }

    async fn start_container(&self, name: &str) -> Result<()> {
        let mut cmd = self.docker_async();
        cmd.arg("start").arg(name);

        let output = cmd.output().await.context("Failed to start container")?;

        if !output.status.success() {
            let stderr = String::from_utf8_lossy(&output.stderr);
            return Err(anyhow::anyhow!("Failed to start container: {}", stderr));
        }

        Ok(())
    }

    async fn stop_container(&self, name: &str) -> Result<()> {
        let mut cmd = self.docker_async();
        cmd.arg("stop").arg(name);

        let output = cmd.output().await.context("Failed to stop container")?;

        if !output.status.success() {
            let stderr = String::from_utf8_lossy(&output.stderr);
            return Err(anyhow::anyhow!("Failed to stop container: {}", stderr));
        }

        Ok(())
    }

    async fn remove_container(&self, name: &str) -> Result<()> {
        let mut cmd = self.docker_async();
        cmd.arg("rm").arg("-f").arg(name);

        let output = cmd.output().await.context("Failed to remove container")?;

        if !output.status.success() {
            let stderr = String::from_utf8_lossy(&output.stderr);
            return Err(anyhow::anyhow!("Failed to remove container: {}", stderr));
        }

        Ok(())
    }

    async fn get_container_status(&self, name: &str) -> Result<ContainerStatus> {
        let mut cmd = self.docker_async();
        cmd.arg("inspect")
            .arg("--format")
            .arg("{{.Id}}|{{.Name}}|{{.State.Status}}|{{.Config.Image}}|{{.Created}}|{{.NetworkSettings.Ports}}")
            .arg(name);

        let output = cmd.output().await.context("Failed to inspect container")?;

        if !output.status.success() {
            return Err(anyhow::anyhow!("Container not found: {}", name));
        }

        let info = String::from_utf8_lossy(&output.stdout);
        let parts: Vec<&str> = info.trim().split('|').collect();

        if parts.len() < 5 {
            return Err(anyhow::anyhow!("Invalid container inspect output"));
        }

        Ok(ContainerStatus {
            id: parts[0].to_string(),
            name: parts[1].to_string(),
            state: crate::podman_client::ContainerState::from(parts[2]),
            health: None,
            exit_code: None,
            started_at: None,
            image: parts[3].to_string(),
            created: parts[4].to_string(),
            ports: vec![],
            lan_address: None,
        })
    }

    async fn get_container_logs(&self, name: &str, lines: u32) -> Result<Vec<String>> {
        let mut cmd = self.docker_async();
        cmd.arg("logs")
            .arg("--tail")
            .arg(lines.to_string())
            .arg(name);

        let output = cmd.output().await.context("Failed to get container logs")?;

        if !output.status.success() {
            let stderr = String::from_utf8_lossy(&output.stderr);
            return Err(anyhow::anyhow!("Failed to get logs: {}", stderr));
        }

        let logs = String::from_utf8_lossy(&output.stdout);
        Ok(logs.lines().map(|s| s.to_string()).collect())
    }

    async fn list_containers(&self) -> Result<Vec<ContainerStatus>> {
        let mut cmd = self.docker_async();
        cmd.arg("ps").arg("-a").arg("--format").arg("json");

        let output = cmd.output().await.context("Failed to list containers")?;

        if !output.status.success() {
            let stderr = String::from_utf8_lossy(&output.stderr);
            return Err(anyhow::anyhow!("Failed to list containers: {}", stderr));
        }

        let json = String::from_utf8_lossy(&output.stdout);
        let mut result = Vec::new();

        // Docker returns NDJSON (newline-delimited JSON), not a JSON array
        for line in json.lines() {
            if line.trim().is_empty() {
                continue;
            }

            let container: serde_json::Value = serde_json::from_str(line)
                .context(format!("Failed to parse container JSON: {}", line))?;

            // Extract ports from JSON
            let ports_value = &container["Ports"];
            let ports_str = ports_value.as_str().unwrap_or("");
            let ports: Vec<String> = if !ports_str.is_empty() {
                ports_str.split(", ").map(|s| s.to_string()).collect()
            } else {
                vec![]
            };

            result.push(ContainerStatus {
                id: container["ID"].as_str().unwrap_or("").to_string(),
                name: container["Names"].as_str().unwrap_or("").to_string(),
                state: ContainerState::from(container["State"].as_str().unwrap_or("unknown")),
                health: None,
                exit_code: container["ExitCode"].as_i64().map(|c| c as i32),
                started_at: None,
                image: container["Image"].as_str().unwrap_or("").to_string(),
                created: container["CreatedAt"].as_str().unwrap_or("").to_string(),
                ports,
                lan_address: None,
            });
        }

        Ok(result)
    }

    async fn image_exists(&self, image_ref: &str) -> Result<bool> {
        // `docker image inspect` exits 1 when the image is absent. Any message
        // to stderr in that case is informational; we swallow it.
        let mut cmd = self.docker_async();
        cmd.arg("image").arg("inspect").arg(image_ref);
        let output = cmd
            .output()
            .await
            .context("failed to execute docker image inspect")?;
        match output.status.code() {
            Some(0) => Ok(true),
            Some(1) => Ok(false),
            Some(code) => {
                let stderr = String::from_utf8_lossy(&output.stderr);
                Err(anyhow::anyhow!(
                    "docker image inspect {image_ref} exited with {code}: {stderr}"
                ))
            }
            None => Err(anyhow::anyhow!(
                "docker image inspect {image_ref} terminated by signal"
            )),
        }
    }

    async fn build_image(&self, config: &BuildConfig) -> Result<()> {
        let mut cmd = self.docker_async();
        cmd.arg("build")
            .arg("-t")
            .arg(&config.tag)
            .arg("-f")
            .arg(&config.dockerfile);
        for (k, v) in &config.build_args {
            cmd.arg("--build-arg").arg(format!("{k}={v}"));
        }
        cmd.arg(&config.context);
        let output = cmd
            .output()
            .await
            .context("failed to execute docker build")?;
        if !output.status.success() {
            let stderr = String::from_utf8_lossy(&output.stderr);
            return Err(anyhow::anyhow!(
                "docker build -t {} failed: {stderr}",
                config.tag
            ));
        }
        Ok(())
    }
}

pub struct AutoRuntime {
    runtime: Box<dyn ContainerRuntime>,
}

impl AutoRuntime {
    pub async fn new(user: String) -> Result<Self> {
        // Try Podman first
        if Self::check_podman_available() {
            Ok(Self {
                runtime: Box::new(PodmanRuntime::new(user)),
            })
        } else if Self::check_docker_available() {
            Ok(Self {
                runtime: Box::new(DockerRuntime::new(user)),
            })
        } else {
            Err(anyhow::anyhow!("Neither Podman nor Docker is available"))
        }
    }

    fn check_podman_available() -> bool {
        Command::new("podman").arg("--version").output().is_ok()
    }

    fn check_docker_available() -> bool {
        Command::new("docker").arg("--version").output().is_ok()
    }
}

#[async_trait]
impl ContainerRuntime for AutoRuntime {
    async fn pull_image(&self, image: &str, signature: Option<&str>) -> Result<()> {
        self.runtime.pull_image(image, signature).await
    }

    async fn create_container(
        &self,
        manifest: &AppManifest,
        name: &str,
        port_offset: u16,
    ) -> Result<String> {
        self.runtime
            .create_container(manifest, name, port_offset)
            .await
    }

    async fn start_container(&self, name: &str) -> Result<()> {
        self.runtime.start_container(name).await
    }

    async fn stop_container(&self, name: &str) -> Result<()> {
        self.runtime.stop_container(name).await
    }

    async fn remove_container(&self, name: &str) -> Result<()> {
        self.runtime.remove_container(name).await
    }

    async fn get_container_status(&self, name: &str) -> Result<ContainerStatus> {
        self.runtime.get_container_status(name).await
    }

    async fn get_container_logs(&self, name: &str, lines: u32) -> Result<Vec<String>> {
        self.runtime.get_container_logs(name, lines).await
    }

    async fn list_containers(&self) -> Result<Vec<ContainerStatus>> {
        self.runtime.list_containers().await
    }

    async fn image_exists(&self, image_ref: &str) -> Result<bool> {
        self.runtime.image_exists(image_ref).await
    }

    async fn build_image(&self, config: &BuildConfig) -> Result<()> {
        self.runtime.build_image(config).await
    }
}

// Runtime factory functions will be provided by the archipelago crate
// that imports this library and has access to Config

#[cfg(test)]
mod tests {
    use super::*;
    use std::collections::HashMap;

    fn cfg(context: &str, tag: &str, dockerfile: &str, args: &[(&str, &str)]) -> BuildConfig {
        BuildConfig {
            context: context.to_string(),
            dockerfile: dockerfile.to_string(),
            tag: tag.to_string(),
            build_args: args
                .iter()
                .map(|(k, v)| (k.to_string(), v.to_string()))
                .collect::<HashMap<_, _>>(),
        }
    }

    #[test]
    fn build_args_minimal() {
        let c = cfg("/tmp/ctx", "archy-bitcoin-ui:local", "Dockerfile", &[]);
        assert_eq!(
            build_args_for_podman(&c),
            vec![
                "build",
                "-t",
                "archy-bitcoin-ui:local",
                "-f",
                "Dockerfile",
                "/tmp/ctx",
            ]
        );
    }

    #[test]
    fn build_args_custom_dockerfile() {
        let c = cfg("/opt/archy/bitcoin-ui", "x:local", "Dockerfile.prod", &[]);
        let got = build_args_for_podman(&c);
        assert_eq!(got[3], "-f");
        assert_eq!(got[4], "Dockerfile.prod");
        assert_eq!(got.last().unwrap(), "/opt/archy/bitcoin-ui");
    }

    #[test]
    fn build_args_are_sorted_deterministically() {
        // HashMap iteration order is nondeterministic; the runtime sorts so that
        // equivalent BuildConfigs produce identical commands (easier to debug,
        // cache-friendly if we ever layer build-cache keys on top).
        let c = cfg(
            "/c",
            "t",
            "Dockerfile",
            &[("BAR", "2"), ("FOO", "1"), ("BAZ", "3")],
        );
        let args = build_args_for_podman(&c);
        let flat: Vec<&str> = args.iter().map(|s| s.as_str()).collect();
        // Build args appear as pairs of --build-arg K=V; locate them:
        let mut pairs: Vec<&str> = Vec::new();
        for w in flat.windows(2) {
            if w[0] == "--build-arg" {
                pairs.push(w[1]);
            }
        }
        assert_eq!(pairs, vec!["BAR=2", "BAZ=3", "FOO=1"]);
    }

    #[test]
    fn build_args_context_is_last() {
        // Context MUST be the final positional argument — podman treats any
        // stray trailing arg after build-args as the context, so placement
        // matters. Regression guard.
        let c = cfg("/final/context", "t", "Dockerfile", &[("K", "V")]);
        let args = build_args_for_podman(&c);
        assert_eq!(args.last().unwrap(), "/final/context");
    }
}