Merge pull request #733 from cgwalters/command-ext

utils: Add a little CommandRunExt helper trait

Author: cgwalters

lib/src/install.rs

@@ -47,7 +47,7 @@ use crate::mount::Filesystem;
 use crate::spec::ImageReference;
 use crate::store::Storage;
 use crate::task::Task;
-use crate::utils::sigpolicy_from_opts;
+use crate::utils::{sigpolicy_from_opts, CommandRunExt};
 
 /// The default "stateroot" or "osname"; see https://github.com/ostreedev/ostree/issues/2794
 const STATEROOT_DEFAULT: &str = "default";
@@ -587,10 +587,9 @@ async fn initialize_ostree_root(state: &State, root_setup: &RootSetup) -> Result
         ("sysroot.bootprefix", "true"),
         ("sysroot.readonly", "true"),
     ] {
-        Task::new("Configuring ostree repo", "ostree")
+        Command::new("ostree")
             .args(["config", "--repo", "ostree/repo", "set", k, v])
-            .cwd(rootfs_dir)?
-            .quiet()
+            .cwd_dir(rootfs_dir.try_clone()?)
             .run()?;
     }
     Task::new("Initializing sysroot", "ostree")

lib/src/utils.rs

@@ -10,6 +10,25 @@ use ostree::glib;
 use ostree_ext::container::SignatureSource;
 use ostree_ext::ostree;
 
+/// Helpers intended for [`std::process::Command`].
+pub(crate) trait CommandRunExt {
+    fn run(&mut self) -> Result<()>;
+}
+
+impl CommandRunExt for Command {
+    /// Synchronously execute the child, and return an error if the child exited unsuccessfully.
+    fn run(&mut self) -> Result<()> {
+        let st = self.status()?;
+        if !st.success() {
+            // Note that we intentionally *don't* include the command string
+            // in the output; we leave it to the caller to add that if they want,
+            // as it may be verbose.
+            anyhow::bail!(format!("Subprocess failed: {st:?}"))
+        }
+        Ok(())
+    }
+}
+
 /// Try to look for keys injected by e.g. rpm-ostree requesting machine-local
 /// changes; if any are present, return `true`.
 pub(crate) fn origin_has_rpmostree_stuff(kf: &glib::KeyFile) -> bool {
@@ -190,3 +209,9 @@ fn test_sigpolicy_from_opts() {
         SignatureSource::ContainerPolicyAllowInsecure
     );
 }
+
+#[test]
+fn command_run_ext() {
+    Command::new("true").run().unwrap();
+    assert!(Command::new("false").run().is_err());
+}