Application: gvSIG desktop

package org.gvsig.fmap.dal.store.h2.expressionbuilderformatter;

import org.apache.commons.lang3.StringUtils;

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

import org.gvsig.expressionevaluator.ExpressionBuilder.Function;

import org.gvsig.expressionevaluator.ExpressionBuilder.Value;

import org.gvsig.expressionevaluator.ExpressionBuilder.Variable;

import org.gvsig.expressionevaluator.ExpressionBuilder.Constant;

import org.gvsig.expressionevaluator.Formatter;

import org.gvsig.fmap.dal.SQLBuilder;

/**

 *

 * @author jjdelcerro

 */

public class Getattr implements Formatter<Value> {

    private final SQLBuilder sqlbuilder;

    private final Formatter<Value> formatter;

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

        this.sqlbuilder = sqlbuilder;

        this.formatter = formatter;

    }

    @Override

    public boolean canApply(Value value) {

        if (value instanceof Function) {

            if( StringUtils.equalsIgnoreCase(FUNCTION_GETATTR, ((Function) value).name()) ) {

              Value obj = ((Function) value).parameters().get(0);

              Value attrname = ((Function) value).parameters().get(1);

              return obj instanceof Variable && attrname instanceof Constant;

            }

        }

        return false;

    }

    @Override

    public String format(Value function0) {

        Function function = (Function) function0;

        Variable obj = (Variable) function.parameters().get(0);

        Constant attrname = (Constant) function.parameters().get(1);

        return this.sqlbuilder.as_identifier(obj.name())+"."+this.sqlbuilder.as_identifier((String) attrname.value());

    }

}