2015-09-19 14:08:28 -04:00
|
|
|
use std::env;
|
|
|
|
use std::fs;
|
2016-07-02 09:15:10 -04:00
|
|
|
use std::path::{Path, PathBuf};
|
2015-09-19 14:08:28 -04:00
|
|
|
use std::io::BufWriter;
|
|
|
|
use std::io::Write;
|
|
|
|
|
|
|
|
fn main() {
|
|
|
|
let out_dir = env::var("OUT_DIR").unwrap();
|
|
|
|
let dest = Path::new(&out_dir);
|
|
|
|
|
2016-03-16 14:15:13 -04:00
|
|
|
let base = Path::new("assets");
|
2015-09-19 14:08:28 -04:00
|
|
|
let mut out = Vec::new();
|
|
|
|
build_map(&mut out, &base);
|
|
|
|
|
|
|
|
let mut file = BufWriter::new(fs::File::create(&dest.join("resources.rs")).unwrap());
|
|
|
|
write!(file, "pub fn get_file(name: &str) -> Option<&'static [u8]> {{\n").unwrap();
|
|
|
|
write!(file, " match name {{\n").unwrap();
|
2016-07-02 09:15:10 -04:00
|
|
|
for path in &out {
|
|
|
|
let mut absolute_path = std::env::current_dir().unwrap();
|
|
|
|
absolute_path.push(path);
|
|
|
|
|
|
|
|
let absolute = absolute_path.to_str().unwrap().replace("\\", "/");
|
|
|
|
let relative = path.to_str().unwrap().replace("\\", "/");
|
|
|
|
|
|
|
|
write!(file, " {:?} => Some(include_bytes!(\"{}\")),\n", relative, absolute).unwrap();
|
2015-09-19 14:08:28 -04:00
|
|
|
}
|
|
|
|
write!(file, " _ => None\n }}\n}}\n").unwrap();
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2016-07-02 09:15:10 -04:00
|
|
|
fn build_map(out: &mut Vec<PathBuf>, path: &Path) {
|
2015-09-19 14:08:28 -04:00
|
|
|
let files = fs::read_dir(path).unwrap();
|
|
|
|
for entry in files {
|
2016-07-02 09:15:10 -04:00
|
|
|
let entry = entry.unwrap();
|
|
|
|
if fs::metadata(entry.path()).unwrap().is_dir() {
|
|
|
|
build_map(out, &entry.path());
|
|
|
|
} else {
|
|
|
|
out.push(entry.path());
|
|
|
|
}
|
2015-09-19 14:08:28 -04:00
|
|
|
}
|
2016-03-16 14:15:13 -04:00
|
|
|
}
|