aboutsummaryrefslogtreecommitdiff
path: root/lib/src
diff options
context:
space:
mode:
authorEric Kidd2016-09-15 10:29:29 -0400
committerEric Kidd2016-09-15 10:29:29 -0400
commitcae983f2cc93258c69e0f28e149d5da7802bb464 (patch)
tree1f175d26bc73dc2f58492188e19bbdbaa3bc52f7 /lib/src
parentb08684329bd0601550e3aa6f0926ce1e755668e8 (diff)
downloadincludedir-cae983f2cc93258c69e0f28e149d5da7802bb464.tar.gz
includedir-cae983f2cc93258c69e0f28e149d5da7802bb464.tar.xz
includedir-cae983f2cc93258c69e0f28e149d5da7802bb464.zip
Allow iteration over available file names
Your example program included an unimplemented `entries` method. I chose not to implement `entries` because iterating over it would require decompressing any large files stored in the map. Instead, I've provided a `file_names` API that provides just the names of the files in the map, which the caller can look up (and potentially decompress) if they wish. I implemented a custom iterator wrapper to hide the implementation details from the caller.
Diffstat (limited to 'lib/src')
-rw-r--r--lib/src/lib.rs21
1 files changed, 21 insertions, 0 deletions
diff --git a/lib/src/lib.rs b/lib/src/lib.rs
index ce1e2ec..57d3610 100644
--- a/lib/src/lib.rs
+++ b/lib/src/lib.rs
@@ -37,6 +37,12 @@ impl Files {
self.files.contains_key(path)
}
+ /// Returns an iterator over all available file names. Does not
+ /// decompress any compressed data.
+ pub fn file_names(&self) -> FileNames {
+ FileNames { iter: self.files.keys() }
+ }
+
pub fn get(&self, path: &str) -> io::Result<Cow<'static, [u8]>> {
let key = as_key(path);
match self.files.get(key.borrow() as &str) {
@@ -83,3 +89,18 @@ impl Files {
}
}
}
+
+/// Iterates over the file names available for `Files` object.
+pub struct FileNames<'a> {
+ /// Our internal iterator. We wrap this in a nice struct so our
+ /// caller doesn't need to know the details.
+ iter: phf::map::Keys<'a, &'static str, (Compression, &'static [u8])>,
+}
+
+impl<'a> Iterator for FileNames<'a> {
+ type Item = &'static str;
+
+ fn next(&mut self) -> Option<Self::Item> {
+ self.iter.next().map(|s| *s)
+ }
+}