Application: gvSIG desktop

package org.gvsig.geodb;

import java.awt.BorderLayout;

import java.awt.Dimension;

import java.util.ArrayList;

import java.util.Collection;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.swing.DefaultListModel;

import javax.swing.ListSelectionModel;

import javax.swing.event.ChangeEvent;

import javax.swing.event.ChangeListener;

import javax.swing.event.ListSelectionEvent;

import org.gvsig.app.gui.WizardPanel;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingLocator;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingManager;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import static org.gvsig.fmap.dal.DataServerExplorer.MODE_ALL;

import org.gvsig.fmap.dal.DataStoreParameters;

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

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

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

import static org.gvsig.fmap.dal.store.jdbc2.JDBCServerExplorer.SHOW_TABLES;

import static org.gvsig.fmap.dal.store.jdbc2.JDBCServerExplorer.SHOW_VIEWS;

import org.gvsig.fmap.dal.swing.DALSwingLocator;

import org.gvsig.fmap.dal.swing.DataSwingManager;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.i18n.I18nManager;

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

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

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

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

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

import org.gvsig.tools.swing.api.pickercontroller.PickerController;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public abstract class AbstractWizardDB extends WizardPanel {

    protected static final Logger LOGGER = LoggerFactory.getLogger(AbstractWizardDB.class);

    protected AbstractWizardDBView view;

    private PickerController<JDBCServerExplorerParameters> pickerConnection;

    protected JListWithCheckbox lwcTables;

    protected Map<String, TableInfo> tablesInfo = null;

    protected FilteredListController tablesFilterController;

    protected TableInfoController tableInfoController;

    public AbstractWizardDB() {

        initComponents();

    }

    private void initComponents() {

        DataSwingManager dataSwingManager = DALSwingLocator.getDataSwingManager();

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        ExpressionEvaluatorSwingManager expressionSwingManager = ExpressionEvaluatorSwingLocator.getManager();

        this.view = new AbstractWizardDBView();

        toolsSwingManager.translate(this.view.lblColumns);

        toolsSwingManager.translate(this.view.lblConnection);

        toolsSwingManager.translate(this.view.lblFilter);

        toolsSwingManager.translate(this.view.lblGeometryField);

        toolsSwingManager.translate(this.view.lblIdField);

        toolsSwingManager.translate(this.view.lblName);

        toolsSwingManager.translate(this.view.lblProjection);

        toolsSwingManager.translate(this.view.lblTable);

        toolsSwingManager.addClearButton(this.view.txtName);

        toolsSwingManager.setDefaultPopupMenu(this.view.txtName);

        this.lwcTables = toolsSwingManager.createJListWithCheckbox(

                this.view.lstTables

        );

        this.lwcTables.addListSelectionListener((ListSelectionEvent e) -> {

            doChangeTableSelected();

        });

        this.initializePickerConnection();

        this.lwcTables.addChecksListener((ListSelectionEvent e) -> {

            doChangeTableChecked(e.getFirstIndex(), e.getLastIndex());

        });

        this.tablesFilterController = toolsSwingManager.createFilteredListController(

                this.view.lstTables, 

                this.view.txtTablesFilter, 

                this.view.btnTablesFilter

        );

        this.tableInfoController = new TableInfoController(

                this.view.btnDeselectAllColumns, 

                this.view.btnSelectAllColumns, 

                this.view.lstColumns, 

                this.view.txtName, 

                this.view.cboIdField, 

                this.view.cboGeometryField, 

                this.view.txtProjection, 

                this.view.btnProjection, 

                this.view.txtFilter, 

                this.view.btnFilter, 

                this.view.btnFilterBookmarks, 

                this.view.btnFilterHistory, 

                this.view.btnAdvancedProperties, 

                this.view.lblReadOnlyNotification,

                this.view.pnlCfgActions

        );

        this.tableInfoController.addChangeListener(new ChangeListener() {

            @Override

            public void stateChanged(ChangeEvent e) {

                checkFinishable();

            }

        });

        this.clearTables();

        this.setEditableTables(false);

        this.setLayout(new BorderLayout());

        this.add(this.view, BorderLayout.CENTER);

//        this.setPreferredSize(new Dimension(500, 400));

        this.setPreferredSize(ToolsSwingUtils.ensureRowsCols(this, 30, 80));

    }

    private void initializePickerConnection() {

        DataSwingManager dataSwingManager = DALSwingLocator.getDataSwingManager();

        this.pickerConnection = dataSwingManager.createJDBCConnectionPickerController(

                this.view.cboConnection,

                this.view.btnConnection

        );

        this.pickerConnection.addChangeListener((ChangeEvent e) -> {

            doUpdateTables();

            setEditableTables(true);

        });

    }

    protected abstract boolean requireGeometry();

    protected Collection<TableInfo> getTablesInformation() {

        this.updateTableInfoFromUI();

        if( this.tablesInfo == null ) {

            this.tablesInfo = new HashMap<>();

        }

        return this.tablesInfo.values();

    }

    private void clearTables() {

        this.lwcTables.setModel(new DefaultListModel());

        this.lwcTables.getCheckedModel().clearSelection();

        this.clearTableConfig();

    }

    private void clearTableConfig() {

        this.tableInfoController.put(null);

    }

    private void setEditableTables(boolean enable) {

        this.view.lstTables.setEnabled(enable);

        this.tableInfoController.setEditable(enable);

    }

    private void doUpdateTables() {

        JDBCServerExplorerParameters connection = this.pickerConnection.get();

        if (connection == null) {

            this.clearTables();

            return;

        }

        DataManager dataManager = DALLocator.getDataManager();

        JDBCServerExplorer explorer = null;

        try {

            explorer = (JDBCServerExplorer) dataManager.openServerExplorer(

                    connection.getDataStoreName(),

                    connection

            );

            List<TableInfo> parameters = new ArrayList<>();

            for (DataStoreParameters params : explorer.list(MODE_ALL, SHOW_TABLES)) {

                JDBCStoreParameters jdbcParams = (JDBCStoreParameters) params;

                parameters.add(new TableInfo(jdbcParams.getTable(), jdbcParams, requireGeometry(), false));

            }

            for (DataStoreParameters params : explorer.list(MODE_ALL, SHOW_VIEWS)) {

                JDBCStoreParameters jdbcParams = (JDBCStoreParameters) params;

                parameters.add(new TableInfo(jdbcParams.getTable(), jdbcParams, requireGeometry(), true));

            }

            parameters.sort((TableInfo o1, TableInfo o2) -> 

                    o1.getLabel().compareTo(o2.getLabel())

            );

            this.tablesInfo = new HashMap<>();

            DefaultListModel<TableInfo> model = new DefaultListModel<>();

            for (TableInfo x : parameters) {

                model.addElement(x);

                this.tablesInfo.put(x.getDocumentName(), x);

            }

            this.lwcTables.setModel(model);

            this.lwcTables.getCheckedModel().clearSelection();

        } catch (Exception ex) {

        } finally {

            DisposeUtils.disposeQuietly(explorer);

        }

    }

    private void doChangeTableSelected() {

        TableInfo tableInfo = (TableInfo) this.lwcTables.getSelectedValue();

        if (tableInfo == null) {

            this.clearTableConfig();

            this.tableInfoController.setEditable(false);

            return;

        }

        this.updateTableInfoFromUI();

        this.tableInfoController.put(tableInfo);

    }

    private void doChangeTableChecked(int firstIndex, int lastIndex) {

        ListSelectionModel checkedModel = this.lwcTables.getCheckedModel();

        for (int i = firstIndex; i <= lastIndex; i++) {

            TableInfo tableInfo = (TableInfo) this.lwcTables.getModel().getElementAt(i);

            tableInfo.setSelected(checkedModel.isSelectedIndex(i));

        }

        this.checkFinishable();

    }

    private void updateTableInfoFromUI() {

        String previousId = this.tableInfoController.getId();

        if (previousId == null) {

            return;

        }

        if (this.tablesInfo != null && !this.tablesInfo.isEmpty()) {

            TableInfo previousInfo = this.tablesInfo.get(previousId);

            if (previousInfo != null) {

                this.tableInfoController.fetch(previousInfo);

            }

        }

    }

    @Override

    public void initWizard() {

        I18nManager i18n = ToolsLocator.getI18nManager();

        setTabName(i18n.getTranslation("_Database"));

    }

    @Override

    public boolean areSettingsValid() {

        boolean hasInvalidValues = false;

        boolean hasSelectedTables = false;

        for (TableInfo tableInfo : this.getTablesInformation() ) {

            if (tableInfo.isSelected()) {

                hasSelectedTables = true;

                if (!tableInfo.hasValidValues()) {

                    hasInvalidValues = true;

                }

            }

        }

        return hasSelectedTables && !hasInvalidValues;

    }

    @Override

    public DataStoreParameters[] getParameters() {

        int sz = 0;

        for (TableInfo tableInfo : this.getTablesInformation() ) {

            if( tableInfo.isSelected() ) {

                sz++;

            }

        }

        if( sz == 0 ) {

            return null;

        }

        DataStoreParameters[] dbParameters = new DataStoreParameters[sz];

        int n = 0;

        for (TableInfo tableInfo : this.getTablesInformation() ) {

            if( tableInfo.isSelected() ) {

                dbParameters[n++] = tableInfo.getParameters();

            }

        }

        return dbParameters;

    }

    @Override

    public void execute() {

        executeWizard();

    }

    @Override

    public void close() {

    }

    protected void checkFinishable() {

        boolean finishable = areSettingsValid();

        callStateChanged(finishable);

    }

}