parent
837a0ca643
commit
2f401b176b
@ -0,0 +1,82 @@ |
||||
//! An adaptation of the assets loader from rst-client, but only for
|
||||
//! maps and unit definitions
|
||||
|
||||
use crate::config::MapCfg; |
||||
use serde_yaml; |
||||
use std::{ |
||||
collections::BTreeMap, |
||||
fs::{self, File, OpenOptions}, |
||||
io::{Read, Write}, |
||||
path::PathBuf, |
||||
}; |
||||
|
||||
/// A unique resource identifier
|
||||
#[derive(Clone, Debug, Eq, PartialEq, Ord, PartialOrd)] |
||||
pub struct Uri(String, String); |
||||
|
||||
impl From<&'static str> for Uri { |
||||
fn from(s: &'static str) -> Self { |
||||
let mut v: Vec<_> = s.split("/").collect(); |
||||
Self(v.remove(0).into(), v.remove(0).into()) |
||||
} |
||||
} |
||||
|
||||
impl From<String> for Uri { |
||||
fn from(s: String) -> Self { |
||||
let mut v: Vec<_> = s.split("/").collect(); |
||||
Self(v.remove(0).into(), v.remove(0).into()) |
||||
} |
||||
} |
||||
|
||||
/// A loader for maps
|
||||
pub struct MapLoader { |
||||
root: PathBuf, |
||||
inner: BTreeMap<String, MapCfg>, |
||||
} |
||||
|
||||
impl MapLoader { |
||||
pub fn load_path(path: PathBuf) -> Self { |
||||
Self { |
||||
root: path.clone(), |
||||
inner: fs::read_dir(&path) |
||||
.unwrap() |
||||
.filter_map(|map| { |
||||
let map = map.unwrap(); |
||||
let name = map.file_name().into_string().unwrap(); |
||||
|
||||
if map.path().is_dir() { |
||||
warn!("Directories in map path will be ignored: {}!", name); |
||||
None |
||||
} else { |
||||
let np = map.path().with_extension(""); |
||||
let name = np.file_name().unwrap().to_str().unwrap(); |
||||
|
||||
debug!("Loading map {}", name); |
||||
let mut c = String::new(); |
||||
let mut f = File::open(map.path()).unwrap(); |
||||
f.read_to_string(&mut c).unwrap(); |
||||
|
||||
Some((name.into(), serde_yaml::from_str(&c).unwrap())) |
||||
} |
||||
}) |
||||
.collect(), |
||||
} |
||||
} |
||||
|
||||
/// Save all maps that are present currently
|
||||
pub fn save_all(&self) { |
||||
self.inner.iter().for_each(|(name, map)| { |
||||
let path = self.root.join(format!("{}.rstmap", name)); |
||||
let mut f = OpenOptions::new() |
||||
.truncate(true) |
||||
.create(true) |
||||
.write(true) |
||||
.open(path.clone()) |
||||
.unwrap(); |
||||
|
||||
debug!("Writing map configuration {}", path.to_str().unwrap()); |
||||
let buf = serde_yaml::to_string(&map).unwrap(); |
||||
f.write_all(buf.as_bytes()).unwrap(); |
||||
}); |
||||
} |
||||
} |
@ -0,0 +1,7 @@ |
||||
use rst_core::loader::MapLoader; |
||||
use std::path::Path; |
||||
|
||||
/// Load a set of maps at launch
|
||||
pub(crate) fn load<'p, P: Into<&'p Path>>(path: P) -> MapLoader { |
||||
MapLoader::load_path(path.into().to_path_buf()) |
||||
} |
Loading…
Reference in new issue