Loop.java

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;
	}
}