krik 0.1.23

A fast static site generator written in Rust with internationalization, theming, and modern web features
Documentation 
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

use crate::parser::Document;
use crate::i18n::I18nManager;
use crate::site::SiteConfig;
use crate::theme::Theme;
use crate::error::{KrikError, KrikResult, TemplateError, TemplateErrorKind};
use chrono::{DateTime, Utc};
use std::fs::File;
use std::io::Write;
use std::path::Path;
use tera::Context;

use super::context::{add_page_links_context, add_site_context, create_post_object, is_post};
use super::paths::get_base_path;

pub fn generate_index(
    documents: &[Document],
    theme: &Theme,
    site_config: &SiteConfig,
    i18n: &I18nManager,
    output_dir: &Path,
) -> KrikResult<()> {
    let mut context = Context::new();
    add_site_context(&mut context, site_config, i18n.default_language(), "index.html");

    let site_description = "Latest posts and articles".to_string();
    context.insert("site_description", &site_description);

    // Choose one document per post base path, prefer default language if available
    let default_lang = i18n.default_language();
    use std::collections::HashMap;
    let mut chosen: HashMap<String, &Document> = HashMap::new();
    for doc in documents.iter().filter(|d| is_post(d)) {
        let base = get_base_path(std::path::Path::new(&doc.file_path));
        match chosen.get(&base) {
            None => {
                chosen.insert(base, doc);
            }
            Some(existing) => {
                // Prefer default language over non-default
                if existing.language != default_lang && doc.language == default_lang {
                    chosen.insert(base, doc);
                }
            }
        }
    }
    let mut post_docs: Vec<&Document> = chosen.values().cloned().collect();
    post_docs.sort_by(|a, b| b.front_matter.date.unwrap_or(DateTime::<Utc>::MIN_UTC).cmp(&a.front_matter.date.unwrap_or(DateTime::<Utc>::MIN_UTC)));

    let posts: Vec<std::collections::HashMap<String, serde_json::Value>> = post_docs
        .iter()
        .map(|doc| create_post_object(doc, "index.html"))
        .collect();
    context.insert("posts", &posts);

    add_page_links_context(&mut context, documents, "index.html");

    // Rendering can be parallelized if needed, but index is a single template render.
    let rendered = theme
        .templates
        .render("index.html", &context)
        .map_err(|e| KrikError::Template(TemplateError {
            kind: TemplateErrorKind::RenderError(e),
            template: "index.html".to_string(),
            context: "Rendering index page".to_string(),
        }))?;
    let index_path = output_dir.join("index.html");
    let mut file = File::create(&index_path)?;
    file.write_all(rendered.as_bytes())?;
    Ok(())
}