Make smaller

[rbdr/page] / src / main.rs

diff --git a/src/main.rs b/src/main.rs
index fdfe8630e51c4989c7749568c8101148c53b5193..fc085d8a39d478970b6a5e1f09222994f530784b 100644 (file)
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,6 +3,7 @@ mod file_finder;
 mod file_handler;
 
 use std::io::Result;
 mod file_handler;
 
 use std::io::Result;

+use std::process::exit;

 use std::env::current_dir;
 use std::fs::{create_dir_all, remove_dir_all};
 
 use std::env::current_dir;
 use std::fs::{create_dir_all, remove_dir_all};
 


@@ -13,23 +14,35 @@ fn main() -> Result<()> {
     let source = current_dir()?;
     let source_name = source.file_name().unwrap().to_string_lossy();
     let parent = source.parent().unwrap();
     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
 
     // Step 1. Identify the files
     let files = find_files(&source);
 
     // Step 2. Prepare the target priority

-    match remove_dir_all(&destination) {
+    match remove_dir_all(&html_destination) {

         _ => {}
     };
         _ => {}
     };

-    create_dir_all(&destination)?;
+    create_dir_all(&html_destination)?;
+    match remove_dir_all(&gemini_destination) {
+        _ => {}
+    };
+    create_dir_all(&gemini_destination)?;

 
     // Step 3. Load the layout
     let mut file_handler = FileHandler::default();
 
     // 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
 
     // Step 4. Process all files

-    file_handler.handle_all(&source, &destination, &files);
+    file_handler.handle_all(&source, &html_destination, &gemini_destination, &files);

     Ok(())
 }
     Ok(())
 }