workmux 0.1.184

An opinionated workflow tool that orchestrates git worktrees and tmux
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

//! Execute a command on the host via RPC.
//!
//! Used by guest shims to transparently proxy toolchain commands
//! (just, cargo, npm, etc.) to the host machine.

use anyhow::{Result, bail};
use std::io::Write;

use crate::sandbox::rpc::{RpcClient, RpcRequest, RpcResponse};

/// Run a command on the host and stream output to local stdout/stderr.
/// Returns the remote process exit code.
pub fn run(command: &str, args: &[String]) -> Result<i32> {
    if !crate::sandbox::guest::is_sandbox_guest() {
        bail!("host-exec only works inside a sandbox guest (WM_SANDBOX_GUEST=1)");
    }

    let mut client = RpcClient::from_env()?;

    // Send exec request
    let request = RpcRequest::Exec {
        command: command.to_string(),
        args: args.to_vec(),
    };
    client.send(&request)?;

    // Stream responses until ExecExit
    let mut stdout = std::io::stdout().lock();
    let mut stderr = std::io::stderr().lock();

    loop {
        let response = client.recv()?;
        match response {
            RpcResponse::ExecOutput { data } => {
                stdout.write_all(data.as_bytes())?;
                stdout.flush()?;
            }
            RpcResponse::ExecError { data } => {
                stderr.write_all(data.as_bytes())?;
                stderr.flush()?;
            }
            RpcResponse::ExecExit { code } => {
                return Ok(code);
            }
            RpcResponse::Error { message } => {
                bail!("Host exec failed: {}", message);
            }
            _ => {
                // Ignore unexpected responses
            }
        }
    }
}