cli.rs

use std::{env::args_os, ffi::OsStr, path::PathBuf, pin::Pin};

use fspy::AccessMode;
use tokio::{
    fs::File,
    io::{AsyncWrite, stdout},
};

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let mut args = args_os();
    let _ = args.next();
    assert_eq!(args.next().as_deref(), Some(OsStr::new("-o")));

    let out_path = args.next().unwrap();

    let program = PathBuf::from(args.next().unwrap());

    let mut command = fspy::Command::new(program);
    command.envs(std::env::vars_os()).args(args);

    let child = command.spawn().await?;
    let termination = child.wait_handle.await?;

    let mut path_count = 0usize;
    let out_file: Pin<Box<dyn AsyncWrite>> =
        if out_path == "-" { Box::pin(stdout()) } else { Box::pin(File::create(out_path).await?) };

    let mut csv_writer = csv_async::AsyncWriter::from_writer(out_file);

    for acc in termination.path_accesses.iter() {
        path_count += 1;
        csv_writer
            .write_record(&[
                acc.path.to_cow_os_str().to_string_lossy().as_ref().as_bytes(),
                match acc.mode {
                    AccessMode::Read => b"read".as_slice(),
                    AccessMode::ReadWrite => b"readwrite",
                    AccessMode::Write => b"write",
                    AccessMode::ReadDir => b"readdir",
                },
            ])
            .await?;
    }
    csv_writer.flush().await?;

    eprintln!("\nfspy: {path_count} paths accessed. status: {}", termination.status);
    Ok(())
}