Files
vnts/build.rs
2024-05-14 13:03:19 +08:00

28 lines
855 B
Rust

use rand::Rng;
use static_files::resource_dir;
use std::fs::File;
use std::io::Write;
fn main() {
std::fs::create_dir_all("src/proto").unwrap();
protobuf_codegen::Codegen::new()
.pure()
.out_dir("src/proto")
.inputs(["proto/message.proto"])
.include("proto")
.run()
.expect("Codegen failed.");
resource_dir("./static").build().unwrap();
let now_time = chrono::Local::now();
let serial_number = format!(
"{}-{}",
&now_time.format("%y%m%d%H%M").to_string(),
rand::thread_rng().gen_range(100..1000)
);
let generated_code = format!(r#"pub const SERIAL_NUMBER: &str = "{}";"#, serial_number);
let dest_path = "src/generated_serial_number.rs";
let mut file = File::create(dest_path).unwrap();
file.write_all(generated_code.as_bytes()).unwrap();
}