-
-
Notifications
You must be signed in to change notification settings - Fork 37
/
Copy pathbuild.rs
85 lines (76 loc) · 2.59 KB
/
build.rs
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
use fancy_regex::Regex as Fancy;
use regex::Regex;
use serde::Deserialize;
use std::fmt::Write;
use std::{env, fs, path::Path};
#[allow(dead_code)]
#[derive(Deserialize, Debug)]
struct Data {
#[serde(rename(deserialize = "Name"))]
name: String,
#[serde(rename(deserialize = "Regex"))]
regex: String,
#[serde(skip_deserializing)]
boundaryless: String,
plural_name: bool,
#[serde(rename(deserialize = "Description"))]
description: Option<&'static str>,
#[serde(rename(deserialize = "Exploit"))]
exploit: Option<String>,
#[serde(rename(deserialize = "Rarity"))]
rarity: f32,
#[serde(rename(deserialize = "URL"))]
url: Option<&'static str>,
#[serde(rename(deserialize = "Tags"))]
tags: Vec<&'static str>,
}
fn main() {
let mut data: Vec<Data> = serde_json::from_str(include_str!("./src/data/regex.json")).unwrap();
data.iter_mut().for_each(|d| {
d.boundaryless = Fancy::new(r"(?<!\\)\^(?![^\[\]]*(?<!\\)\])")
.expect("can't compile for boundaryless")
.replace(&d.regex, "")
.to_string();
d.boundaryless = Fancy::new(r"(?<!\\)\$(?![^\[\]]*(?<!\\)\])")
.expect("can't compile for boundaryless")
.replace(&d.boundaryless, "")
.to_string();
});
data.retain(|r| Regex::new(&r.regex).is_ok() && Regex::new(&r.boundaryless).is_ok());
let mut data_str = format!("{:?}", data);
// we want reference to [], i.e. &[]
data_str = data_str.replace("tags: [", "tags: &[");
let regex_str: String = data.iter().fold(String::new(), |mut output, d| {
let _ = write!(
output,
r#"Lazy::new(|| Regex::new({:?}).unwrap()),"#,
d.regex
);
output
});
let boundaryless_regex_str: String = data.iter().fold(String::new(), |mut output, d| {
let _ = write!(
output,
r#"Lazy::new(|| Regex::new({:?}).unwrap()),"#,
d.boundaryless
);
output
});
let count = data.len();
let final_str = format!(
r#"
const DATA: [Data; {count}] = {data_str};
"#
);
let out_dir = env::var_os("OUT_DIR").unwrap();
let dest_path = Path::new(&out_dir).join("data.rs");
fs::write(dest_path, final_str).unwrap();
let final_str = format!(
r#"
static REGEX: [Lazy<Regex>; {count}] = [{regex_str}];
static BOUNDARYLESS_REGEX: [Lazy<Regex>; {count}] = [{boundaryless_regex_str}];
"#
);
let regex_dest_path = Path::new(&out_dir).join("regex_data.rs");
fs::write(regex_dest_path, final_str).unwrap();
}