main.rs

#[macro_use]
extern crate log;

mod boot_manager;
mod cli;
mod config_types;
mod daemon;
mod daemon_id;
mod daemon_list;
mod daemon_status;
mod deps;
mod env;
mod error;
mod ipc;
mod logger;
mod pitchfork_toml;
mod procs;
mod proxy;
mod settings;
mod shell;
mod state_file;
mod supervisor;
mod tui;
mod ui;
mod watch_files;
mod web;

pub use miette::Result;
#[cfg(unix)]
use tokio::signal;
#[cfg(unix)]
use tokio::signal::unix::SignalKind;

#[tokio::main]
async fn main() -> Result<()> {
    // Install ring as the default rustls crypto provider.
    // Required because reqwest is built with `rustls-tls-*-no-provider`, which
    // avoids pulling in aws-lc-sys but requires the caller to install a provider.
    let _ = rustls::crypto::ring::default_provider().install_default();
    logger::init();
    // Re-apply log levels now that settings (env + config files) are loaded.
    // logger::init() only sees env vars; this picks up pitchfork.toml values.
    logger::apply_settings();
    #[cfg(unix)]
    handle_epipe();
    cli::run().await
}

#[cfg(unix)]
fn handle_epipe() {
    match signal::unix::signal(SignalKind::pipe()) {
        Ok(mut pipe_stream) => {
            tokio::spawn(async move {
                pipe_stream.recv().await;
                debug!("received SIGPIPE");
            });
        }
        Err(e) => {
            warn!("Could not set up SIGPIPE handler: {e}");
        }
    }
}