Add static file copying

[rbdr/page] / src / main.rs

diff --git a/src/main.rs b/src/main.rs
index 4b8a79f42a71cc030c5658c200f7fc1fca1e15b0..7ab009e862e9b7fd6c34d81244076addbfbff2fe 100644 (file)
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,16 +2,41 @@ mod gemini_parser;
 mod file_finder;
 mod file_handler;
 
+use std::io::Result;
 use std::env::current_dir;
+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() {
+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 files = find_files(current_dir().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);
+
+    // 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)?;
+
     println!("Found {} files", files.len());
+
+    // Step 3. Load the layout
+    let mut file_handler = FileHandler::default();
+    file_handler.get_layout_or_panic(&files);
+
+    // Step 4. Process all files
+    file_handler.handle_all(&source, &destination, &files);
+    Ok(())
 }