frontend: pass unknown subcommands to python CLI (#55)

* rust code to passthrough unknown subcommands

* make probe-py avalible to probe frontend wrapper

flake.nix

@@ -75,7 +75,8 @@
                   makeWrapper \
                     ${self.packages.${system}.probe-cli}/bin/probe \
                     $out/bin/probe \
-                    --set __PROBE_LIB ${libprobe}/lib
+                    --set __PROBE_LIB ${libprobe}/lib \
+                    --prefix PATH : ${probe-py}/bin
                 '';
               };
             probe-py-manual = python.pkgs.buildPythonPackage rec {
@@ -96,6 +97,7 @@
               ];
               pythonImportsCheck = [pname];
             };
+            probe-py = python.withPackages (pypkgs: [probe-py-manual]);
             default = probe-bundled;
           }
           // frontend.packages;

probe_src/frontend/cli/src/main.rs

@@ -24,6 +24,7 @@ fn main() -> Result<()> {
     let matches = command!()
         .about("Generate or manipulate Provenance for Replay OBservation Engine (PROBE) logs.")
         .propagate_version(true)
+        .allow_external_subcommands(true)
         .subcommands([
             Command::new("record")
                 .args([
@@ -142,7 +143,28 @@ fn main() -> Result<()> {
 
             Err(e).wrap_err("Shim failed to exec")
         }
+        Some((subcommand, args)) => {
+            let args = args
+                .get_many::<OsString>("")
+                .unwrap()
+                .cloned()
+                .collect::<Vec<_>>();
+
+            let exit = std::process::Command::new("python3")
+                .arg("-m")
+                .arg("probe_py.manual.cli")
+                .arg(subcommand)
+                .args(&args)
+                .spawn()
+                .wrap_err("Unknown subcommand")?
+                .wait()
+                .wrap_err("Wait on subcommand failed")?;
+
+            match exit.success() {
+                true => Ok(()),
+                false => Err(eyre!("Subcommand exited with code: {}", exit)),
+            }
+        }
         None => Err(eyre!("Subcommand expected, try --help for more info")),
-        _ => Err(eyre!("Unknown subcommand")),
     }
 }