help-probe 0.1.0

CLI tool discovery and automation framework that extracts structured information from command help text
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use std::process::{Output, Stdio};
use std::time::Duration;

use anyhow::{Context, Result, anyhow};
use tokio::process::Command;
use tokio::time;

/// Outcome of running a command with timeout.
pub enum RunOutcome {
    /// Command completed (may have non-zero exit code).
    Completed(Output),
    /// Command timed out before completion.
    TimedOut,
}

/// Run a command with arguments, capturing stdout/stderr, with a timeout.
pub async fn run_with_timeout(
    program: &str,
    args: &[String],
    timeout: Duration,
) -> Result<RunOutcome> {
    let mut cmd = Command::new(program);
    cmd.args(args)
        .stdin(Stdio::null())
        .stdout(Stdio::piped())
        .stderr(Stdio::piped());

    let child = cmd
        .spawn()
        .with_context(|| format!("Failed to spawn command: {} {:?}", program, args))?;

    match time::timeout(timeout, child.wait_with_output()).await {
        Ok(Ok(output)) => Ok(RunOutcome::Completed(output)),
        Ok(Err(e)) => Err(anyhow!("Error while running command: {}", e)),
        Err(_) => Ok(RunOutcome::TimedOut),
    }
}