50 lines
1.3 KiB
Rust
50 lines
1.3 KiB
Rust
use std::env;
|
|
use std::fs;
|
|
use std::io::BufWriter;
|
|
use std::io::Write;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
fn main() {
|
|
let out_dir = env::var("OUT_DIR").unwrap();
|
|
let dest = Path::new(&out_dir);
|
|
|
|
let base = Path::new("assets");
|
|
let mut out = Vec::new();
|
|
build_map(&mut out, &base);
|
|
|
|
let mut file = BufWriter::new(fs::File::create(&dest.join("resources.rs")).unwrap());
|
|
writeln!(
|
|
file,
|
|
"pub fn get_file(name: &str) -> Option<&'static [u8]> {{"
|
|
)
|
|
.unwrap();
|
|
writeln!(file, " match name {{").unwrap();
|
|
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("\\", "/");
|
|
|
|
writeln!(
|
|
file,
|
|
" {:?} => Some(include_bytes!(\"{}\")),",
|
|
relative, absolute
|
|
)
|
|
.unwrap();
|
|
}
|
|
write!(file, " _ => None\n }}\n}}\n").unwrap();
|
|
}
|
|
|
|
fn build_map(out: &mut Vec<PathBuf>, path: &Path) {
|
|
let files = fs::read_dir(path).unwrap();
|
|
for entry in files {
|
|
let entry = entry.unwrap();
|
|
if fs::metadata(entry.path()).unwrap().is_dir() {
|
|
build_map(out, &entry.path());
|
|
} else {
|
|
out.push(entry.path());
|
|
}
|
|
}
|
|
}
|