ratto 0.8.0

mod cli;
mod color;
mod commands;
mod core;
mod exit;
mod style_spec;
mod term;
mod theme;

mod ui;

use clap::Parser;

use crate::cli::{Cli, Command};
use crate::color::ColorProfile;
use crate::exit::{AppError, OK};
use crate::term::tty::UiStream;
use crate::theme::Palette;

fn main() {
    // Die quietly on a closed pipe (`rat ... | head`) like other unix
    // tools, instead of panicking on EPIPE.
    #[cfg(unix)]
    unsafe {
        libc::signal(libc::SIGPIPE, libc::SIG_DFL);
    }
    // Windows has no SIGPIPE: a closed pipe surfaces as a println! panic.
    // Exit quietly instead of spraying a backtrace. (Kept on unix too as a
    // belt for writes that race the signal disposition.)
    let default_hook = std::panic::take_hook();
    std::panic::set_hook(Box::new(move |info| {
        let payload = info
            .payload()
            .downcast_ref::<String>()
            .map(String::as_str)
            .or_else(|| info.payload().downcast_ref::<&str>().copied())
            .unwrap_or("");
        // Windows closed-pipe errors: 109 ERROR_BROKEN_PIPE ("the pipe has
        // been ended"), 232 ERROR_NO_DATA ("the pipe is being closed").
        if payload.contains("Broken pipe")
            || payload.contains("os error 109")
            || payload.contains("os error 232")
        {
            std::process::exit(0);
        }
        default_hook(info);
    }));
    std::process::exit(real_main());
}

fn real_main() -> i32 {
    let cli = Cli::parse();
    // Capability comes from the UI stream's ttyness, never stdout's, so
    // command substitution keeps color.
    let is_tty = UiStream::open().is_tty();
    let profile = color::resolve_profile(cli.color, &color::SystemEnv, is_tty);
    // Asked at most once per process, here, before any command runs and long
    // before anything claims raw mode. Commands never ask.
    let detected = if theme::may_detect(cli.appearance, profile) {
        term::appearance::probe(theme::PROBE_TIMEOUT)
            .map(|a| (a, theme::AppearanceSource::Osc))
            .or_else(|| {
                theme::appearance_from_colorfgbg(&color::SystemEnv)
                    .map(|a| (a, theme::AppearanceSource::ColorFgBg))
            })
    } else {
        None
    };
    let (appearance, source) = theme::resolve_appearance(cli.appearance, detected);
    let palette = theme::Palette::builtin(appearance, source);

    match dispatch(cli.command, profile, palette) {
        Ok(()) => OK,
        Err(err) => {
            match &err {
                AppError::Fail(inner) => eprintln!("rat: {inner:#}"),
                AppError::Timeout => eprintln!("timed out"),
                AppError::NoSelection | AppError::Aborted | AppError::Child(_) => {}
            }
            err.code()
        }
    }
}

fn dispatch(command: Command, profile: ColorProfile, palette: Palette) -> exit::AppResult {
    match command {
        Command::Style(args) => commands::style::run(args, profile, palette),
        Command::Bar(args) => commands::bar::run(args, profile, palette),
        Command::Table(args) => commands::table::run(args, profile, palette),
        Command::Join(args) => commands::join::run(args, profile, palette),
        Command::Duration(args) => commands::duration::run(args, profile, palette),
        Command::Date(args) => commands::date::run(args, profile, palette),
        Command::Spark(args) => commands::spark::run(args, profile, palette),
        Command::Log(args) => commands::log::run(args, profile, palette),
        Command::Frame(args) => commands::frame::run(args, profile, palette),
        Command::Watch(args) => commands::watch::run(args, profile, palette),
        Command::Dashboard(args) => commands::dashboard::run(args, profile, palette),
        Command::Doctor(args) => commands::doctor::run(args, profile, palette),
        Command::Choose(args) => commands::choose::run(args, profile, palette),
        Command::Confirm(args) => commands::confirm::run(args, profile, palette),
        Command::Input(args) => commands::input::run(args, profile, palette),
        Command::Filter(args) => commands::filter::run(args, profile, palette),
        Command::Spin(args) => commands::spin::run(args, profile, palette),
        Command::Completion(args) => commands::completion::run(args, profile, palette),
        #[cfg(debug_assertions)]
        Command::Env(args) => {
            match std::env::var(&args.name) {
                Ok(value) => println!("{value}"),
                Err(_) => println!("unset"),
            }
            Ok(())
        }
        #[cfg(debug_assertions)]
        Command::ExitCode(args) => {
            if let Some(msg) = &args.stderr_msg {
                eprintln!("{msg}");
            }
            match args.code {
                0 => Ok(()),
                1 => Err(AppError::NoSelection),
                124 => Err(AppError::Timeout),
                130 => Err(AppError::Aborted),
                n => Err(AppError::Child(n)),
            }
        }
        #[cfg(debug_assertions)]
        Command::Sleep(args) => {
            std::thread::sleep(std::time::Duration::from_millis(args.millis));
            if let Some(text) = &args.text {
                println!("{text}");
            }
            Ok(())
        }
        #[cfg(debug_assertions)]
        Command::Cat(args) => {
            use std::io::Write;

            use anyhow::Context as _;
            let bytes = std::fs::read(&args.file).context("reading")?;
            let mut out = std::io::stdout().lock();
            out.write_all(&bytes).context("writing")?;
            out.flush().context("flushing")?;
            Ok(())
        }
        #[cfg(debug_assertions)]
        Command::Lines(args) => {
            use std::io::Write;

            use anyhow::Context as _;
            let mut out = std::io::BufWriter::new(std::io::stdout().lock());
            for i in 0..args.count {
                writeln!(out, "{i}").context("writing")?;
            }
            out.flush().context("flushing")?;
            Ok(())
        }
    }
}