Add logic to read and assign files / layout

[rbdr/page] / src / file_finder.rs

diff --git a/src/file_finder.rs b/src/file_finder.rs
index fc32fa92d16e11d8e5cdeadbe26ceaccc078004e..105a99928a0df5fd9b04921f275df189d22bc776 100644 (file)
--- a/src/file_finder.rs
+++ b/src/file_finder.rs
@@ -2,17 +2,22 @@ use crate::file_handler::{File, FileHandler};
 use std::fs::read_dir;
 use std::path::PathBuf;
 
-pub fn find_files(directory_path: PathBuf) -> Vec<File> {
+pub fn find_files(directory_path: &PathBuf) -> Vec<File> {
+    return find_files_recursively(directory_path, directory_path);
+}
+
+fn find_files_recursively(root_path: &PathBuf, directory_path: &PathBuf) -> Vec<File> {
     let mut result: Vec<File> = vec![];
     let file_handler = FileHandler::default();
-    let entries = read_dir(directory_path).unwrap();
+    let entries = read_dir(&directory_path).unwrap();
     for entry in entries {
         let path = entry.unwrap().path();
-        if path.starts_with(".") && !path.starts_with(".well-known") {
+        let relative_path = path.strip_prefix(&root_path).unwrap();
+        if relative_path.starts_with(".git") {
             continue;
         }
         if path.is_dir() {
-            result.append(&mut find_files(path))
+            result.append(&mut find_files_recursively(&root_path, &path))
         } else {
             let file_type = file_handler.identify(&path);
             result.push(File {