| gvSIG desktop 1

/* 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 com.iver.cit.gvsig.sde.gui.sdewizard2;

import java.awt.BorderLayout;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.awt.geom.Rectangle2D;

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.TreeMap;

import java.util.Vector;

import javax.swing.DefaultListModel;

import javax.swing.ImageIcon;

import javax.swing.JComboBox;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import javax.swing.JScrollPane;

import javax.swing.ListSelectionModel;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import org.apache.log4j.Logger;

import org.cresques.cts.IProjection;

import org.gvsig.gui.beans.swing.JButton;

import com.esri.sde.sdk.client.SeConnection;

import com.esri.sde.sdk.client.SeException;

import com.esri.sde.sdk.client.SeLayer;

import com.hardcode.driverManager.DriverLoadException;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.ui.mdiManager.IWindow;

import com.iver.cit.gvsig.fmap.MapContext;

import com.iver.cit.gvsig.fmap.core.ICanReproject;

import com.iver.cit.gvsig.fmap.crs.CRSFactory;

import com.iver.cit.gvsig.fmap.drivers.sde.ArcSDELayerDefinition;

import com.iver.cit.gvsig.fmap.drivers.sde.VectorialSDEDriver;

import com.iver.cit.gvsig.fmap.drivers.sde.VectorialSDELayerDefinition;

import com.iver.cit.gvsig.fmap.layers.FLayer;

import com.iver.cit.gvsig.fmap.layers.FLayers;

import com.iver.cit.gvsig.fmap.layers.LayerFactory;

import com.iver.cit.gvsig.gui.WizardPanel;

import com.iver.cit.gvsig.project.documents.view.gui.View;

import com.iver.utiles.swing.JPasswordDlg;

/**

 * Driver-independent GeoBD wizard. Queries the drivers to fill GUI controls.

 * Multi-table selection available.

 *

 * @author jldominguez

 *

 */

