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
use std::fs::File;
use std::path::{Path, PathBuf};
use ::abc::{Loader, LoadResult, LoadError};
#[derive(Debug)]
pub struct FilesystemLoader {
path: PathBuf,
}
impl FilesystemLoader {
pub fn new(path: &Path) -> Box<Self> {
Box::new(FilesystemLoader { path: path.to_owned() })
}
}
impl Loader for FilesystemLoader {
fn load(&self, name: &str) -> LoadResult {
if name.contains("..") { return Err(LoadError::BadName("\"..\" is not supported in a path".to_owned())) }
let mut path = self.path.join(name);
if path.extension().is_none() { path.set_extension("tpl"); }
if !path.exists() || !path.is_file() { return Err(LoadError::NotFound); }
let load = || -> ::std::io::Result<String> {
use std::io::Read;
let mut buf = String::new();
let mut f = File::open(path)?;
f.read_to_string(&mut buf)?;
Ok(buf)
};
load().map_err(|err| LoadError::IoError(format!("{:?}", err)))
}
}