use std::io::Write;
use std::{env, path::PathBuf};
fn main() -> Result<(), anyhow::Error> {
let root = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("protos");
let proto_files = ["mehari/txs.proto", "mehari/server.proto"]
.iter()
.map(|f| root.join(f))
.collect::<Vec<_>>();
for proto_file in &proto_files {
println!("cargo:rerun-if-changed={}", proto_file.display());
}
let descriptor_path: PathBuf =
PathBuf::from(env::var("OUT_DIR").unwrap()).join("proto_descriptor.bin");
prost_build::Config::new()
.file_descriptor_set_path(&descriptor_path)
.compile_well_known_types()
.extern_path(".google.protobuf", "::pbjson_types")
.compile_protos(&proto_files, &[root])?;
let descriptor_set = std::fs::read(descriptor_path).unwrap();
pbjson_build::Builder::new()
.register_descriptors(&descriptor_set)?
.build(&[".mehari"])?;
let src = env::var("CARGO_MANIFEST_DIR").unwrap();
let dst = PathBuf::from(env::var("OUT_DIR").unwrap()).join("built.rs");
let manifest_path = std::path::Path::new(&src);
if let Err(e) = built::write_built_file_with_opts(Some(manifest_path), &dst) {
println!(
"cargo:warning=Failed to write built file with manifest: {}",
e
);
println!("cargo:warning=Falling back to building without manifest (likely Python sdist)");
built::write_built_file_with_opts(None, &dst)
.map_err(|e| anyhow::anyhow!("Failed to write built file: {}", e))?;
let mut file = std::fs::OpenOptions::new()
.append(true)
.open(&dst)
.expect("Failed to open built.rs for appending");
writeln!(
file,
"pub const DEPENDENCIES: [(&str, &str); 1] = [(\"hgvs\", \"unknown-sdist\")];"
)
.expect("Failed to append dummy dependencies");
}
Ok(())
}