public class WizardSDE extends WizardPanel implements ActionListener,

    ListSelectionListener {

    private static Logger logger = Logger.getLogger(WizardSDE.class.getName());

    private SeConnection conex = null;

    private ConnectionWithParamsSDE selectedDataSource = null;

    private JPanel namePanel = null;

    private JPanel tablesPanel = null;

    private JScrollPane tablesScrollPane = null;

    private AvailableTablesCheckBoxList tablesList = null;

    private JComboBox datasourceComboBox = null;

    private UserTableSettingsPanel settingsPanel = null;

    private UserSelectedFieldsPanel fieldsPanel = null;

    private UserTableSettingsPanel emptySettingsPanel = null;

    private UserSelectedFieldsPanel emptyFieldsPanel = null;

    private JButton jdbcButton = null;

    private View view = null;

    public WizardSDE() {

        super();

        initialize();

    }

    /**

     * This method initializes this

     *

     * @return void

     */

    private void initialize() {

        setTabName("SDE");

        setLayout(null);

        setSize(512, 478);

        IWindow iw = PluginServices.getMDIManager().getActiveWindow();

        if (iw == null) {

            return;

        }

        if (!(iw instanceof View)) {

            return;

        }

        view = (View) iw;

        setMapCtrl(view.getMapControl());

        emptySettingsPanel = new UserTableSettingsPanel(null, null, "",

                getMapCtrl(), true, this);

        emptyFieldsPanel = new UserSelectedFieldsPanel(null, null, true, this);

        add(getNamePanel(), null);

        loadJdbcDatasourcesCombo();

        add(getTablesPanel(), null);

    }

    private void loadJdbcDatasourcesCombo() {

        getDatasourceComboBox().removeAllItems();

        getDatasourceComboBox().addItem(new ConnectionWithParamsSDE());

        ConnectionWithParamsSDE[] conn = SingleSDEConnectionManager.instance()

                                                                 .getAllConnections();

        if (conn == null) {

            return;

        }

        for (int i = 0; i < conn.length; i++) {

            getDatasourceComboBox().addItem(conn[i]);

        }

    }

    private String[] getDriverNames() {

        Class[] classes = new Class[] { VectorialSDEDriver.class };

        ArrayList ret = new ArrayList();

        String[] driverNames = LayerFactory.getDM().getDriverNames();

        for (int i = 0; i < driverNames.length; i++) {

            boolean is = false;

            for (int j = 0; j < classes.length; j++) {

                if (LayerFactory.getDM().isA(driverNames[i], classes[j])) {

                    ret.add(driverNames[i]);

                }

            }

        }

        return (String[]) ret.toArray(new String[0]);

    }

    /**

     * Utility method to instantiate a VectorialJDBCDriver from its name.

     *

     * @param drvname driver's name

     * @return a VectorialJDBCDriver

     *

     * @throws DriverLoadException

     */

    public VectorialSDEDriver getDriverFromName(String drvname)

        throws DriverLoadException {

        VectorialSDEDriver resp = (VectorialSDEDriver) LayerFactory.getDM()

                                                                     .getDriver(drvname);

        return resp;

    }

    public void initWizard() {

    }

    public void execute() {

    }

    /**

     * Return FLayers if user performs multi selection.

     */

    public FLayer getLayer() {

        try {

            IProjection proj = null;

            TablesListItem[] selected = getSelectedTables();

            int count = selected.length;

            String groupName = selectedDataSource.getDb() + " (" +

                conex.getSdeDbaName() + ")";

            FLayer[] all_layers = new FLayer[count];

            String strEPSG = getMapCtrl().getViewPort().getProjection()

                                 .getAbrev().substring(5);

            for (int i = 0; i < count; i++) {

                TablesListItem item = selected[i];

                VectorialSDEDriver driver = null;

                driver = (VectorialSDEDriver) getDriverFromName(selectedDataSource.getDrvName());

                Rectangle2D _wa = item.getUserTableSettingsPanel()

                                      .getWorkingArea();

                if (_wa != null) {

                    driver.setWorkingArea(_wa);

                }

                String layerName = item.getUserTableSettingsPanel()

                                       .getUserLayerName();

                String tableName = item.getTableName();

                String fidField = item.getUserTableSettingsPanel()

                                      .getIdFieldName();

                String geomField = item.getUserTableSettingsPanel()

                                       .getGeoFieldName();

                String[] fields = item.getUserSelectedFieldsPanel()

                                      .getUserSelectedFields(fidField, geomField);

                //                        fields = driver.manageGeometryField(fields, geomField);

                // driver.manageGeometryField(geomField);

                VectorialSDELayerDefinition lyrDef = new ArcSDELayerDefinition();

                lyrDef.setHost(selectedDataSource.getHost());

                lyrDef.setPort(selectedDataSource.getPort());

                lyrDef.setSchema(selectedDataSource.getDb());

                lyrDef.setUser(selectedDataSource.getUser());

                lyrDef.setPassword(selectedDataSource.getPw());

                lyrDef.setName(layerName);

                lyrDef.setTableName(tableName);

//                lyrDef.setWhereClause(whereClause);

//                lyrDef.setFieldGeometry(arcsde_wizard.getGeomField());

//                lyrDef.setFieldID(arcsde_wizard.getFID());

//                lyrDef.setFieldNames(arcsde_wizard.getFields());

//                lyrDef.setSRID_EPSG(strEPSG);

                lyrDef.setConnectionName(selectedDataSource.getName());

                if (item.getUserTableSettingsPanel().isSqlActive()) {

                    String whereClause = item.getUserTableSettingsPanel()

                                             .getWhereClause();

                    lyrDef.setWhereClause(whereClause);

                }

                else {

                    lyrDef.setWhereClause("");

                }

                lyrDef.setFieldGeometry(geomField);

                String[] fieldsAux=null;

                if ((fields != null) && (fields.length != 0)) {

                    if (!fields[0].equals(lyrDef.getFieldGeometry())) {

                        fieldsAux = new String[fields.length + 1];

                        fieldsAux[0] = lyrDef.getFieldGeometry();

                        for (int k = 1; k < fields.length+1; k++) {

                            fieldsAux[k] = fields[k - 1];

                        }

                    }

                }

                fields=fieldsAux;

//                else {

//                    SeSqlConstruct sqlConstruct = new SeSqlConstruct(tableName);

//                    sqlTotal = sqlConstruct.getWhere();

//

//                    SeTable table1 = new SeTable(conn, tableName);

//

//                    /*

//                     *   Get the table's column definition.

//                     */

//                    SeColumnDefinition[] tableDef = table1.describe();

//                    this.fields = new String[tableDef.length];

//

//                    /*

//                     *   Store the names of all the table's columns in the

//                     *   String array cols. This array specifies the columns

//                     *   to be retrieved from the database.

//                     */

//                    int idField = 1;

//

//                    for (int i = 0; i < tableDef.length; i++) {

//                        if (tableDef[i].getType() == SeColumnDefinition.TYPE_SHAPE) {

//                            this.fields[0] = tableDef[i].getName();

//                        } else {

//                            this.fields[idField] = tableDef[i].getName();

//                            idField++;

//                        }

//                    }

//

//                }

                lyrDef.setFieldNames(fields);

                lyrDef.setFieldID(fidField);

                if (_wa != null) {

                    lyrDef.setWorkingArea(_wa);

                }

                lyrDef.setSRID_EPSG(strEPSG);

                if (driver instanceof ICanReproject) {

                    ((ICanReproject) driver).setDestProjection(strEPSG);

                }

                driver.setData(null, lyrDef);

                if (driver instanceof ICanReproject) {

                    proj = CRSFactory.getCRS("EPSG:" +

                            ((ICanReproject) driver).getSourceProjection());

                }

//                all_layers[i] = SDELayerFactory.getLayerSDE(driver, layerName,

//                        proj);

                all_layers[i] = LayerFactory.createDBLayer(driver, layerName,

                        proj);

            }

            return layerArrayToGroup(all_layers, groupName);

        }

        catch (Exception e) {

            logger.error("While creating jdbc layer: " + e.getMessage(), e);

            NotificationManager.addError("Error al cargar la capa: " +

                e.getMessage(), e);

        }

        return null;

    }

    private FLayer layerArrayToGroup(FLayer[] all_layers, String name) {

        if (all_layers.length == 1) {

            return all_layers[0];

        }

        MapContext mc = view.getMapControl().getMapContext();

        FLayers root = view.getMapControl().getMapContext().getLayers();

        FLayers group = new FLayers(mc, root);

        group.setName(name);

        for (int i = 0; i < all_layers.length; i++) {

            group.addLayer(all_layers[i]);

        }

        return group;

    }

    private TablesListItem[] getSelectedTables() {

        int count = tablesList.getModel().getSize();

        ArrayList resp = new ArrayList();

        for (int i = 0; i < count; i++) {

            TablesListItem item = (TablesListItem) tablesList.getModel()

                                                             .getElementAt(i);

            if (item.isSelected()) {

                resp.add(item);

            }

        }

        return (TablesListItem[]) resp.toArray(new TablesListItem[0]);

    }

    /**

     * This method initializes namePanel

     *

     * @return javax.swing.JPanel

     */

    private JPanel getNamePanel() {

        if (namePanel == null) {

            namePanel = new JPanel();

            namePanel.setLayout(null);

            namePanel.setBounds(new java.awt.Rectangle(5, 5, 501, 51));

            namePanel.setBorder(javax.swing.BorderFactory.createTitledBorder(

                    null, PluginServices.getText(this, "choose_connection"),

                    javax.swing.border.TitledBorder.DEFAULT_JUSTIFICATION,

                    javax.swing.border.TitledBorder.DEFAULT_POSITION, null, null));

            namePanel.add(getDatasourceComboBox(), null);

            namePanel.add(getJdbcButton(), null);

        }

        return namePanel;

    }

    /**

     * This method initializes tablesPanel

     *

     * @return javax.swing.JPanel

     */

    private JPanel getTablesPanel() {

        if (tablesPanel == null) {

            tablesPanel = new JPanel();

            tablesPanel.setLayout(new BorderLayout());

            tablesPanel.setBorder(javax.swing.BorderFactory.createTitledBorder(

                    null, PluginServices.getText(this, "choose_table"),

                    javax.swing.border.TitledBorder.DEFAULT_JUSTIFICATION,

                    javax.swing.border.TitledBorder.DEFAULT_POSITION, null, null));

            tablesPanel.setBounds(new java.awt.Rectangle(5, 55, 246, 191));

            tablesPanel.add(getTablesScrollPane(), java.awt.BorderLayout.CENTER);

        }

        return tablesPanel;

    }

    /**

     * This method initializes settingsPanel

     *

     * @return javax.swing.JPanel

     */

    /**

     * This method initializes tablesScrollPane

     *

     * @return javax.swing.JScrollPane

     */

    private JScrollPane getTablesScrollPane() {

        if (tablesScrollPane == null) {

            tablesScrollPane = new JScrollPane();

            tablesScrollPane.setViewportView(getTablesList());

        }

        return tablesScrollPane;

    }

    /**

     * This method initializes tablesList

     *

     * @return javax.swing.JList

     */

    private AvailableTablesCheckBoxList getTablesList() {

        if (tablesList == null) {

            tablesList = new AvailableTablesCheckBoxList(this);

            tablesList.addListSelectionListener(this);

            tablesList.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

        }

        return tablesList;

    }

    /**

     * This method initializes layerNameTextField

     *

     * @return javax.swing.JTextField

     */

    /**

     * This method initializes jComboBox

     *

     * @return javax.swing.JComboBox

     */

    private JComboBox getDatasourceComboBox() {

        if (datasourceComboBox == null) {

            datasourceComboBox = new JComboBox();

            datasourceComboBox.setBounds(new java.awt.Rectangle(10, 20, 446, 21));

            datasourceComboBox.addActionListener(this);

        }

        return datasourceComboBox;

    }

    public void actionPerformed(ActionEvent arg0) {

        Object src = arg0.getSource();

        if (src == datasourceComboBox) {

            Object sel_obj = datasourceComboBox.getSelectedItem();

            if (sel_obj == null) {

                return;

            }

            if (!(sel_obj instanceof ConnectionWithParamsSDE)) {

                return;

            }

            selectedDataSource = (ConnectionWithParamsSDE) sel_obj;

            if (selectedDataSource.isNull()) {

                updateTableList(selectedDataSource);

                setEmptyPanels();

                return;

            }

            if (!selectedDataSource.isConnected()) {

                if (!tryToConnect(selectedDataSource)) {

                    datasourceComboBox.setSelectedIndex(0);

                    return;

                }

            }

            getDatasourceComboBox().repaint();

            updateTableList(selectedDataSource);

            // setEmptyPanels();

            return;

        }

        if (src == jdbcButton) {

            ConnectionWithParamsSDE sel = addNewConnection();

            if (sel != null) {

                loadJdbcDatasourcesCombo();

                getDatasourceComboBox().setSelectedItem(sel);

            }

        }

    }

    private boolean tryToConnect(ConnectionWithParamsSDE _cwp) {

        JPasswordDlg dlg = new JPasswordDlg();

        String strMessage = PluginServices.getText(this, "conectar_jdbc");

        String strPassword = PluginServices.getText(this, "password");

        dlg.setMessage(strMessage + " [" + _cwp.getDrvName() + ", " +

            _cwp.getHost() + ", " + _cwp.getPort() + ", " + _cwp.getDb() +

            ", " + _cwp.getUser() + "]. " + strPassword + "?");

        dlg.show();

        String clave = dlg.getPassword();

        if (clave == null) {

            return false;

        }

        try {

            _cwp.connect(clave);

        }

        catch (SQLException e) {

            showConnectionErrorMessage(e.getMessage());

            return false;

        }

        return true;

    }

    private void updateTableList(ConnectionWithParamsSDE src) {

        if (src.isNull()) {

            getTablesList().setModel(new DefaultListModel());

            getTablesScrollPane().setViewportView(tablesList);

            tablesScrollPane.updateUI();

            return;

        }

        conex = src.getConnection();

        String drvName = src.getDrvName();

        String dbName = "";

        VectorialSDEDriver drv = null;

        try {

            dbName = src.getDb();//getConnection().getDatabaseName();

            drv = (VectorialSDEDriver) LayerFactory.getDM().getDriver(drvName);

        } catch (Exception e) {

            logger.error("While getting driver instance: " + e.getMessage(), e);

        }

        if (!(drv instanceof VectorialSDEDriver)) {

            logger.error(

                "Unexpected driver type (not a ArcSdeDriver driver)");

            return;

        }

        String[] tablnames = null;

        try {

            tablnames = getTableNames(conex, dbName);

        } catch (SQLException e) {

                logger.error("While getting table names: " + e.getMessage(), e);

            return;

                }

        DefaultListModel lmodel = new DefaultListModel();

        for (int i = 0; i < tablnames.length; i++) {

            lmodel.addElement(new TablesListItem(tablnames[i], drv, conex,

                    getMapCtrl(), this));

        }

        getTablesList().setModel(lmodel);

        getTablesScrollPane().setViewportView(tablesList);

        tablesScrollPane.updateUI();

    }

    public void valueChanged(ListSelectionEvent arg0) {

        Object src = arg0.getSource();

        if (src == tablesList) {

            TablesListItem selected = (TablesListItem) tablesList.getSelectedValue();

            try {

                setSettingsPanels(selected);

            }

            catch (SQLException e) {

                showConnectionErrorMessage(e.getMessage());

                tablesList.clearSelection();

                setEmptyPanels();

            }

            checkFinishable();

        }

    }

    private boolean validFormSettings() {

        int count = tablesList.getModel().getSize();

        boolean at_least_one = false;

        boolean resp = true;

        for (int i = 0; i < count; i++) {

            TablesListItem item = (TablesListItem) tablesList.getModel()

                                                             .getElementAt(i);

            if (item.isSelected()) {

                at_least_one = true;

            }

            if (item.disturbsWizardValidity()) {

                resp = false;

            }

        }

        return (at_least_one && resp);

    }

    public void checkFinishable() {

        boolean finishable = validFormSettings();

        callStateChanged(finishable);

    }

    /**

     * This method initializes jdbcButton

     *

     * @return javax.swing.JButton

     */

    private JButton getJdbcButton() {

        if (jdbcButton == null) {

            jdbcButton = new JButton();

            jdbcButton.addActionListener(this);

            jdbcButton.setToolTipText(PluginServices.getText(this,

                    "add_connection"));

            jdbcButton.setBounds(new java.awt.Rectangle(465, 20, 26, 21));

            String _file = createResourceUrl("images/jdbc.png").getFile();

            jdbcButton.setIcon(new ImageIcon(_file));

        }

        return jdbcButton;

    }

    private ConnectionWithParamsSDE addNewConnection() {

        ConnectionWithParamsSDE resp = null;

        SDEConnectionParamsDialog newco = new SDEConnectionParamsDialog();

        newco.showDialog();

        if (newco.isOkPressed()) {

            String _drvname = newco.getConnectionDriverName();

            String _host = newco.getConnectionServerUrl();

            String _port = newco.getConnectionPort();

            String _dbname = newco.getConnectionDBName();

            String _user = newco.getConnectionUser();

            String _pw = newco.getConnectionPassword();

            String _conn_usr_name = newco.getConnectionName();

            boolean hasToBeCon = newco.hasToBeConnected();

            try {

                resp = SingleSDEConnectionManager.instance()

                                                  .getConnection(_drvname,

                        _user, _pw, _conn_usr_name, _host, _port, _dbname,

                        hasToBeCon);

            }

            catch (SQLException e) {

                showConnectionErrorMessage(e.getMessage());

                return null;

            }

            return resp;

        }

        else {

            return null;

        }

    }

    private void showConnectionErrorMessage(String _msg) {

        String msg = (_msg.length() > 300) ? "" : (": " + _msg);

        String title = PluginServices.getText(this, "connection_error");

        JOptionPane.showMessageDialog(this, title + msg, title,

            JOptionPane.ERROR_MESSAGE);

    }

    private java.net.URL createResourceUrl(String path) {

        return getClass().getClassLoader().getResource(path);

    }

    public void setSettingsPanels(TablesListItem actTable)

        throws SQLException {

        if (actTable == null) {

            setEmptyPanels();

            return;

        }

        settingsPanel = actTable.getUserTableSettingsPanel();

        fieldsPanel = actTable.getUserSelectedFieldsPanel();

        removeFieldPanels();

        add(fieldsPanel);

        fieldsPanel.repaint();

        removeSettingsPanels();

        add(settingsPanel);

        settingsPanel.repaint();

        repaint();

    }

    private void setEmptyPanels() {

        removeFieldPanels();

        add(emptyFieldsPanel);

        removeSettingsPanels();

        add(emptySettingsPanel);

        settingsPanel = emptySettingsPanel;

        fieldsPanel = emptyFieldsPanel;

        repaint();

    }

    private void removeFieldPanels() {

        for (int i = 0; i < getComponentCount(); i++) {

            if (getComponent(i) instanceof UserSelectedFieldsPanel) {

                remove(i);

            }

        }

    }

    private void removeSettingsPanels() {

        for (int i = 0; i < getComponentCount(); i++) {

            if (getComponent(i) instanceof UserTableSettingsPanel) {

                remove(i);

            }

        }

    }

    public String[] getTableNames(Object conex, String dbName) throws SQLException {

                Vector theLayers=null;

                try {

                        theLayers = ((SeConnection)conex).getLayers();

                } catch (SeException e) {

                        throw new SQLException(e.getMessage());

                }

                TreeMap ret = new TreeMap();

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

                        SeLayer layer = (SeLayer) theLayers.elementAt(i);

                        ret.put(layer.getTableName(), layer.getTableName());

                }

                return (String[]) ret.keySet().toArray(new String[0]);

        }

} //  @jve:decl-index=0:visual-constraint="10,10"