Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

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

import java.util.Objects;

import java.util.function.Function;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.Codes;

import org.gvsig.expressionevaluator.ExpressionBuilder;

import static org.gvsig.expressionevaluator.impl.function.dataaccess.SelectFunction.PROP_FEATURETYPESUPPLIER;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.SQLBuilder;

import static org.gvsig.fmap.dal.SQLBuilder.PROP_SQLBUILDER;

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

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

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

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

/**

 *

 * @author fdiaz

 */

public class GetattrFunction extends org.gvsig.expressionevaluator.impl.function.programming.GetattrFunction {

    @Override

    public ExpressionBuilder.Value toValue(ExpressionBuilder builder, Codes args) {

        try {

            SQLBuilder sqlBuilder = (SQLBuilder) builder.getProperty(PROP_SQLBUILDER);

            if(sqlBuilder == null){

                return super.toValue(builder, args);

            }

            if(!(args.get(0) instanceof Code.Identifier)){

                return super.toValue(builder, args);

            }

            if(!(args.get(1) instanceof Code.Constant)){

                return super.toValue(builder, args);

            }

            Function<String, FeatureType> featureTypeSupplier = (Function<String, FeatureType>) builder.getProperty(PROP_FEATURETYPESUPPLIER);

            if (featureTypeSupplier == null) {

                featureTypeSupplier = (String tableName) -> {

                    DataManager dataManager = DALLocator.getDataManager();

                    FeatureType featureType = dataManager.getStoresRepository().getFeatureType(tableName);

                    return featureType;

                };

            }

            String tableName = ((Code.Identifier)args.get(0)).name();

            String columnName = Objects.toString(((Code.Constant)args.get(1)).value(), null);

            if(StringUtils.isBlank(tableName) || StringUtils.isBlank(columnName)){

                return super.toValue(builder, args);

            }

            FeatureAttributeDescriptor attr = null;

            FeatureQuery query = (FeatureQuery) builder.getProperty(SQLBuilder.PROP_QUERY);

            if(query != null){

                attr = query.getExtraColumns().get(columnName);

            }

            if(attr == null) {

                String builderTableName = (String) builder.getProperty(SQLBuilder.PROP_TABLENAME);

                FeatureType featureType = null;

                if (StringUtils.equalsIgnoreCase(builderTableName, tableName)) {

                    featureType = (FeatureType) builder.getProperty(SQLBuilder.PROP_FEATURE_TYPE);

                } else {

                    featureType = featureTypeSupplier.apply(tableName);

                }

                if (featureType == null) {

                    return super.toValue(builder, args);

                }

                attr = featureType.getAttributeDescriptor(columnName);

                if (attr == null) {

                    return super.toValue(builder, args);

                }

            }

            SQLBuilder.TableNameBuilder table = sqlBuilder.createTableNameBuilder();

            table.name(tableName);

            SQLBuilder.Column column = sqlBuilder.column(table, columnName);

            return column;

        } catch (Exception ex) {

            return super.toValue(builder, args);

        }

    }

}