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 org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.Codes;

import org.gvsig.expressionevaluator.ExpressionBuilder;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.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("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);

            }

            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("Query");

            if(query != null){

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

            }

            if(attr == null) {

                String builderTable = (String) builder.getProperty("Table");

                FeatureType featureType = null;

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

                    featureType = (FeatureType) builder.getProperty("FeatureType");

                } else {

                    DataManager dataManager = DALLocator.getDataManager();

                    featureType = dataManager.getStoresRepository().getFeatureType(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);

        }

    }

}