mirror of https://github.com/realaravinth/gitpad
feat: embed templates in binary
parent
d0dbfd9729
commit
551a48d0c5
|
@ -63,8 +63,7 @@ lazy_static! {
|
|||
async fn main() -> std::io::Result<()> {
|
||||
let settings = Settings::new().unwrap();
|
||||
pretty_env_logger::init();
|
||||
pages::init(&settings);
|
||||
|
||||
lazy_static::initialize(&pages::TEMPLATES);
|
||||
info!(
|
||||
"{}: {}.\nFor more information, see: {}\nBuild info:\nVersion: {} commit: {}",
|
||||
PKG_NAME,
|
||||
|
|
|
@ -16,6 +16,7 @@
|
|||
*/
|
||||
use actix_web::*;
|
||||
use lazy_static::lazy_static;
|
||||
use rust_embed::RustEmbed;
|
||||
use serde::*;
|
||||
use tera::*;
|
||||
|
||||
|
@ -30,19 +31,40 @@ pub mod routes;
|
|||
pub use routes::get_auth_middleware;
|
||||
pub use routes::PAGES;
|
||||
|
||||
pub struct TemplateFile {
|
||||
pub name: &'static str,
|
||||
pub path: &'static str,
|
||||
}
|
||||
|
||||
impl TemplateFile {
|
||||
pub const fn new(name: &'static str, path: &'static str) -> Self {
|
||||
Self { name, path }
|
||||
}
|
||||
|
||||
pub fn register(&self, t: &mut Tera) -> std::result::Result<(), tera::Error> {
|
||||
t.add_raw_template(self.name, &Templates::get_template(self).expect(self.name))
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
#[allow(dead_code)]
|
||||
pub fn register_from_file(&self, t: &mut Tera) -> std::result::Result<(), tera::Error> {
|
||||
use std::path::Path;
|
||||
t.add_template_file(Path::new("templates/").join(self.path), Some(self.name))
|
||||
}
|
||||
}
|
||||
|
||||
pub const PAYLOAD_KEY: &str = "payload";
|
||||
|
||||
pub const BASE: TemplateFile = TemplateFile::new("base", "components/base.html");
|
||||
pub const FOOTER: TemplateFile = TemplateFile::new("footer", "components/footer.html");
|
||||
pub const PUB_NAV: TemplateFile = TemplateFile::new("pub_nav", "components/pub-nav.html");
|
||||
|
||||
lazy_static! {
|
||||
pub static ref TEMPLATES: Tera = {
|
||||
let mut tera = Tera::default();
|
||||
if let Err(e) = tera.add_template_files(vec![
|
||||
("templates/components/base.html", Some("base")),
|
||||
("templates/components/footer.html", Some("footer")),
|
||||
("templates/components/nav.html", Some("nav")),
|
||||
]) {
|
||||
println!("Parsing error(s): {}", e);
|
||||
::std::process::exit(1);
|
||||
};
|
||||
for t in [BASE, FOOTER, PUB_NAV].iter() {
|
||||
t.register(&mut tera).unwrap();
|
||||
}
|
||||
errors::register_templates(&mut tera);
|
||||
tera.autoescape_on(vec![".html", ".sql"]);
|
||||
auth::register_templates(&mut tera);
|
||||
|
@ -50,6 +72,19 @@ lazy_static! {
|
|||
};
|
||||
}
|
||||
|
||||
#[derive(RustEmbed)]
|
||||
#[folder = "templates/"]
|
||||
pub struct Templates;
|
||||
|
||||
impl Templates {
|
||||
pub fn get_template(t: &TemplateFile) -> Option<String> {
|
||||
match Self::get(t.path) {
|
||||
Some(file) => Some(String::from_utf8_lossy(&file.data).into_owned()),
|
||||
None => None,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn context(s: &Settings) -> Context {
|
||||
let mut ctx = Context::new();
|
||||
let footer = Footer::new(s);
|
||||
|
@ -59,11 +94,6 @@ pub fn context(s: &Settings) -> Context {
|
|||
ctx
|
||||
}
|
||||
|
||||
pub fn init(s: &Settings) {
|
||||
auth::login::Login::page(s);
|
||||
auth::register::Register::page(s);
|
||||
}
|
||||
|
||||
#[derive(Serialize)]
|
||||
pub struct Footer<'a> {
|
||||
version: &'a str,
|
||||
|
@ -95,12 +125,23 @@ pub async fn home() -> impl Responder {
|
|||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::{init, Settings};
|
||||
use super::{auth, errors, BASE, FOOTER, PUB_NAV};
|
||||
use tera::Tera;
|
||||
|
||||
#[test]
|
||||
fn templates_work() {
|
||||
let settings = Settings::new().unwrap();
|
||||
init(&settings);
|
||||
let mut tera = Tera::default();
|
||||
let mut tera2 = Tera::default();
|
||||
for t in [
|
||||
BASE,
|
||||
FOOTER,
|
||||
PUB_NAV,
|
||||
]
|
||||
.iter()
|
||||
{
|
||||
t.register_from_file(&mut tera2).unwrap();
|
||||
t.register(&mut tera).unwrap();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue