xtask: basic tool to build Helios agent in a job

Author: jclulow

xtask/scripts/build_helios_agent.sh

@@ -0,0 +1,57 @@
+#!/bin/bash
+
+#
+# This job script is run inside a buildomat ephemeral VM.
+#
+if [[ -z $BUILDOMAT_JOB_ID ]]; then
+	printf 'ERROR: this is supposed to be run under buildomat.\n' >&2
+	exit 1
+fi
+
+set -o errexit
+set -o pipefail
+set -o xtrace
+
+#
+# Make sure we can locate the C compiler:
+#
+export PATH=/usr/bin:/usr/sbin:/sbin:/opt/ooce/sbin:/opt/ooce/bin
+
+#
+# Install a stable Rust toolchain:
+#
+RUSTUP_INIT_SKIP_PATH_CHECK=yes \
+curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | bash -s - \
+    --default-toolchain stable \
+    --profile minimal \
+    --no-modify-path \
+    -y -q
+
+. "$HOME/.cargo/env"
+
+mkdir -p /work
+mkdir -p /out
+
+cd /work
+
+#
+# Unpack the templates and scripts we included when kicking off the job:
+#
+cpio -idv < '/input/src.cpio'
+
+cargo build --features vendored-openssl --release --locked -p buildomat-agent
+
+#
+# Copy rather than moving, because we're on the same file system and gzip
+# complains about a link count issue otherwise:
+#
+cp target/release/buildomat-agent /out/buildomat-agent
+chmod 0755 /out/buildomat-agent
+
+sha256sum /out/buildomat-agent \
+    >/out/buildomat-agent.sha256.txt
+gzip /out/buildomat-agent
+sha256sum /out/buildomat-agent.gz \
+    >/out/buildomat-agent.gz.sha256.txt
+
+find '/out' -type f -ls

xtask/src/main.rs

@@ -91,7 +91,52 @@ fn openapi() -> Result<()> {
     Ok(())
 }
 
-fn build_linux_agent() -> Result<()> {
+#[derive(Copy, Clone, Debug)]
+enum AgentBuild {
+    Linux,
+    Helios,
+}
+
+impl AgentBuild {
+    fn job_name(&self) -> &str {
+        match self {
+            AgentBuild::Linux => "build-linux-agent",
+            AgentBuild::Helios => "build-helios-agent",
+        }
+    }
+
+    fn script(&self) -> &str {
+        match self {
+            AgentBuild::Linux => {
+                include_str!("../scripts/build_linux_agent.sh")
+            }
+            AgentBuild::Helios => {
+                include_str!("../scripts/build_helios_agent.sh")
+            }
+        }
+    }
+
+    fn output_filename(&self) -> &str {
+        match self {
+            AgentBuild::Linux => "buildomat-agent-linux.gz",
+            AgentBuild::Helios => "buildomat-agent.gz",
+        }
+    }
+
+    fn use_target(&self) -> &str {
+        /*
+         * We should use the earliest version of a target when building the
+         * agent.  Using a newer target may result in binaries that don't work
+         * in older images.
+         */
+        match self {
+            AgentBuild::Linux => "ubuntu-18.04",
+            AgentBuild::Helios => "helios-2.0-20240204",
+        }
+    }
+}
+
+fn build_agent(ab: AgentBuild) -> Result<()> {
     let files = Command::new("git").arg("ls-files").output()?;
 
     if !files.status.success() {
@@ -132,13 +177,13 @@ fn build_linux_agent() -> Result<()> {
         .arg("run")
         .arg("-W")
         .arg("-n")
-        .arg("build-linux-agent")
+        .arg(ab.job_name())
         .arg("-c")
-        .arg(include_str!("../scripts/build_linux_agent.sh"))
+        .arg(ab.script())
         .arg("-t")
-        .arg("ubuntu-18.04")
+        .arg(ab.use_target())
         .arg("-O")
-        .arg("=/out/buildomat-agent-linux.gz")
+        .arg(format!("=/out/{}", ab.output_filename()))
         .arg("-O")
         .arg("=/out/*.sha256.txt")
         .arg("-i")
@@ -182,8 +227,8 @@ fn build_linux_agent() -> Result<()> {
         .arg("job")
         .arg("copy")
         .arg(&jid)
-        .arg("/out/buildomat-agent-linux.gz")
-        .arg("./buildomat-agent-linux.gz")
+        .arg(format!("/out/{}", ab.output_filename()))
+        .arg(format!("./{}", ab.output_filename()))
         .stdin(Stdio::inherit())
         .stdout(Stdio::inherit())
         .stderr(Stdio::inherit())
@@ -192,7 +237,7 @@ fn build_linux_agent() -> Result<()> {
     println!("unpacking agent...");
 
     Command::new("gunzip")
-        .arg("./buildomat-agent-linux.gz")
+        .arg(format!("./{}", ab.output_filename()))
         .stdin(Stdio::inherit())
         .stdout(Stdio::inherit())
         .stderr(Stdio::inherit())
@@ -309,7 +354,8 @@ fn crates() -> Result<()> {
 fn main() -> Result<()> {
     match std::env::args().nth(1).as_deref() {
         Some("openapi") => openapi(),
-        Some("build-linux-agent") => build_linux_agent(),
+        Some("build-linux-agent") => build_agent(AgentBuild::Linux),
+        Some("build-agent") => build_agent(AgentBuild::Helios),
         Some("crates") => crates(),
         Some(_) | None => {
             bail!("do not know how to do that");