add entrypoint/init-system functionality

Author: Panaetius

Cargo.lock

@@ -74,10 +74,11 @@ tui-realm-treeview = "3.0.0"
 aws-sdk-s3 = "1.115.0"
 toml_edit = "0.23.9"
 clap_complete = "4.5.61"
+pid1 = "0.1.5"
+rust_supervisor = "0.2.0"
 
 [build-dependencies]
 anyhow = "1.0.90"
-
 vergen-gix = { version = "1.0.2", features = ["build", "cargo"] }
 
 [dev-dependencies]

coman/Cargo.toml

@@ -1,8 +1,10 @@
-use std::{error::Error, path::PathBuf};
+use std::{error::Error, path::PathBuf, process::Stdio, thread, time::Duration};
 
 use clap::{Args, Command, Parser, Subcommand, ValueHint, builder::TypedValueParser};
 use clap_complete::{Generator, Shell, generate};
 use color_eyre::Result;
+use pid1::Pid1Settings;
+use rust_supervisor::{ChildType, Supervisor, SupervisorConfig};
 use strum::VariantNames;
 
 use crate::{
@@ -63,6 +65,11 @@ pub enum CliCommands {
         #[clap(value_enum)]
         generator: Shell,
     },
+    #[clap(about = "Execute a process/command through coman, with additional monitoring and side processes")]
+    Exec {
+        #[clap(trailing_var_arg = true, help = "The command to run", value_hint=ValueHint::Other)]
+        command: Vec<String>,
+    },
 }
 
 #[derive(Subcommand, Debug)]
@@ -372,3 +379,41 @@ fn is_bare_string(value_str: &str) -> bool {
 pub fn print_completions<G: Generator>(generator: G, cmd: &mut Command) {
     generate(generator, cmd, cmd.get_name().to_string(), &mut std::io::stdout());
 }
+pub(crate) async fn cli_exec_command(command: Vec<String>) -> Result<()> {
+    // Pid1 takes care of proper terminating of processes and signal handling when running in a container
+    Pid1Settings::new()
+        .enable_log(true)
+        .timeout(Duration::from_secs(2))
+        .launch()
+        .expect("Launch failed");
+
+    let mut supervisor = Supervisor::new(SupervisorConfig::default());
+    supervisor.add_process("iroh-ssh", ChildType::Permanent, || {
+        thread::spawn(|| {
+            // iroh!
+        })
+    });
+    supervisor.add_process("main-process", ChildType::Temporary, move || {
+        let command = command.clone();
+        thread::spawn(move || {
+            let mut child = std::process::Command::new(command[0].clone())
+                .args(&command[1..])
+                .spawn()
+                .expect("Failed to start compute job");
+            child.wait().expect("Failed to wait on compute job");
+        })
+    });
+
+    let supervisor = supervisor.start_monitoring();
+    loop {
+        thread::sleep(Duration::from_secs(1));
+
+        if let Some(state) = supervisor.get_process_state("main-process") {
+            match state {
+                rust_supervisor::ProcessState::Failed | rust_supervisor::ProcessState::Stopped => break,
+                _ => {}
+            }
+        }
+    }
+    Ok(())
+}

coman/src/cli.rs

@@ -17,7 +17,7 @@ use crate::{
         model::Model,
         user_events::{CscsEvent, FileEvent, StatusEvent, UserEvent},
     },
-    cli::{Cli, get_config, print_completions, set_config, version},
+    cli::{Cli, cli_exec_command, get_config, print_completions, set_config, version},
     components::{
         file_tree::FileTree, global_listener::GlobalListener, status_bar::StatusBar, toolbar::Toolbar,
         workload_list::WorkloadList,
@@ -135,6 +135,7 @@ async fn main() -> Result<()> {
                 },
             },
             cli::CliCommands::Init { destination, name } => Config::create_project_config(destination, name)?,
+            cli::CliCommands::Exec { command } => cli_exec_command(command).await?,
         },
         None => run_tui(args.tick_rate)?,
     }