X-Git-Url: https://git.r.bdr.sh/rbdr/page/blobdiff_plain/260e8ec69b8e08b9fd105bf688e7a3a9fafecd61..5732d284ebc2cc2cbde0f050443b8f137dbf585b:/src/main.rs

diff --git a/src/main.rs b/src/main.rs
index 533a6bb..cedd557 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,12 +1,12 @@
-mod gemini_parser;
-mod html_renderer;
 mod file_finder;
 mod file_handler;
+mod gemini_parser;
+mod html_renderer;
 
-use std::io::Result;
-use std::process::exit;
 use std::env::current_dir;
 use std::fs::{create_dir_all, remove_dir_all};
+use std::io::Result;
+use std::process::exit;
 
 use crate::file_finder::find_files;
 use crate::file_handler::FileHandler;
@@ -23,26 +23,23 @@ 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(_) => {},
+        Ok(_) => {}
         Err(error) => {
             eprintln!("{}", error);
             exit(1);
         }
     }
 
+    // Step 3. Prepare the target priority
+    let _ = remove_dir_all(&html_destination);
+    let _ = 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(())