nix_builder/src/main.rs

use std::process::ExitCode;

use clap::Parser;

use self::cli::parameters::Cli;
use self::cli::parameters::Commands;
use self::command::build::run_build;
use self::command::daemon::start_daemon;
use self::command::feed_logs::feed_logs;
use self::error::CustomError;
use self::init_tracing::init_telemetry;
use self::init_tracing::shutdown_telemetry;
mod cli;
mod command;
mod config;
mod database;
mod error;
mod fs_util;
mod git_util;
mod init_tracing;
mod nix_util;

pub(crate) type Result<T> = std::result::Result<T, CustomError>;

fn main() -> Result<ExitCode> {
    let rt = tokio::runtime::Runtime::new()?;
    rt.block_on(async { main_body().await })
}

async fn main_body() -> Result<ExitCode> {
    init_telemetry().expect("Telemetry should initialize successfully.");
    let args = Cli::parse();
    match args.command {
        Commands::Build(args) => {
            run_build(args).await?;
        }
        Commands::Daemon(args) => {
            start_daemon(args).await?;
        }
        Commands::FeedLog(args) => {
            feed_logs(args).await?;
        }
    };
    shutdown_telemetry().expect("Telemetry should shutdown successfully.");
    Ok(ExitCode::SUCCESS)
}