git.r.bdr.sh - rbdr/page/blob - src/main.rs

   1 mod file_finder;
   2 mod file_handler;
   3 mod gemini_parser;
   4 mod html_renderer;
   5
   6 use std::env::current_dir;
   7 use std::fs::{create_dir_all, remove_dir_all};
   8 use std::io::Result;
   9 use std::process::exit;
  10
  11 use crate::file_finder::find_files;
  12 use crate::file_handler::FileHandler;
  13
  14 fn main() -> Result<()> {
  15     let source = current_dir()?;
  16     let source_name = source.file_name().unwrap().to_string_lossy();
  17     let parent = source.parent().unwrap();
  18     let gemini_destination_name = format!("{}_gemini", source_name);
  19     let gemini_destination = parent.join(gemini_destination_name);
  20     let html_destination_name = format!("{}_html", source_name);
  21     let html_destination = parent.join(html_destination_name);
  22
  23     // Step 1. Identify the files
  24     let files = find_files(&source);
  25
  26     // Step 2. Load the layout
  27     let mut file_handler = FileHandler::default();
  28     match file_handler.get_layout_or_panic(&files) {
  29         Ok(_) => {}
  30         Err(error) => {
  31             eprintln!("{}", error);
  32             exit(1);
  33         }
  34     }
  35
  36     // Step 3. Prepare the target priority
  37     let _ = remove_dir_all(&html_destination);
  38     let _ = remove_dir_all(&gemini_destination);
  39
  40     create_dir_all(&html_destination).expect("Could not create HTML directory.");
  41     create_dir_all(&gemini_destination).expect("Could not create Gemini directory.");
  42
  43     // Step 4. Process all files
  44     file_handler.handle_all(&source, &html_destination, &gemini_destination, &files);
  45     Ok(())
  46 }