brokk-mj-controller 2.23.2

Daemon-side controller, session manager, and web server for Mjolnir
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

use super::*;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum TargetOs {
    Linux,
    Darwin,
}

/// Platform of the execution boundary, which may differ from both the
/// controller and the host of a Linux container.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct TargetPlatform {
    pub os: TargetOs,
    pub architecture: &'static str,
}

impl TargetPlatform {
    pub fn parse(uname: &str) -> Result<Self> {
        let mut fields = uname.split_whitespace();
        let os = match fields.next() {
            Some("Linux") => TargetOs::Linux,
            Some("Darwin") => TargetOs::Darwin,
            other => bail!("unsupported target operating system {other:?}"),
        };
        let architecture = normalize_architecture(fields.next().unwrap_or_default())?;
        ensure!(
            fields.next().is_none(),
            "invalid target platform response {uname:?}"
        );
        Ok(Self { os, architecture })
    }
}

pub fn normalize_architecture(architecture: &str) -> Result<&'static str> {
    match architecture {
        "x86_64" | "amd64" => Ok("x86_64"),
        "aarch64" | "arm64" => Ok("aarch64"),
        other => bail!("unsupported target architecture {other:?}"),
    }
}

pub fn platform_probe(locator: &TargetLocator) -> CommandSpec {
    locator_command(locator, vec!["uname".into(), "-sm".into()]).purpose("detect target platform")
}

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

    #[test]
    fn detects_supported_platforms_and_rejects_unknown_responses() {
        for (text, os, architecture) in [
            ("Darwin arm64\n", TargetOs::Darwin, "aarch64"),
            ("Darwin x86_64\n", TargetOs::Darwin, "x86_64"),
            ("Linux aarch64\n", TargetOs::Linux, "aarch64"),
            ("Linux x86_64\n", TargetOs::Linux, "x86_64"),
        ] {
            assert_eq!(
                TargetPlatform::parse(text).unwrap(),
                TargetPlatform { os, architecture }
            );
        }
        for text in [
            "FreeBSD arm64",
            "Linux riscv64",
            "Darwin",
            "",
            "Linux x86_64 junk",
        ] {
            assert!(TargetPlatform::parse(text).is_err(), "{text}");
        }
    }
}