use std::collections::btree_map;
use std::collections::BTreeMap;
use std::path::PathBuf;
use std::rc::Rc;
use super::translate::OpsMap;
use super::Error;
use super::OpPointer;
pub struct Ops {
ops: BTreeMap<PathBuf, Rc<OpsMap>>,
}
impl Ops {
pub fn new() -> Self {
Self {
ops: BTreeMap::new(),
}
}
pub fn entry<'a, P: Into<PathBuf>>(&'a mut self, path: P) -> Entry<'a> {
Entry(self.ops.entry(path.into()))
}
}
pub struct Entry<'a>(btree_map::Entry<'a, PathBuf, Rc<OpsMap>>);
impl<'a> Entry<'a> {
pub fn get_pointer_or_else<F: FnOnce() -> Result<OpsMap, Error>, P: Into<PathBuf>>(
self,
f: F,
path: P,
) -> Result<OpPointer, Error> {
let cached = match self.0 {
btree_map::Entry::Occupied(e) => e.get().clone(),
btree_map::Entry::Vacant(e) => {
let v = Rc::new(f()?);
e.insert(v.clone());
v
}
};
let mut ptr = OpPointer::new(cached);
ptr.set_path(path.into());
Ok(ptr)
}
}