Application: gvSIG desktop

package org.gvsig.fmap.dal.store.jdbc2.spi.expressionbuilderformatter;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.ExpressionBuilder;

import static org.gvsig.expressionevaluator.ExpressionBuilder.FUNCTION_TUPLE;

import org.gvsig.expressionevaluator.ExpressionBuilder.Function;

import org.gvsig.expressionevaluator.ExpressionBuilder.Value;

import org.gvsig.expressionevaluator.Formatter;

import static org.gvsig.fmap.dal.DataManager.FUNCTION_SELECT;

import org.gvsig.fmap.dal.SQLBuilder;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class Select implements Formatter<Value> {

  private final SQLBuilder sqlbuilder;

  private final Formatter<Value> formatter;

  public Select(SQLBuilder sqlbuilder, Formatter<Value> formatter) {

    this.sqlbuilder = sqlbuilder;

    this.formatter = formatter;

  }

  @Override

  public boolean canApply(ExpressionBuilder.Value value) {

    if (value instanceof ExpressionBuilder.Function) {

      return StringUtils.equalsIgnoreCase(FUNCTION_SELECT, ((Function) value).name());

    }

    return false;

  }

  private static final int COLUMNS = 0;

  private static final int TABLE = 1;

  private static final int WHERE = 2;

  private static final int ORDER = 3;

  private static final int ORDER_MODE = 4;

  private static final int LIMIT = 5;

  @Override

  public String format(Value function0) {

    Function function = (Function) function0;

    String tableName = (String) function.parameters().get(TABLE).toString();

    Value where = function.parameters().get(WHERE);

    Value order = function.parameters().get(ORDER);

    Value limit = function.parameters().get(LIMIT);

    Value order_mode = function.parameters().get(ORDER_MODE);

    StringBuilder builder = new StringBuilder();

    builder.append("(SELECT ");

    try {

      Function columns = (Function) function.parameters().get(COLUMNS);

      if (!StringUtils.equalsIgnoreCase(((ExpressionBuilder.Function) columns).name(), FUNCTION_TUPLE)) {

        throw new IllegalArgumentException();

      }

      if( columns.parameters().isEmpty() ) {

            builder.append("* ");

      } else {

        boolean needcomma = false;

        for (Value parameter : columns.parameters()) {

          if(needcomma) {

            builder.append(", ");

          } else {

            needcomma = true;

          }

          builder.append(parameter.toString(formatter));

        }

      }

    } catch (Exception ex) {

      throw new IllegalArgumentException("Invalid columns, need a tuple function as columns.");

    }

    builder.append(" FROM ");

    builder.append(tableName);

    if (where != null) {

      builder.append(" WHERE ");

      builder.append(where.toString(formatter));

    }

//        for( int n=0 ; function.containsParameter("ORDER", n); n++) {

//          if( n>0 ) {

//            builder.append(", ");

//          }

//          builder.append(function.getParameter("ORDER",n).toString(formatter));

//          if( StringUtils.equalsIgnoreCase(function.getParameter("ORDER_MODE",n).toString(),"true") ) {

//            builder.append(" ASC ");

//          } else {

//            builder.append(" DESC ");

//          }

//        }

    if (limit != null) {

      builder.append(" LIMIT ");

      builder.append(limit.toString(formatter));

    }

    builder.append(")");

    return builder.toString();

  }

}