/**
* MVEL 2.0
* Copyright (C) 2007 The Codehaus
* Mike Brock, Dhanji Prasanna, John Graham, Mark Proctor
*
* 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 org.mvel2.templates.res;
import org.mvel2.CompileException;
import org.mvel2.MVEL;
import org.mvel2.ParserContext;
import org.mvel2.integration.VariableResolverFactory;
import org.mvel2.integration.impl.MapVariableResolverFactory;
import org.mvel2.templates.TemplateRuntime;
import org.mvel2.templates.TemplateRuntimeError;
import org.mvel2.templates.util.ArrayIterator;
import org.mvel2.templates.util.CountIterator;
import org.mvel2.templates.util.TemplateOutputStream;
import org.mvel2.util.ParseTools;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
public class CompiledForEachNode extends Node {
public Node nestedNode;
private Serializable[] ce;
private String[] item;
private char[] sepExpr;
private Serializable cSepExpr;
private ParserContext context;
public CompiledForEachNode(int begin, String name, char[] template, int start, int end, ParserContext context) {
super(begin, name, template, start, end);
this.context = context;
configure();
}
public Node getNestedNode() {
return nestedNode;
}
public void setNestedNode(Node nestedNode) {
this.nestedNode = nestedNode;
}
public boolean demarcate(Node terminatingnode, char[] template) {
nestedNode = next;
next = terminus;
sepExpr = terminatingnode.getContents();
if (sepExpr.length == 0) {
sepExpr = null;
}
else {
cSepExpr = MVEL.compileExpression(sepExpr, context);
}
return false;
}
public Object eval(TemplateRuntime runtime, TemplateOutputStream appender, Object ctx, VariableResolverFactory factory) {
Iterator[] iters = new Iterator[item.length];
Object o;
for (int i = 0; i < iters.length; i++) {
if ((o = MVEL.executeExpression(ce[i], ctx, factory)) instanceof Iterable) {
iters[i] = ((Iterable) o).iterator();
}
else if (o instanceof Object[]) {
iters[i] = new ArrayIterator((Object[]) o);
}
else if (o instanceof Integer) {
iters[i] = new CountIterator((Integer) o);
}
else {
throw new TemplateRuntimeError("cannot iterate object type: " + o.getClass().getName());
}
}
Map<String, Object> locals = new HashMap<String, Object>();
MapVariableResolverFactory localFactory = new MapVariableResolverFactory(locals, factory);
int iterate = iters.length;
while (true) {
for (int i = 0; i < iters.length; i++) {
if (!iters[i].hasNext()) {
iterate--;
locals.put(item[i], "");
}
else {
locals.put(item[i], iters[i].next());
}
}
if (iterate != 0) {
nestedNode.eval(runtime, appender, ctx, localFactory);
if (sepExpr != null) {
for (Iterator it : iters) {
if (it.hasNext()) {
appender.append(String.valueOf(MVEL.executeExpression(cSepExpr, ctx, factory)));
break;
}
}
}
}
else break;
}
return next != null ? next.eval(runtime, appender, ctx, factory) : null;
}
private void configure() {
ArrayList<String> items = new ArrayList<String>();
ArrayList<String> expr = new ArrayList<String>();
int start = cStart;
for (int i = start; i < cEnd; i++) {
switch (contents[i]) {
case '(':
case '[':
case '{':
case '"':
case '\'':
i = ParseTools.balancedCapture(contents, i, contents[i]);
break;
// if (expr.size() < items.size()) {
// start = i;
// i = ParseTools.balancedCapture(contents, i, contents[i]);
// expr.add(ParseTools.createStringTrimmed(contents, start, i - start + 1));
// start = i + 1;
// }
// else {
// throw new CompileException("unexpected character '" + contents[i] + "' in foreach tag", contents,cStart + 1);
// }
// break;
case ':':
items.add(ParseTools.createStringTrimmed(contents, start, i - start));
start = i + 1;
break;
case ',':
if (expr.size() != (items.size() - 1)) {
throw new CompileException("unexpected character ',' in foreach tag", contents, cStart + i);
}
expr.add(ParseTools.createStringTrimmed(contents, start, i - start));
start = i + 1;
break;
}
}
if (start < cEnd) {
if (expr.size() != (items.size() - 1)) {
throw new CompileException("expected character ':' in foreach tag", contents, cEnd);
}
expr.add(ParseTools.createStringTrimmed(contents, start, cEnd - start));
}
item = new String[items.size()];
int i = 0;
for (String s : items) item[i++] = s;
String[] expression;
ce = new Serializable[(expression = new String[expr.size()]).length];
i = 0;
for (String s : expr) {
ce[i] = MVEL.compileExpression(expression[i++] = s, context);
}
}
}