Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * 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 3

 * 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., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.newlayer.prov.jdbc;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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

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

import org.gvsig.newlayer.NewLayerException;

import org.gvsig.newlayer.NewLayerProviderPanel;

import org.gvsig.newlayer.NewLayerService;

import org.gvsig.newlayer.NewLayerWizard;

import org.gvsig.newlayer.prov.jdbc.panels.GeometryIndexPanel;

import org.gvsig.newlayer.prov.jdbc.panels.IdentifiersOptionsPanel;

import org.gvsig.newlayer.prov.jdbc.panels.JDBCConnectionPanel;

import org.gvsig.newlayer.prov.jdbc.panels.PermissionsPanel;

import org.gvsig.newlayer.prov.jdbc.panels.PostCreatingStatementPanel;

import org.gvsig.newlayer.prov.jdbc.panels.SelectTableNamePanel;

import org.gvsig.newlayer.spi.AbstractNewLayerProvider;

/**

 * NewLayer provider to create a JDBC table.

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class NewLayerJDBCProvider extends AbstractNewLayerProvider {

    private final NewLayerJDBCProviderParameters parameters;

    protected NewLayerJDBCProvider(NewLayerService service) {

        super(service);

        this.parameters = new NewLayerJDBCProviderParameters();

    }

    @Override

    public List<NewLayerProviderPanel> createPanels(NewLayerWizard wizard) {

        List<NewLayerProviderPanel> panels = new ArrayList<>();

        panels.add(new IdentifiersOptionsPanel(wizard, this.parameters));

        panels.add(new JDBCConnectionPanel(wizard, this.parameters));

        panels.add(new SelectTableNamePanel(wizard, this.parameters));

        panels.add(wizard.createFeatureTypePanel());

        panels.add(new GeometryIndexPanel(wizard, this.parameters));

        panels.add(new PermissionsPanel(wizard, this.parameters));

        panels.add(new PostCreatingStatementPanel(wizard, this.parameters));

        return panels;

    }

    @Override

    public EditableFeatureType getFeatureType() {

        return this.parameters.getFeatureType();

    }

    public NewLayerJDBCProviderParameters getParameters() {

        return this.parameters;

    }

    private JDBCServerExplorer createServerExplorer() throws NewLayerException {

        try {

            DataManager dataManager = DALLocator.getDataManager();

            JDBCServerExplorer explorer = (JDBCServerExplorer) dataManager.openServerExplorer(

                    this.getParameters().getExplorerParameters().getExplorerName(),

                    this.getParameters().getExplorerParameters()

            );

            return explorer;

        } catch (Exception e) {

            throw new NewLayerException(e);

        }

    }

    @Override

    public NewFeatureStoreParameters getNewStoreParameters() throws NewLayerException {

        try {

            EditableFeatureType featureType = this.parameters.getFeatureType();

            JDBCServerExplorer explorer = this.createServerExplorer();

            String pkname = getParameters().getPrimaryKey();

            if (pkname != null) {

                // pkname it's not part of exportattributes list.

                EditableFeatureAttributeDescriptor pk = (EditableFeatureAttributeDescriptor) featureType.get(pkname);

                if (pk == null) {

                    pk = featureType.add(pkname, DataTypes.LONG);

                    pk.setIsPrimaryKey(true);

                    pk.setIsAutomatic(true);

                } else {

                    pk.setIsPrimaryKey(true);

                }

            }

            EditableFeatureAttributeDescriptor attrdescriptor = (EditableFeatureAttributeDescriptor) featureType.getDefaultGeometryAttribute();

            if (attrdescriptor != null && this.getParameters().getCreateIndexInGeometryRow()) {

                attrdescriptor.setIsIndexed(true);

            }

            // ======================================================

            // Reprojection: set SRS of geometry field to target SRS

            if (attrdescriptor != null) {

                attrdescriptor.setSRS(this.getParameters().getProjection());

            }

            // ======================================

            JDBCNewStoreParameters createTableParams = (JDBCNewStoreParameters) explorer.getAddParameters();

            createTableParams.setSelectRole(this.getParameters().getSelectRole());

            createTableParams.setInsertRole(this.getParameters().getInsertRole());

            createTableParams.setUpdateRole(this.getParameters().getUpdateRole());

            createTableParams.setDeleteRole(this.getParameters().getDeleteRole());

            createTableParams.setTruncateRole(this.getParameters().getTruncateRole());

            createTableParams.setReferenceRole(this.getParameters().getReferenceRole());

            createTableParams.setTriggerRole(this.getParameters().getTriggerRole());

            createTableParams.setAllRole(this.getParameters().getAllRole());

            createTableParams.setSchema(this.getParameters().getSchema());

            createTableParams.setPostCreatingStatement(this.getParameters().getPostCreatingStatement());

            createTableParams.setDefaultFeatureType(featureType);

            createTableParams.setTable(this.getParameters().getTableName());

            return createTableParams;

        } catch (Exception ex) {

            throw new NewLayerException(ex);

        }

    }

    @Override

    public DataStoreParameters getOpenStoreParameters() throws NewLayerException {

        try {

            DataManager dataManager = DALLocator.getDataManager();

            JDBCServerExplorer explorer = this.createServerExplorer();

            JDBCStoreParameters openStoreParameters = (JDBCStoreParameters) explorer.getOpenParameters();

            openStoreParameters.setSchema(this.getParameters().getSchema());

            openStoreParameters.setTable(this.getParameters().getTableName());

            openStoreParameters.setCRS(this.getParameters().getProjection());

            openStoreParameters.setDefaultGeometryField(

                    this.getParameters().getFeatureType().getDefaultGeometryAttributeName()

            );

            return openStoreParameters;

        } catch (DataException ex) {

            throw new NewLayerException(ex);

        }

    }

    @Override

    public DataServerExplorer getExplorer() throws NewLayerException {

        return this.createServerExplorer();

    }

    @Override

    public String getStoreName() {

        return this.parameters.getExplorerParameters().getProviderName();

    }

}