SQLPrettyfier.java

package railo.runtime.db;

import java.util.Locale;

import railo.commons.lang.ParserString;
import railo.runtime.exp.PageException;
import railo.runtime.format.DateFormat;
import railo.runtime.format.TimeFormat;
import railo.runtime.op.date.DateCaster;
import railo.runtime.type.dt.DateTime;

public class SQLPrettyfier {
	
	public static final String PLACEHOLDER_COUNT="placeholder_count";	
	public static final String PLACEHOLDER_ASTERIX="placeholder_asterix";
	public static final String PLACEHOLDER_QUESTION="QUESTION_MARK_SIGN";
	
	
	
	public static String prettyfie(String sql){
		return prettyfie(sql,false);
	}
	
	public static String prettyfie(String sql, boolean validZql){
	
		ParserString ps=new ParserString(sql.trim());
		boolean insideString=false;
		//short insideKlammer=0;
		StringBuffer sb=new StringBuffer(sql.length());
		//char last=0;
		
		outer:while(!ps.isAfterLast()) {
			if(insideString) {
		        if(ps.isCurrent('\'')) {
		            if(!ps.hasNext() || !ps.isNext('\''))insideString=false;
		        }
		    }
			else {
				if(ps.isCurrent('\''))insideString=true;
		        else if(ps.isCurrent('?')) {
		            sb.append(" "+PLACEHOLDER_QUESTION+" ");
		            ps.next();
		            continue;
		        }
		        else if(ps.isCurrent('{')) {
		        	StringBuffer date=new StringBuffer();
		        	int pos=ps.getPos();
		        	while(true) {
		        		if(ps.isAfterLast()){
		        			ps.setPos(pos);
		        			break;
		        		}
		        		else if(ps.isCurrent('}')) {
		        			date.append('}');
		        			DateTime d;
							try {
								d = DateCaster.toDateAdvanced(date.toString(), null);
							} 
							catch (PageException e) {
								ps.setPos(pos);
			        			break;
							}
							sb.append('\'');
		        			sb.append(new DateFormat(Locale.US).format(d,"yyyy-mm-dd"));
		        			sb.append(' ');
		        			sb.append(new TimeFormat(Locale.US).format(d,"HH:mm:ss"));
		        			sb.append('\'');
		        			ps.next();
		        			continue outer;
		        		}
		        		else {
		        			date.append(ps.getCurrent());
		        			ps.next();
		        		}
		        	}
		        }
		        else if(ps.isCurrent('*')) {
		        	sb.append(" "+PLACEHOLDER_ASTERIX+" ");
		        	ps.next();
	            	//last=ps.getCurrent();
				    continue;
		        }
		        else if(validZql && ps.isCurrent('a')) {
		        		if(ps.isPreviousWhiteSpace() && ps.isNext('s') && ps.isNextNextWhiteSpace())	{
			            	ps.next();
			            	ps.next();
			            	ps.removeSpace();
			            	
			            	continue;
			            }
			        
		        }
		        /*for(int i=0;i<reseved_words.length;i++) {
		        	if(ps.isCurrent(reseved_words[i])) {
			        	int pos=ps.getPos();
			        	ps.setPos(pos+4);
			        	if(ps.isCurrentWhiteSpace()) {
			        		sb.append(" placeholder_"+reseved_words[i]+" ");
			        		continue;
			        	}
			        	if(ps.isCurrent(',')) {
			        		sb.append(" placeholder_"+reseved_words[i]+",");
			        		continue;
			        	}
			        	ps.setPos(pos);
			        }
		        }*/
				/*if(ps.isCurrent("char")) {
		        	int pos=ps.getPos();
		        	ps.setPos(pos+4);
		        	if(ps.isCurrentWhiteSpace()) {
		        		sb.append(" "+PLACEHOLDER_CHAR+" ");
		        		continue;
		        	}
		        	if(ps.isCurrent(',')) {
		        		sb.append(" "+PLACEHOLDER_CHAR+",");
		        		continue;
		        	}
		        	ps.setPos(pos);
		        }*/
			}
		    sb.append(ps.getCurrent());
			ps.next();
		}
		;	
		if(!ps.isLast(';'))sb.append(';');
		
		//print.err(sb.toString());
		//print.err("---------------------------------------------------------------------------------");
		return sb.toString();
	}
	
	
}