From 6c387e893c3421cd68ca8b5251d09614fc68f5ba Mon Sep 17 00:00:00 2001 From: Matthias Beyer Date: Sat, 14 Oct 2017 13:36:04 +0200 Subject: Add RefStore::get_by_partial_hash() --- lib/entry/libimagentryref/src/refstore.rs | 24 +++++++++++++++++++++++- 1 file changed, 23 insertions(+), 1 deletion(-) (limited to 'lib/entry/libimagentryref') diff --git a/lib/entry/libimagentryref/src/refstore.rs b/lib/entry/libimagentryref/src/refstore.rs index acfb2b8c..b1ac9381 100644 --- a/lib/entry/libimagentryref/src/refstore.rs +++ b/lib/entry/libimagentryref/src/refstore.rs @@ -22,7 +22,6 @@ use std::collections::BTreeMap; use std::fs::File; use libimagstore::store::FileLockEntry; -use libimagstore::storeid::StoreId; use libimagstore::storeid::IntoStoreId; use libimagstore::storeid::StoreIdIterator; use libimagstore::store::Store; @@ -48,6 +47,12 @@ pub trait RefStore { /// Returns None if the hash cannot be found. fn get_by_hash<'a>(&'a self, hash: String) -> Result>>; + /// Get a Ref object from the store by (eventually partial) hash. + /// + /// If the hash is complete, `RefStore::get_by_hash()` should be used as it is cheaper. + /// If the hash comes from user input and thus might be abbreviated, this function can be used. + fn get_by_partitial_hash<'a>(&'a self, hash: &String) -> Result>>; + /// Delete a ref by hash /// /// If the returned Result contains an error, the ref might not be deleted. @@ -118,6 +123,23 @@ impl RefStore for Store { .map_err(From::from) } + /// Get a Ref object from the store by (eventually partial) hash. + /// + /// If the hash is complete, `RefStore::get_by_hash()` should be used as it is cheaper. + /// If the hash comes from user input and thus might be abbreviated, this function can be used. + fn get_by_partitial_hash<'a>(&'a self, hash: &String) -> Result>> { + for id in self.retrieve_for_module("ref")? { + let components_have_hash = id + .components() + .any(|c| c.as_os_str().to_str().map(|s| s.contains(hash)).unwrap_or(false)); + + if components_have_hash { + return self.get(id).map_err(From::from); + } + } + Ok(None) + } + /// Delete a ref by hash /// /// If the returned Result contains an error, the ref might not be deleted. -- cgit v1.2.3