Application: gvSIG desktop

package org.gvsig.fmap.dal.swing.impl.jdbc;

import org.gvsig.fmap.dal.swing.jdbc.JDBCServerExplorerParametersController;

import java.awt.event.ActionEvent;

import java.awt.event.ItemEvent;

import java.util.ArrayList;

import java.util.List;

import javax.swing.ComboBoxModel;

import javax.swing.JButton;

import javax.swing.JComboBox;

import javax.swing.SwingUtilities;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

import org.gvsig.fmap.dal.DataServerExplorerPool;

import org.gvsig.fmap.dal.DataServerExplorerPoolEntry;

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

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

import static org.gvsig.fmap.dal.swing.impl.DefaultDALSwingLibrary.LIBRARY_NAME;

import org.gvsig.fmap.dal.swing.jdbc.JDBCConnectionDialog;

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

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

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

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

import org.gvsig.tools.swing.icontheme.IconTheme;

/**

 *

 * @author jjdelcerro

 */

public class JDBCConnectionPickerControllerImpl

        extends AbstractPickerController<JDBCServerExplorerParameters>

        implements JDBCServerExplorerParametersController

    {

    private final JComboBox cboConnection;

    private final JButton btnConnection;

    public JDBCConnectionPickerControllerImpl(JComboBox cboConnection, JButton btnConnection) {

        this.btnConnection = btnConnection;

        this.cboConnection = cboConnection;

        this.fillConnections(this.cboConnection);

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

            doAddConnection();

        });

        if( "...".equals(this.btnConnection.getText()) ) {

            this.btnConnection.setText("");

        }

        IconTheme theme = ToolsSwingLocator.getIconThemeManager().getCurrent();

        this.btnConnection.setIcon(theme.get("picker-dbconnection"));

        this.cboConnection.addItemListener((ItemEvent e) -> {

            if(!this.isEnabled()){

                return;

            }

            if (e.getStateChange() == ItemEvent.SELECTED) {

                SwingUtilities.invokeLater(() -> { fireChangeEvent(); });

            }

        });

    }

    private void fillConnections(JComboBox combo) {

        DataManager dataManager = DALLocator.getDataManager();

        DataServerExplorerPool pool = dataManager.getDataServerExplorerPool();

        DataServerExplorerParameters params;

        combo.removeAllItems();

        List<DataServerExplorerPoolEntry>entries = new ArrayList<>();

        pool.iterator().forEachRemaining((DataServerExplorerPoolEntry e) -> {entries.add(e); });

        entries.sort((DataServerExplorerPoolEntry o1, DataServerExplorerPoolEntry o2) -> {

            return StringUtils.compare(o1.getName(), o2.getName());

        });

        for (DataServerExplorerPoolEntry entry : entries) {

            if (entry.getExplorerParameters() instanceof JDBCServerExplorerParameters) {

                JDBCServerExplorerParameters dbParams = (JDBCServerExplorerParameters) entry.getExplorerParameters();

                combo.addItem(

                        new ListElement<>(entry.getName(), dbParams)

                );

            }

        }

        combo.setSelectedIndex(-1);

    }

    private void doAddConnection() {

        JDBCConnectionDialog dialog = DALSwingLocator.getSwingManager().createJDBCConectionDialog();

        dialog.showDialog();

        if (dialog.isCanceled()) {

            return;

        }

//        DataManager dataManager = DALLocator.getDataManager();

//        DataServerExplorerPool pool = dataManager.getDataServerExplorerPool();

        JDBCServerExplorerParameters connectionParameters = dialog.getServerExplorerParameters();

//        pool.add(dialog.getConnectionName(), connectionParameters);

        this.fillConnections(this.cboConnection);

        ListElement.setSelected(cboConnection, connectionParameters);

    }

    @Override

    public JDBCServerExplorerParameters get() {

        ListElement<JDBCServerExplorerParameters> item = (ListElement<JDBCServerExplorerParameters>) this.cboConnection.getSelectedItem();

        if (item == null) {

            return null;

        }

        JDBCServerExplorerParameters conn = item.getValue();

        return conn;

    }

    @Override

    public String getName() {

        ListElement<JDBCServerExplorerParameters> item = (ListElement<JDBCServerExplorerParameters>) this.cboConnection.getSelectedItem();

        if (item == null) {

            return null;

        }

        String name = item.getLabel();

        return name;

    }

    @Override

    public void set(JDBCServerExplorerParameters value) {

        if(value == null){

            this.cboConnection.setSelectedIndex(-1);

            this.cboConnection.setSelectedItem(null);

            return;

        }

        ComboBoxModel<ListElement<JDBCServerExplorerParameters>> model = this.cboConnection.getModel();

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

            JDBCServerExplorerParameters params = model.getElementAt(i).getValue();

            if( params == value ) {

                this.cboConnection.setSelectedIndex(i);

                return;

            }

        }

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

            JDBCServerExplorerParameters params = model.getElementAt(i).getValue();

            if( StringUtils.equals(params.getUrl(), value.getUrl())){

                this.cboConnection.setSelectedIndex(i);

                return;

            }

        }

        //Si llega aqu? a?adimos un nuevo item

        cboConnection.addItem(

                new ListElement<>(value.getHost()+":"+value.getDBName(), value)

        );

        this.cboConnection.setSelectedIndex(this.cboConnection.getModel().getSize()-1);

    }

    @Override

    public void coerceAndSet(Object value) {

        JDBCServerExplorerParameters params;

        try {

            params = (JDBCServerExplorerParameters) value;

        } catch(Exception ex) {

            throw new IllegalArgumentException("Can't coerce value to JDBCServerExplorerParameters", ex);

        }

        this.set(params);

    }

    @Override

    public void setEnabled(boolean enabled) {

        this.cboConnection.setEnabled(enabled);

        this.btnConnection.setEnabled(enabled);

    }

    @Override

    public boolean isEnabled() {

        return this.cboConnection.isEnabled();

    }

    public static void selfRegister() {

        boolean n = ToolsSwingUtils.registerIcons(JDBCConnectionPickerControllerImpl.class,

                null,

                LIBRARY_NAME,

                new String[]{ "picker", "picker-dbconnection", "picker-dbconnection"}

        );

    }

}