Sienna Meridian Satterwhite
20f32531b7
Nextest cover profile for cargo llvm-cov integration. Update existing test imports from load_workflow_from_str to load_single_workflow_from_str for backward compatibility.
51 lines
1.4 KiB
Rust
51 lines
1.4 KiB
Rust
use std::path::PathBuf;
|
|
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let api_dir = PathBuf::from("vendor/containerd/api");
|
|
|
|
// Collect all .proto files, excluding internal runtime shim protos
|
|
let proto_files: Vec<PathBuf> = walkdir(&api_dir)?
|
|
.into_iter()
|
|
.filter(|p| {
|
|
let s = p.to_string_lossy();
|
|
!s.contains("/runtime/task/") && !s.contains("/runtime/sandbox/")
|
|
})
|
|
.collect();
|
|
|
|
println!(
|
|
"cargo:warning=Compiling {} containerd proto files",
|
|
proto_files.len()
|
|
);
|
|
|
|
let _out_dir = PathBuf::from(std::env::var("OUT_DIR")?);
|
|
|
|
// Use tonic-prost-build (the tonic 0.14 way)
|
|
let mut prost_config = prost_build::Config::new();
|
|
prost_config.include_file("mod.rs");
|
|
|
|
tonic_prost_build::configure()
|
|
.build_server(false)
|
|
.compile_with_config(
|
|
prost_config,
|
|
&proto_files,
|
|
&[api_dir, PathBuf::from("proto")],
|
|
)?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Recursively collect all .proto files under a directory.
|
|
fn walkdir(dir: &PathBuf) -> Result<Vec<PathBuf>, Box<dyn std::error::Error>> {
|
|
let mut protos = Vec::new();
|
|
for entry in std::fs::read_dir(dir)? {
|
|
let entry = entry?;
|
|
let path = entry.path();
|
|
if path.is_dir() {
|
|
protos.extend(walkdir(&path)?);
|
|
} else if path.extension().is_some_and(|ext| ext == "proto") {
|
|
protos.push(path);
|
|
}
|
|
}
|
|
Ok(protos)
|
|
}
|