package org.groovymud.utils.text;
import groovy.lang.Binding;
import groovy.lang.GroovyRuntimeException;
import groovy.lang.GroovyShell;
import groovy.lang.Script;
import groovy.lang.Writable;
import groovy.text.Template;
import groovy.text.TemplateEngine;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.io.Reader;
import java.io.StringWriter;
import java.io.Writer;
import java.util.Map;
import org.codehaus.groovy.control.CompilationFailedException;
import org.codehaus.groovy.runtime.InvokerHelper;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
/**
* This simple template engine uses JSP <% %> script, <%= %> and <@ @>
* expression syntax. It also lets you use normal groovy expressions in the
* template text much like the new JSP EL functionality. The variable 'out' is
* bound to the writer that the template is being written to.
*
* @author sam
* @author Christian Stein
* @author Paul King
* @see http://www.antoniogoncalves.org/xwiki/bin/view/Blog/
* TemplateWithTemplatesInGroovy
*/
public class IncludeTemplateEngine extends TemplateEngine {
private boolean verbose;
private static int counter = 1;
private GroovyShell groovyShell;
public IncludeTemplateEngine() {
this(GroovyShell.class.getClassLoader());
}
public IncludeTemplateEngine(boolean verbose) {
this(GroovyShell.class.getClassLoader());
setVerbose(verbose);
}
public IncludeTemplateEngine(ClassLoader parentLoader) {
this(new GroovyShell(parentLoader));
}
public IncludeTemplateEngine(GroovyShell groovyShell) {
this.groovyShell = groovyShell;
}
public Template createTemplate(Reader reader) throws CompilationFailedException, IOException {
SimpleTemplate template = new SimpleTemplate();
String script = template.parse(reader, true);
if (verbose) {
System.out.println("\n-- script source --");
System.out.print(script);
System.out.println("\n-- script end --\n");
}
try {
template.script = groovyShell.parse(script, "SimpleTemplateScript" + counter++ + ".groovy");
} catch (Exception e) {
throw new GroovyRuntimeException("Failed to parse template script (your template may contain an error or be trying to use expressions not currently supported): " + e.getMessage());
}
return template;
}
public void setVerbose(boolean verbose) {
this.verbose = verbose;
}
public boolean isVerbose() {
return verbose;
}
private static class SimpleTemplate implements Template {
protected Script script;
public Writable make() {
return make(null);
}
public Writable make(final Map map) {
return new Writable() {
/**
* Write the template document with the set binding applied to
* the writer.
*
* @see groovy.lang.Writable#writeTo(java.io.Writer)
*/
public Writer writeTo(Writer writer) {
Binding binding;
if (map == null)
binding = new Binding();
else
binding = new Binding(map);
Script scriptObject = InvokerHelper.createScript(script.getClass(), binding);
PrintWriter pw = new PrintWriter(writer);
scriptObject.setProperty("out", pw);
scriptObject.run();
pw.flush();
return writer;
}
/**
* Convert the template and binding into a result String.
*
* @see java.lang.Object#toString()
*/
public String toString() {
StringWriter sw = new StringWriter();
writeTo(sw);
return sw.toString();
}
};
}
/**
* Parse the text document looking for <% or <%= and then call out to
* the appropriate handler, otherwise copy the text directly into the
* script while escaping quotes.
*
* @param reader
* a reader for the template text
* @return the parsed text
* @throws IOException
* if something goes wrong
*/
protected String parse(Reader reader, boolean rootTemplate) throws IOException {
if (!reader.markSupported()) {
reader = new BufferedReader(reader);
}
StringWriter sw = new StringWriter();
if (rootTemplate)
startScript(sw);
int c;
while ((c = reader.read()) != -1) {
if (c == '<') {
reader.mark(1);
c = reader.read();
if (c != '%') {
if (c == '@') { // 64
c = reader.read();
processInclude(reader, sw);
} else {
sw.write('<');
reader.reset();
}
} else {
reader.mark(1);
c = reader.read();
if (c == '=') {
groovyExpression(reader, sw);
} else {
reader.reset();
groovySection(reader, sw);
}
}
continue; // at least '<' is consumed � read next chars.
}
if (c == '$') {
reader.mark(1);
c = reader.read();
if (c != '{') {
sw.write('$');
reader.reset();
} else {
reader.mark(1);
sw.write("${");
processGSstring(reader, sw);
}
continue; // at least '$' is consumed � read next chars.
}
if (c == '\"') {
sw.write('\\');
}
/*
* Handle raw new line characters.
*/
if (c == '\n' || c == '\r') {
if (c == '\r') { // on Windows, "\r\n" is a new line.
reader.mark(1);
c = reader.read();
if (c != '\n') {
reader.reset();
}
}
sw.write("\\n\");\nout.print(\"");
continue;
}
sw.write(c);
}
endScript(sw);
return sw.toString();
}
private void startScript(StringWriter sw) {
sw.write("/* Generated by SimpleTemplateEngine */\n");
sw.write("out.print(\"");
}
private void endScript(StringWriter sw) {
sw.write("\");\n");
}
private void processInclude(Reader reader, StringWriter sw) throws IOException {
int c;
StringWriter templateName = new StringWriter();
while ((c = reader.read()) != -1) {
if (c == '@') {
c = reader.read();
if (c == '>')
break;
}
if (c != '\n' && c != '\r') {
templateName.write(c);
}
}
// using spring power :)
String script = "#an error has occured whilst resolving the include:" + templateName.toString() + "#";
ResourcePatternResolver resolver = createResourceResolver();
Resource[] groovyFiles = resolver.getResources(templateName.toString());
for (Resource resource : groovyFiles) {
SimpleTemplate template = new SimpleTemplate();
InputStreamReader isReader = new InputStreamReader(resource.getInputStream());
script = template.parse(isReader, false);
}
sw.write(script);
sw.write("out.print(\"");
}
protected ResourcePatternResolver createResourceResolver() {
return new PathMatchingResourcePatternResolver();
}
private void processGSstring(Reader reader, StringWriter sw) throws IOException {
int c;
while ((c = reader.read()) != -1) {
if (c != '\n' && c != '\r') {
sw.write(c);
}
if (c == '}') {
break;
}
}
}
/**
* Closes the currently open write and writes out the following text as
* a GString expression until it reaches an end %>.
*
* @param reader
* a reader for the template text
* @param sw
* a StringWriter to write expression content
* @throws IOException
* if something goes wrong
*/
private void groovyExpression(Reader reader, StringWriter sw) throws IOException {
sw.write("\");out.print(\"${");
int c;
while ((c = reader.read()) != -1) {
if (c == '%') {
c = reader.read();
if (c != '>') {
sw.write('%');
} else {
break;
}
}
if (c != '\n' && c != '\r') {
sw.write(c);
}
}
sw.write("}\");\nout.print(\"");
}
/**
* Closes the currently open write and writes the following text as
* normal Groovy script code until it reaches an end %>.
*
* @param reader
* a reader for the template text
* @param sw
* a StringWriter to write expression content
* @throws IOException
* if something goes wrong
*/
private void groovySection(Reader reader, StringWriter sw) throws IOException {
sw.write("\");");
int c;
while ((c = reader.read()) != -1) {
if (c == '%') {
c = reader.read();
if (c != '>') {
sw.write('%');
} else {
break;
}
}
/*
* Don't eat EOL chars in sections - as they are valid
* instruction separators. See
* http://jira.codehaus.org/browse/GROOVY-980
*/
// if (c != '\n' && c != '\r') {
sw.write(c);
// }
}
sw.write(";\nout.print(\"");
}
}
}