refactor: add SourceDir

Sponsored-by: author

2 files changed, 39 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 9f955d8..868ddfb 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -7,6 +7,7 @@
 //! little slow. This care implements a subset of the functionality of
 //! ikiwiki in Rust, for speed.
 
+pub mod srcdir;
 pub mod name;
 pub mod directive;
 pub mod error;
diff --git a/src/srcdir.rs b/src/srcdir.rs
new file mode 100644
index 0000000..f8d3949
--- /dev/null
+++ b/src/srcdir.rs
@@ -0,0 +1,38 @@
+use crate::error::SiteError;
+use log::trace;
+use std::path::{Path, PathBuf};
+use walkdir::WalkDir;
+
+pub struct SouceDir {
+    path: PathBuf,
+    files: Vec<PathBuf>,
+}
+
+impl SouceDir {
+    pub fn new(path: &Path) -> Self {
+        Self {
+            path: path.into(),
+            files: vec![],
+        }
+    }
+
+    pub fn insert(&mut self, path: &Path) {
+        trace!("Source Dir::insert: path={}", path.display());
+        self.files.push(path.into());
+    }
+
+    pub fn scan(&mut self) -> Result<(), SiteError> {
+        trace!("SourceDir::scan: find files in {}", self.path.display());
+        for e in WalkDir::new(&self.path) {
+            let e = e.map_err(|err| SiteError::WalkDir(self.path.clone(), err))?;
+            let path = e.path();
+            trace!("SourceDir::scan: found {}", path.display());
+            self.insert(path);
+        }
+        Ok(())
+    }
+
+    pub fn files(&self) -> &[PathBuf] {
+        &self.files
+    }
+}