Application: gvSIG desktop

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

import javax.json.JsonObject;

import org.apache.commons.lang3.Range;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.I18N;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.spi.AbstractFunction;

import org.gvsig.expressionevaluator.spi.JsonUtils;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

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

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

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.feature.EditableFeatureType;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.dal.feature.NewFeatureStoreParameters;

import org.gvsig.fmap.dal.spi.DataStoreProviderServices;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public abstract class AbstractFeatureFunction extends AbstractFunction {

  protected AbstractFeatureFunction(String group, String name, Range argc, String description, String template, String[] descriptionArgs, String returnType, boolean sqlCompatible) {

    super(group, name, argc, description, template, descriptionArgs, returnType, sqlCompatible);

  }

  protected AbstractFeatureFunction(String group, String name, Range argc, String description, String template, String[] descriptionArgs, String returnType) {

    super(group, name, argc, description, template, descriptionArgs, returnType);

  }

  public Feature current_row(Interpreter interpreter) {

    try {

      Feature f = (Feature) interpreter.call(FUNCTION_CURRENT_ROW);

      return f;

    } catch (Exception ex) {

      return null;

    }

  }

  public FeatureStore current_store(Interpreter interpreter) {

    try {

        FeatureStore store = (FeatureStore) interpreter.call(FUNCTION_CURRENT_STORE);

      return store;

    } catch (Exception ex) {

      return null;

    }

  }

  protected NewFeatureStoreParameters getNewFeatureStoreParameters(Object[] args, int n) {      

      Object v = getObject(args, n);

      if( v instanceof NewFeatureStoreParameters ) {

          return (NewFeatureStoreParameters) v;

      }

      NewFeatureStoreParameters parameters;

      JsonObject jsonObj = getJsonObject(args, n);

      try {

        String providerName = jsonObj.getString(DataStoreProviderServices.PROVIDER_PARAMTER_NAME);

        DataManager dataManager = DALLocator.getDataManager();

        parameters = (NewFeatureStoreParameters) dataManager.createNewStoreParameters(providerName);

        for (String paramName : jsonObj.keySet()) {

            if( StringUtils.equalsIgnoreCase(paramName, DataStoreProviderServices.PROVIDER_PARAMTER_NAME)) {

                continue;

            }

            parameters.setDynValue(paramName, JsonUtils.getitem(jsonObj, paramName));

        }

      } catch (Exception ex) {

        String type = v.getClass().getCanonicalName();

        throw new IllegalArgumentException(

                I18N.The_type_of_the_argument_XargnX_for_the_XIdentifierX_function_is_incorrect(name(), n) + " " +

                I18N.Expected_XexpectedX_and_found_XfoundX("NewFeatureStoreParameters",type)

        );

      }

      return parameters;

  }

  protected FeatureType getFeatureType(Object[] args, int n) {      

      Object v = getObject(args, n);

      if( v instanceof FeatureType ) {

          return (FeatureType) v;

      }

      EditableFeatureType featureType;

      JsonObject jsonObj = getJsonObject(args, n);

      try {

        DataManager dataManager = DALLocator.getDataManager();

        featureType = dataManager.createFeatureType(jsonObj);

      } catch (Exception ex) {

        String type = v.getClass().getCanonicalName();

        throw new IllegalArgumentException(

                I18N.The_type_of_the_argument_XargnX_for_the_XIdentifierX_function_is_incorrect(name(), n) + " " +

                I18N.Expected_XexpectedX_and_found_XfoundX("NewFeatureStoreParameters",type)

        );

      }

      return featureType;      

  }

}