mirror of
https://github.com/a-b-street/abstreet.git
synced 2024-12-22 13:51:33 +03:00
101 lines
3.2 KiB
Rust
101 lines
3.2 KiB
Rust
use std::collections::BTreeMap;
|
|
use std::env;
|
|
use std::fs::File;
|
|
use std::io::{Read, Write};
|
|
use walkdir::WalkDir;
|
|
|
|
// TODO See https://github.com/dtolnay/inventory for an alternate approach.
|
|
fn main() {
|
|
let mut mapping: BTreeMap<String, String> = BTreeMap::new();
|
|
for entry in WalkDir::new("src") {
|
|
let path = format!("{}", entry.unwrap().into_path().display());
|
|
if path.ends_with(".rs") && path != "src/colors.rs" {
|
|
for (k, v) in read_file(&path) {
|
|
if mapping.contains_key(&k) {
|
|
panic!("Color {} defined twice", k);
|
|
}
|
|
mapping.insert(k, v);
|
|
}
|
|
}
|
|
}
|
|
|
|
let mut f = File::create(format!("{}/init_colors.rs", env::var("OUT_DIR").unwrap())).unwrap();
|
|
writeln!(f, "fn default_colors() -> HashMap<String, Color> {{").unwrap();
|
|
writeln!(f, " let mut m = HashMap::new();").unwrap();
|
|
for (k, v) in mapping {
|
|
writeln!(f, " m.insert(\"{}\".to_string(), {});", k, v).unwrap();
|
|
}
|
|
writeln!(f, " m").unwrap();
|
|
writeln!(f, "}}").unwrap();
|
|
}
|
|
|
|
fn read_file(path: &str) -> Vec<(String, String)> {
|
|
let mut src = {
|
|
let mut s = String::new();
|
|
let mut f = File::open(path).unwrap();
|
|
f.read_to_string(&mut s).unwrap();
|
|
s
|
|
};
|
|
|
|
let mut entries: Vec<(String, String)> = Vec::new();
|
|
|
|
while !src.is_empty() {
|
|
if src.starts_with("get_def(") {
|
|
src = src["get_def(".len()..].to_string();
|
|
|
|
// Look for the opening "
|
|
while !src.starts_with("\"") {
|
|
src = src[1..].to_string();
|
|
}
|
|
src = src[1..].to_string();
|
|
|
|
// Read the key until the closing "
|
|
let mut key = String::new();
|
|
while !src.starts_with("\"") {
|
|
key.push(src.chars().next().unwrap());
|
|
src = src[1..].to_string();
|
|
}
|
|
src = src[1..].to_string();
|
|
|
|
// Look for the ,
|
|
while !src.starts_with(",") {
|
|
src = src[1..].to_string();
|
|
}
|
|
src = src[1..].to_string();
|
|
|
|
// Look for the Color
|
|
while !src.starts_with("Color") {
|
|
src = src[1..].to_string();
|
|
}
|
|
|
|
// Wait for the ()'s to be mismatched, meaning we found the ) of the get_def()
|
|
let mut counter = 0;
|
|
let mut value = String::new();
|
|
loop {
|
|
value.push(src.chars().next().unwrap());
|
|
if src.starts_with("(") {
|
|
counter += 1;
|
|
} else if src.starts_with(")") {
|
|
counter -= 1;
|
|
if counter == -1 {
|
|
value.pop();
|
|
entries.push((key, value));
|
|
src = src[1..].to_string();
|
|
break;
|
|
}
|
|
} else if src.starts_with(",") && counter == 0 {
|
|
value.pop();
|
|
entries.push((key, value));
|
|
src = src[1..].to_string();
|
|
break;
|
|
}
|
|
src = src[1..].to_string();
|
|
}
|
|
} else {
|
|
src = src[1..].to_string();
|
|
}
|
|
}
|
|
|
|
entries
|
|
}
|