Update main and file_finder tests

[rbdr/page] / src / main.rs

mod gemini_parser;
mod html_renderer;
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 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. 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 3. Prepare the target priority
    match remove_dir_all(&html_destination) {
        _ => {}
    };
    match remove_dir_all(&gemini_destination) {
        _ => {}
    };

    create_dir_all(&html_destination)
        .expect("Could not create HTML directory.");
    create_dir_all(&gemini_destination)
        .expect("Could not create Gemini directory.");

    // Step 4. Process all files
    file_handler.handle_all(&source, &html_destination, &gemini_destination, &files);
    Ok(())
}