Application: gvSIG desktop

package org.gvsig.expressionevaluator.impl.function.programming;

import java.lang.reflect.Field;

import org.apache.commons.lang3.Range;

import org.gvsig.expressionevaluator.AttributeHandler;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.Codes;

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

import org.gvsig.expressionevaluator.Formatter;

import org.gvsig.expressionevaluator.Function;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.spi.AbstractFunction;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.util.GetItemByKey;

public class GetattrFunction extends AbstractFunction {

    public GetattrFunction() {

        super(

                Function.GROUP_PROGRAMMING, 

                FUNCTION_GETATTR, 

                Range.is(2),

                null,

                "GETATTR({{object}}, attribute_name_as_string)",

                null,

                null,

                true

        );

    }

    @Override

    public boolean allowConstantFolding() {

        return false;

    }

    @Override

    public Object call(Interpreter interpreter, Object[] args) throws Exception {

        Object obj = getObject(args, 0);

        String attrname = getStr(args,1);

        if( attrname == null ) {

            throw new NullPointerException("A attribute name was expected and a null was received");

        }

        if( obj==null ) {

            return null;

            //throw new NullPointerException("An object pointer was expected to access attribute "+attrname+" and a null was received");

        }

        Object value;

        if( obj instanceof DynObject ) {

            value = ((DynObject) obj).getDynValue(attrname);

        } else if( obj instanceof AttributeHandler ) {

            value = ((AttributeHandler) obj).get(attrname);

        } else if( obj instanceof GetItemByKey ) {

            value = ((GetItemByKey) obj).get(attrname);

        } else {

            Class<?> theClass = obj.getClass();

            Field field = theClass.getField(attrname);

            value = field.get(obj);

        }

        return value;

    }

    @Override

    public String toString(Codes args, Formatter<Code> formatter) {

        StringBuilder builder = new StringBuilder();

        Code arg0 = args.get(0);

        Code arg1 = args.get(1);

//        if( arg0 instanceof Code.Identifier && arg1 instanceof Code.Identifier ) {

//          builder.append(arg0.toString(formatter));

//          builder.append("->");

//          builder.append(arg1.toString());

//        } else 

        if( arg0 instanceof Code.Identifier && arg1 instanceof Code.Constant && ((Code.Constant)arg1).value() instanceof CharSequence ) {

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

          builder.append(".\"");

          builder.append(((Code.Constant)arg1).value().toString());

          builder.append("\"");

        } else {

          builder.append(this.name());

          builder.append("(");

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

          builder.append(", ");

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

          builder.append(")");

        }

        return builder.toString();

    }

}