package railo.runtime.db;
import java.io.ByteArrayInputStream;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.Set;
import java.util.Vector;
import railo.runtime.sql.old.ParseException;
import railo.runtime.sql.old.ZExp;
import railo.runtime.sql.old.ZExpression;
import railo.runtime.sql.old.ZFromItem;
import railo.runtime.sql.old.ZQuery;
import railo.runtime.sql.old.ZStatement;
import railo.runtime.sql.old.ZqlParser;
/**
* utilities for sql statements
*/
public final class HSQLUtil {
private ZqlParser parser;
private String sql;
private boolean isUnion;
/**or of the class
* construct
* @param sql SQl Statement as String
*/
public HSQLUtil(String sql) {
this.sql=SQLPrettyfier.prettyfie(sql,true);//sqlToZQL(sql,true);
parser = new ZqlParser(new ByteArrayInputStream(this.sql.getBytes()));
}
/* *
* transalte SQL syntax to a ZQL combatible form
* @param sql sql to transalte
* @param changePlaceHolder
* @return translated sql
* /
private static String sqlToZQL(String sql,boolean changePlaceHolder) {
sql=sql.trim();
char c=' ';//,last=' ';
int len=sql.length();
boolean insideString=false;
StringBuffer sb=new StringBuffer(len);
for(int i=0;i<len;i++) {
c=sql.charAt(i);
if(insideString) {
if(c=='\'') {
if(i+1>=len || sql.charAt(i+1)!='\'')insideString=false;
}
}
else {
if(c=='\'')insideString=true;
else if(changePlaceHolder && c=='?') {
sb.append("QUESTION_MARK_SIGN");
//last=c;
continue;
}
else if(c=='a'|| c=='A') {
if(
(i!=0 && isWhiteSpace(sql.charAt(i-1)))
&&
(i+1<len && (sql.charAt(i+1)=='s' || sql.charAt(i+1)=='S'))
&&
(i+2<len && isWhiteSpace(sql.charAt(i+2)))
) {
i++;
//last=c;
continue;
}
}
else if(c=='*') {
}
}
//last=c;
sb.append(c);
}
if(c!=';')sb.append(';');
return sb.toString();
}*/
/*private static boolean isWhiteSpace(char c) {
return (c==' ' || c=='\t' || c=='\b' || c=='\r' || c=='\n');
}*/
/**
* @return return the sql state inside
*/
public String getSQL() {
return sql;
}
/**
* return all invoked tables by a sql statement
* @return invoked tables in a ArrayList
* @throws ParseException
*/
public Set<String> getInvokedTables() throws ParseException {
// Read all SQL statements from input
ZStatement st;
Set<String> tables=new HashSet<String>();
while((st = parser.readStatement()) != null) {
this.sql=st.toString();
if(st instanceof ZQuery) { // An SQL query: query the DB
getInvokedTables((ZQuery)st,tables);
}
break;
}
return tables;
}
private void getInvokedTables(ZQuery query, Set<String> tablesNames) {
//print.out("qry:"+query.getSet());
Vector tables=query.getFrom();
Enumeration e = tables.elements();
// from
while(e.hasMoreElements()) {
ZFromItem fromItem=(ZFromItem) e.nextElement();
tablesNames.add(fromItem.getFullName());
}
// where
ZExp where = query.getWhere();
if(where instanceof ZExpression) {
parseZExpression((ZExpression) where, tablesNames);
}
// set
ZExpression set = query.getSet();
if(set!=null){
isUnion=true;
ZExp op = set.getOperand(0);
if(op instanceof ZQuery) getInvokedTables((ZQuery)op, tablesNames);
}
}
public boolean isUnion() {
return isUnion;
}
private void parseZExpression(ZExpression expression, Set tablesNames) {
Vector operands = expression.getOperands();
Enumeration e = operands.elements();
while(e.hasMoreElements()) {
Object el=e.nextElement();
if(el instanceof ZExpression)parseZExpression((ZExpression)el,tablesNames);
else if(el instanceof ZQuery)getInvokedTables((ZQuery)el,tablesNames);
}
}
}