2016-02-03 14:47:14 +00:00
|
|
|
use libimagstore::store::EntryHeader;
|
|
|
|
|
2016-02-03 18:59:22 +00:00
|
|
|
use error::{LinkError, LinkErrorKind};
|
2016-02-03 14:47:14 +00:00
|
|
|
use link::{Link, Links};
|
|
|
|
use result::Result;
|
|
|
|
|
2016-02-03 18:59:22 +00:00
|
|
|
use toml::Value;
|
|
|
|
use toml::Table;
|
|
|
|
|
|
|
|
pub fn get_link(header: &EntryHeader) -> Result<Option<Link>> {
|
|
|
|
let uri = header.read("imag.content.uri");
|
|
|
|
|
|
|
|
if uri.is_err() {
|
|
|
|
let kind = LinkErrorKind::EntryHeaderReadError;
|
|
|
|
let lerr = LinkError::new(kind, Some(Box::new(uri.err().unwrap())));
|
|
|
|
return Err(lerr);
|
|
|
|
}
|
|
|
|
let uri = uri.unwrap();
|
|
|
|
|
|
|
|
match uri {
|
|
|
|
Some(Value::String(s)) => Ok(Some(Link::new(s))),
|
|
|
|
_ => Err(LinkError::new(LinkErrorKind::ExistingLinkTypeWrong, None)),
|
|
|
|
}
|
2016-02-03 14:47:14 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Set an external link in the header
|
|
|
|
///
|
|
|
|
/// Return the previous set link if there was any
|
2016-02-07 03:40:43 +00:00
|
|
|
pub fn set_link(header: &mut EntryHeader, l: Link) -> Result<Option<Link>> {
|
|
|
|
let old_link = header.set("imag.content.uri", Value::String(l.into()));
|
|
|
|
|
|
|
|
if old_link.is_err() {
|
|
|
|
let kind = LinkErrorKind::EntryHeaderWriteError;
|
|
|
|
let lerr = LinkError::new(kind, Some(Box::new(old_link.err().unwrap())));
|
|
|
|
return Err(lerr);
|
|
|
|
}
|
|
|
|
let old_link = old_link.unwrap();
|
|
|
|
|
|
|
|
if old_link.is_none() {
|
|
|
|
return Ok(None);
|
|
|
|
}
|
|
|
|
|
|
|
|
match old_link.unwrap() {
|
|
|
|
Value::String(s) => Ok(Some(Link::new(s))),
|
|
|
|
|
|
|
|
// We don't do anything in this case and be glad we corrected the type error with this set()
|
|
|
|
_ => Ok(None),
|
|
|
|
}
|
2016-02-03 14:47:14 +00:00
|
|
|
}
|
|
|
|
|