diff --git a/lib/entry/libimagentryref/src/refstore.rs b/lib/entry/libimagentryref/src/refstore.rs index 5f1c780d..69a96eb8 100644 --- a/lib/entry/libimagentryref/src/refstore.rs +++ b/lib/entry/libimagentryref/src/refstore.rs @@ -93,30 +93,48 @@ pub trait RefStore<'a> { impl<'a> RefStore<'a> for Store { - fn get_ref(&self) -> Result>, RPG::Error> { - unimplemented!() - } - - fn create_ref(&self) -> Result, RPG::Error> { - unimplemented!() - } - - fn retrieve_ref(&self) Result, RPG::Error> { - unimplemented!() - } - - fn delete_ref(&self) -> Result<(), RPG::Error> { - unimplemented!() - } - - fn ref_exists(&self) -> Result { - unimplemented!() - } - - fn move_ref_by_id(&self) - -> Result<(), Either> + fn get_ref(&'a self, hash: &String) + -> Result>, RPG::Error> { - unimplemented!() + let sid = StoreId::new_baseless(PathBuf::from(format!("{}/{}", RPG::collection(), hash))) + .map_err(RE::from)?; + + debug!("Getting: {:?}", sid); + self.get(sid) + .map_err(RE::from) + .map_err(RPG::Error::from) + } + + fn create_ref>(&'a self, path: A) + -> Result, RPG::Error> + { + let path_str = path.as_ref().to_str().map(String::from).ok_or(REK::PathUTF8Error.into())?; + let hash = RPG::unique_hash(path)?; + let pathbuf = PathBuf::from(format!("{}/{}", RPG::collection(), hash)); + let sid = StoreId::new_baseless(pathbuf).map_err(RE::from)?; + + debug!("Creating: {:?}", sid); + self.create(sid) + .map_err(RE::from) + .and_then(|mut fle| { + let _ = fle.set_isflag::()?; + { + let hdr = fle.get_header_mut(); + hdr.insert("ref.path", Value::String(String::from(path_str)))?; + hdr.insert("ref.hash", Value::String(hash))?; + } + Ok(fle) + }) + .map_err(RPG::Error::from) + } + + fn retrieve_ref>(&'a self, path: A) + -> Result, RPG::Error> + { + match self.get_ref::(&RPG::unique_hash(path.as_ref())?)? { + Some(r) => Ok(r), + None => self.create_ref::(path), + } } }