haddock 0.2.1

Docker Compose for Podman
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use std::fmt::{self, Display, Formatter};

use anyhow::{anyhow, bail, Result};
use clap::ValueEnum;

use crate::{
    compose::types::Compose,
    podman::{types::Container, Podman},
};

/// Print the public port for a port binding
#[derive(clap::Args, Debug)]
#[command(next_display_order = None)]
pub(crate) struct Args {
    service: String,
    port: u16,

    #[arg(long, value_enum, default_value_t = Protocol::Tcp)]
    protocol: Protocol,

    /// Index of the container if service has multiple replicas
    #[arg(long, default_value_t = 1)]
    index: usize,
}

#[derive(ValueEnum, Clone, Debug)]
enum Protocol {
    Tcp,
    Udp,
}

impl Display for Protocol {
    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
        write!(f, "{}", format!("{self:?}").to_ascii_lowercase())
    }
}

pub(crate) async fn run(args: Args, podman: &Podman, file: &Compose) -> Result<()> {
    let output = podman
        .force_run([
            "ps",
            "--all",
            "--format",
            "json",
            "--filter",
            "label=io.podman.compose.oneoff=false",
            "--filter",
            &format!("pod={}", file.name.as_ref().unwrap()),
            "--filter",
            &format!("label=io.podman.compose.service={}", args.service),
        ])
        .await?;
    let containers = serde_json::from_str::<Vec<Container>>(&output)?;

    if containers.is_empty() {
        bail!("No container found for service \"{}\"", args.service);
    }

    let container = containers
        .into_iter()
        .find_map(|mut container| {
            container
                .labels
                .and_then(|labels| labels.container_number)
                .and_then(|n| {
                    if n == args.index {
                        container.names.pop_front()
                    } else {
                        None
                    }
                })
        })
        .ok_or_else(|| {
            anyhow!(
                "Service \"{}\" is not running container #{}",
                args.service,
                args.index
            )
        })?;

    print!(
        "{}",
        podman
            .run([
                "port",
                &container,
                &format!("{}/{}", args.port, args.protocol)
            ])
            .await?
    );

    Ok(())
}