Update layout tests

[rbdr/page] / src / main.rs

diff --git a/src/main.rs b/src/main.rs
index c6de435878859cdfebf369748e292e9491c24142..902f3c3a757747e08581999b0e57d1c85f5cd732 100644 (file)
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,4 +1,5 @@
 mod gemini_parser;
+mod html_renderer;
 mod file_finder;
 mod file_handler;
 
@@ -14,19 +15,15 @@ fn main() -> Result<()> {
     let source = current_dir()?;
     let source_name = source.file_name().unwrap().to_string_lossy();
     let parent = source.parent().unwrap();
-    let destination_name = format!("{}_html", source_name);
-    let destination = parent.join(destination_name);
+    let gemini_destination_name = format!("{}_gemini", source_name);
+    let gemini_destination = parent.join(gemini_destination_name);
+    let html_destination_name = format!("{}_html", source_name);
+    let html_destination = parent.join(html_destination_name);
 
     // Step 1. Identify the files
     let files = find_files(&source);
 
-    // Step 2. Prepare the target priority
-    match remove_dir_all(&destination) {
-        _ => {}
-    };
-    create_dir_all(&destination)?;
-
-    // Step 3. Load the layout
+    // Step 2. Load the layout
     let mut file_handler = FileHandler::default();
     match file_handler.get_layout_or_panic(&files) {
         Ok(_) => {},
@@ -36,7 +33,20 @@ fn main() -> Result<()> {
         }
     }
 
+    // Step 3. Prepare the target priority
+    match remove_dir_all(&html_destination) {
+        _ => {}
+    };
+    match remove_dir_all(&gemini_destination) {
+        _ => {}
+    };
+
+    create_dir_all(&html_destination)
+        .expect("Could not create HTML directory.");
+    create_dir_all(&gemini_destination)
+        .expect("Could not create Gemini directory.");
+
     // Step 4. Process all files
-    file_handler.handle_all(&source, &destination, &files);
+    file_handler.handle_all(&source, &html_destination, &gemini_destination, &files);
     Ok(())
 }