Read the files

7 files changed, 112 insertions, 1 deletions

diff --git a/src/file_finder.rs b/src/file_finder.rs
new file mode 100644
index 0000000..fc32fa9
--- /dev/null
+++ b/src/file_finder.rs
@@ -0,0 +1,25 @@
+use crate::file_handler::{File, FileHandler};
+use std::fs::read_dir;
+use std::path::PathBuf;
+
+pub fn find_files(directory_path: PathBuf) -> Vec<File> {
+    let mut result: Vec<File> = vec![];
+    let file_handler = FileHandler::default();
+    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") {
+            continue;
+        }
+        if path.is_dir() {
+            result.append(&mut find_files(path))
+        } else {
+            let file_type = file_handler.identify(&path);
+            result.push(File {
+                path: path,
+                file_type: file_type,
+            });
+        }
+    }
+    return result;
+}
diff --git a/src/file_handler/file_strategies/file.rs b/src/file_handler/file_strategies/file.rs
new file mode 100644
index 0000000..530bbd6
--- /dev/null
+++ b/src/file_handler/file_strategies/file.rs
@@ -0,0 +1,23 @@
+pub struct Strategy {}
+
+use std::path::PathBuf;
+
+use crate::file_handler::{FileType, FileHandlerStrategy};
+
+impl FileHandlerStrategy for Strategy {
+    fn is(&self, path: &PathBuf) -> bool {
+        !path.is_dir()
+    }
+
+    fn identify(&self) -> FileType {
+        FileType::File
+    }
+
+    fn can_handle(&self, path: &PathBuf) -> bool {
+        !path.is_dir()
+    }
+
+    fn handle(&self, path: &PathBuf) {
+        println!("Should copy {}", path.display())
+    }
+}
diff --git a/src/file_handler/file_strategies/gemini.rs b/src/file_handler/file_strategies/gemini.rs
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/src/file_handler/file_strategies/gemini.rs
diff --git a/src/file_handler/file_strategies/layout.rs b/src/file_handler/file_strategies/layout.rs
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/src/file_handler/file_strategies/layout.rs
diff --git a/src/file_handler/file_strategies/mod.rs b/src/file_handler/file_strategies/mod.rs
new file mode 100644
index 0000000..2e172cd
--- /dev/null
+++ b/src/file_handler/file_strategies/mod.rs
@@ -0,0 +1 @@
+pub mod file;
diff --git a/src/file_handler/mod.rs b/src/file_handler/mod.rs
new file mode 100644
index 0000000..8038b3c
--- /dev/null
+++ b/src/file_handler/mod.rs
@@ -0,0 +1,54 @@
+mod file_strategies;
+
+use file_strategies::file::Strategy as FileStrategy;
+use std::path::PathBuf;
+
+pub struct FileHandler {
+   pub strategies: Vec<Box<dyn FileHandlerStrategy>>
+}
+
+impl Default for FileHandler {
+    fn default() -> FileHandler {
+        FileHandler {
+            strategies: vec![Box::new(FileStrategy{})]
+        }
+    }
+}
+
+impl FileHandler {
+    pub fn identify(&self, path: &PathBuf) -> FileType {
+        for strategy in self.strategies.iter() {
+            if strategy.is(&path) {
+                return strategy.identify();
+            }
+        }
+        FileType::Unknown
+    }
+
+    pub fn handle(&self, path: &PathBuf) {
+        for strategy in self.strategies.iter() {
+            if strategy.can_handle(path) {
+                return strategy.handle(path);
+            }
+        }
+    }
+}
+
+pub trait FileHandlerStrategy {
+    fn is(&self, path: &PathBuf) -> bool;
+    fn identify(&self) -> FileType;
+    fn can_handle(&self, path: &PathBuf) -> bool;
+    fn handle(&self, path: &PathBuf);
+}
+
+pub enum FileType {
+    Gemini,
+    File,
+    Layout,
+    Unknown,
+}
+
+pub struct File {
+    pub path: PathBuf,
+    pub file_type: FileType,
+}
diff --git a/src/main.rs b/src/main.rs
index aa3fb93..4b8a79f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,9 +1,17 @@
 mod gemini_parser;
+mod file_finder;
+mod file_handler;
+
+use std::env::current_dir;
 
 use crate::gemini_parser::parse;
+use crate::file_finder::find_files;
 
 fn main() {
     let gemini_source = "# Test\n## 2nd H\na line\n another line\n```\npreformat\n=> preformat\n```\n=> http://lol.com\n=> http://lol.com lol\n* lol\nbla\n* lol\n* lmao\n> blabla\n> blabla";
     let html = parse(gemini_source);
-    println!("{}", html)
+    println!("{}", html);
+
+    let files = find_files(current_dir().unwrap());
+    println!("Found {} files", files.len());
 }