Application: gvSIG desktop

package org.gvsig.fmap.dal.complements.relatedfeatures;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Collections;

import java.util.List;

import org.apache.commons.lang3.ArrayUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.StoresRepository;

import org.gvsig.fmap.dal.complements.RelatedFeatures;

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.complement.AbstractComplementFactory;

import org.gvsig.tools.complement.Complement;

import org.gvsig.tools.complement.ComplementsManager;

import org.gvsig.tools.dynobject.AbstractDynMethod;

import org.gvsig.tools.dynobject.DynMethod;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.dynobject.DynObjectManager;

import org.gvsig.tools.dynobject.DynObjectValueItem;

import org.gvsig.tools.dynobject.Tagged;

import org.gvsig.tools.dynobject.Tags;

import org.gvsig.tools.dynobject.exception.DynMethodException;

import org.gvsig.tools.dynobject.exception.DynMethodNotSupportedException;

/**

 *

 * @author jjdelcerro

 */

public class RelatedFeaturesFactory extends AbstractComplementFactory<Tagged>{

    public RelatedFeaturesFactory() {

        super(RelatedFeatures.COMPLEMENT_MANE);

    }

    @Override

    public Complement<Tagged> createComplement(Tagged object) {

        return new RelatedFeaturesImpl(this, object);

    }

    private static class RelatedTableTagValues extends AbstractDynMethod {

        public RelatedTableTagValues() {

            super("RelatedTableTagValues");

        }

        @Override

        public Object invoke(DynObject self, Object[] args) throws DynMethodException {

            StoresRepository repository = DALLocator.getDataManager().getStoresRepository();

            List<String> tableNames = new ArrayList<>();

            for (String tableName : repository.keySet()) {

                tableNames.add(tableName);

            }

            Collections.sort(tableNames);

            DynObjectValueItem[] values = new DynObjectValueItem[tableNames.size()];

            for (int i = 0; i < tableNames.size(); i++) {

                values[i] = new DynObjectValueItem(tableNames.get(i),tableNames.get(i));

            }

            return values;

        }

    }

    private static class RelatedUniqueFieldNameTagValues extends AbstractDynMethod {

        public RelatedUniqueFieldNameTagValues() {

            super("RelatedUniqueFieldNameTagValues");

        }

        @Override

        public Object invoke(DynObject self, Object[] args) throws DynMethodException {

            if( ArrayUtils.isEmpty(args) ) {

                return null;

            }

            if( !(args[0] instanceof Tags) ) {

                return null;

            }

            Tags tags = (Tags) args[0];

            String tableName = tags.getString(RelatedFeatures.DAL_RELATED_TABLE, null);

            if( StringUtils.isBlank(tableName) ) {

                return null;

            }

            StoresRepository repository = DALLocator.getDataManager().getStoresRepository();

            FeatureType ft = repository.getFeatureType(tableName);

            if( ft == null ) {

                return null;

            }

            List<String> attrNames = new ArrayList<>();

            for (FeatureAttributeDescriptor attr : ft) {

                attrNames.add(attr.getName());

            }

            Collections.sort(attrNames);

            DynObjectValueItem[] values = new DynObjectValueItem[attrNames.size()];

            for (int i = 0; i < attrNames.size(); i++) {

                values[i] = new DynObjectValueItem(attrNames.get(i),attrNames.get(i));

            }

            return values;

        }

    }

    public static void selfRegister() {

        ComplementsManager complementsManager = ToolsLocator.getComplementsManager();

        complementsManager.registerComplement(new RelatedFeaturesFactory());

        DynObjectManager dynObjectManager = ToolsLocator.getDynObjectManager();

        dynObjectManager.registerTag(

                RelatedFeatures.DAL_RELATED_COLUMNS, 

                "Part of a table of related features definition, stores the name of the fields to be shown in the table separated by two points."

        );

        dynObjectManager.registerTag(

                RelatedFeatures.DAL_RELATED_TABLE, 

                "Part of a table of related features definition, store the table name of features.",

                new RelatedTableTagValues()

        );

        dynObjectManager.registerTag(

                RelatedFeatures.DAL_RELATED_UNIQUE_FIELD_NAME, 

                "Part of a table of related features definition, store the name of the field that is a unique key in the features table.",

                new RelatedUniqueFieldNameTagValues()

        );

//        dynObjectManager.registerTag(

//                RelatedFeatures.DAL_RELATED_FOREING_KEY_NAME, 

//                "Part of a table of related features definition, store the name of the field that is the foreing key in the features table."

//        );

    }

}