poet 0.7.0

Static site generator with optional MCP server and LLM SEO optimizations
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
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

use std::path::PathBuf;
use std::sync::Arc;

use anyhow::Result;
use rhai::Engine;
use rhai::module_resolvers::FileModuleResolver;
use rhai_components::builds_engine::BuildsEngine;
use rhai_components::component_syntax::component_reference::ComponentReference;
use rhai_components::component_syntax::component_registry::ComponentRegistry;
use rhai_components::rhai_template_renderer::RhaiTemplateRenderer;
use rhai_components::rhai_template_renderer_params::RhaiTemplateRendererParams;

use crate::asset_manager::AssetManager;
use crate::author::Author;
use crate::author_collection::AuthorCollection;
use crate::author_data::AuthorData;
use crate::content_document_collection_ranked::ContentDocumentCollectionRanked;
use crate::content_document_component_context::ContentDocumentComponentContext;
use crate::content_document_front_matter::ContentDocumentFrontMatter;
use crate::content_document_hierarchy::ContentDocumentHierarchy;
use crate::content_document_reference::ContentDocumentReference;
use crate::content_document_tree_node::ContentDocumentTreeNode;
use crate::filesystem::file_entry::FileEntry;
use crate::prompt_document_component_context::PromptDocumentComponentContext;
use crate::prompt_document_front_matter::PromptDocumentFrontMatter;
use crate::prompt_document_front_matter::argument_with_input::ArgumentWithInput;
use crate::rhai_helpers::render_hierarchy;
use crate::table_of_contents::TableOfContents;
use crate::table_of_contents::heading::Heading;

pub struct RhaiTemplateRendererFactory {
    base_directory: PathBuf,
    component_registry: Arc<ComponentRegistry>,
    shortcodes_subdirectory: PathBuf,
}

impl RhaiTemplateRendererFactory {
    pub fn new(base_directory: PathBuf, shortcodes_subdirectory: PathBuf) -> Self {
        Self {
            base_directory,
            component_registry: Default::default(),
            shortcodes_subdirectory,
        }
    }

    pub fn register_component_file(&self, file_entry: FileEntry) {
        let component_name = file_entry.get_stem_relative_to(&self.shortcodes_subdirectory);

        self.component_registry
            .register_component(ComponentReference {
                name: component_name.clone(),
                path: component_name,
            });
    }
}

impl BuildsEngine for RhaiTemplateRendererFactory {
    fn component_registry(&self) -> Arc<ComponentRegistry> {
        self.component_registry.clone()
    }

    fn prepare_engine(&self, engine: &mut Engine) -> Result<()> {
        engine.set_module_resolver(FileModuleResolver::new_with_path(
            self.base_directory.join(&self.shortcodes_subdirectory),
        ));

        engine.build_type::<ArgumentWithInput>();
        engine.build_type::<AssetManager>();
        engine.build_type::<Author>();
        engine.build_type::<AuthorCollection>();
        engine.build_type::<AuthorData>();
        engine.build_type::<ContentDocumentCollectionRanked>();
        engine.build_type::<ContentDocumentComponentContext>();
        engine.build_type::<ContentDocumentFrontMatter>();
        engine.build_type::<ContentDocumentHierarchy>();
        engine.build_type::<ContentDocumentReference>();
        engine.build_type::<ContentDocumentTreeNode>();
        engine.build_type::<FileEntry>();
        engine.build_type::<Heading>();
        engine.build_type::<PromptDocumentComponentContext>();
        engine.build_type::<PromptDocumentFrontMatter>();
        engine.build_type::<TableOfContents>();

        engine.register_fn("render_hierarchy", render_hierarchy);

        Ok(())
    }
}

impl TryInto<RhaiTemplateRenderer> for RhaiTemplateRendererFactory {
    type Error = anyhow::Error;

    fn try_into(self) -> Result<RhaiTemplateRenderer, Self::Error> {
        let expression_engine = self.create_engine()?;

        RhaiTemplateRenderer::build(RhaiTemplateRendererParams {
            component_registry: self.component_registry,
            expression_engine,
        })
    }
}