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.export.jdbc.swing.panels;

import java.awt.event.ActionEvent;

import java.util.ArrayList;

import java.util.Collections;

import java.util.List;

import javax.swing.DefaultListModel;

import javax.swing.JComponent;

import javax.swing.JOptionPane;

import javax.swing.ListModel;

import javax.swing.SwingUtilities;

import javax.swing.event.AncestorEvent;

import javax.swing.event.AncestorListener;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.export.jdbc.service.ExportJDBCAttributeNamesTranslator;

import org.gvsig.export.jdbc.service.ExportJDBCParameters;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.SQLBuilder;

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.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.api.threadsafedialogs.ThreadSafeDialogsManager;

import org.gvsig.tools.task.AbstractMonitorableTask;

import org.gvsig.tools.task.SimpleTaskStatus;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.export.swing.JExportProcessPanel;

import org.gvsig.export.swing.spi.ExportPanel;

import org.gvsig.export.swing.spi.ExportPanelValidationException;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingLocator;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingManager;

import org.gvsig.expressionevaluator.swing.ExpressionPickerController;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.tools.swing.api.ToolsSwingManager;

/**

 * @author gvSIG Team

 * @version $Id$

 *

 */

@SuppressWarnings("UseSpecificCatch")

public class SelectTableNamePanel

        extends SelectTableNamePanelLayout

        implements ExportPanel {

    private static final Logger logger = LoggerFactory.getLogger(SelectTableNamePanel.class);

    private static final long serialVersionUID = 6269512983586358017L;

    private ExpressionPickerController deleteRowsFilterPicker;

    private static class TableItem implements Comparable<TableItem> {

        private JDBCStoreParameters params;

        private String label;

        public TableItem(String label, JDBCStoreParameters params) {

            this.params = params;

            this.label = label;

        }

        public TableItem(JDBCStoreParameters params) {

            this(StringUtils.isEmpty(params.getSchema())

                    ? params.getTable() : params.getSchema() + "." + params.getTable(),

                    params);

        }

        @Override

        public String toString() {

            return this.label;

        }

        public JDBCStoreParameters getParams() {

            return this.params;

        }

        @Override

        public int compareTo(TableItem o) {

            return this.label.compareTo(o.label);

        }

    }

    private FillTablesListTask task = null;

    private SQLBuilder sqlbuilder;

    private final ExportJDBCParameters parameters;

    private final JExportProcessPanel processPanel;

    @SuppressWarnings("OverridableMethodCallInConstructor")

    public SelectTableNamePanel(

            JExportProcessPanel processPanel,

            ExportJDBCParameters parameters

    ) {

        this.processPanel = processPanel;

        this.parameters = parameters;

        initComponents();

        this.addAncestorListener(new AncestorListener() {

            @Override

            public void ancestorAdded(AncestorEvent ae) {

            }

            @Override

            public void ancestorRemoved(AncestorEvent ae) {

                cancelTask();

            }

            @Override

            public void ancestorMoved(AncestorEvent ae) {

            }

        });

    }

    private void initComponents() {

        this.rdoCreateTable.addActionListener((ActionEvent e) -> {

            onChangeRadioSelecion();

        });

        this.rdoInsert.addActionListener((ActionEvent e) -> {

            onChangeRadioSelecion();

        });

        this.rdoCreateTable.setSelected(true);

        this.rdoInsert.setEnabled(false);

        this.lstTables.setEnabled(false);

        this.chkDeleteRows.setEnabled(false);

        try {

            this.txtTableName.setText(this.parameters.getSourceFeatureStore().getName());

        } catch (Exception ex) {

            logger.warn("Can't set the default value for the table name", ex);

        }

        ExpressionEvaluatorSwingManager expressionEvaluatorSwingManager = ExpressionEvaluatorSwingLocator.getManager();

        this.deleteRowsFilterPicker = expressionEvaluatorSwingManager.createExpressionPickerController(txtDeleteRowsFilter, btnDeleteRowsFilter);

        this.deleteRowsFilterPicker.setEnabled(false);

        ToolsSwingManager swingManager = ToolsSwingLocator.getToolsSwingManager();

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        swingManager.translate(this.lblHeader);

        swingManager.translate(this.lblWarningUseExistingTable, true);

        swingManager.translate(this.rdoInsert);

        swingManager.translate(this.lblSelectTableName);

        swingManager.translate(this.rdoCreateTable);

        swingManager.translate(this.lblSchema);

        swingManager.translate(this.lblTableName);

        swingManager.translate(this.lblCaseSensitiveNote,true);

        swingManager.translate(this.chkDeleteRows);

        swingManager.translate(this.lblDeleteRowsFilter);

        swingManager.translate(this.txtDeleteRowsFilter);

        swingManager.translate(this.btnDeleteRowsFilter);

//        this.lblHeader.setText(i18nManager.getTranslation("_Indique_donde_desea_insertar_los_datos"));

//        this.lblWarningUseExistingTable.setText(

//                "<html>\n"

//                + i18nManager.getTranslation("_Los_datos_se_insertaran_usando_los_nombres_de_columna_que_coincidan_con_la_tabla_origen_dejandose_al_valor_por_defecto_para_los_que_no_haya_valores_en_la_tabla_origen")

//                + "\n</html>"

//        );

//        this.rdoInsert.setText(i18nManager.getTranslation("_Insertar_registros_en_una_tabla_existente"));

//        this.lblSelectTableName.setText(i18nManager.getTranslation("_Seleccione_la_tabla_a_usar"));

//        this.rdoCreateTable.setText(i18nManager.getTranslation("_Crear_una_tabla_nueva"));

//        this.lblSchema.setText(i18nManager.getTranslation("_Indique_el_esquema_en_el_que_desea_crear_la_tabla"));

//        this.lblTableName.setText(i18nManager.getTranslation("_Indique_el_nombre_de_la_tabla"));

    }

    private void cancelTask() {

        if (task != null) {

            task.cancelRequest();

            task.getSimpleTaskStatus().remove();

            task = null;

        }

    }

    public boolean canCreateTable() {

        return this.rdoCreateTable.isSelected();

    }

    public String getSchema() {

        if (this.canCreateTable()) {

            return StringUtils.defaultIfBlank(this.txtSchema.getText(), null);

        }

        TableItem item = (TableItem) this.lstTables.getSelectedValue();

        JDBCStoreParameters tableParameter = item.getParams();

        if (tableParameter == null) {

            return null;

        }

        return tableParameter.getSchema();

    }

    public String getTableName() {

        if (this.canCreateTable()) {

            return StringUtils.defaultIfBlank(this.txtTableName.getText(), null);

        }

        TableItem item = (TableItem) this.lstTables.getSelectedValue();

        if (item == null) {

            return null;

        }

        JDBCStoreParameters tableParameter = item.getParams();

        if (tableParameter == null) {

            return null;

        }

        return tableParameter.getTable();

    }

    @Override

    public String getIdPanel() {

        return this.getClass().getCanonicalName();

    }

    @Override

    public String getTitlePanel() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        return i18nManager.getTranslation("_Tablename");

    }

    @Override

    public void previousPanel() {

    }

    @Override

    public void nextPanel() {

        this.parameters.setSchema(this.getSchema());

        this.parameters.setTableName(this.getTableName());

        this.parameters.setCanCreatetable(this.canCreateTable());

        if (this.canCreateTable()) {

            this.parameters.setDeleteRows(false);

            this.parameters.setDeleteRowsFilter(null);

        } else {

            this.parameters.setDeleteRows(this.chkDeleteRows.isSelected());

            this.parameters.setDeleteRowsFilter(this.deleteRowsFilterPicker.get());

        }

    }

    @Override

    public boolean validatePanel() throws ExportPanelValidationException {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        String tablename = this.getTableName();

        if (tablename == null) {

            throw new ExportPanelValidationException(

                    i18nManager.getTranslation(

                            "_The_name_of_table_cannot_be_empty"

                    )

            );

        }

        String schema = this.getSchema();

        if (sqlbuilder.support_schemas()) {

            if (schema == null) {

                throw new ExportPanelValidationException(

                        i18nManager.getTranslation(

                                "_The_name_of_schema_cannot_be_empty"

                        )

                );

            }

        }

        if (this.rdoCreateTable.isSelected()) {

            ExportJDBCAttributeNamesTranslator nameTranslator

                    = (ExportJDBCAttributeNamesTranslator) this.parameters.getExportAttributes().getNamesTranslator();

            String tablename_tr = nameTranslator.getSuggestion(tablename);

            if (!tablename_tr.equals(tablename)) {

                String msg = i18nManager.getTranslation(

                        "_The_name_entered_for_the_table_does_not_correspond_to_the_conditions_selected_in_the_previous_panel_You_want_it_to_be_corrected_automatically"

                );

                ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

                int resp = dialogs.confirmDialog(

                        msg,

                        i18nManager.getTranslation("_Warning"),

                        JOptionPane.YES_NO_OPTION,

                        JOptionPane.WARNING_MESSAGE,

                        "Exportto_Table_name_with_spaces_or_mixed_case"

                );

                if (resp != JOptionPane.YES_OPTION) {

                    msg = i18nManager.getTranslation(

                            "El_nombre_de_tabla_contiene_caracteres no_validos"

                    );

                    throw new ExportPanelValidationException(msg);

                }

                tablename = tablename_tr;

                this.txtTableName.setText(tablename);

            }

            ListModel model = this.lstTables.getModel();

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

                TableItem item = (TableItem) model.getElementAt(i);

                if (StringUtils.equals(schema, item.getParams().getSchema())

                        && StringUtils.equals(tablename, item.getParams().getTable())) {

                    String msg = i18nManager.getTranslation(

                            "_La_tabla_{0}_{1}_ya_existe_en_la_base_de_datos_Seleccione_la_opcion_de_insertar_registros_en_una_tabla_existente_para_a?adir_los_datos_a_esta_o_indique_otro_nombre",

                            new String[]{schema, tablename}

                    );

                    throw new ExportPanelValidationException(msg);

                }

            }

        }

        return true;

    }

    @Override

    public void enterPanel() {

        JDBCServerExplorerParameters explorerParameters = parameters.getExplorerParameters();

        if (explorerParameters == null) {

            this.processPanel.setButtonEnabled(JExportProcessPanel.BUTTON_NEXT, false);

            return;

        }

        this.processPanel.setButtonEnabled(JExportProcessPanel.BUTTON_NEXT, true);

        try {

            DataManager dataManager = DALLocator.getDataManager();

            JDBCServerExplorer explorer = (JDBCServerExplorer) dataManager.openServerExplorer(

                    explorerParameters.getExplorerName(),

                    explorerParameters

            );

            this.sqlbuilder = explorer.createSQLBuilder();

        } catch (Exception ex) {

            throw new RuntimeException("Can't retrieve the sqlbuilder", ex);

        }

        this.fillTablesList();

        // set values from params

        this.rdoCreateTable.setSelected(this.parameters.canCreatetable());

        if (this.parameters.canCreatetable() == true) {

            if (this.parameters.getSchema() != null) {

                this.txtSchema.setText(this.parameters.getSchema());

                this.txtTableName.setText(this.parameters.getTableName());

            }

            this.chkDeleteRows.setSelected(false);

            this.deleteRowsFilterPicker.set(null);

        } else {

            String schema = this.parameters.getSchema();

            if (schema != null) {

                //this.lstTables;

                for (int i = 0; i < this.lstTables.getModel().getSize(); i++) {

                    TableItem item = (TableItem) this.lstTables.getModel().getElementAt(i);

                    JDBCStoreParameters tableParameter = item.getParams();

                    if ( StringUtils.equals(tableParameter.getSchema(),schema) ) {

                        this.lstTables.setSelectedIndex(i);

                    }

                }

            }

            this.chkDeleteRows.setSelected(this.parameters.isDeleteRows());

            this.deleteRowsFilterPicker.set(this.parameters.getDeleteRowsFilter());

        }

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    public void onChangeRadioSelecion() {

        if (this.rdoCreateTable.isSelected()) {

            this.txtSchema.setEnabled(true);

            this.txtTableName.setEnabled(true);

            this.lstTables.setEnabled(false);

            this.chkDeleteRows.setEnabled(false);

            this.txtDeleteRowsFilter.setEnabled(false);

            this.btnDeleteRowsFilter.setEnabled(false);

        } else {

            this.txtSchema.setEnabled(false);

            this.txtTableName.setEnabled(false);

            this.lstTables.setEnabled(true);

            this.chkDeleteRows.setEnabled(true);

            this.txtDeleteRowsFilter.setEnabled(true);

            this.btnDeleteRowsFilter.setEnabled(true);

        }

    }

    private void fillTablesList() {

        JDBCServerExplorerParameters explorerParameters = this.parameters.getExplorerParameters();

        if (explorerParameters == null) {

            return;

        }

        cancelTask();

        this.task = new FillTablesListTask();

        task.setDaemon(true);

        task.start();

    }

    private class FillTablesListTask extends AbstractMonitorableTask {

        public FillTablesListTask() {

            super("Export");

        }

        @Override

        protected SimpleTaskStatus getSimpleTaskStatus() {

            return (SimpleTaskStatus) this.getTaskStatus();

        }

        @Override

        public void run() {

            JDBCServerExplorerParameters explorerParameters = parameters.getExplorerParameters();

            if (parameters.getExplorerParameters() == null) {

                return;

            }

            final SimpleTaskStatus status = this.getSimpleTaskStatus();

            try {

                status.setAutoremove(true);

                DataManager dataManager = DALLocator.getDataManager();

                this.getSimpleTaskStatus().message("Connecting server");

                explorerParameters.setShowInformationDBTables(false);

                final JDBCServerExplorer explorer = (JDBCServerExplorer) dataManager.openServerExplorer(

                        explorerParameters.getExplorerName(),

                        explorerParameters

                );

                SwingUtilities.invokeAndWait(() -> {

                    if (sqlbuilder.support_schemas()) {

                        txtSchema.setText(sqlbuilder.default_schema());

                    } else {

                        txtSchema.setText("");

                        txtSchema.setEnabled(false);

                    }

                });

                this.getSimpleTaskStatus().message("Retrieving tables");

                final List<DataStoreParameters> parameters = explorer.list();

                final List<TableItem> tables = new ArrayList<>();

                for (DataStoreParameters table : parameters) {

                    if (status.isCancelled()) {

                        status.cancel();

                        break;

                    }

                    tables.add(new TableItem((JDBCStoreParameters)table));

                }

                Collections.sort(tables);

                this.getSimpleTaskStatus().message("Add tables");

                final DefaultListModel lmodel = new DefaultListModel();

                for (TableItem table : tables) {

                    lmodel.addElement(table);

                }

                SwingUtilities.invokeAndWait(() -> {

                    lstTables.setModel(lmodel);

                    //lstTables.setEnabled(true);

                    rdoInsert.setEnabled(true);

                });

                status.message("finish");

                status.terminate();

            } catch (final Exception ex) {

                logger.warn("Fail to fill tables list", ex);

                if (status.isCancellationRequested()) {

                    status.cancel();

                }

                if (status.isRunning()) {

                    I18nManager i18nManager = ToolsLocator.getI18nManager();

                    ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

                    dialogs.messageDialog(

                            i18nManager.getTranslation("_There_have_been_problems_filling_data_in_panel")

                            + " (" + getTitlePanel() + ")",

                            null,

                            i18nManager.getTranslation("_Warning"),

                            JOptionPane.WARNING_MESSAGE,

                            "ProblemsFillingTableNamePanel"

                    );

                }

            } finally {

                status.terminate();

                status.remove();

            }

        }

    }

}