com.aspectran.pebble.PebbleEngineFactory Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2023 The Aspectran Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.aspectran.pebble;
import com.aspectran.core.adapter.ApplicationAdapter;
import com.aspectran.core.component.bean.aware.ApplicationAdapterAware;
import com.aspectran.core.util.LocaleUtils;
import com.aspectran.core.util.ResourceUtils;
import com.aspectran.core.util.logging.Logger;
import com.aspectran.core.util.logging.LoggerFactory;
import io.pebbletemplates.pebble.PebbleEngine;
import io.pebbletemplates.pebble.loader.ClasspathLoader;
import io.pebbletemplates.pebble.loader.DelegatingLoader;
import io.pebbletemplates.pebble.loader.FileLoader;
import io.pebbletemplates.pebble.loader.Loader;
import io.pebbletemplates.pebble.loader.StringLoader;
import java.io.File;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Locale;
/**
* Factory that configures a Pebble Engine Configuration.
*
* Created: 2016. 1. 9.
*/
public class PebbleEngineFactory implements ApplicationAdapterAware {
private static final Logger logger = LoggerFactory.getLogger(PebbleEngineFactory.class);
private ApplicationAdapter applicationAdapter;
private Locale defaultLocale;
private boolean strictVariables;
private String[] templateLoaderPaths;
private Loader[] templateLoaders;
@Override
public void setApplicationAdapter(ApplicationAdapter applicationAdapter) {
this.applicationAdapter = applicationAdapter;
}
public void setDefaultLocale(String defaultLocale) {
this.defaultLocale = LocaleUtils.parseLocaleString(defaultLocale);
}
public void setStrictVariables(boolean strictVariables) {
this.strictVariables = strictVariables;
}
public void setTemplateLoaderPath(String templateLoaderPath) {
this.templateLoaderPaths = new String[] { templateLoaderPath };
}
public void setTemplateLoaderPath(String[] templateLoaderPaths) {
this.templateLoaderPaths = templateLoaderPaths;
}
public void setTemplateLoaderPath(List templateLoaderPathList) {
this.templateLoaderPaths = templateLoaderPathList.toArray(new String[0]);
}
public void setTemplateLoader(Loader templateLoaders) {
this.templateLoaders = new Loader[] {templateLoaders};
}
public void setTemplateLoader(Loader[] templateLoaders) {
this.templateLoaders = templateLoaders;
}
public void setTemplateLoader(List> templateLoaderList) {
this.templateLoaders = templateLoaderList.toArray(new Loader[0]);
}
/**
* Creates a PebbleEngine instance.
* @return a PebbleEngine object that can be used to create PebbleTemplate objects
*/
public PebbleEngine createPebbleEngine() {
PebbleEngine.Builder builder = new PebbleEngine.Builder();
builder.strictVariables(strictVariables);
if (defaultLocale != null) {
builder.defaultLocale(defaultLocale);
}
if (templateLoaders == null) {
if (templateLoaderPaths != null && templateLoaderPaths.length > 0) {
List> templateLoaderList = new ArrayList<>();
for (String path : templateLoaderPaths) {
templateLoaderList.add(getTemplateLoaderForPath(path));
}
setTemplateLoader(templateLoaderList);
}
}
Loader templateLoader = getAggregateTemplateLoader(templateLoaders);
builder.loader(templateLoader);
return builder.build();
}
/**
* Return a Template Loader based on the given Template Loader list.
* If more than one Template Loader has been registered, a DelegatingLoader needs to be created.
* @param templateLoaders the final List of TemplateLoader instances
* @return the aggregate TemplateLoader
*/
protected Loader getAggregateTemplateLoader(Loader[] templateLoaders) {
int loaderCount = (templateLoaders == null) ? 0 : templateLoaders.length;
switch (loaderCount) {
case 0:
// Register default template loaders.
Loader stringLoader = new StringLoader();
if (logger.isDebugEnabled()) {
logger.debug("Pebble Engine Template Loader not specified. Default Template Loader registered: " + stringLoader);
}
return stringLoader;
case 1:
if (logger.isDebugEnabled()) {
logger.debug("One Pebble Engine Template Loader registered: " + templateLoaders[0]);
}
return templateLoaders[0];
default:
List> defaultLoadingStrategies = new ArrayList<>();
Collections.addAll(defaultLoadingStrategies, templateLoaders);
Loader delegatingLoader = new DelegatingLoader(defaultLoadingStrategies);
if (logger.isDebugEnabled()) {
logger.debug("Multiple Pebble Engine Template Loader registered: " + delegatingLoader);
}
return delegatingLoader;
}
}
/**
* Determine a Pebble Engine Template Loader for the given path.
* @param templateLoaderPath the path to load templates from
* @return an appropriate Template Loader
*/
protected Loader getTemplateLoaderForPath(String templateLoaderPath) {
if (templateLoaderPath.startsWith(ResourceUtils.CLASSPATH_URL_PREFIX)) {
String basePackagePath = templateLoaderPath.substring(ResourceUtils.CLASSPATH_URL_PREFIX.length());
if (logger.isDebugEnabled()) {
logger.debug("Template loader path [" + templateLoaderPath + "] resolved to class path [" + basePackagePath + "]");
}
ClasspathLoader loader = new ClasspathLoader(applicationAdapter.getClassLoader());
loader.setPrefix(basePackagePath);
return loader;
} else if (templateLoaderPath.startsWith(ResourceUtils.FILE_URL_PREFIX)) {
File file = new File(templateLoaderPath.substring(ResourceUtils.FILE_URL_PREFIX.length()));
String prefix = file.getAbsolutePath();
if (logger.isDebugEnabled()) {
logger.debug("Template loader path [" + templateLoaderPath + "] resolved to file path [" + prefix + "]");
}
FileLoader loader = new FileLoader();
loader.setPrefix(prefix);
return loader;
} else {
File file = new File(applicationAdapter.getBasePath(), templateLoaderPath);
String prefix = file.getAbsolutePath();
if (logger.isDebugEnabled()) {
logger.debug("Template loader path [" + templateLoaderPath + "] resolved to file path [" + prefix + "]");
}
FileLoader loader = new FileLoader();
loader.setPrefix(prefix);
return loader;
}
}
}