44 lines
1.4 KiB
Rust
44 lines
1.4 KiB
Rust
use async_trait::async_trait;
|
|
use axum::{Extension, Router as AxumRouter};
|
|
use fluent_templates::{ArcLoader, FluentLoader};
|
|
use loco_rs::{
|
|
app::{AppContext, Initializer},
|
|
controller::views::{engines, ViewEngine},
|
|
Error, Result,
|
|
};
|
|
use tracing::info;
|
|
|
|
const I18N_DIR: &str = "assets/i18n";
|
|
const I18N_SHARED: &str = "assets/i18n/_shared.ftl";
|
|
#[allow(clippy::module_name_repetitions)]
|
|
pub struct ViewEngineInitializer;
|
|
|
|
#[async_trait]
|
|
impl Initializer for ViewEngineInitializer {
|
|
fn name(&self) -> String {
|
|
"view-engine".to_string()
|
|
}
|
|
|
|
async fn after_routes(&self, router: AxumRouter, _ctx: &AppContext) -> Result<AxumRouter> {
|
|
let tera_engine = if std::path::Path::new(I18N_DIR).exists() {
|
|
let arc = std::sync::Arc::new(
|
|
ArcLoader::builder(&I18N_DIR, unic_langid::langid!("en-US"))
|
|
.shared_resources(Some(&[I18N_SHARED.into()]))
|
|
.customize(|bundle| bundle.set_use_isolating(false))
|
|
.build()
|
|
.map_err(|e| Error::string(&e.to_string()))?,
|
|
);
|
|
info!("locales loaded");
|
|
|
|
engines::TeraView::build_with_post_process(move |tera| {
|
|
tera.register_function("t", FluentLoader::new(arc.clone()));
|
|
Ok(())
|
|
})?
|
|
} else {
|
|
engines::TeraView::build()?
|
|
};
|
|
|
|
Ok(router.layer(Extension(ViewEngine::from(tera_engine))))
|
|
}
|
|
}
|