io.honeybadger.com.github.mustachejava.DefaultMustacheFactory Maven / Gradle / Ivy
package com.github.mustachejava;
import com.github.mustachejava.codes.DefaultMustache;
import com.github.mustachejava.reflect.ReflectionObjectHandler;
import com.github.mustachejava.resolver.DefaultResolver;
import java.io.File;
import java.io.Reader;
import java.io.StringReader;
import java.io.Writer;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ExecutorService;
import java.util.function.Function;
import static com.github.mustachejava.util.HtmlEscaper.escape;
/**
* Simplest possible code factory
*/
public class DefaultMustacheFactory implements MustacheFactory {
/**
* Create the default cache for mustache compilations. This is basically
* required by the specification to handle recursive templates.
*/
protected final ConcurrentHashMap mustacheCache = createMustacheCache();
/**
* This is the default object handler.
*/
protected ObjectHandler oh = new ReflectionObjectHandler();
/**
* This parser should work with any MustacheFactory
*/
protected final MustacheParser mc = createParser();
/**
* New templates that are generated at runtime are cached here. The template key
* includes the text of the template and the context so we get proper error
* messages and debugging information.
*/
protected final ConcurrentHashMap templateCache = createLambdaCache();
protected int recursionLimit = 100;
private final MustacheResolver mustacheResolver;
protected ExecutorService es;
public DefaultMustacheFactory() {
this.mustacheResolver = new DefaultResolver();
}
public DefaultMustacheFactory(MustacheResolver mustacheResolver) {
this.mustacheResolver = mustacheResolver;
}
/**
* Use the classpath to resolve mustache templates.
*
* @param classpathResourceRoot the location in the resources where templates are stored
*/
public DefaultMustacheFactory(String classpathResourceRoot) {
this.mustacheResolver = new DefaultResolver(classpathResourceRoot);
}
/**
* Use the file system to resolve mustache templates.
*
* @param fileRoot the root of the file system where templates are stored
*/
public DefaultMustacheFactory(File fileRoot) {
this.mustacheResolver = new DefaultResolver(fileRoot);
}
/**
* Using the directory, namd and extension, resolve a partial to a name.
*
* @param dir
* @param name
* @param extension
* @return
*/
public String resolvePartialPath(String dir, String name, String extension) {
String filePath = name;
// Do not prepend directory if it is already defined
if (!name.startsWith("/")) {
filePath = dir + filePath;
}
int sepIndex = name.lastIndexOf("/");
name = sepIndex == -1 ? name : name.substring(sepIndex);
// Do not append extension if it has the same extension or original one
if (!(name.endsWith(extension) || name.contains("."))) {
filePath = filePath + extension;
}
String path = new File(filePath).getPath();
return ensureForwardSlash(path);
}
private static String ensureForwardSlash(String path) {
return path.replace('\\', '/');
}
@Override
public MustacheVisitor createMustacheVisitor() {
return new DefaultMustacheVisitor(this);
}
@Override
public Reader getReader(String resourceName) {
Reader reader = mustacheResolver.getReader(resourceName);
if (reader == null) {
throw new MustacheNotFoundException(resourceName);
}
return reader;
}
@Override
public void encode(String value, Writer writer) {
escape(value, writer);
}
@Override
public ObjectHandler getObjectHandler() {
return oh;
}
/**
* You can override the default object handler post construction.
*
* @param oh The object handler to use.
*/
public void setObjectHandler(ObjectHandler oh) {
this.oh = oh;
}
/**
* There is an ExecutorService that is used when executing parallel
* operations when a Callable is returned from a mustache value or iterable.
*
* @return the executor service
*/
public ExecutorService getExecutorService() {
return es;
}
/**
* If you need to specify your own executor service you can.
*
* @param es The executor service to use for Future evaluation
*/
public void setExecutorService(ExecutorService es) {
this.es = es;
}
public Mustache getFragment(FragmentKey templateKey) {
Mustache mustache = templateCache.computeIfAbsent(templateKey, getFragmentCacheFunction());
mustache.init();
return mustache;
}
protected Function getFragmentCacheFunction() {
return (fragmentKey) -> {
StringReader reader = new StringReader(fragmentKey.templateText);
TemplateContext tc = fragmentKey.tc;
return mc.compile(reader, tc.file(), tc.startChars(), tc.endChars(), tc.startOfLine());
};
}
@Override
public Mustache compile(String name) {
Mustache mustache = mustacheCache.computeIfAbsent(name, getMustacheCacheFunction());
mustache.init();
return mustache;
}
@Override
public Mustache compile(Reader reader, String name) {
return compile(reader, name, "{{", "}}");
}
// Template functions need this to comply with the specification
public Mustache compile(Reader reader, String file, String sm, String em) {
Mustache compile = mc.compile(reader, file, sm, em);
compile.init();
partialCache.remove();
return compile;
}
@Override
public String translate(String from) {
return from;
}
/**
* Override this method to apply any filtering to text that will appear
* verbatim in the output template.
*
* @param appended The text to be appended to the output
* @param startOfLine Are we at the start of the line?
* @return the filtered string
*/
public String filterText(String appended, boolean startOfLine) {
return appended;
}
/**
* Maximum recursion limit for partials.
*
* @param recursionLimit the number of recursions we will attempt before failing
*/
public void setRecursionLimit(int recursionLimit) {
this.recursionLimit = recursionLimit;
}
public int getRecursionLimit() {
return recursionLimit;
}
private final ThreadLocal> partialCache = ThreadLocal.withInitial(HashMap::new);
/**
* In order to handle recursion, we need a temporary thread local cache during compilation
* that is ultimately thrown away after the top level partial is complete.
*
* @param s the name of the partial to compile
* @return the compiled partial
*/
public Mustache compilePartial(String s) {
final Map cache = partialCache.get();
final Mustache cached = cache.get(s);
if (cached != null) {
// Our implementation supports this but I
// don't think it makes sense in the interface
if (cached instanceof DefaultMustache) {
((DefaultMustache)cached).setRecursive();
}
return cached;
}
try {
final Mustache mustache = mc.compile(s);
cache.put(s, mustache);
mustache.init();
return mustache;
} finally {
cache.remove(s);
}
}
protected MustacheParser createParser() {
return new MustacheParser(this);
}
protected Function getMustacheCacheFunction() {
return mc::compile;
}
protected ConcurrentHashMap createMustacheCache() {
return new ConcurrentHashMap<>();
}
protected ConcurrentHashMap createLambdaCache() {
return new ConcurrentHashMap<>();
}
}