1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
//! Module containing utilities for the filestore implementation
//!
use std::collections::BTreeMap;
use std::path::Path;
use anyhow::Error;
use anyhow::Result;
use anyhow::anyhow;
use indicatif::ProgressBar;
use resiter::AndThen;
use resiter::Map;
use walkdir::WalkDir;
use crate::filestore::Artifact;
use crate::filestore::path::*;
/// The actual filestore implementation
///
/// Because the "staging" filestore and the "release" filestore function the same underneath, we
/// provide this type as the implementation.
///
/// It can then be wrapped into the actual interface of this module with specialized functionality.
pub struct FileStoreImpl {
pub(in crate::filestore) root: StoreRoot,
store: BTreeMap<ArtifactPath, Artifact>,
}
impl FileStoreImpl {
/// Loads the passed path recursively into a Path => Artifact mapping
pub fn load(path: &Path, progress: ProgressBar) -> Result<Self> {
if path.is_dir() {
let root = StoreRoot::new(path.to_path_buf());
let store = WalkDir::new(&path)
.follow_links(false)
.into_iter()
.filter_entry(|e| e.file_type().is_file())
.map_err(Error::from)
.and_then_ok(|f| {
progress.tick();
let p = root.stripped_from(f.path())?;
Artifact::load(&root, p.clone()).map(|a| (p, a))
})
.collect::<Result<BTreeMap<ArtifactPath, Artifact>>>()?;
Ok(FileStoreImpl { root, store })
} else {
Err(anyhow!("File store cannot be loaded from non-directory: {}", path.display()))
}
}
pub fn root_path(&self) -> &StoreRoot {
&self.root
}
pub fn path_exists_in_store_root(&self, p: &Path) -> bool {
self.root.join_path(p).is_file()
}
pub (in crate::filestore) fn values(&self) -> impl Iterator<Item = &Artifact> {
self.store.values()
}
pub (in crate::filestore) fn load_from_path(&mut self, pb: &FullArtifactPath) -> Result<&Artifact> {
if !self.is_sub_path(pb.as_path())? {
Err(anyhow!("Not a sub-path of {}: {}", self.root.display(), pb.display()))
} else {
let art_path = self.root.stripped_from(pb.as_path())?;
if self.store.get(&art_path).is_some() {
Err(anyhow!("Entry exists: {}", pb.display()))
} else {
Ok(self.store.entry(art_path.clone()).or_insert(Artifact::load(&self.root, art_path)?))
}
}
}
fn is_sub_path(&self, p: &Path) -> Result<bool> {
p.canonicalize()
.map(|c| c.starts_with(self.root.as_ref()))
.map_err(Error::from)
}
}
|
