package railo.transformer.cfml.evaluator.impl;
import railo.commons.lang.StringUtil;
import railo.runtime.config.Config;
import railo.runtime.config.ConfigImpl;
import railo.runtime.engine.ThreadLocalPageContext;
import railo.runtime.exp.TemplateException;
import railo.transformer.bytecode.Statement;
import railo.transformer.bytecode.cast.CastBoolean;
import railo.transformer.bytecode.cast.CastString;
import railo.transformer.bytecode.expression.Expression;
import railo.transformer.bytecode.literal.LitString;
import railo.transformer.bytecode.statement.tag.Attribute;
import railo.transformer.bytecode.statement.tag.Tag;
import railo.transformer.bytecode.statement.tag.TagLoop;
import railo.transformer.bytecode.util.ASMUtil;
import railo.transformer.cfml.Data;
import railo.transformer.cfml.ExprTransformer;
import railo.transformer.cfml.TransfomerSettings;
import railo.transformer.cfml.evaluator.EvaluatorException;
import railo.transformer.cfml.evaluator.EvaluatorSupport;
import railo.transformer.library.function.FunctionLib;
import railo.transformer.library.tag.TagLib;
import railo.transformer.library.tag.TagLibTag;
import railo.transformer.util.CFMLString;
public final class Loop extends EvaluatorSupport {
public TagLib execute(Config config,Tag tag, TagLibTag libTag, FunctionLib[] flibs,Data data) throws TemplateException {
TagLoop loop=(TagLoop) tag;
// label
try {
if(ASMUtil.isLiteralAttribute(tag, "label", ASMUtil.TYPE_STRING, false, true)) {
LitString ls=(LitString) CastString.toExprString(tag.getAttribute("label").getValue());
String l = ls.getString();
if(!StringUtil.isEmpty(l,true)) {
loop.setLabel(l.trim());
tag.removeAttribute("label");
}
}
}
catch (EvaluatorException e) {
throw new TemplateException(null, e);
}
return null;
}
/**
*
* @see railo.transformer.cfml.evaluator.EvaluatorSupport#evaluate(railo.transformer.bytecode.statement.tag.Tag, railo.transformer.library.tag.TagLibTag, railo.transformer.library.function.FunctionLib[])
*/
public void evaluate(Tag tag,TagLibTag tagLibTag,FunctionLib[] flibs) throws EvaluatorException {
TagLoop loop=(TagLoop) tag;
// attribute maxrows and endrow not allowd at the same time
if(tag.containsAttribute("maxrows") && tag.containsAttribute("endrow"))
throw new EvaluatorException("Wrong Context, you cannot use attribute maxrows and endrow at the same time.");
// file loop
if(tag.containsAttribute("file")) {
if(!tag.containsAttribute("index") && !tag.containsAttribute("item"))
throw new EvaluatorException("Wrong Context, when you use attribute file you must also use attribute index and/or item");
loop.setType(TagLoop.TYPE_FILE);
return;
}
// list loop
if(tag.containsAttribute("list")){
if(!tag.containsAttribute("index") && !tag.containsAttribute("item"))
throw new EvaluatorException("Wrong Context, when you use attribute list,you must define attribute index and/or item");
loop.setType(TagLoop.TYPE_LIST);
return;
}
// array loop
if(tag.containsAttribute("array")){
if(!tag.containsAttribute("index") && !tag.containsAttribute("item"))
throw new EvaluatorException("Wrong Context, when you use attribute array, you must define attribute index and/or item");
loop.setType(TagLoop.TYPE_ARRAY);
return;
}
// collection loop
if(tag.containsAttribute("collection")) {
if(!tag.containsAttribute("index") && !tag.containsAttribute("item"))
throw new EvaluatorException("Wrong Context, when you use attribute collection,you must define attribute index and/or item");
loop.setType(TagLoop.TYPE_COLLECTION);
return;
}
// index loop
if(tag.containsAttribute("index")) {
if(!tag.containsAttribute("from") || !tag.containsAttribute("to"))
throw new EvaluatorException("Wrong Context, when you use attribute index you must also use attribute from and to or list or file");
loop.setType(TagLoop.TYPE_INDEX);
return;
}
// condition loop
if(tag.containsAttribute("condition")){
if(tag.isScriptBase())
throw new EvaluatorException("tag loop-condition is not supported within cfscript, use instead a while statement.");
TagLib tagLib=tagLibTag.getTagLib();
ExprTransformer transformer;
String text=ASMUtil.getAttributeString(tag, "condition");
try {
ConfigImpl config=(ConfigImpl) ThreadLocalPageContext.getConfig();
transformer = tagLib.getExprTransfomer();
Expression expr=transformer.transform(ASMUtil.getAncestorPage(tag),null,flibs,config.getCoreTagLib().getScriptTags(),new CFMLString(text,"UTF-8"),TransfomerSettings.toSetting(ThreadLocalPageContext.getConfig(),null));
tag.addAttribute(new Attribute(false,"condition",CastBoolean.toExprBoolean(expr),"boolean"));
}
catch (Exception e) {
throw new EvaluatorException(e.getMessage());
}
loop.setType(TagLoop.TYPE_CONDITION);
return;
}
// query loop
if(tag.containsAttribute("query")){
loop.setType(TagLoop.TYPE_QUERY);
return;
}
Info info=getParentInfo(loop);
// query group
if(tag.containsAttribute("group") && info.hasParentWithQuery){
loop.setType(TagLoop.TYPE_GROUP);
return;
}
if(info.hasParentWithQuery) {
if(info.hasParentWithGroup) loop.setType(TagLoop.TYPE_INNER_GROUP);
else loop.setType(TagLoop.TYPE_INNER_QUERY);
return;
}
/*
if(hasQuery)
output.setType(TagOutput.TYPE_QUERY);
else if(tag.containsAttribute("group") && hasParentWithQuery)
output.setType(TagOutput.TYPE_GROUP);
else if(hasParentWithQuery) {
if(hasParentWithGroup) output.setType(TagOutput.TYPE_INNER_GROUP);
else output.setType(TagOutput.TYPE_INNER_QUERY);
}
else
output.setType(TagOutput.TYPE_NORMAL);
*/
loop.setType(TagLoop.TYPE_NOTHING);
//throw new EvaluatorException("Wrong Context, invalid attributes in tag cfloop");
}
private Info getParentInfo(TagLoop loop) {
// check if inside a query tag
TagLoop parent = loop;
Info info=new Info();
info.hasParentWithGroup=false;
info.hasParentWithQuery=false;
//boolean hasQuery=loop.containsAttribute("query");
while((parent=getParentTagLoop(parent))!=null) {
if(!info.hasParentWithQuery)info.hasParentWithQuery=parent.hasQuery();
if(!info.hasParentWithGroup)info.hasParentWithGroup=parent.hasGroup();
if(info.hasParentWithQuery && info.hasParentWithGroup)break;
}
return info;
}
private static TagLoop getParentTagLoop(TagLoop stat) {
Statement parent = stat;
while(true) {
parent=parent.getParent();
if(parent==null)return null;
if(parent instanceof TagLoop) return (TagLoop) parent;
}
}
class Info {
private boolean hasParentWithGroup=false;
private boolean hasParentWithQuery=false;
}
}