1 files changed, 162 insertions, 0 deletions
diff --git a/xtask/src/command.rs b/xtask/src/command.rs
new file mode 100644
index 00000000..8bf49849
--- /dev/null
+++ b/xtask/src/command.rs
@@ -0,0 +1,162 @@
+use crate::RunResult;
+use core::fmt;
+use os_pipe::pipe;
+use std::{fs::File, io::Read, path::Path, process::Command};
+
+#[derive(Debug, Clone, Copy, PartialEq)]
+pub enum BuildMode {
+    Release,
+    Debug,
+}
+
+pub enum CargoCommand<'a> {
+    Run {
+        example: &'a str,
+        target: &'a str,
+        features: Option<&'a str>,
+        mode: BuildMode,
+    },
+    Build {
+        example: &'a str,
+        target: &'a str,
+        features: Option<&'a str>,
+        mode: BuildMode,
+    },
+    Objcopy {
+        example: &'a str,
+        target: &'a str,
+        features: Option<&'a str>,
+        ihex: &'a str,
+    },
+    Size {
+        example_paths: Vec<&'a Path>,
+    },
+    Clean,
+}
+
+impl<'a> CargoCommand<'a> {
+    fn name(&self) -> &str {
+        match self {
+            CargoCommand::Run { .. } => "run",
+            CargoCommand::Size { example_paths: _ } => "rust-size",
+            CargoCommand::Clean => "clean",
+            CargoCommand::Build { .. } => "build",
+            CargoCommand::Objcopy { .. } => "objcopy",
+        }
+    }
+
+    pub fn args(&self) -> Vec<&str> {
+        match self {
+            CargoCommand::Run {
+                example,
+                target,
+                features,
+                mode,
+            }
+            | CargoCommand::Build {
+                example,
+                target,
+                features,
+                mode,
+            } => {
+                let mut args = vec![self.name(), "--example", example, "--target", target];
+
+                if let Some(feature_name) = features {
+                    args.extend_from_slice(&["--features", feature_name]);
+                }
+                if let Some(flag) = mode.to_flag() {
+                    args.push(flag);
+                }
+                args
+            }
+            CargoCommand::Size { example_paths } => {
+                example_paths.iter().map(|p| p.to_str().unwrap()).collect()
+            }
+            CargoCommand::Clean => vec!["clean"],
+            CargoCommand::Objcopy {
+                example,
+                target,
+                features,
+                ihex,
+            } => {
+                let mut args = vec![self.name(), "--example", example, "--target", target];
+
+                if let Some(feature_name) = features {
+                    args.extend_from_slice(&["--features", feature_name]);
+                }
+
+                // this always needs to go at the end
+                args.extend_from_slice(&["--", "-O", "ihex", ihex]);
+                args
+            }
+        }
+    }
+
+    pub fn command(&self) -> &str {
+        match self {
+            // we need to cheat a little here:
+            // `cargo size` can't be ran on multiple files, so we're using `rust-size` instead –
+            // which isn't a command that starts wizh `cargo`. So we're sneakily swapping them out :)
+            CargoCommand::Size { .. } => "rust-size",
+            _ => "cargo",
+        }
+    }
+}
+
+impl BuildMode {
+    pub fn to_flag(&self) -> Option<&str> {
+        match self {
+            BuildMode::Release => Some("--release"),
+            BuildMode::Debug => None,
+        }
+    }
+}
+
+impl fmt::Display for BuildMode {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        let cmd = match self {
+            BuildMode::Release => "release",
+            BuildMode::Debug => "debug",
+        };
+
+        write!(f, "{}", cmd)
+    }
+}
+
+pub fn run_command(command: &CargoCommand) -> anyhow::Result<RunResult> {
+    let (mut reader, writer) = pipe()?;
+    println!("👟 {} {}", command.command(), command.args().join(" "));
+
+    let mut handle = Command::new(command.command())
+        .args(command.args())
+        .stdout(writer)
+        .spawn()?;
+
+    // retrieve output and clean up
+    let mut output = String::new();
+    reader.read_to_string(&mut output)?;
+    let exit_status = handle.wait()?;
+
+    Ok(RunResult {
+        exit_status,
+        output,
+    })
+}
+
+/// Check if `run` was sucessful.
+/// returns Ok in case the run went as expected,
+/// Err otherwise
+pub fn run_successful(run: &RunResult, expected_output_file: String) -> anyhow::Result<()> {
+    let mut file_handle = File::open(expected_output_file)?;
+    let mut expected_output = String::new();
+    file_handle.read_to_string(&mut expected_output)?;
+    if expected_output == run.output && run.exit_status.success() {
+        Ok(())
+    } else {
+        Err(anyhow::anyhow!(
+            "Run failed with exit status {}: {}",
+            run.exit_status,
+            run.output
+        ))
+    }
+}