1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
use std::env;
use std::fs;
use std::io::prelude::*;
use std::io::BufReader;
use std::iter::FromIterator;
use std::path::Path;
use std::string::String;
fn main() -> Result<(), Box<dyn std::error::Error>> {
let out_dir = env::var_os("OUT_DIR").unwrap();
let field_def_file = "src/audit-specs/fields/field-dictionary.csv";
let msgtype_def_file = "src/audit-specs/messages/message-dictionary.csv";
let const_file = Path::new(&out_dir).join("const.rs");
let msgtype_file = Path::new(&out_dir).join("message_type_impl.rs");
let constants: Vec<(String, String)> = BufReader::new(fs::File::open(msgtype_def_file)?)
.lines()
.skip(1) // skip over header
.map(|line| {
line.unwrap()
.split(',')
.map(|x| x.to_string())
.collect::<Vec<_>>()
})
.map(|fields| {
(
fields[0].strip_prefix("AUDIT_").unwrap().to_string(),
fields[1].clone(),
)
})
.collect();
let fields: Vec<(String, String)> = BufReader::new(fs::File::open(field_def_file)?)
.lines()
.skip(3) // skip over heder and regex describing a* mess
.map(|line| {
line.unwrap()
.split(',')
.map(|x| x.to_string())
.collect::<Vec<_>>()
})
.map(|fields| (fields[0].clone(), fields[1].clone()))
.collect();
let mut template = Vec::new();
fs::File::open("src/const.rs.in")?.read_to_end(&mut template)?;
let template = String::from_utf8(template)?;
let buf = template
.replace(
"/* @EVENT_CONST@ */",
&String::from_iter(
constants
.iter()
.map(|(name, value)| format!(r#"("{name}", {value}), "#)),
),
)
.replace(
"/* @FIELD_TYPES@ */",
&String::from_iter(
fields
.iter()
.filter(|(_, typ)| typ == "encoded" || typ.starts_with("numeric"))
.map(|(name, typ)| match typ.as_str() {
"numeric hexadecimal" => format!(r#"("{name}", FieldType::NumericHex),"#),
"numeric decimal" => format!(r#"("{name}", FieldType::NumericDec),"#),
"numeric octal" => format!(r#"("{name}", FieldType::NumericOct),"#),
"numeric" => format!(r#"("{name}", FieldType::Numeric),"#),
"encoded" => format!(r#"("{name}", FieldType::Encoded),"#),
_ => format!(r#"("{name}", FieldType::Invalid),"#),
}),
),
)
.into_bytes();
fs::write(const_file, buf)?;
let mut template = Vec::new();
fs::File::open("src/message_type_impl.rs.in")?.read_to_end(&mut template)?;
let template = String::from_utf8(template)?;
let buf = template.replace(
"/* @MSG_TYPE_CONST@ */",
&String::from_iter(
constants
.iter()
.map(|(name, value)| format!("pub const {name}: Self = Self({value});\n",)),
),
);
fs::write(msgtype_file, buf)?;
println!("cargo:rerun-if-changed=build.rs");
println!("cargo:rerun-if-changed=const.rs.in");
println!("cargo:rerun-if-changed={msgtype_def_file}");
println!("cargo:rerun-if-changed={field_def_file}");
Ok(())
}
|
