port-plumber 0.3.0

use std::path::Path;
use std::process::{Child, Command, Output, Stdio};

use anyhow::Result;

pub struct CmdRunner {
    command: Command,
    process: Option<Child>
}

impl CmdRunner {
    pub fn build(cmd_name: &str, args: &[String], dir: impl AsRef<Path>) -> Result<Self> {
        let mut command = Command::new(cmd_name);
        command.current_dir(dir);
        command.args(args);
        command.stdin(Stdio::piped());

        Ok(Self {
            command,
            process: None
        })
    }

    pub fn start(&mut self) -> Result<()> {
        log::debug!("Starting command {:?} with args {:?}", self.command.get_program(), self.command.get_args().collect::<Vec<_>>());
        let process = self.command.spawn()?;
        self.process = Some(process);
        Ok(())
    }

    pub fn run(&mut self) -> Result<Output> {
        log::debug!("Starting command {:?} with args {:?}", self.command.get_program(), self.command.get_args().collect::<Vec<_>>());
        let out = self.command.output()?;
        Ok(out)
    }

    pub fn stop(&mut self) -> Result<()> {
        let Some(ref mut process) = self.process else {
            return Ok(())
        };
        if let Some(_status) = process.try_wait()? {
            self.process = None;
            Ok(())
        } else {
            process.kill()?;
            let _status = process.wait()?;
            self.process = None;
            Ok(())
        }
    }

    pub fn is_running(&mut self) -> Result<bool> {
        let running = self.process
            .as_mut()
            .map(|p| p.try_wait().map(|out| out.is_none()))
            .transpose()?
            .unwrap_or(false);
        Ok(running)
    }
}