Handle layout error better

[rbdr/page] / src / main.rs

diff --git a/src/main.rs b/src/main.rs
index e04da8abb815f787d599ef457fe23c70b7a0d6d2..c6de435878859cdfebf369748e292e9491c24142 100644 (file)
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,18 +3,14 @@ mod file_finder;
 mod file_handler;
 
 use std::io::Result;
+use std::process::exit;
 use std::env::current_dir;
-use std::fs::create_dir_all;
+use std::fs::{create_dir_all, remove_dir_all};
 
-use crate::gemini_parser::parse;
 use crate::file_finder::find_files;
 use crate::file_handler::FileHandler;
 
 fn main() -> Result<()> {
-    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);
-
     let source = current_dir()?;
     let source_name = source.file_name().unwrap().to_string_lossy();
     let parent = source.parent().unwrap();
@@ -25,13 +21,20 @@ fn main() -> Result<()> {
     let files = find_files(&source);
 
     // Step 2. Prepare the target priority
+    match remove_dir_all(&destination) {
+        _ => {}
+    };
     create_dir_all(&destination)?;
 
-    println!("Found {} files", files.len());
-
     // Step 3. Load the layout
     let mut file_handler = FileHandler::default();
-    file_handler.get_layout_or_panic(&files);
+    match file_handler.get_layout_or_panic(&files) {
+        Ok(_) => {},
+        Err(error) => {
+            eprintln!("{}", error);
+            exit(1);
+        }
+    }
 
     // Step 4. Process all files
     file_handler.handle_all(&source, &destination, &files);