Application: gvSIG desktop

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2006 Prodevelop and Generalitat Valenciana.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *   Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *   +34 963862235

 *   gvsig@gva.es

 *   www.gvsig.gva.es

 *

 *    or

 *

 *   Prodevelop Integraci?n de Tecnolog?as SL

 *   Conde Salvatierra de ?lava , 34-10

 *   46004 Valencia

 *   Spain

 *

 *   +34 963 510 612

 *   +34 963 510 968

 *   gis@prodevelop.es

 *   http://www.prodevelop.es

 */

package org.gvsig.oracle.extension;

import java.awt.Component;

import javax.swing.JComponent;

import javax.swing.JOptionPane;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.messages.NotificationManager;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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

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

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

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

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.oracle.OracleServerExplorer;

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

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

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.FLayer;

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

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

import org.gvsig.geodb.vectorialdb.wizard.MyExplorer;

import org.gvsig.oracle.gui.OracleConnectionChooserPanel;

import org.gvsig.utils.PostProcessSupport;

import org.jfree.util.Log;

import org.opengis.feature.FeatureAttributeDescriptor;

/**

 * Writes a layer as an Oracle table.

 * 

 * @author jldominguez, vsanjaime

 * 

 */

public class ExportToOracle {

        /**

         * Constructor

         */

        public ExportToOracle() {

                System.out.println("Start export process...");

        }

        /**

         * run export process

         * 

         * @param mapContext

         * @param layer

         */

        public void toOracle(MapContext mapContext, FLyrVect layer) {

                // name of oracle table

                String tableName = this.getNewOracleTableName();

                if (tableName == null) {

                        return;

                }

                // name of oracle table

                String pkFieldName = "";

                try {

                        pkFieldName = this.getNewOracleTablePK(layer);

                } catch (Exception e) {

                        Log.warn("There are not primakey fields in this data store");

                        pkFieldName = null;

                }

                // Selected explorer (connection)

                MyExplorer explorer = this.getSelectedExplorer();

                if (explorer == null) {

                        return;

                }

                //

                OracleServerExplorerParameters explorerParams = (OracleServerExplorerParameters) explorer

                                .getDbSeverExplorerParameters();

                this.createNewOracleStore(mapContext, layer, explorerParams, tableName,

                                pkFieldName);

        }

        /**

         * create new oracle store

         * 

         * @param mapContext

         * @param layer

         * @param exParams

         * @param tableName

         * @param pkName

         */

        private void createNewOracleStore(MapContext mapContext, FLyrVect layer,

                        OracleServerExplorerParameters explorerParams, String tableName,

                        String pkName) {

                try {

                        DataManager dataManager = DALLocator.getDataManager();

                        DataServerExplorer explorer = dataManager

                                        .createServerExplorer(explorerParams);

                        OracleServerExplorer oraExplorer = (OracleServerExplorer) explorer;

                        NewFeatureStoreParameters newOracleStoreParams = (NewFeatureStoreParameters) oraExplorer

                                        .getAddParameters();

                        ((OracleStoreParameters) newOracleStoreParams).setTable(tableName);

                        boolean exportOk = export(oraExplorer, newOracleStoreParams, layer

                                        .getFeatureStore(), pkName);

                        if (exportOk) {

                                openLayer(newOracleStoreParams, tableName, mapContext);

                        }

                } catch (DataException e) {

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

                } catch (ValidateDataParametersException e) {

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

                        e.printStackTrace();

                }

        }

        /**

         * show layer in the view

         * 

         * @param newParams

         * @param name

         * @param mapContext

         * @throws ReadException

         */

        private void openLayer(NewFeatureStoreParameters newParams, String name,

                        MapContext mapContext) throws ReadException {

                int res = JOptionPane.showConfirmDialog((JComponent) PluginServices

                                .getMDIManager().getActiveWindow(), PluginServices.getText(

                                this, "insertar_en_la_vista_la_capa_creada"), PluginServices

                                .getText(this, "insertar_capa"), JOptionPane.YES_NO_OPTION);

                if (res == JOptionPane.YES_OPTION) {

                        PostProcessSupport.executeCalls();

                        try {

                                FLayer newLayer = LayerFactory.getInstance().createLayer(name,

                                                newParams);

                                mapContext.getLayers().addLayer(newLayer);

                        } catch (LoadLayerException e) {

                                throw new ReadException("Load layer", e);

                        }

                }

        }

        /**

         * Create new fields and fill data

         * 

         * @param explorer

         * @param params

         * @param set

         * @param featureStore

         * @param pkName

         * @return

         */

