use std::collections::BTreeMap;
use femstache_eval::{TemplateEnv, TemplateInput};
use femstache_loader::{TemplateLoader, TemplateSource};
use femstache_parser::Template;
use femstache_types::{Ty, TyContext, TyError};
use crate::{
FemstacheError, FemstacheResult, builtins,
output::{EscapeMode, Output},
renderer::TemplateRenderer,
};
#[derive(Debug)]
pub struct Femstache {
loader: TemplateLoader,
escape: EscapeMode,
}
impl Femstache {
pub fn new(source: TemplateSource, escape: EscapeMode) -> Self {
let loader = TemplateLoader::new(source);
Self { loader, escape }
}
pub fn render(&self, path: &str, inputs: &impl TemplateInput) -> FemstacheResult<String> {
let template = self.loader.load(path)?;
self.render_direct(&template, inputs)
}
pub fn render_direct(
&self,
template: &Template,
inputs: &impl TemplateInput,
) -> FemstacheResult<String> {
let mut ctx = TyContext::new(self.loader.clone());
let builtins = builtins::make_builtins();
for (name, value) in &builtins {
let Some(builtin) = value.as_builtin() else {
panic!("only builtins are supported as builtins");
};
let (arg_id, ret_id) = builtin.types(ctx.store());
let fun_ty_id = ctx.store_mut().intern(Ty::Fun(arg_id, ret_id));
ctx.add_builtin(name, fun_ty_id);
}
ctx.infer_document(template.document())?;
for (name, ty_id) in ctx.inputs() {
let Some(value) = inputs.get(name) else {
return Err(FemstacheError::TypeError(TyError::VarNotFound(
name.to_owned(),
)));
};
if !value.conforms_to(ctx.store(), *ty_id) {
return Err(FemstacheError::TypeError(TyError::VarInvalidType(
name.to_owned(),
)));
}
}
let mut env = TemplateEnv::new(inputs);
for (name, value) in &builtins {
env.set(name, value.clone());
}
let mut renderer = TemplateRenderer::new(env, BTreeMap::new(), Some(self.loader.clone()));
let mut output = Output::string(self.escape);
renderer.render_document(template.document(), &mut output)?;
Ok(output.into_inner())
}
}
pub fn render(template: &Template, inputs: &impl TemplateInput) -> FemstacheResult<String> {
let femstache = Femstache::new(TemplateSource::none(), EscapeMode::None);
femstache.render_direct(template, inputs)
}