CustomTagUtil.java

package railo.runtime.customtag;

import railo.commons.io.res.Resource;
import railo.commons.io.res.util.ResourceUtil;
import railo.runtime.Mapping;
import railo.runtime.MappingImpl;
import railo.runtime.PageContext;
import railo.runtime.PageContextImpl;
import railo.runtime.PageSource;
import railo.runtime.config.Config;
import railo.runtime.config.ConfigImpl;
import railo.runtime.config.ConfigWeb;
import railo.runtime.exp.ExpressionException;
import railo.runtime.exp.PageException;
import railo.runtime.type.util.ArrayUtil;
import railo.runtime.type.util.ListUtil;

public class CustomTagUtil {
	



	public static InitFile loadInitFile(PageContext pc, String name) throws PageException  {
		InitFile initFile = loadInitFile(pc, name, null);
		if(initFile!=null) {
    		return initFile;
    	}
    	 // EXCEPTION
    	ConfigWeb config = pc.getConfig();
        // message
        StringBuffer msg=new StringBuffer("custom tag \"");
        msg.append(getDisplayName(config,name));
        msg.append("\" is not defined in directory \"");
        msg.append(ResourceUtil.getResource(pc, pc.getCurrentPageSource()).getParent());
        msg.append('"');
        
        Mapping[] actms = pc.getApplicationContext().getCustomTagMappings();
        Mapping[] cctms = config.getCustomTagMappings();
        int asize=ArrayUtil.size(actms);
        int csize=ArrayUtil.size(cctms);
        int size=asize+csize;
        
        if(size>0){
        	if(size==1)msg.append(" and directory ");
        	else msg.append(" and directories ");
        	msg.append("\"");
        	
        	String list;
        	if(asize>0) {
        		list=toString(actms);
        		if(csize>0) list+=", "+toString(cctms);
        	}
        	else {
        		list=toString(cctms);
        	}
        	
        	
        	msg.append(list);
        	msg.append("\"");
        }
        throw new ExpressionException(msg.toString(),getDetail(config));
    	
	}
	
	public static InitFile loadInitFile(PageContext pc, String name, InitFile defaultValue) throws PageException  {
    	ConfigImpl config=(ConfigImpl) pc.getConfig();
    	String[] filenames=getFileNames(config, name);
    	boolean doCache=config.useCTPathCache();
    	
    	
    	boolean doCustomTagDeepSearch = config.doCustomTagDeepSearch();
        PageSource ps=null;
    	InitFile initFile;
    
   // CACHE
    	// check local
    	String localCacheName=null;    
    	Mapping[] actms = pc.getApplicationContext().getCustomTagMappings();
    	Mapping[] cctms = config.getCustomTagMappings();
        
    	if(doCache) {
	    	if(pc.getConfig().doLocalCustomTag()){
		    	localCacheName=pc.getCurrentPageSource().getDisplayPath().replace('\\', '/');
		    	localCacheName="local:"+localCacheName.substring(0,localCacheName.lastIndexOf('/')+1).concat(name);
		    	initFile=config.getCTInitFile(pc, localCacheName);
		    	if(initFile!=null) return initFile;
	    	}
	        
	    	// cache application mapping
	        if(actms!=null)for(int i=0;i<actms.length;i++){
	        	initFile=config.getCTInitFile(pc,"application:"+actms[i].hashCode()+"/"+name);
	        	if(initFile!=null)return initFile;
	        }
	        
	    	// cache config mapping
	        if(cctms!=null)for(int i=0;i<cctms.length;i++){
	        	initFile=config.getCTInitFile(pc,"config:"+cctms[i].hashCode()+"/"+name);
	        	if(initFile!=null)return initFile;
	        }
    	}
        
     // SEARCH
        // search local
        if(pc.getConfig().doLocalCustomTag()){
		    for(int i=0;i<filenames.length;i++){
	            PageSource[] arr = ((PageContextImpl)pc).getRelativePageSources(filenames[i]);
		    	//ps=pc.getRelativePageSource(filenames[i]);
	            ps=MappingImpl.isOK(arr);
				if(ps !=null) {
					initFile= new InitFile(ps,filenames[i],filenames[i].endsWith('.'+config.getCFCExtension()));
					if(doCache)config.putCTInitFile(localCacheName, initFile);
	        		return initFile;
				}
	        }
        }
    	
        // search application custom tag mapping
        if(actms!=null){
        	for(int i=0;i<filenames.length;i++){
            	ps=getMapping(actms, filenames[i],doCustomTagDeepSearch);
            	if(ps!=null) {
            		initFile=new InitFile(ps,filenames[i],filenames[i].endsWith('.'+config.getCFCExtension()));
            		if(doCache)config.putCTInitFile("application:"+ps.getMapping().hashCode()+"/"+name, initFile);
            		return initFile;
            	}
            }
        }
    	
        // search custom tag mappings
        for(int i=0;i<filenames.length;i++){
        	ps=getMapping(cctms, filenames[i], doCustomTagDeepSearch);
        	if(ps!=null) {
        		initFile=new InitFile(ps,filenames[i],filenames[i].endsWith('.'+config.getCFCExtension()));
        		if(doCache)config.putCTInitFile("config:"+ps.getMapping().hashCode()+"/"+name, initFile);
        		return initFile;
        	}
        }
        
        return defaultValue;
	}
	
	
	public static String[] getFileNames(Config config, String name) throws ExpressionException {
		String[] extensions=config.getCustomTagExtensions();
        if(extensions.length==0) throw new ExpressionException("Custom Tags are disabled");
        String[] fileNames=new String[extensions.length];
		
    	for(int i =0;i<fileNames.length;i++){
    		fileNames[i]=name+'.'+extensions[i];
    	}
    	return fileNames;
	}
	
	 private static PageSource getMapping(Mapping[] ctms, String filename, boolean doCustomTagDeepSearch) {
	    	PageSource ps;
	    	for(int i=0;i<ctms.length;i++){
	    		ps = ((MappingImpl) ctms[i]).getCustomTagPath(filename, doCustomTagDeepSearch);
				if(ps!=null) return ps;
	        }
			return null;
		}
	 
	 public static  String getDisplayName(Config config,String name) {
		String[] extensions=config.getCustomTagExtensions();
        if(extensions.length==0) return name;
        
		return name+".["+ListUtil.arrayToList(extensions, "|")+"]";
	}
	 
	 public static String getDetail(Config config) {
    	boolean hasCFC=false,hasCFML=false;
    	
    	String[] extensions=config.getCustomTagExtensions();
        for(int i =0;i<extensions.length;i++){
    		if(extensions[i].equalsIgnoreCase(config.getCFCExtension())) hasCFC=true;
    		else hasCFML=true;
    	}
    	StringBuffer sb=new StringBuffer();
    	if(!hasCFC)sb.append("Component based Custom Tags are not enabled;");
    	if(!hasCFML)sb.append("CFML based Custom Tags are not enabled;");
    	return sb.toString();
	}
	 
	 public static String toString(Mapping[] ctms) {
		if(ctms==null) return "";
    	StringBuffer sb=new StringBuffer();
    	Resource p;
    	for(int i=0;i<ctms.length;i++){
    		if(sb.length()!=0)sb.append(", ");
    		p = ctms[i].getPhysical();
    		if(p!=null)
    			sb.append(p.toString());
        }
        return sb.toString();
	}
}