Application: gvSIG desktop

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

import java.util.List;

import java.util.Map;

import javax.json.JsonArray;

import javax.json.JsonObject;

import org.apache.commons.lang3.Range;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.Codes;

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

import org.gvsig.expressionevaluator.ExpressionRuntimeException;

import org.gvsig.expressionevaluator.Formatter;

import org.gvsig.expressionevaluator.Function;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.spi.JsonUtils;

import org.gvsig.expressionevaluator.spi.AbstractFunction;

import org.gvsig.tools.util.GetItem;

import org.gvsig.tools.util.GetItem64;

import org.gvsig.tools.util.GetItemByKey;

import org.gvsig.tools.util.PropertiesSupport;

public class GetitemFunction extends AbstractFunction {

    public GetitemFunction() {

        super(Function.GROUP_PROGRAMMING, 

            FUNCTION_GETITEM, 

            Range.is(2),

            null,

            FUNCTION_GETITEM+"({{object}}, index_or_key)",

            null,

            "Object",

            false

        );

    }

    @Override

    public boolean allowConstantFolding() {

        return false;

    }

    @Override

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

        Object value;

        Object obj = getObject(args, 0);

        if( obj == null ) {

            throw new ExpressionRuntimeException("Null is not valid in "+FUNCTION_GETITEM+" function.");

        }

        if( obj instanceof List ) {

            List l = ((List)obj);

            int index = getInt(args,1);

            if( index<0 ) {

                index = l.size() + index;

            }

            value = l.get(index);

        } else if( obj instanceof Object[] ) {

            Object[] a = ((Object[])obj);

            int index = getInt(args,1);

            if( index<0 ) {

                index = a.length + index;

            }

            value = a[index];

        } else if( obj instanceof CharSequence ) {

            CharSequence s = (CharSequence)obj;

            int index = getInt(args,1);

            if( index<0 ) {

                index = s.length() + index;

            }

            value = s.subSequence(index, index+1);

        } else if( obj instanceof Map ) {

            Map m = (Map)obj;

            Object index = getObject(args,1);

            value = m.get(index);

        } else if( obj instanceof JsonArray ) {

            JsonArray jsonArray = ((JsonArray)obj);

            int index = getInt(args,1);

            if( index<0 ) {

                index = jsonArray.size() + index;

            }

            value = JsonUtils.getitem(jsonArray, index);

        } else if( obj instanceof JsonObject ) {

            JsonObject jsonobj = ((JsonObject)obj);

            String name = getStr(args,1);

            value = JsonUtils.getitem(jsonobj, name);

        } else if( obj instanceof GetItem ) {

            GetItem l = (GetItem)obj;

            int index = getInt(args,1);

            value = l.get(index);

        } else if( obj instanceof GetItem64 ) {

            GetItem64 l = (GetItem64)obj;

            long index = getLong(args,1);

            value = l.get64(index);

        } else if( obj instanceof GetItemByKey ) {

            GetItemByKey l = (GetItemByKey)obj;

            Object key = getObject(args,1);

            value = l.get(key);

        } else if( obj instanceof PropertiesSupport ) {

            PropertiesSupport l = (PropertiesSupport)obj;

            String propName = getStr(args,1);

            value = l.getProperty(propName);

        } else {

            throw new ExpressionRuntimeException("The "+FUNCTION_GETITEM+" function require a String, List or Map and a received a '"+obj.getClass().getSimpleName()+"'.");

        }

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

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

        builder.append("[");

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

        builder.append("]");

        return builder.toString();

    }

}