io.vertx.ext.web.templ.pug.impl.PugTemplateEngineImpl Maven / Gradle / Ivy
/*
* Copyright 2014 Red Hat, Inc.
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* and Apache License v2.0 which accompanies this distribution.
*
* The Eclipse Public License is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* The Apache License v2.0 is available at
* http://www.opensource.org/licenses/apache2.0.php
*
* You may elect to redistribute this code under either of these licenses.
*/
package io.vertx.ext.web.templ.pug.impl;
import de.neuland.pug4j.PugConfiguration;
import de.neuland.pug4j.template.PugTemplate;
import de.neuland.pug4j.template.TemplateLoader;
import io.vertx.core.Future;
import io.vertx.core.Vertx;
import io.vertx.core.buffer.Buffer;
import io.vertx.ext.web.common.template.CachingTemplateEngine;
import io.vertx.ext.web.common.template.CachedTemplate;
import io.vertx.ext.web.templ.pug.PugTemplateEngine;
import java.io.IOException;
import java.io.Reader;
import java.io.StringReader;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.Map;
/**
* This implementation has been copied from
*
* JadeTemplateEngineImpl.java.
* Authors of JadeTemplateEngineImpl.java are Paulo Lopes,
* Tim Fox, Julien Viet (vietj), Roman Novikov (mystdeim), nEJC (mrnejc), Yunyu Lin,
* Kevin Macksamie (k-mack), Clement Escoffier (cescoffier), Geoffrey Clements (baldmountain).
*
* For authors of this file see git history.
*/
public class PugTemplateEngineImpl extends CachingTemplateEngine implements PugTemplateEngine {
private final PugConfiguration config = new PugConfiguration();
private final Charset encoding;
/**
* Constructor that reads the template file with UTF-8 encoding.
*/
public PugTemplateEngineImpl(Vertx vertx, String extension) {
this(vertx, extension, StandardCharsets.UTF_8.name());
}
public PugTemplateEngineImpl(Vertx vertx, String extension, String encoding) {
super(vertx, extension);
config.setTemplateLoader(new PugTemplateLoader(vertx));
config.setCaching(false);
this.encoding = Charset.forName(encoding);
}
@Override
public T unwrap() {
return (T) config;
}
@Override
public Future render(Map context, String templateFile) {
try {
String src = adjustLocation(templateFile);
CachedTemplate template = getTemplate(src);
if (template == null) {
synchronized (this) {
// Compile
template = new CachedTemplate<>(config.getTemplate(src));
}
putTemplate(src, template);
}
return Future.succeededFuture(Buffer.buffer(config.renderTemplate(template.template(), context)));
} catch (Exception ex) {
return Future.failedFuture(ex);
}
}
private class PugTemplateLoader implements TemplateLoader {
private final Vertx vertx;
PugTemplateLoader(Vertx vertx) {
this.vertx = vertx;
}
@Override
public long getLastModified(String name) throws IOException {
name = adjustLocation(name);
try {
if (vertx.fileSystem().existsBlocking(name)) {
return vertx.fileSystem().propsBlocking(name).lastModifiedTime();
} else {
throw new IOException("Cannot find resource " + name);
}
} catch (RuntimeException e) {
throw new IOException("Unexpected exception", e);
}
}
@Override
public String getExtension() {
return extension;
}
@Override
public Reader getReader(String name) throws IOException {
// the internal loader will always resolve with .pug extension
name = adjustLocation(name);
String templ = null;
if (vertx.fileSystem().existsBlocking(name)) {
templ = vertx.fileSystem()
.readFileBlocking(name)
.toString(encoding);
}
if (templ == null) {
throw new IOException("Cannot find resource " + name);
}
return new StringReader(templ);
}
@Override
public String getBase() {
return "";
}
}
}