io.vertx.ext.web.handler.impl.TemplateHandlerImpl 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.handler.impl;
import io.vertx.core.http.HttpHeaders;
import io.vertx.core.json.JsonObject;
import io.vertx.ext.web.LanguageHeader;
import io.vertx.ext.web.handler.TemplateHandler;
import io.vertx.ext.web.impl.Utils;
import io.vertx.ext.web.common.template.TemplateEngine;
import io.vertx.ext.web.RoutingContext;
import java.util.Locale;
/**
*
* @author Tim Fox
*/
public class TemplateHandlerImpl implements TemplateHandler {
private final TemplateEngine engine;
private final String templateDirectory;
private final String contentType;
private String indexTemplate;
public TemplateHandlerImpl(TemplateEngine engine, String templateDirectory, String contentType) {
this.engine = engine;
this.templateDirectory = templateDirectory == null || templateDirectory.isEmpty() ? "." : templateDirectory;
this.contentType = contentType;
this.indexTemplate = DEFAULT_INDEX_TEMPLATE;
}
@Override
public void handle(RoutingContext context) {
String file = Utils.pathOffset(context.normalizedPath(), context);
if (file.endsWith("/") && null != indexTemplate) {
file += indexTemplate;
}
// files are always normalized (start with /)
// however if there's no base strip / to avoid making the path absolute
if (templateDirectory == null || "".equals(templateDirectory)) {
// strip the leading slash from the filename
file = file.substring(1);
}
// put the locale if present and not on the context yet into the context.
if (!context.data().containsKey("lang")) {
for (LanguageHeader acceptableLocale : context.acceptableLanguages()) {
try {
Locale.forLanguageTag(acceptableLocale.value());
} catch (RuntimeException e) {
// we couldn't parse the locale so it's not valid or unknown
continue;
}
context.data().put("lang", acceptableLocale.value());
break;
}
}
if (!context.request().isEnded()) {
context.request().pause();
}
// render using the engine
engine.render(new JsonObject(context.data()), templateDirectory + file, res -> {
if (res.succeeded()) {
if (!context.request().isEnded()) {
context.request().resume();
}
context.response().putHeader(HttpHeaders.CONTENT_TYPE, contentType).end(res.result());
} else {
if (!context.request().isEnded()) {
context.request().resume();
}
context.fail(res.cause());
}
});
}
@Override
public TemplateHandler setIndexTemplate(String indexTemplate) {
this.indexTemplate = indexTemplate;
return this;
}
}