marmite 0.4.2

Marmite is the easiest static site generator.
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

#![allow(clippy::module_name_repetitions)]
use crate::embedded::{generate_static, Templates, ThemeTemplate};
use log::{error, info};
use std::fs;
use std::path::Path;

/// Creates the `templates/` folder and writes embedded templates to it.
pub fn initialize_templates(input_folder: &Path) {
    let templates_path = input_folder.join("templates");
    if let Err(e) = fs::create_dir(&templates_path) {
        error!("Failed to create templates directory: {e:?}");
        return;
    }

    for name in Templates::iter() {
        if let Some(template) = Templates::get(name.as_ref()) {
            if let Ok(template_str) = std::str::from_utf8(template.data.as_ref()) {
                let template_path = templates_path.join(name.as_ref());
                if let Err(e) = fs::write(&template_path, template_str) {
                    error!("Failed to write template '{}': {e:?}", name.as_ref());
                } else {
                    info!("Generated {}", template_path.display());
                }
            } else {
                error!("Failed to parse template '{}' as UTF-8", name.as_ref());
            }
        } else {
            error!(
                "Template '{}' not found in embedded resources",
                name.as_ref()
            );
        }
    }
}

/// Uses `generate_static` to populate the `static/` folder with embedded content.
#[allow(dead_code)]
pub fn initialize_theme_assets(input_folder: &Path) {
    let static_path = input_folder.join("static");

    if let Err(e) = fs::create_dir(&static_path) {
        error!("Failed to create static directory: {e:?}");
        return;
    }
    generate_static(&static_path);
}

/// Creates a new theme with templates and static assets from the embedded theme template
pub fn initialize_theme(input_folder: &Path, theme_name: &str) {
    // Validate theme name
    if theme_name.is_empty() || theme_name.contains('/') || theme_name.contains('\\') {
        error!("Invalid theme name: '{theme_name}'");
        return;
    }

    let theme_path = input_folder.join(theme_name);

    // Check if theme directory already exists
    if theme_path.exists() {
        error!("Theme directory already exists: {}", theme_path.display());
        return;
    }

    // Create theme directory structure
    let templates_path = theme_path.join("templates");
    let static_path = theme_path.join("static");

    if let Err(e) = fs::create_dir_all(&templates_path) {
        error!("Failed to create theme templates directory: {e:?}");
        return;
    }

    if let Err(e) = fs::create_dir_all(&static_path) {
        error!("Failed to create theme static directory: {e:?}");
        return;
    }

    // Copy theme template files from embedded ThemeTemplate
    for name in ThemeTemplate::iter() {
        if let Some(file) = ThemeTemplate::get(name.as_ref()) {
            let file_path = Path::new(name.as_ref());
            let dest_path = theme_path.join(file_path);

            // Ensure parent directory exists
            if let Some(parent) = dest_path.parent() {
                if let Err(e) = fs::create_dir_all(parent) {
                    error!("Failed to create directory {}: {e:?}", parent.display());
                    continue;
                }
            }

            // Write file data
            if let Err(e) = fs::write(&dest_path, file.data.as_ref()) {
                error!("Failed to copy theme file '{}': {e:?}", name.as_ref());
            } else {
                info!("Generated {}", dest_path.display());
            }
        } else {
            error!(
                "Theme template file '{}' not found in embedded resources",
                name.as_ref()
            );
        }
    }

    info!(
        "Theme '{}' created successfully at {}",
        theme_name,
        theme_path.display()
    );
    info!("To use this theme, add 'theme: {theme_name}' to your marmite.yaml config file");
}

#[cfg(test)]
#[path = "tests/templates.rs"]
mod tests;