Handle layout error better

[rbdr/page] / src / main.rs

mod gemini_parser;
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, remove_dir_all};

use crate::file_finder::find_files;
use crate::file_handler::FileHandler;

fn main() -> Result<()> {
    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)?;

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