prikk 0.28.0

//! RFC 121 §2.1: a closed stdout is not an error.
//!
//! `std::println!`/`std::print!` panic when the write returns `EPIPE` (`ErrorKind::BrokenPipe`),
//! which is Rust's default `SIGPIPE` disposition (`SIG_IGN`) turning what would otherwise silently
//! kill the process into an `io::Error` instead. This is the ordinary shape of `| head`, quitting
//! `less` early, or `| grep -q .` -- the consumer asked for less output than there was and got it,
//! which is success (`0`), not a panic and a backtrace hint.
//!
//! Every `print!`/`println!` call site in this crate should route through the macros here instead
//! of the standard library's own, via `use crate::stdout::{print, println};` at the top of the
//! file: importing an item named `println`/`print` into scope shadows the language prelude's macros
//! of the same name for that file, so no call site itself needs to change -- only the import at the
//! top, which is what keeps this a small diff over ~400 call sites rather than a mechanical rewrite
//! of every one of them.

use std::io::{self, ErrorKind, Write as _};

macro_rules! println {
    () => {
        $crate::stdout::write_line(::std::string::String::new())
    };
    ($($arg:tt)*) => {
        $crate::stdout::write_line(::std::format!($($arg)*))
    };
}

macro_rules! print {
    () => {
        $crate::stdout::write_str(::std::string::String::new())
    };
    ($($arg:tt)*) => {
        $crate::stdout::write_str(::std::format!($($arg)*))
    };
}

pub(crate) use print;
pub(crate) use println;

/// What a failed stdout write means: silence (the reader is gone and does not care), or a real
/// failure that must not be swallowed alongside it. Kept separate from acting on the outcome so the
/// classification itself -- the part that decides whether this fix is narrow -- is unit-testable
/// without touching a real file descriptor or `std::process::exit`.
enum WriteOutcome {
    Ok,
    ClosedPipe,
    Failed(io::Error),
}

fn classify(result: io::Result<()>) -> WriteOutcome {
    match result {
        Ok(()) => WriteOutcome::Ok,
        Err(err) if err.kind() == ErrorKind::BrokenPipe => WriteOutcome::ClosedPipe,
        Err(err) => WriteOutcome::Failed(err),
    }
}

/// Writes one line to stdout. See the module doc for the `BrokenPipe` behaviour.
pub(crate) fn write_line(line: String) {
    write_and_handle(|stdout| writeln!(stdout, "{line}"));
}

/// As [`write_line`], without the trailing newline.
pub(crate) fn write_str(text: String) {
    write_and_handle(|stdout| write!(stdout, "{text}"));
}

fn write_and_handle(write: impl FnOnce(&mut io::StdoutLock<'_>) -> io::Result<()>) {
    let stdout = io::stdout();
    let mut lock = stdout.lock();
    match classify(write(&mut lock)) {
        WriteOutcome::Ok => {}
        // A closed reader is not this process's failure -- exit exactly as if the requested output
        // had been produced in full and the process ended normally.
        WriteOutcome::ClosedPipe => std::process::exit(0),
        // Anything else (e.g. `ENOSPC` on a redirected stdout) is a genuine failure and must not be
        // swallowed by the same arm that swallows `BrokenPipe`. RFC 121 §6a's exit-code contract
        // rules 0/1/2 as the whole vocabulary -- a panic's exit 101 fell outside it, so this now
        // reports and exits 1 like every other operational failure, instead of a panic banner and a
        // backtrace hint. `eprintln!` (not the shadowed `println!` above) matches `main()`'s own
        // `"error: {msg}"` wording for every other `CliError::Failure`, and is never itself
        // BrokenPipe-fed here: a write failure on stdout says nothing about whether stderr is still
        // open. Deliberately `std::process::exit`, not a threaded `Result` -- this runs from macro
        // expansions at ~400 call sites, exactly why `ClosedPipe` already exits in place instead of
        // propagating.
        WriteOutcome::Failed(err) => {
            eprintln!("error: failed printing to stdout: {err}");
            std::process::exit(1);
        }
    }
}

#[cfg(test)]
#[path = "stdout/tests.rs"]
mod tests;