        private boolean export(OracleServerExplorer explorer,

                        NewFeatureStoreParameters params, FeatureStore featureStore,

                        String pkName) {

                DisposableIterator it1 = null;

                try {

                        EditableFeatureType efType = featureStore.getDefaultFeatureType()

                                        .getEditable();

                        // if GID field not exists, create it

                        boolean gidFound = false;

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

                                EditableFeatureAttributeDescriptor edesc = (EditableFeatureAttributeDescriptor) efType

                                                .getAttributeDescriptor(i);

                                if (edesc.getName().compareTo(

                                                OracleValues.DEFAULT_ID_FIELD_CASE_SENSITIVE) == 0) {

                                        gidFound = true;

                                        break;

                                }

                        }

                        if (!gidFound) {

                                EditableFeatureAttributeDescriptor gid = efType.add(

                                                OracleValues.DEFAULT_ID_FIELD_CASE_SENSITIVE,

                                                DataTypes.INT);

                                gid.setIsAutomatic(true);

                                gid.setSize(7);

                                gid.setPrecision(0);

                        }

                        if (pkName != null) {

                                EditableFeatureAttributeDescriptor pk = efType.add(pkName,

                                                DataTypes.LONG);

                                pk.setIsPrimaryKey(true);

                                pk.setIsAutomatic(true);

                        }

                        efType

                                        .setDefaultGeometryAttributeName(OracleValues.DEFAULT_GEO_FIELD);

                        params.setDefaultFeatureType(efType.getNotEditableCopy());

                        explorer.add(params, true);

                        DataManager manager = DALLocator.getDataManager();

                        FeatureStore target = (FeatureStore) manager.createStore(params);

                        FeatureType targetType = target.getDefaultFeatureType();

                        target.edit(FeatureStore.MODE_APPEND);

                        FeatureSet set = featureStore.getFeatureSet();

                        it1 = set.iterator();

                        while (it1.hasNext()) {

                                Feature feature = (Feature) it1.next();

                                EditableFeature efeat = target.createNewFeature(targetType,

                                                feature);

                                target.insert(efeat);

                        }

                        set.dispose();

                        target.finishEditing();

                        target.dispose();

                        return true;

                } catch (Exception e) {

                        NotificationManager.showMessageError(e.getLocalizedMessage(), e);

                        return false;

                } finally {

                        if (it1 != null) {

                                it1.dispose();

                        }

                }

        }

        /**

         * get new oracle table name

         * 

         * @return

         */

        private String getNewOracleTableName() {

                String tableName = "";

                boolean valid_name = false;

                while (!valid_name) {

                        tableName = JOptionPane.showInputDialog(((Component) PluginServices

                                        .getMainFrame()), PluginServices.getText(this,

                                        "intro_tablename"));

                        valid_name = ((tableName == null) || ((tableName.length() <= (OracleValues.MAX_ID_LENGTH - 3))

                                        && (tableName.indexOf(" ") == -1) && (tableName.length() > 0)));

                        if (!valid_name) {

                                if (tableName.length() > (OracleValues.MAX_ID_LENGTH - 3)) {

                                        JOptionPane.showMessageDialog(null, PluginServices.getText(

                                                        this, "nombre_demasiado_largo"), PluginServices

                                                        .getText(this, "error"), JOptionPane.ERROR_MESSAGE);

                                } else {

                                        JOptionPane.showMessageDialog(null, PluginServices.getText(

                                                        this, "nombre_no_valido"), PluginServices.getText(

                                                        this, "error"), JOptionPane.ERROR_MESSAGE);

                                }

                        }

                }

                return tableName.toUpperCase();

        }

        /**

         * get new oracle table pk field

         * 

         * @return

         * @throws DataException

         * @throws ReadException

         */

        private String getNewOracleTablePK(FLyrVect layer) throws ReadException,

                        DataException {

                FeatureStore store = layer.getFeatureStore();

                FeatureType type = store.getDefaultFeatureType();

                FeatureAttributeDescriptor[] pks = (FeatureAttributeDescriptor[]) type

                                .getPrimaryKey();

                String pkName = null;

                if (pks == null || pks.length < 1) {

                        pkName = JOptionPane.showInputDialog(PluginServices.getText(this,

                                        "input_pk_field_name"));

                }

                return pkName;

        }

        /**

         * get parameters of selected explorer

         * 

         * @return

         */

        private MyExplorer getSelectedExplorer() {

                OracleConnectionChooserPanel dlg = new OracleConnectionChooserPanel();

                PluginServices.getMDIManager().addWindow(dlg);

                dlg.validate();

                if (!dlg.isOkPressed()) {

                        return null;

                }

                return dlg.getSelectedServerExplorer();

        }

}