All Downloads are FREE. Search and download functionalities are using the official Maven repository.

freemarker.core.Environment Maven / Gradle / Ivy

There is a newer version: 0.4.3
Show newest version
/*
 * Copyright (c) 2003 The Visigoth Software Society. All rights
 * reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * 3. The end-user documentation included with the redistribution, if
 *    any, must include the following acknowledgement:
 *       "This product includes software developed by the
 *        Visigoth Software Society (http://www.visigoths.org/)."
 *    Alternately, this acknowledgement may appear in the software itself,
 *    if and wherever such third-party acknowledgements normally appear.
 *
 * 4. Neither the name "FreeMarker", "Visigoth", nor any of the names of the
 *    project contributors may be used to endorse or promote products derived
 *    from this software without prior written permission. For written
 *    permission, please contact [email protected].
 *
 * 5. Products derived from this software may not be called "FreeMarker" or "Visigoth"
 *    nor may "FreeMarker" or "Visigoth" appear in their names
 *    without prior written permission of the Visigoth Software Society.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED.  IN NO EVENT SHALL THE VISIGOTH SOFTWARE SOCIETY OR
 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 * ====================================================================
 *
 * This software consists of voluntary contributions made by many
 * individuals on behalf of the Visigoth Software Society. For more
 * information on the Visigoth Software Society, please see
 * http://www.visigoths.org/
 */

package freemarker.core;

import java.io.*;
import java.text.*;
import java.util.*;

import freemarker.ext.beans.BeansWrapper;
import freemarker.log.Logger;
import freemarker.template.*;
import freemarker.template.utility.UndeclaredThrowableException;

