Application: gvSIG desktop

package org.gvsig.oracle.extension;

import javax.swing.table.TableModel;

import jwizardcomponent.FinishAction;

import jwizardcomponent.JWizardComponents;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.messages.NotificationManager;

import org.gvsig.andami.ui.mdiManager.IWindow;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataTypes;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.exception.ValidateDataParametersException;

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

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

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.store.jdbc.JDBCServerExplorer;

import org.gvsig.fmap.dal.store.jdbc.JDBCServerExplorerParameters;

import org.gvsig.fmap.dal.store.jdbc.JDBCStoreParameters;

import org.gvsig.fmap.dal.store.oracle.OracleNewStoreParameters;

import org.gvsig.fmap.dal.store.oracle.OracleValues;

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.exceptions.LoadLayerException;

import org.gvsig.fmap.mapcontext.layers.LayerFactory;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLayerVectorialDB;

import org.gvsig.oracle.gui.NewOracleTableWizard;

import org.gvsig.oracle.gui.RepeatedChooseGeometryTypePanel;

import org.gvsig.oracle.gui.RepeatedFieldDefinitionPanel;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * 

 * @author vsanjaime

 * 

 */

public class NewOracleTableFinishAction extends FinishAction {

        @SuppressWarnings("unused")

        private static Logger logger = LoggerFactory

                        .getLogger(NewOracleTableFinishAction.class.getName());

        private NewOracleTableWizard connectionPanel;

        private RepeatedChooseGeometryTypePanel geoTypePanel;

        private RepeatedFieldDefinitionPanel fieldsPanel;

        private MapContext theMapContext;

        private IWindow closeThis;

        /**

         * Constructor

         * 

         * @param wc

         * @param closeit

         * @param connPanel

         * @param mc

         */

        public NewOracleTableFinishAction(JWizardComponents wc, IWindow closeit,

                        NewOracleTableWizard connPanel, MapContext mc) {

                super(wc);

                theMapContext = mc;

                closeThis = closeit;

                connectionPanel = connPanel;

                geoTypePanel = (RepeatedChooseGeometryTypePanel) wc.getWizardPanel(0);

                fieldsPanel = (RepeatedFieldDefinitionPanel) wc.getWizardPanel(1);

        }

        /**

         * run process

         */

        public void performAction() {

                PluginServices.getMDIManager().closeWindow(closeThis);

                // Table name

                String tableName = geoTypePanel.getLayerName().toUpperCase();

                // Connection

                JDBCServerExplorerParameters explorerParams = (JDBCServerExplorerParameters) connectionPanel

                                .getSelectedServerExplorer().getDbSeverExplorerParameters();

                FeatureStore target = null;

                try {

                        DataManager dataManager = DALLocator.getDataManager();

                        JDBCServerExplorer explorer = (JDBCServerExplorer) dataManager

                                        .createServerExplorer(explorerParams);

                        NewFeatureStoreParameters newStoreParams = (NewFeatureStoreParameters) explorer

                                        .getAddParameters();

                        ((JDBCStoreParameters) newStoreParams).setTable(tableName);

                        EditableFeatureType eFType = (EditableFeatureType) newStoreParams

                                        .getDefaultFeatureType();

                        TableModel model = fieldsPanel.getFieldsDescription();

                        FeatureType fType = createDefaultFeatureType(eFType, model);

                        ((OracleNewStoreParameters) newStoreParams)

                                        .setDefaultFeatureType(fType);

                        explorer.add((OracleNewStoreParameters) newStoreParams, true);

                        target = (FeatureStore) dataManager

                                        .createStore((OracleNewStoreParameters) newStoreParams);

                } catch (DataException e) {

                        NotificationManager.addError(e.getMessage(), e);

                } catch (ValidateDataParametersException e) {

                        NotificationManager.addError(e.getMessage(), e);

                        e.printStackTrace();

                }

                LayerFactory factory = LayerFactory.getInstance();

                FLayerVectorialDB lyr = null;

                try {

                        lyr = (FLayerVectorialDB) factory.createLayer(tableName, target);

                } catch (LoadLayerException e) {

                        NotificationManager.addError(e.getMessage(), e);

                        e.printStackTrace();

                }

                lyr.setVisible(true);

                theMapContext.getLayers().addLayer(lyr);

        }

        /**

         * Validate field name

         * 

         * @param field

         *            name

         * @return field name

         */

        private String validFieldName(String name) {

                if ((name

                                .compareToIgnoreCase(OracleValues.DEFAULT_ID_FIELD_CASE_SENSITIVE) == 0)

                                || (name.compareToIgnoreCase(OracleValues.ORACLE_ID_FIELD) == 0)) {

                        return name + "_";

                }

                return name;

        }

        /**

         * Build feature type from field table model

         * 

         * @param eFType

         * @param model

         * @return

         */

        private FeatureType createDefaultFeatureType(EditableFeatureType eFType,

                        TableModel model) {

                // GID Field

                EditableFeatureAttributeDescriptor gid = eFType.add(

                                OracleValues.DEFAULT_ID_FIELD_CASE_SENSITIVE, DataTypes.INT);

                gid.setIsAutomatic(true);

                gid.setSize(7);

                gid.setPrecision(0);

                // Other fields

                for (int i = 0; i < model.getRowCount(); i++) {

                        String fieldName = (String) model.getValueAt(i, 0);

                        fieldName = validFieldName(fieldName);

                        String fieldType = (String) model.getValueAt(i, 1);

                        String fieldSize = (String) model.getValueAt(i, 2);

                        int intFieldSize = new Integer(fieldSize).intValue();

                        EditableFeatureAttributeDescriptor newfield = null;

                        if (fieldType.compareToIgnoreCase("String") == 0) {

                                newfield = eFType.add(fieldName, DataTypes.STRING);

                                newfield.setSize(intFieldSize);

                        } else if (fieldType.compareToIgnoreCase("Boolean") == 0) {

                                newfield = eFType.add(fieldName, DataTypes.BOOLEAN);

                                newfield.setSize(intFieldSize);

                        } else if (fieldType.compareToIgnoreCase("Date") == 0) {

                                newfield = eFType.add(fieldName, DataTypes.DATE);

                                newfield.setSize(intFieldSize);

                        } else if (fieldType.compareToIgnoreCase("Integer") == 0) {

                                newfield = eFType.add(fieldName, DataTypes.INT);

                                newfield.setSize(intFieldSize);

                        } else if (fieldType.compareToIgnoreCase("Double") == 0) {

                                newfield = eFType.add(fieldName, DataTypes.DOUBLE);

                                newfield.setSize(intFieldSize);

                        }

                }

                // GEOMETRY Field

                EditableFeatureAttributeDescriptor geometry = eFType.add(

                                OracleValues.DEFAULT_GEO_FIELD, DataTypes.GEOMETRY);

                int geom = geoTypePanel.getSelectedGeometryType();

                geometry.setGeometryType(geom);

                eFType.setDefaultGeometryAttributeName(OracleValues.DEFAULT_GEO_FIELD);

                return eFType.getNotEditableCopy();

        }

}