build.rs 534 B

1234567891011121314151617
  1. use std::io::Result;
  2. fn main() -> Result<()> {
  3. const PROTOC_ENVAR: &str = "PROTOC";
  4. if std::env::var(PROTOC_ENVAR).is_err() {
  5. #[cfg(not(windows))]
  6. std::env::set_var(PROTOC_ENVAR, protobuf_src::protoc());
  7. }
  8. let proto_base_path = std::path::PathBuf::from("proto");
  9. let proto = proto_base_path.join("wen_restart.proto");
  10. println!("cargo:rerun-if-changed={}", proto.display());
  11. // Generate rust files from protos.
  12. prost_build::compile_protos(&[proto], &[proto_base_path])?;
  13. Ok(())
  14. }