use cli_table::{print_stdout, Cell, Style, Table};
use furiosa_device::blocking::{get_status_all, list_devices};
use furiosa_device::{Device, DeviceError};
use itertools::join;

fn main() -> Result<(), DeviceError> {
    tracing_subscriber::fmt::init();

    let mut found: Vec<Device> = Vec::new();
    for device in list_devices()? {
        found.push(device);
    }

    for device in found.iter() {
        println!("{:?}", device);
    }

    let mut rows = vec![];

    for device in found.iter() {
        let status = get_status_all(device)?;
        let mut status: Vec<(u8, _)> = status.into_iter().collect();
        status.sort_by(|a, b| a.0.cmp(&b.0));
        let msg = join(
            status
                .iter()
                .map(|(k, v)| k.to_string() + " (" + &v.to_string() + ")"),
            ", ",
        );
        rows.push(vec![device.to_string().cell(), msg.cell()]);
    }
    let table = rows
        .table()
        .title(vec!["NPU".cell().bold(true), "Cores".cell().bold(true)]);
    print_stdout(table)?;

    Ok(())
}