parent
2511939f4e
commit
e2eee6beb1
GPG Key ID:
284488987C31F630
@ -0,0 +1,107 @@
|
||||
use super::*;
|
||||
use std::{
|
||||
error,
|
||||
fmt,
|
||||
};
|
||||
use std::ffi::OsStr;
|
||||
|
||||
/// Builder for creating [`Entry`] instances.
|
||||
#[derive(Debug, Clone, Eq, PartialEq, Hash, Default)]
|
||||
pub struct EntryBuilder<'a>
|
||||
{
|
||||
name: Option<&'a str>,
|
||||
desc: Option<&'a str>,
|
||||
|
||||
tags: Option<&'a[&'a str]>,
|
||||
|
||||
key: Option<aes::AesKey>,
|
||||
hash: Option<sha256::Sha256Hash>,
|
||||
|
||||
location: Option<&'a Path>,
|
||||
filename: Option<&'a OsStr>,
|
||||
}
|
||||
|
||||
macro_rules! build_fn {
|
||||
($name:ident: $ty:ty $(, $com:literal)?) => {
|
||||
$(#[doc=$com])?
|
||||
#[inline] pub const fn $name(self, $name: $ty) -> Self
|
||||
{
|
||||
Self {
|
||||
$name: Some($name),
|
||||
..self
|
||||
}
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
impl<'a> EntryBuilder<'a>
|
||||
{
|
||||
/// Create a new entry builder
|
||||
#[inline] pub const fn new() -> Self
|
||||
{
|
||||
macro_rules! default {
|
||||
($($name:ident),*) => {
|
||||
Self {
|
||||
$(
|
||||
$name: None
|
||||
),*
|
||||
}
|
||||
}
|
||||
}
|
||||
default!{
|
||||
name, desc, tags, key, hash, location, filename
|
||||
}
|
||||
}
|
||||
|
||||
build_fn!(name: &'a str, "Insert the name of this entry");
|
||||
build_fn!(desc: &'a str, "Insert a description for this entry");
|
||||
build_fn!(tags: &'a [&'a str], "Insert the tags for this entry");
|
||||
build_fn!(key: aes::AesKey, "Insert the encryption key for this entry");
|
||||
build_fn!(hash: sha256::Sha256Hash, "Insert the hash for this entry");
|
||||
build_fn!(location: &'a Path, "Insert the location for this entry");
|
||||
build_fn!(filename: &'a OsStr, "Insert the original filename for this entry");
|
||||
|
||||
/// Try to build an `Entry` from this builder.
|
||||
#[inline] pub fn build(self) -> Result<Entry, EntryBuildError>
|
||||
{
|
||||
macro_rules! fail_on {
|
||||
($name:ident) => {
|
||||
self.$name.ok_or(EntryBuildError)?
|
||||
};
|
||||
($name:ident: $or:expr) =>{
|
||||
if let Some(name) = self.$name {
|
||||
name
|
||||
} else {
|
||||
$or
|
||||
}
|
||||
};
|
||||
}
|
||||
Ok(
|
||||
Entry {
|
||||
name: fail_on!(name).into(),
|
||||
description: fail_on!(desc: "").into(),
|
||||
tags: fail_on!(tags: &[]).iter().map(|&x| x.to_owned()).collect(),
|
||||
key: self.key,
|
||||
hash: fail_on!(hash),
|
||||
location: fail_on!(location).into(),
|
||||
filename: fail_on!(filename: OsStr::new("")).into(),
|
||||
|
||||
cache: Default::default(),
|
||||
}
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
/// Error for when an improporly configured [`EntryBuilder`] called `.build()`.
|
||||
/// Usually this is because of missing fields.
|
||||
#[derive(Debug)]
|
||||
pub struct EntryBuildError;
|
||||
|
||||
impl error::Error for EntryBuildError{}
|
||||
impl fmt::Display for EntryBuildError
|
||||
{
|
||||
#[inline] fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||
{
|
||||
write!(f, "improporly configured builder")
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,61 @@
|
||||
use super::*;
|
||||
|
||||
#[inline] fn gen_meta() -> StoreMetadata
|
||||
{
|
||||
StoreMetadata{
|
||||
name: "test".to_string(),
|
||||
root: "./test/db".to_owned().into(),
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn create()
|
||||
{
|
||||
let store = Store::new(gen_meta());
|
||||
let freeze = store.into_freeze();
|
||||
let _ = freeze.into_new();
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn entries()
|
||||
{
|
||||
let mut store = Store::new(gen_meta());
|
||||
macro_rules! build {
|
||||
() => {
|
||||
EntryBuilder::new()
|
||||
.name("Hello")
|
||||
.desc("Desc")
|
||||
.tags(&["Tag1", "Tag2", "Tag3"])
|
||||
.hash(sha256::compute_slice(b"hello world"))
|
||||
.location(Path::new("test_file"))
|
||||
};
|
||||
}
|
||||
let entry = build!()
|
||||
.build().expect("Failed to build entry");
|
||||
|
||||
println!("Entry: {:#?}", entry);
|
||||
let entry = store.insert(entry).clone();
|
||||
store.insert(build!()
|
||||
.hash(sha256::compute_slice("something else"))
|
||||
.name("Entry 2")
|
||||
.build().expect("Failed to build entry 2"));
|
||||
println!("Entry ref: {:#?}", entry);
|
||||
println!("Store: {:#?}", store);
|
||||
|
||||
let freeze = store.freeze();
|
||||
println!("Frozen: {:#?}", freeze);
|
||||
let store2 = Store::unfreeze(&freeze);
|
||||
|
||||
|
||||
assert_eq!(store2.tag_search("Tag2").filter(|x| x.name() == entry.name()).next().unwrap().hash(), entry.hash());
|
||||
assert_eq!(store.tag_search_any(vec!["Tag2", "Tag3"]).filter(|x| x.name() == entry.name()).next().unwrap().hash(), entry.hash());
|
||||
assert_eq!(store2.tag_search_all(vec!["Tag1", "Tag3"]).filter(|x| x.name() == entry.name()).next().unwrap().hash(), entry.hash());
|
||||
|
||||
assert!(store.tag_search_all(vec!["Tag1", "Nope"]).next().is_none());
|
||||
|
||||
let json = serde_json::to_string_pretty(&freeze).unwrap();
|
||||
println!("Freeze serialised ({} bytes): '{}'", json.len(), json);
|
||||
let ufreeze: Freeze = serde_json::from_str(json.as_str()).unwrap();
|
||||
assert_eq!(ufreeze, freeze);
|
||||
println!("Freeze unseralised as store: {:#?}", ufreeze.into_new());
|
||||
}
|
||||
Loading…
Reference in new issue