mirror of
https://github.com/lbl8603/vnts.git
synced 2025-12-24 12:47:51 +08:00
28 lines
855 B
Rust
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();
|
|
}
|