use ffiber::CDylibCompiler;
use ffiber::types::{ArgType, SelfArgType};
use color_eyre::eyre::Result;

fn main() -> Result<()> {
    let mut compiler = CDylibCompiler::new("mlx5-datapath", ".");
    compiler.add_dependency("bumpalo", vec![
        ("git", "\"https://github.com/deeptir18/bumpalo\""),
        ("features", "[\"collections\"]"),
    ])?;
    compiler.add_dependency("mlx5-datapath",
        vec![("path", "\"/users/ygina/cornflakes/mlx5-datapath\"")])?;
    compiler.add_dependency("cornflakes-libos",
        vec![("path", "\"/users/ygina/cornflakes/cornflakes-libos\"")])?;
    compiler.import("bumpalo::Bump")?;
    compiler.import(
        "mlx5_datapath::datapath::connection::Mlx5Connection")?;
    compiler.import("cornflakes_libos::datapath::{{Datapath, \
        ReceivedPkt}}")?;

    // Bump
    compiler.add_extern_c_function(
        ArgType::new_struct("Bump"),
        SelfArgType::Mut,
        "reset",
        vec![],
        None,
        false,
    )?;

    // ReceivedPkt
    let struct_ty = ArgType::Struct {
        name: "ReceivedPkt".to_string(),
        params: vec![Box::new(ArgType::new_struct("Mlx5Connection"))],
    };
    compiler.add_extern_c_function(
        struct_ty.clone(),
        SelfArgType::Value,
        "msg_id",
        vec![],
        Some(ArgType::Primitive("u32".to_string())),
        false,
    )?;
    compiler.add_extern_c_function(
        struct_ty.clone(),
        SelfArgType::Value,
        "conn_id",
        vec![],
        Some(ArgType::Primitive("usize".to_string())),
        false,
    )?;

    // Mlx5Connection
    compiler.add_extern_c_function(
        ArgType::new_struct("Mlx5Connection"),
        SelfArgType::Mut,
        "set_copying_threshold",
        vec![
            ("copying_threshold", ArgType::Primitive("usize".to_string())),
        ],
        None,
        false,
    )?;
    compiler.add_extern_c_function(
        ArgType::new_struct("Mlx5Connection"),
        SelfArgType::Mut,
        "add_memory_pool",
        vec![
            ("buf_size", ArgType::Primitive("usize".to_string())),
            ("min_elts", ArgType::Primitive("usize".to_string())),
        ],
        None,
        true,
    )?;
    compiler.add_extern_c_function(
        ArgType::new_struct("Mlx5Connection"),
        SelfArgType::Mut,
        "add_tx_mempool",
        vec![
            ("size", ArgType::Primitive("usize".to_string())),
            ("min_elts", ArgType::Primitive("usize".to_string())),
        ],
        None,
        true,
    )?;

    compiler.flush()?;
    Ok(())
}