abl-exorcist-assembler 0.0.1

use std::{
    fs,
    io::{self, BufWriter, Write},
    path::Path,
    process::ExitCode,
};

use clap::Parser;

mod cli;

use cli::Cli;

fn main() -> ExitCode {
    let cli = Cli::parse();
    let result = if cli.ramdisk {
        run_ramdisk(&cli.kernel, &cli.input)
    } else {
        run(&cli.kernel, &cli.input)
    };
    match result {
        Ok(()) => ExitCode::SUCCESS,
        Err(err) => {
            eprintln!("error: {err}");
            ExitCode::FAILURE
        }
    }
}

fn run(kernel: &Path, shim: &Path) -> Result<(), String> {
    let kernel = fs::read(kernel).map_err(|err| format!("read {}: {err}", kernel.display()))?;
    let shim = fs::read(shim).map_err(|err| format!("read {}: {err}", shim.display()))?;
    let kernel =
        abl_exorcist_assembler::canonicalize_kernel(&kernel).map_err(|err| err.to_string())?;
    let assembled =
        abl_exorcist_assembler::assemble(&kernel, &shim).map_err(|err| err.to_string())?;

    let stdout = io::stdout();
    let mut out = BufWriter::new(stdout.lock());
    out.write_all(&assembled)
        .map_err(|err| format!("write assembled image to stdout: {err}"))?;
    out.flush().map_err(|err| format!("flush stdout: {err}"))?;
    Ok(())
}

fn run_ramdisk(kernel: &Path, initrd: &Path) -> Result<(), String> {
    let kernel = fs::read(kernel).map_err(|err| format!("read {}: {err}", kernel.display()))?;
    let initrd = fs::read(initrd).map_err(|err| format!("read {}: {err}", initrd.display()))?;
    let kernel =
        abl_exorcist_assembler::canonicalize_kernel(&kernel).map_err(|err| err.to_string())?;
    let assembled = abl_exorcist_assembler::assemble_ramdisk(&kernel, &initrd)
        .map_err(|err| err.to_string())?;

    let stdout = io::stdout();
    let mut out = BufWriter::new(stdout.lock());
    out.write_all(&assembled)
        .map_err(|err| format!("write ramdisk container to stdout: {err}"))?;
    out.flush().map_err(|err| format!("flush stdout: {err}"))?;
    Ok(())
}