/**
 * Object that represents the runtime environment during template processing.
 * For every invocation of a Template.process() method, a new instance
 * of this object is created, and then discarded when process() returns.
 * This object stores the set of temporary variables created by the template,
 * the value of settings set by the template, the reference to the data model root,
 * etc. Everything that is needed to fulfill the template processing job.
 *
 * 

Data models that need to access the Environment * object that represents the template processing on the current thread can use * the {@link #getCurrentEnvironment()} method. * *

If you need to modify or read this object before or after the process * call, use {@link Template#createProcessingEnvironment(Object rootMap, Writer out, ObjectWrapper wrapper)} * * @author Jonathan Revusky * @author Attila Szegedi */ public final class Environment extends Configurable { private static final ThreadLocal threadEnv = new ThreadLocal(); private static final Logger logger = Logger.getLogger("freemarker.runtime"); private static final Map localizedNumberFormats = new HashMap(); private static final Map localizedDateFormats = new HashMap(); // Do not use this object directly; clone it first! DecimalFormat isn't // thread-safe. private static final DecimalFormat C_NUMBER_FORMAT = new DecimalFormat( "0.################", new DecimalFormatSymbols(Locale.US)); static { C_NUMBER_FORMAT.setGroupingUsed(false); C_NUMBER_FORMAT.setDecimalSeparatorAlwaysShown(false); } private final TemplateHashModel rootDataModel; private final ArrayList elementStack = new ArrayList(); private final ArrayList recoveredErrorStack = new ArrayList(); private NumberFormat numberFormat; private Map numberFormats; private DateFormat timeFormat, dateFormat, dateTimeFormat; private Map[] dateFormats; private NumberFormat cNumberFormat; private Collator collator; private Writer out; private Macro.Context currentMacroContext; private ArrayList localContextStack; private Namespace mainNamespace, currentNamespace, globalNamespace; private HashMap loadedLibs; private Throwable lastThrowable; private TemplateModel lastReturnValue; private HashMap macroToNamespaceLookup = new HashMap(); private TemplateNodeModel currentVisitorNode; private TemplateSequenceModel nodeNamespaces; // Things we keep track of for the fallback mechanism. private int nodeNamespaceIndex; private String currentNodeName, currentNodeNS; private String cachedURLEscapingCharset; private boolean urlEscapingCharsetCached; /** * Retrieves the environment object associated with the current * thread. Data model implementations that need access to the * environment can call this method to obtain the environment object * that represents the template processing that is currently running * on the current thread. */ public static Environment getCurrentEnvironment() { return (Environment)threadEnv.get(); } public Environment(Template template, final TemplateHashModel rootDataModel, Writer out) { super(template); this.globalNamespace = new Namespace(null); this.currentNamespace = mainNamespace = new Namespace(template); this.out = out; this.rootDataModel = rootDataModel; importMacros(template); } /** * Retrieves the currently processed template. */ public Template getTemplate() { return (Template)getParent(); } /** * Deletes cached values that meant to be valid only during a single * template execution. */ private void clearCachedValues() { numberFormats = null; numberFormat = null; dateFormats = null; collator = null; cachedURLEscapingCharset = null; urlEscapingCharsetCached = false; } /** * Processes the template to which this environment belongs. */ public void process() throws TemplateException, IOException { Object savedEnv = threadEnv.get(); threadEnv.set(this); try { // Cached values from a previous execution are possibly outdated. clearCachedValues(); try { doAutoImportsAndIncludes(this); visit(getTemplate().getRootTreeNode()); // Do not flush if there was an exception. out.flush(); } finally { // It's just to allow the GC to free memory... clearCachedValues(); } } finally { threadEnv.set(savedEnv); } } /** * "Visit" the template element. */ void visit(TemplateElement element) throws TemplateException, IOException { pushElement(element); try { element.accept(this); } catch (TemplateException te) { handleTemplateException(te); } finally { popElement(); } } private static final TemplateModel[] NO_OUT_ARGS = new TemplateModel[0]; public void visit(final TemplateElement element, TemplateDirectiveModel directiveModel, Map args, final List bodyParameterNames) throws TemplateException, IOException { TemplateDirectiveBody nested; if(element == null) { nested = null; } else { nested = new TemplateDirectiveBody() { public void render(Writer newOut) throws TemplateException, IOException { Writer prevOut = out; out = newOut; try { Environment.this.visit(element); } finally { out = prevOut; } } }; } final TemplateModel[] outArgs; if(bodyParameterNames == null || bodyParameterNames.isEmpty()) { outArgs = NO_OUT_ARGS; } else { outArgs = new TemplateModel[bodyParameterNames.size()]; } if(outArgs.length > 0) { pushLocalContext(new LocalContext() { public TemplateModel getLocalVariable(String name) { int index = bodyParameterNames.indexOf(name); return index != -1 ? outArgs[index] : null; } public Collection getLocalVariableNames() { return bodyParameterNames; } }); } try { directiveModel.execute(this, args, outArgs, nested); } finally { if(outArgs.length > 0) { popLocalContext(); } } } /** * "Visit" the template element, passing the output * through a TemplateTransformModel * @param element the element to visit through a transform * @param transform the transform to pass the element output * through * @param args optional arguments fed to the transform */ void visit(TemplateElement element, TemplateTransformModel transform, Map args) throws TemplateException, IOException { try { Writer tw = transform.getWriter(out, args); if (tw == null) tw = EMPTY_BODY_WRITER; TransformControl tc = tw instanceof TransformControl ? (TransformControl)tw : null; Writer prevOut = out; out = tw; try { if(tc == null || tc.onStart() != TransformControl.SKIP_BODY) { do { if(element != null) { visit(element); } } while(tc != null && tc.afterBody() == TransformControl.REPEAT_EVALUATION); } } catch(Throwable t) { try { if(tc != null) { tc.onError(t); } else { throw t; } } catch(TemplateException e) { throw e; } catch(IOException e) { throw e; } catch(RuntimeException e) { throw e; } catch(Error e) { throw e; } catch(Throwable e) { throw new UndeclaredThrowableException(e); } } finally { out = prevOut; tw.close(); } } catch(TemplateException te) { handleTemplateException(te); } } /** * Visit a block using buffering/recovery */ void visit(TemplateElement attemptBlock, TemplateElement recoveryBlock) throws TemplateException, IOException { Writer prevOut = this.out; StringWriter sw = new StringWriter(); this.out = sw; TemplateException thrownException = null; try { visit(attemptBlock); } catch (TemplateException te) { thrownException = te; } finally { this.out = prevOut; } if (thrownException != null) { if (logger.isErrorEnabled()) { String msg = "Error in attempt block " + attemptBlock.getStartLocation(); logger.error(msg, thrownException); } try { recoveredErrorStack.add(thrownException.getMessage()); visit(recoveryBlock); } finally { recoveredErrorStack.remove(recoveredErrorStack.size() -1); } } else { out.write(sw.toString()); } } String getCurrentRecoveredErrorMesssage() throws TemplateException { if(recoveredErrorStack.isEmpty()) { throw new TemplateException( ".error is not available outside of a <#recover> block", this); } return (String) recoveredErrorStack.get(recoveredErrorStack.size() -1); } void visit(BodyInstruction.Context bctxt) throws TemplateException, IOException { Macro.Context invokingMacroContext = getCurrentMacroContext(); ArrayList prevLocalContextStack = localContextStack; TemplateElement body = invokingMacroContext.body; if (body != null) { this.currentMacroContext = invokingMacroContext.prevMacroContext; currentNamespace = invokingMacroContext.bodyNamespace; Configurable prevParent = getParent(); setParent(currentNamespace.getTemplate()); this.localContextStack = invokingMacroContext.prevLocalContextStack; if (invokingMacroContext.bodyParameterNames != null) { pushLocalContext(bctxt); } try { visit(body); } finally { if (invokingMacroContext.bodyParameterNames != null) { popLocalContext(); } this.currentMacroContext = invokingMacroContext; currentNamespace = getMacroNamespace(invokingMacroContext.getMacro()); setParent(prevParent); this.localContextStack = prevLocalContextStack; } } } /** * "visit" an IteratorBlock */ void visit(IteratorBlock.Context ictxt) throws TemplateException, IOException { pushLocalContext(ictxt); try { ictxt.runLoop(this); } catch (BreakInstruction.Break br) { } catch (TemplateException te) { handleTemplateException(te); } finally { popLocalContext(); } } /** * "Visit" A TemplateNodeModel */ void visit(TemplateNodeModel node, TemplateSequenceModel namespaces) throws TemplateException, IOException { if (nodeNamespaces == null) { SimpleSequence ss = new SimpleSequence(1); ss.add(currentNamespace); nodeNamespaces = ss; } int prevNodeNamespaceIndex = this.nodeNamespaceIndex; String prevNodeName = this.currentNodeName; String prevNodeNS = this.currentNodeNS; TemplateSequenceModel prevNodeNamespaces = nodeNamespaces; TemplateNodeModel prevVisitorNode = currentVisitorNode; currentVisitorNode = node; if (namespaces != null) { this.nodeNamespaces = namespaces; } try { TemplateModel macroOrTransform = getNodeProcessor(node); if (macroOrTransform instanceof Macro) { visit((Macro) macroOrTransform, null, null, null, null); } else if (macroOrTransform instanceof TemplateTransformModel) { visit(null, (TemplateTransformModel) macroOrTransform, null); } else { String nodeType = node.getNodeType(); if (nodeType != null) { // If the node's type is 'text', we just output it. if ((nodeType.equals("text") && node instanceof TemplateScalarModel)) { out.write(((TemplateScalarModel) node).getAsString()); } else if (nodeType.equals("document")) { recurse(node, namespaces); } // We complain here, unless the node's type is 'pi', or "comment" or "document_type", in which case // we just ignore it. else if (!nodeType.equals("pi") && !nodeType.equals("comment") && !nodeType.equals("document_type")) { String nsBit = ""; String ns = node.getNodeNamespace(); if (ns != null) { if (ns.length() >0) { nsBit = " and namespace " + ns; } else { nsBit = " and no namespace"; } } throw new TemplateException("No macro or transform defined for node named " + node.getNodeName() + nsBit + ", and there is no fallback handler called @" + nodeType + " either.", this); } } else { String nsBit = ""; String ns = node.getNodeNamespace(); if (ns != null) { if (ns.length() >0) { nsBit = " and namespace " + ns; } else { nsBit = " and no namespace"; } } throw new TemplateException("No macro or transform defined for node with name " + node.getNodeName() + nsBit + ", and there is no macro or transform called @default either.", this); } } } finally { this.currentVisitorNode = prevVisitorNode; this.nodeNamespaceIndex = prevNodeNamespaceIndex; this.currentNodeName = prevNodeName; this.currentNodeNS = prevNodeNS; this.nodeNamespaces = prevNodeNamespaces; } } void fallback() throws TemplateException, IOException { TemplateModel macroOrTransform = getNodeProcessor(currentNodeName, currentNodeNS, nodeNamespaceIndex); if (macroOrTransform instanceof Macro) { visit((Macro) macroOrTransform, null, null, null, null); } else if (macroOrTransform instanceof TemplateTransformModel) { visit(null, (TemplateTransformModel) macroOrTransform, null); } } /** * "visit" a macro. */ void visit(Macro macro, Map namedArgs, List positionalArgs, List bodyParameterNames, TemplateElement nestedBlock) throws TemplateException, IOException { if (macro == Macro.DO_NOTHING_MACRO) { return; } pushElement(macro); try { Macro.Context previousMacroContext = currentMacroContext; Macro.Context mc = macro.new Context(this, nestedBlock, bodyParameterNames); String catchAll = macro.getCatchAll(); TemplateModel unknownVars = null; if (namedArgs != null) { if (catchAll != null) unknownVars = new SimpleHash(); for (Iterator it = namedArgs.entrySet().iterator(); it.hasNext();) { Map.Entry entry = (Map.Entry) it.next(); String varName = (String) entry.getKey(); boolean hasVar = macro.hasArgNamed(varName); if (hasVar || catchAll != null) { Expression arg = (Expression) entry.getValue(); TemplateModel value = arg.getAsTemplateModel(this); if (hasVar) { mc.setLocalVar(varName, value); } else { ((SimpleHash)unknownVars).put(varName, value); } } else { String msg = "Macro " + macro.getName() + " has no such argument: " + varName; throw new TemplateException(msg, this); } } } else if (positionalArgs != null) { if (catchAll != null) unknownVars = new SimpleSequence(); String[] argumentNames = macro.getArgumentNames(); int size = positionalArgs.size(); if (argumentNames.length < size && catchAll == null) { throw new TemplateException("Macro " + macro.getName() + " only accepts " + argumentNames.length + " parameters.", this); } for (int i = 0; i < size; i++) { Expression argExp = (Expression) positionalArgs.get(i); TemplateModel argModel = argExp.getAsTemplateModel(this); try { if (i < argumentNames.length) { String argName = argumentNames[i]; mc.setLocalVar(argName, argModel); } else { ((SimpleSequence)unknownVars).add(argModel); } } catch (RuntimeException re) { throw new TemplateException(re, this); } } } if (catchAll != null) { mc.setLocalVar(catchAll, unknownVars); } ArrayList prevLocalContextStack = localContextStack; localContextStack = null; Namespace prevNamespace = currentNamespace; Configurable prevParent = getParent(); currentNamespace = (Namespace) macroToNamespaceLookup.get(macro); currentMacroContext = mc; try { mc.runMacro(this); } catch (ReturnInstruction.Return re) { } catch (TemplateException te) { handleTemplateException(te); } finally { currentMacroContext = previousMacroContext; localContextStack = prevLocalContextStack; currentNamespace = prevNamespace; setParent(prevParent); } } finally { popElement(); } } void visitMacroDef(Macro macro) { macroToNamespaceLookup.put(macro, currentNamespace); currentNamespace.put(macro.getName(), macro); } Namespace getMacroNamespace(Macro macro) { return (Namespace) macroToNamespaceLookup.get(macro); } void recurse(TemplateNodeModel node, TemplateSequenceModel namespaces) throws TemplateException, IOException { if (node == null) { node = this.getCurrentVisitorNode(); if (node == null) { throw new TemplateModelException( "The target node of recursion is missing or null."); } } TemplateSequenceModel children = node.getChildNodes(); if (children == null) return; for (int i=0; isetting directive, it still must be allowed to set it from Java * code while the template executes, since some frameworks allow templates * to actually change the output encoding on-the-fly. */ public void setOutputEncoding(String outputEncoding) { urlEscapingCharsetCached = false; super.setOutputEncoding(outputEncoding); } /** * Returns the name of the charset that should be used for URL encoding. * This will be null if the information is not available. * The function caches the return value, so it is quick to call it * repeately. */ String getEffectiveURLEscapingCharset() { if (!urlEscapingCharsetCached) { cachedURLEscapingCharset = getURLEscapingCharset(); if (cachedURLEscapingCharset == null) { cachedURLEscapingCharset = getOutputEncoding(); } urlEscapingCharsetCached = true; } return cachedURLEscapingCharset; } Collator getCollator() { if(collator == null) { collator = Collator.getInstance(getLocale()); } return collator; } public void setOut(Writer out) { this.out = out; } public Writer getOut() { return out; } String formatNumber(Number number) { if(numberFormat == null) { numberFormat = getNumberFormatObject(getNumberFormat()); } return numberFormat.format(number); } public void setNumberFormat(String formatName) { super.setNumberFormat(formatName); numberFormat = null; } String formatDate(Date date, int type) throws TemplateModelException { DateFormat df = getDateFormatObject(type); if(df == null) { throw new TemplateModelException("Can't convert the date to string, because it is not known which parts of the date variable are in use. Use ?date, ?time or ?datetime built-in, or ?string. or ?string(format) built-in with this date."); } return df.format(date); } public void setTimeFormat(String formatName) { super.setTimeFormat(formatName); timeFormat = null; } public void setDateFormat(String formatName) { super.setDateFormat(formatName); dateFormat = null; } public void setDateTimeFormat(String formatName) { super.setDateTimeFormat(formatName); dateTimeFormat = null; } public Configuration getConfiguration() { return getTemplate().getConfiguration(); } TemplateModel getLastReturnValue() { return lastReturnValue; } void setLastReturnValue(TemplateModel lastReturnValue) { this.lastReturnValue = lastReturnValue; } void clearLastReturnValue() { this.lastReturnValue = null; } NumberFormat getNumberFormatObject(String pattern) { if(numberFormats == null) { numberFormats = new HashMap(); } NumberFormat format = (NumberFormat) numberFormats.get(pattern); if(format != null) { return format; } // Get format from global format cache synchronized(localizedNumberFormats) { Locale locale = getLocale(); NumberFormatKey fk = new NumberFormatKey(pattern, locale); format = (NumberFormat)localizedNumberFormats.get(fk); if(format == null) { // Add format to global format cache. Note this is // globally done once per locale per pattern. if("number".equals(pattern)) { format = NumberFormat.getNumberInstance(locale); } else if("currency".equals(pattern)) { format = NumberFormat.getCurrencyInstance(locale); } else if("percent".equals(pattern)) { format = NumberFormat.getPercentInstance(locale); } else if ("computer".equals(pattern)) { format = getCNumberFormat(); } else { format = new DecimalFormat(pattern, new DecimalFormatSymbols(getLocale())); } localizedNumberFormats.put(fk, format); } } // Clone it and store the clone in the local cache format = (NumberFormat)format.clone(); numberFormats.put(pattern, format); return format; } DateFormat getDateFormatObject(int dateType) throws TemplateModelException { switch(dateType) { case TemplateDateModel.UNKNOWN: { return null; } case TemplateDateModel.TIME: { if(timeFormat == null) { timeFormat = getDateFormatObject(dateType, getTimeFormat()); } return timeFormat; } case TemplateDateModel.DATE: { if(dateFormat == null) { dateFormat = getDateFormatObject(dateType, getDateFormat()); } return dateFormat; } case TemplateDateModel.DATETIME: { if(dateTimeFormat == null) { dateTimeFormat = getDateFormatObject(dateType, getDateTimeFormat()); } return dateTimeFormat; } default: { throw new TemplateModelException("Unrecognized date type " + dateType); } } } DateFormat getDateFormatObject(int dateType, String pattern) throws TemplateModelException { if(dateFormats == null) { dateFormats = new Map[4]; dateFormats[TemplateDateModel.UNKNOWN] = new HashMap(); dateFormats[TemplateDateModel.TIME] = new HashMap(); dateFormats[TemplateDateModel.DATE] = new HashMap(); dateFormats[TemplateDateModel.DATETIME] = new HashMap(); } Map typedDateFormat = dateFormats[dateType]; DateFormat format = (DateFormat) typedDateFormat.get(pattern); if(format != null) { return format; } // Get format from global format cache synchronized(localizedDateFormats) { Locale locale = getLocale(); TimeZone timeZone = getTimeZone(); DateFormatKey fk = new DateFormatKey(dateType, pattern, locale, timeZone); format = (DateFormat)localizedDateFormats.get(fk); if(format == null) { // Add format to global format cache. Note this is // globally done once per locale per pattern. StringTokenizer tok = new StringTokenizer(pattern, "_"); int style = tok.hasMoreTokens() ? parseDateStyleToken(tok.nextToken()) : DateFormat.DEFAULT; if(style != -1) { switch(dateType) { case TemplateDateModel.UNKNOWN: { throw new TemplateModelException( "Can't convert the date to string using a " + "built-in format, because it is not known which " + "parts of the date variable are in use. Use " + "?date, ?time or ?datetime built-in, or " + "?string. or ?string() built-in "+ "with explicit formatting pattern with this date."); } case TemplateDateModel.TIME: { format = DateFormat.getTimeInstance(style, locale); break; } case TemplateDateModel.DATE: { format = DateFormat.getDateInstance(style, locale); break; } case TemplateDateModel.DATETIME: { int timestyle = tok.hasMoreTokens() ? parseDateStyleToken(tok.nextToken()) : style; if(timestyle != -1) { format = DateFormat.getDateTimeInstance(style, timestyle, locale); } break; } } } if(format == null) { try { format = new SimpleDateFormat(pattern, locale); } catch(IllegalArgumentException e) { throw new TemplateModelException("Can't parse " + pattern + " to a date format.", e); } } format.setTimeZone(timeZone); localizedDateFormats.put(fk, format); } } // Clone it and store the clone in the local cache format = (DateFormat)format.clone(); typedDateFormat.put(pattern, format); return format; } int parseDateStyleToken(String token) { if("short".equals(token)) { return DateFormat.SHORT; } if("medium".equals(token)) { return DateFormat.MEDIUM; } if("long".equals(token)) { return DateFormat.LONG; } if("full".equals(token)) { return DateFormat.FULL; } return -1; } /** * Returns the {@link NumberFormat} used for the c built-in. * This is always US English "0.################", without * grouping and without superfluous decimal separator. */ public NumberFormat getCNumberFormat() { // It can't be cached in a static field, because DecimalFormat-s aren't // thread-safe. if (cNumberFormat == null) { cNumberFormat = (DecimalFormat) C_NUMBER_FORMAT.clone(); } return cNumberFormat; } TemplateTransformModel getTransform(Expression exp) throws TemplateException { TemplateTransformModel ttm = null; TemplateModel tm = exp.getAsTemplateModel(this); if (tm instanceof TemplateTransformModel) { ttm = (TemplateTransformModel) tm; } else if (exp instanceof Identifier) { tm = getConfiguration().getSharedVariable(exp.toString()); if (tm instanceof TemplateTransformModel) { ttm = (TemplateTransformModel) tm; } } return ttm; } /** * Returns the loop or macro local variable corresponding to this * variable name. Possibly null. * (Note that the misnomer is kept for backward compatibility: loop variables * are not local variables according to our terminology.) */ public TemplateModel getLocalVariable(String name) throws TemplateModelException { if (localContextStack != null) { for (int i = localContextStack.size()-1; i>=0; i--) { LocalContext lc = (LocalContext) localContextStack.get(i); TemplateModel tm = lc.getLocalVariable(name); if (tm != null) { return tm; } } } return currentMacroContext == null ? null : currentMacroContext.getLocalVariable(name); } /** * Returns the variable that is visible in this context. * This is the correspondent to an FTL top-level variable reading expression. * That is, it tries to find the the variable in this order: *

    *
  1. An loop variable (if we're in a loop or user defined directive body) such as foo_has_next *
  2. A local variable (if we're in a macro) *
  3. A variable defined in the current namespace (say, via <#assign ...>) *
  4. A variable defined globally (say, via <#global ....>) *
  5. Variable in the data model: *
      *
    1. A variable in the root hash that was exposed to this rendering environment in the Template.process(...) call *
    2. A shared variable set in the configuration via a call to Configuration.setSharedVariable(...) *
    *
  6. *
*/ public TemplateModel getVariable(String name) throws TemplateModelException { TemplateModel result = getLocalVariable(name); if (result == null) { result = currentNamespace.get(name); } if (result == null) { result = getGlobalVariable(name); } return result; } /** * Returns the globally visible variable of the given name (or null). * This is correspondent to FTL .globals.name. * This will first look at variables that were assigned globally via: * <#global ...> and then at the data model exposed to the template. */ public TemplateModel getGlobalVariable(String name) throws TemplateModelException { TemplateModel result = globalNamespace.get(name); if (result == null) { result = rootDataModel.get(name); } if (result == null) { result = getConfiguration().getSharedVariable(name); } return result; } /** * Sets a variable that is visible globally. * This is correspondent to FTL <#global name=model>. * This can be considered a convenient shorthand for: * getGlobalNamespace().put(name, model) */ public void setGlobalVariable(String name, TemplateModel model) { globalNamespace.put(name, model); } /** * Sets a variable in the current namespace. * This is correspondent to FTL <#assign name=model>. * This can be considered a convenient shorthand for: * getCurrentNamespace().put(name, model) */ public void setVariable(String name, TemplateModel model) { currentNamespace.put(name, model); } /** * Sets a local variable (one effective only during a macro invocation). * This is correspondent to FTL <#local name=model>. * @param name the identifier of the variable * @param model the value of the variable. * @throws IllegalStateException if the environment is not executing a * macro body. */ public void setLocalVariable(String name, TemplateModel model) { if(currentMacroContext == null) { throw new IllegalStateException("Not executing macro body"); } currentMacroContext.setLocalVar(name, model); } /** * Returns a set of variable names that are known at the time of call. This * includes names of all shared variables in the {@link Configuration}, * names of all global variables that were assigned during the template processing, * names of all variables in the current name-space, names of all local variables * and loop variables. If the passed root data model implements the * {@link TemplateHashModelEx} interface, then all names it retrieves through a call to * {@link TemplateHashModelEx#keys()} method are returned as well. * The method returns a new Set object on each call that is completely * disconnected from the Environment. That is, modifying the set will have * no effect on the Environment object. */ public Set getKnownVariableNames() throws TemplateModelException { // shared vars. Set set = getConfiguration().getSharedVariableNames(); // root hash if (rootDataModel instanceof TemplateHashModelEx) { TemplateModelIterator rootNames = ((TemplateHashModelEx) rootDataModel).keys().iterator(); while(rootNames.hasNext()) { set.add(((TemplateScalarModel)rootNames.next()).getAsString()); } } // globals for (TemplateModelIterator tmi = globalNamespace.keys().iterator(); tmi.hasNext();) { set.add(((TemplateScalarModel) tmi.next()).getAsString()); } // current name-space for (TemplateModelIterator tmi = currentNamespace.keys().iterator(); tmi.hasNext();) { set.add(((TemplateScalarModel) tmi.next()).getAsString()); } // locals and loop vars if(currentMacroContext != null) { set.addAll(currentMacroContext.getLocalVariableNames()); } if (localContextStack != null) { for (int i = localContextStack.size()-1; i>=0; i--) { LocalContext lc = (LocalContext) localContextStack.get(i); set.addAll(lc.getLocalVariableNames()); } } return set; } /** * Outputs the instruction stack. Useful for debugging. * {@link TemplateException}s incorporate this information in their stack * traces. */ public void outputInstructionStack(PrintWriter pw) { pw.println("----------"); ListIterator iter = elementStack.listIterator(elementStack.size()); if(iter.hasPrevious()) { pw.print("==> "); TemplateElement prev = (TemplateElement) iter.previous(); pw.print(prev.getDescription()); pw.print(" ["); pw.print(prev.getStartLocation()); pw.println("]"); } while(iter.hasPrevious()) { TemplateElement prev = (TemplateElement) iter.previous(); if (prev instanceof UnifiedCall || prev instanceof Include) { String location = prev.getDescription() + " [" + prev.getStartLocation() + "]"; if(location != null && location.length() > 0) { pw.print(" in "); pw.println(location); } } } pw.println("----------"); pw.flush(); } private void pushLocalContext(LocalContext localContext) { if (localContextStack == null) { localContextStack = new ArrayList(); } localContextStack.add(localContext); } private void popLocalContext() { localContextStack.remove(localContextStack.size() - 1); } ArrayList getLocalContextStack() { return localContextStack; } /** * Returns the name-space for the name if exists, or null. * @param name the template path that you have used with the import directive * or {@link #importLib(String, String)} call, in normalized form. That is, the path must be an absolute * path, and it must not contain "/../" or "/./". The leading "/" is optional. */ public Namespace getNamespace(String name) { if (name.startsWith("/")) name = name.substring(1); if (loadedLibs != null) { return (Namespace) loadedLibs.get(name); } else { return null; } } /** * Returns the main name-space. * This is correspondent of FTL .main hash. */ public Namespace getMainNamespace() { return mainNamespace; } /** * Returns the main name-space. * This is correspondent of FTL .namespace hash. */ public Namespace getCurrentNamespace() { return currentNamespace; } /** * Returns a fictitious name-space that contains the globally visible variables * that were created in the template, but not the variables of the data-model. * There is no such thing in FTL; this strange method was added because of the * JSP taglib support, since this imaginary name-space contains the page-scope * attributes. */ public Namespace getGlobalNamespace() { return globalNamespace; } public TemplateHashModel getDataModel() { final TemplateHashModel result = new TemplateHashModel() { public boolean isEmpty() { return false; } public TemplateModel get(String key) throws TemplateModelException { TemplateModel value = rootDataModel.get(key); if (value == null) { value = getConfiguration().getSharedVariable(key); } return value; } }; if (rootDataModel instanceof TemplateHashModelEx) { return new TemplateHashModelEx() { public boolean isEmpty() throws TemplateModelException { return result.isEmpty(); } public TemplateModel get(String key) throws TemplateModelException { return result.get(key); } //NB: The methods below do not take into account // configuration shared variables even though // the hash will return them, if only for BWC reasons public TemplateCollectionModel values() throws TemplateModelException { return ((TemplateHashModelEx) rootDataModel).values(); } public TemplateCollectionModel keys() throws TemplateModelException { return ((TemplateHashModelEx) rootDataModel).keys(); } public int size() throws TemplateModelException { return ((TemplateHashModelEx) rootDataModel).size(); } }; } return result; } /** * Returns the read-only hash of globally visible variables. * This is the correspondent of FTL .globals hash. * That is, you see the variables created with * <#global ...>, and the variables of the data-model. * To create new global variables, use {@link #setGlobalVariable setGlobalVariable}. */ public TemplateHashModel getGlobalVariables() { return new TemplateHashModel() { public boolean isEmpty() { return false; } public TemplateModel get(String key) throws TemplateModelException { TemplateModel result = globalNamespace.get(key); if (result == null) { result = rootDataModel.get(key); } if (result == null) { result = getConfiguration().getSharedVariable(key); } return result; } }; } private void pushElement(TemplateElement element) { elementStack.add(element); } private void popElement() { elementStack.remove(elementStack.size() - 1); } public TemplateNodeModel getCurrentVisitorNode() { return currentVisitorNode; } /** * sets TemplateNodeModel as the current visitor node. .current_node */ public void setCurrentVisitorNode(TemplateNodeModel node) { currentVisitorNode = node; } TemplateModel getNodeProcessor(TemplateNodeModel node) throws TemplateException { String nodeName = node.getNodeName(); if (nodeName == null) { throw new TemplateException("Node name is null.", this); } TemplateModel result = getNodeProcessor(nodeName, node.getNodeNamespace(), 0); if (result == null) { String type = node.getNodeType(); /* DD: Original version: */ if (type == null) { type = "default"; } result = getNodeProcessor("@" + type, null, 0); /* DD: Jonathan's non-BC version and IMHO otherwise wrong version: if (type != null) { result = getNodeProcessor("@" + type, null, 0); } if (result == null) { result = getNodeProcessor("@default", null, 0); } */ } return result; } private TemplateModel getNodeProcessor(final String nodeName, final String nsURI, int startIndex) throws TemplateException { TemplateModel result = null; int i; for (i = startIndex; i0) { result = ns.get(prefix + ":" + localName); if (!(result instanceof Macro) && !(result instanceof TemplateTransformModel)) { result = null; } } else { if (nsURI.length() == 0) { result = ns.get(Template.NO_NS_PREFIX + ":" + localName); if (!(result instanceof Macro) && !(result instanceof TemplateTransformModel)) { result = null; } } if (nsURI.equals(template.getDefaultNS())) { result = ns.get(Template.DEFAULT_NAMESPACE_PREFIX + ":" + localName); if (!(result instanceof Macro) && !(result instanceof TemplateTransformModel)) { result = null; } } if (result == null) { result = ns.get(localName); if (!(result instanceof Macro) && !(result instanceof TemplateTransformModel)) { result = null; } } } } return result; } /** * Emulates include directive, except that name must be tempate * root relative. * *

It's the same as include(getTemplateForInclusion(name, encoding, parse)). * But, you may want to separately call these two methods, so you can determine the source of * exceptions more precisely, and thus achieve more intelligent error handling. * * @see #getTemplateForInclusion(String name, String encoding, boolean parse) * @see #include(Template includedTemplate) */ public void include(String name, String encoding, boolean parse) throws IOException, TemplateException { include(getTemplateForInclusion(name, encoding, parse)); } /** * Gets a template for inclusion; used with {@link #include(Template includedTemplate)}. * The advantage over simply using config.getTemplate(...) is that it chooses * the default encoding as the include directive does. * * @param name the name of the template, relatively to the template root directory * (not the to the directory of the currently executing template file!). * (Note that you can use {@link freemarker.cache.TemplateCache#getFullTemplatePath} * to convert paths to template root relative paths.) * @param encoding the encoding of the obtained template. If null, * the encoding of the Template that is currently being processed in this * Environment is used. * @param parse whether to process a parsed template or just include the * unparsed template source. */ public Template getTemplateForInclusion(String name, String encoding, boolean parse) throws IOException { if (encoding == null) { encoding = getTemplate().getEncoding(); } if (encoding == null) { encoding = getConfiguration().getEncoding(this.getLocale()); } return getConfiguration().getTemplate(name, getLocale(), encoding, parse); } /** * Processes a Template in the context of this Environment, including its * output in the Environment's Writer. * * @param includedTemplate the template to process. Note that it does not need * to be a template returned by * {@link #getTemplateForInclusion(String name, String encoding, boolean parse)}. */ public void include(Template includedTemplate) throws TemplateException, IOException { Template prevTemplate = getTemplate(); setParent(includedTemplate); importMacros(includedTemplate); try { visit(includedTemplate.getRootTreeNode()); } finally { setParent(prevTemplate); } } /** * Emulates import directive, except that name must be tempate * root relative. * *

It's the same as importLib(getTemplateForImporting(name), namespace). * But, you may want to separately call these two methods, so you can determine the source of * exceptions more precisely, and thus achieve more intelligent error handling. * * @see #getTemplateForImporting(String name) * @see #importLib(Template includedTemplate, String namespace) */ public Namespace importLib(String name, String namespace) throws IOException, TemplateException { return importLib(getTemplateForImporting(name), namespace); } /** * Gets a template for importing; used with * {@link #importLib(Template importedTemplate, String namespace)}. The advantage * over simply using config.getTemplate(...) is that it chooses the encoding * as the import directive does. * * @param name the name of the template, relatively to the template root directory * (not the to the directory of the currently executing template file!). * (Note that you can use {@link freemarker.cache.TemplateCache#getFullTemplatePath} * to convert paths to template root relative paths.) */ public Template getTemplateForImporting(String name) throws IOException { return getTemplateForInclusion(name, null, true); } /** * Emulates import directive. * * @param loadedTemplate the template to import. Note that it does not need * to be a template returned by {@link #getTemplateForImporting(String name)}. */ public Namespace importLib(Template loadedTemplate, String namespace) throws IOException, TemplateException { if (loadedLibs == null) { loadedLibs = new HashMap(); } String templateName = loadedTemplate.getName(); Namespace existingNamespace = (Namespace) loadedLibs.get(templateName); if (existingNamespace != null) { if (namespace != null) { setVariable(namespace, existingNamespace); } } else { Namespace newNamespace = new Namespace(loadedTemplate); if (namespace != null) { currentNamespace.put(namespace, newNamespace); if (currentNamespace == mainNamespace) { globalNamespace.put(namespace, newNamespace); } } Namespace prevNamespace = this.currentNamespace; this.currentNamespace = newNamespace; loadedLibs.put(templateName, currentNamespace); Writer prevOut = out; this.out = NULL_WRITER; try { include(loadedTemplate); } finally { this.out = prevOut; this.currentNamespace = prevNamespace; } } return (Namespace) loadedLibs.get(templateName); } String renderElementToString(TemplateElement te) throws IOException, TemplateException { Writer prevOut = out; try { StringWriter sw = new StringWriter(); this.out = sw; visit(te); return sw.toString(); } finally { this.out = prevOut; } } void importMacros(Template template) { for (Iterator it = template.getMacros().values().iterator(); it.hasNext();) { visitMacroDef((Macro) it.next()); } } /** * @return the namespace URI registered for this prefix, or null. * This is based on the mappings registered in the current namespace. */ public String getNamespaceForPrefix(String prefix) { return currentNamespace.getTemplate().getNamespaceForPrefix(prefix); } public String getPrefixForNamespace(String nsURI) { return currentNamespace.getTemplate().getPrefixForNamespace(nsURI); } /** * @return the default node namespace for the current FTL namespace */ public String getDefaultNS() { return currentNamespace.getTemplate().getDefaultNS(); } /** * A hook that Jython uses. */ public Object __getitem__(String key) throws TemplateModelException { return BeansWrapper.getDefaultInstance().unwrap(getVariable(key)); } /** * A hook that Jython uses. */ public void __setitem__(String key, Object o) throws TemplateException { setGlobalVariable(key, getObjectWrapper().wrap(o)); } private static final class NumberFormatKey { private final String pattern; private final Locale locale; NumberFormatKey(String pattern, Locale locale) { this.pattern = pattern; this.locale = locale; } public boolean equals(Object o) { if(o instanceof NumberFormatKey) { NumberFormatKey fk = (NumberFormatKey)o; return fk.pattern.equals(pattern) && fk.locale.equals(locale); } return false; } public int hashCode() { return pattern.hashCode() ^ locale.hashCode(); } } private static final class DateFormatKey { private final int dateType; private final String pattern; private final Locale locale; private final TimeZone timeZone; DateFormatKey(int dateType, String pattern, Locale locale, TimeZone timeZone) { this.dateType = dateType; this.pattern = pattern; this.locale = locale; this.timeZone = timeZone; } public boolean equals(Object o) { if(o instanceof DateFormatKey) { DateFormatKey fk = (DateFormatKey)o; return dateType == fk.dateType && fk.pattern.equals(pattern) && fk.locale.equals(locale) && fk.timeZone.equals(timeZone); } return false; } public int hashCode() { return dateType ^ pattern.hashCode() ^ locale.hashCode() ^ timeZone.hashCode(); } } public class Namespace extends SimpleHash { private Template template; Namespace() { this.template = Environment.this.getTemplate(); } Namespace(Template template) { this.template = template; } /** * @return the Template object with which this Namespace is associated. */ public Template getTemplate() { return template == null ? Environment.this.getTemplate() : template; } } static final Writer NULL_WRITER = new Writer() { public void write(char cbuf[], int off, int len) {} public void flush() {} public void close() {} }; private static final Writer EMPTY_BODY_WRITER = new Writer() { public void write(char[] cbuf, int off, int len) throws IOException { if (len > 0) { throw new IOException( "This transform does not allow nested content."); } } public void flush() { } public void close() { } }; }





© 2015 - 2024 Weber Informatics LLC | Privacy Policy