src/main.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

mod file_finder;
mod file_handler;

use std::env::current_dir;
use std::fs::{create_dir_all, remove_dir_all};
use std::io::{Error, ErrorKind, Result};

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

fn main() -> Result<()> {
    let source = current_dir()?;
    let source_name = source
        .file_name()
        .ok_or_else(|| {
            Error::new(
                ErrorKind::InvalidData,
                "Could not read current directory name.",
            )
        })?
        .to_string_lossy();
    let parent = source.parent().ok_or_else(|| {
        Error::new(
            ErrorKind::InvalidData,
            "Current directory parent was not readable.",
        )
    })?;
    let gemini_destination_name = format!("{source_name}_gemini");
    let gemini_destination = parent.join(gemini_destination_name);
    let html_destination_name = format!("{source_name}_html");
    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();
    file_handler.get_layout(&files)?;

    // Step 3. Prepare the target priority
    let _ = remove_dir_all(&html_destination);
    let _ = remove_dir_all(&gemini_destination);

    create_dir_all(&html_destination)
        .map_err(|e| Error::new(e.kind(), "Could not create HTML directory."))?;
    create_dir_all(&gemini_destination)
        .map_err(|e| Error::new(e.kind(), "Could not create Gemini directory."))?;

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