Application: gvSIG desktop

package org.gvsig.expressionevaluator.impl.expressionbuilder.formatters;

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;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class Select implements Formatter<Value> {

  private final Formatter<Value> formatter;

  public Select(Formatter<Value> formatter) {

    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(formatter);

        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 {

            Value c = function.parameters().get(COLUMNS);

            if (c instanceof ExpressionBuilder.Variable) {

                builder.append('"');

                builder.append(((ExpressionBuilder.Variable)c).name());

                builder.append("\" ");

            } else {

                Function columns = (Function) c;

                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();

    }

}