fn main()

in build.rs [20:56]

• 28 lines of code
• 2 McCabe index (conditional complexity)

fn main() -> Result<(), String> {
    use std::io::Write;

    let out = std::path::PathBuf::from(std::env::var("OUT_DIR").unwrap());

    // for use in docker build where file changes can be wonky
    println!("cargo:rerun-if-env-changed=FORCE_REBUILD");

    let version = rustc_version::version().unwrap();
    println!("cargo:rustc-env=RUSTC_VERSION={version}");

    let path = "src/proto/generated/protobuf.rs";

    // We don't include the proto files in releases so that downstreams
    // do not need to have PROTOC included
    if Path::new("src/proto/datafusion_ray.proto").exists() {
        println!("cargo:rerun-if-changed=src/proto/datafusion_common.proto");
        println!("cargo:rerun-if-changed=src/proto/datafusion.proto");
        println!("cargo:rerun-if-changed=src/proto/datafusion_ray.proto");
        tonic_build::configure()
            .extern_path(".datafusion", "::datafusion_proto::protobuf")
            .extern_path(".datafusion_common", "::datafusion_proto::protobuf")
            .compile(&["src/proto/datafusion_ray.proto"], &["src/proto"])
            .map_err(|e| format!("protobuf compilation failed: {e}"))?;
        let generated_source_path = out.join("datafusion_ray.protobuf.rs");
        let code = std::fs::read_to_string(generated_source_path).unwrap();
        let mut file = std::fs::OpenOptions::new()
            .write(true)
            .truncate(true)
            .create(true)
            .open(path)
            .unwrap();
        file.write_all(code.as_str().as_ref()).unwrap();
    }

    Ok(())
}