Add logic to read and assign files / layout

3 files changed, 23 insertions, 17 deletions

diff --git a/src/file_handler/file_strategies/file.rs b/src/file_handler/file_strategies/file.rs
index c9e8c96..b1de596 100644
--- a/src/file_handler/file_strategies/file.rs
+++ b/src/file_handler/file_strategies/file.rs
@@ -2,7 +2,7 @@ pub struct Strategy {}
 
 use std::path::PathBuf;
 
-use crate::file_handler::{FileType, FileHandlerStrategy};
+use crate::file_handler::{File, FileType, FileHandlerStrategy};
 
 impl FileHandlerStrategy for Strategy {
     fn is(&self, path: &PathBuf) -> bool {
@@ -13,11 +13,14 @@ impl FileHandlerStrategy for Strategy {
         FileType::File
     }
 
-    fn can_handle(&self, path: &PathBuf) -> bool {
-        self.is(path)
+    fn can_handle(&self, file_type: &FileType) -> bool {
+        match file_type {
+            FileType::File => true,
+            _ => false,
+        }
     }
 
-    fn handle(&self, path: &PathBuf) {
-        println!("Should copy {}", path.display())
+    fn handle(&self, file: &File) {
+        println!("Should copy {}", file.path.display())
     }
 }
diff --git a/src/file_handler/file_strategies/gemini.rs b/src/file_handler/file_strategies/gemini.rs
index ffa2a9f..905e1a9 100644
--- a/src/file_handler/file_strategies/gemini.rs
+++ b/src/file_handler/file_strategies/gemini.rs
@@ -2,7 +2,7 @@ pub struct Strategy {}
 
 use std::path::PathBuf;
 
-use crate::file_handler::{FileType, FileHandlerStrategy};
+use crate::file_handler::{File, FileType, FileHandlerStrategy};
 
 impl FileHandlerStrategy for Strategy {
     fn is(&self, path: &PathBuf) -> bool {
@@ -16,11 +16,14 @@ impl FileHandlerStrategy for Strategy {
         FileType::Gemini
     }
 
-    fn can_handle(&self, path: &PathBuf) -> bool {
-        self.is(path)
+    fn can_handle(&self, file_type: &FileType) -> bool {
+        match file_type {
+            FileType::Gemini => true,
+            _ => false,
+        }
     }
 
-    fn handle(&self, path: &PathBuf) {
-        println!("Should convert {}", path.display())
+    fn handle(&self, file: &File) {
+        println!("Should parse and copy {}", file.path.display())
     }
 }
diff --git a/src/file_handler/file_strategies/layout.rs b/src/file_handler/file_strategies/layout.rs
index cf1bb9f..bac1059 100644
--- a/src/file_handler/file_strategies/layout.rs
+++ b/src/file_handler/file_strategies/layout.rs
@@ -2,7 +2,7 @@ pub struct Strategy {}
 
 use std::path::PathBuf;
 
-use crate::file_handler::{FileType, FileHandlerStrategy};
+use crate::file_handler::{File, FileType, FileHandlerStrategy};
 
 impl FileHandlerStrategy for Strategy {
     fn is(&self, path: &PathBuf) -> bool {
@@ -13,12 +13,12 @@ impl FileHandlerStrategy for Strategy {
         FileType::Layout
     }
 
-    fn can_handle(&self, path: &PathBuf) -> bool {
-        self.is(path)
+    fn can_handle(&self, file_type: &FileType) -> bool {
+        match file_type {
+            FileType::Layout => true,
+            _ => false,
+        }
     }
 
-    fn handle(&self, path: &PathBuf) {
-        println!("Should convert {}", path.display())
-    }
+    fn handle(&self, _file: &File) {}
 }
-