babysit 0.7.0

mod attach;
mod cli;
mod control;
mod pane;
mod paths;
mod render;
mod run;
mod session;
mod sub;
mod upgrade;

use anyhow::Result;
use clap::Parser;

#[tokio::main]
async fn main() -> Result<()> {
    // Rust sets SIGPIPE to SIG_IGN at startup, which turns a closed pipe into
    // a panic on the next write (`babysit log | head`) and is also inherited
    // by the wrapped command. Restore the default so we exit quietly on a
    // broken pipe and children get normal SIGPIPE behavior.
    #[cfg(unix)]
    unsafe {
        use nix::sys::signal::{SigHandler, Signal, signal};
        let _ = signal(Signal::SIGPIPE, SigHandler::SigDfl);
    }

    // Short wrap forms: `babysit [-d] -- <cmd> [args…]`. Handled before clap
    // so that `babysit listt` (a typo of `list`) goes through clap and gets a
    // proper "did you mean 'list'?" error instead of silently being treated
    // as a wrap of the non-existent command `listt`.
    let raw: Vec<String> = std::env::args().collect();
    let short = match raw.get(1).map(String::as_str) {
        Some("--") => Some((false, 2)),
        Some("-d") | Some("--detach") if raw.get(2).map(String::as_str) == Some("--") => {
            Some((true, 3))
        }
        _ => None,
    };
    if let Some((detach, start)) = short {
        let cmd: Vec<String> = raw[start..].to_vec();
        if cmd.is_empty() {
            eprintln!("babysit: empty command after `--`");
            std::process::exit(2);
        }
        let code = run::run(cmd, None, detach, None, false, None, None, None, false).await?;
        std::process::exit(code);
    }

    let cli = cli::Cli::parse();

    match cli.command {
        cli::Command::Run {
            id,
            detach,
            detached_id,
            no_tty,
            timeout,
            idle_timeout,
            size,
            json,
            cmd,
        } => {
            let code = run::run(
                cmd,
                id,
                detach,
                detached_id,
                no_tty,
                timeout,
                idle_timeout,
                size,
                json,
            )
            .await?;
            std::process::exit(code);
        }
        cli::Command::List { json } => sub::list(json).await,
        cli::Command::Status { sel, json } => sub::status(sel.session, json).await,
        cli::Command::Log {
            sel,
            tail,
            grep,
            raw,
            since,
            follow,
            json,
        } => sub::log(sel.session, tail, grep, raw, since, follow, json).await,
        cli::Command::Screenshot { sel, format, trim } => {
            sub::screenshot(sel.session, format, trim).await
        }
        cli::Command::Restart { sel, json } => sub::restart(sel.session, json).await,
        cli::Command::Kill { sel, json } => sub::kill(sel.session, json).await,
        cli::Command::Send {
            sel,
            text,
            no_newline,
            json,
        } => sub::send(sel.session, text, !no_newline, json).await,
        cli::Command::Key { sel, keys, json } => sub::key(sel.session, keys, json).await,
        cli::Command::Expect {
            sel,
            pattern,
            timeout,
            since,
            from_now,
            raw,
            json,
        } => {
            let code =
                sub::expect(sel.session, pattern, timeout, since, from_now, raw, json).await?;
            std::process::exit(code);
        }
        cli::Command::WaitIdle {
            sel,
            settle,
            timeout,
        } => {
            let code = sub::wait_idle(sel.session, settle, timeout).await?;
            std::process::exit(code);
        }
        cli::Command::Resize { sel, size, json } => sub::resize(sel.session, size, json).await,
        cli::Command::Flag { sel, message, json } => sub::flag(sel.session, message, json).await,
        cli::Command::Unflag { sel, json } => sub::unflag(sel.session, json).await,
        cli::Command::Wait { sel, timeout } => {
            let code = sub::wait(sel.session, timeout).await?;
            std::process::exit(code);
        }
        cli::Command::Attach { sel } => {
            let code = attach::attach(sel.session).await?;
            std::process::exit(code);
        }
        cli::Command::Detach { sel, json } => attach::detach(sel.session, json).await,
        cli::Command::Prune { dry_run, json } => sub::prune(dry_run, json).await,
        cli::Command::Upgrade => {
            let code = upgrade::run()?;
            std::process::exit(code);
        }
        cli::Command::Config { shell } => {
            match shell {
                cli::Shell::Zsh => print!("{}", include_str!("completions/babysit.zsh")),
                cli::Shell::Bash => print!("{}", include_str!("completions/babysit.bash")),
            }
            Ok(())
        }
    }
}