From 1c5797fadeea6be505c01f13508203ba234cbdfa Mon Sep 17 00:00:00 2001
From: Ruben Beltran del Rio <git@r.bdr.sh>
Date: Fri, 3 Jan 2025 23:17:06 +0100
Subject: Update main and file_finder tests

---
 src/main.rs | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index 533a6bb..902f3c3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -23,17 +23,7 @@ fn main() -> Result<()> {
     // Step 1. Identify the files
     let files = find_files(&source);
 
-    // Step 2. Prepare the target priority
-    match remove_dir_all(&html_destination) {
-        _ => {}
-    };
-    create_dir_all(&html_destination)?;
-    match remove_dir_all(&gemini_destination) {
-        _ => {}
-    };
-    create_dir_all(&gemini_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(_) => {},
@@ -43,6 +33,19 @@ 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, &html_destination, &gemini_destination, &files);
     Ok(())
-- 
cgit