build.rs 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. extern crate protobuf_macros;
  2. extern crate rand;
  3. extern crate vergen;
  4. use rand::Rng;
  5. use std::env;
  6. use std::fs::OpenOptions;
  7. use std::io::Write;
  8. use std::path::PathBuf;
  9. fn main() {
  10. let out = PathBuf::from(env::var("OUT_DIR").unwrap());
  11. vergen::vergen(vergen::OutputFns::all()).unwrap();
  12. let build_id: String = rand::thread_rng().gen_ascii_chars().take(8).collect();
  13. let mut version_file = OpenOptions::new()
  14. .write(true)
  15. .append(true)
  16. .open(&out.join("version.rs"))
  17. .unwrap();
  18. let build_id_fn = format!(
  19. "
  20. /// Generate a random build id.
  21. pub fn build_id() -> &'static str {{
  22. \"{}\"
  23. }}
  24. ",
  25. build_id
  26. );
  27. if let Err(e) = version_file.write_all(build_id_fn.as_bytes()) {
  28. println!("{}", e);
  29. }
  30. protobuf_macros::expand("src/lib.in.rs", &out.join("lib.rs")).unwrap();
  31. println!("cargo:rerun-if-changed=src/lib.in.rs");
  32. println!("cargo:rerun-if-changed=src/connection/mod.rs");
  33. println!("cargo:rerun-if-changed=src/connection/codec.rs");
  34. println!("cargo:rerun-if-changed=src/connection/handshake.rs");
  35. }