gvSIG projects pool

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (c) 2007-2020 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 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., 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.vcsgis.swing.impl.addToWorkspace;

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import javax.swing.ComboBoxModel;

import javax.swing.DefaultComboBoxModel;

import javax.swing.DefaultListCellRenderer;

import javax.swing.JComponent;

import javax.swing.JLabel;

import javax.swing.JList;

import javax.swing.JTree;

import javax.swing.ListSelectionModel;

import javax.swing.SwingUtilities;

import javax.swing.event.ChangeEvent;

import javax.swing.event.DocumentEvent;

import javax.swing.event.DocumentListener;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.TreeSelectionEvent;

import javax.swing.tree.DefaultTreeCellRenderer;

import javax.swing.tree.TreePath;

import javax.swing.tree.TreeSelectionModel;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

import org.gvsig.fmap.dal.swing.featuretype.AttributeDescriptorPickerController;

import org.gvsig.tools.ToolsLocator;

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

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

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

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

import org.gvsig.tools.swing.api.task.TaskStatusController;

import org.gvsig.tools.swing.api.task.TaskStatusSwingManager;

import org.gvsig.tools.swing.api.windowmanager.Dialog;

import org.gvsig.tools.swing.api.windowmanager.WindowManager_v2;

import org.gvsig.tools.util.LabeledValue;

import org.gvsig.vcsgis.lib.VCSGisManager;

import static org.gvsig.vcsgis.lib.VCSGisManager.ERR_OK;

import org.gvsig.vcsgis.lib.VCSGisRuntimeException;

import org.gvsig.vcsgis.lib.workspace.VCSGisWorkspace;

import org.gvsig.vcsgis.swing.VCSGisJAddToWorkspace;

import org.gvsig.vcsgis.swing.VCSGisSwingLocator;

import org.gvsig.vcsgis.swing.VCSGisSwingManager;

import org.gvsig.vcsgis.swing.VCSGisSwingServices;

import static org.gvsig.vcsgis.swing.impl.VCSGisSwingCommons.notInSwingThreadInvokeLater;

import org.slf4j.LoggerFactory;

import static org.gvsig.vcsgis.swing.impl.VCSGisSwingCommons.showAuthenticationErrors;

import static org.gvsig.vcsgis.swing.impl.VCSGisSwingCommons.showAuthenticationErrors;

import static org.gvsig.vcsgis.swing.impl.VCSGisSwingCommons.showAuthenticationErrors;

/**

 *

 * @author gvSIG Team

 */

public class VCSGisJAddToWorkspaceImpl extends VCSGisJAddToWorkspaceView

        implements Component, VCSGisJAddToWorkspace {

    private static final org.slf4j.Logger LOGGER = LoggerFactory.getLogger(VCSGisJAddToWorkspaceImpl.class);

    private static final int TAB_LAYERS_INDEX = 0;

    private static final int TAB_TABLES_INDEX = 1;

    private PickerController<VCSGisWorkspace> workspacePicker;

    private Dialog dialog;

    private AttributeDescriptorPickerController fieldForLabelPicker;

    private boolean processing;

    private TaskStatusController taskStatusController;

    public VCSGisJAddToWorkspaceImpl() {

        processing = false;

        initializeComponents();

    }

    private void initializeComponents() {

        VCSGisSwingManager swingManager = VCSGisSwingLocator.getVCSGisSwingManager();

        TaskStatusSwingManager taskStatusManager = ToolsSwingLocator.getTaskStatusSwingManager();

        translate();

        this.taskStatusController = taskStatusManager.createTaskStatusController(

                this.lblStatusTitle,

                this.lblStatusMessages,

                this.pbStatus

        );

        this.taskStatusController.setShowCancelButton(false);

        this.taskStatusController.setShowRemoveTaskButton(false);

        this.taskStatusController.bind(ToolsLocator.getTaskStatusManager());

        SwingUtilities.invokeLater(() -> {this.setVisibleStatus(false); });

        this.workspacePicker = swingManager.createWorkspacePickerController(

                cboWorkspace,

                btnWorkspace

        );

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

            doChangeWorkspace();

        });

        this.setPreferredSize(new Dimension(520, 340));

        tabAddToWorkspace.addChangeListener((ChangeEvent e) -> {

            updateTextName();

            updateFieldsForLabel();

            doUpdateComponents();

        });

        VCSGisSwingServices services = swingManager.getDefaultServices();

        ComboBoxModel<LabeledValue> viewDocumentsModel = services.getViewDocumentsComboBoxModel();

        cboView.setModel(viewDocumentsModel);

        cboView.addActionListener((ActionEvent e) -> {

            doUpdateComponents();

        });

        chkAddLayerToView.addChangeListener((ChangeEvent e) -> {

            doUpdateComponents();

        });

        treeLayers.setModel(services.getFeatureStoresTreeModel());

        treeLayers.getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);

        treeLayers.getSelectionModel().addTreeSelectionListener((TreeSelectionEvent e) -> {

            updateTextName();

            updateFieldsForLabel();

            doUpdateComponents();

        });

        treeLayers.setCellRenderer(new DefaultTreeCellRenderer() {

            @Override

            public java.awt.Component getTreeCellRendererComponent(JTree tree, Object value, boolean sel, boolean expanded, boolean leaf, int row, boolean hasFocus) {

                JLabel label = (JLabel) super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, row, hasFocus); //To change body of generated methods, choose Tools | Templates.

                if (value instanceof FeatureStore) {

                    label.setText(((FeatureStore) value).getName());

                }

                return label;

            }

        });

        lstTables.setModel(services.getFeatureStoresListModel());

        lstTables.getSelectionModel().setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

        lstTables.getSelectionModel().addListSelectionListener((ListSelectionEvent e) -> {

            updateTextName();

            updateFieldsForLabel();

            doUpdateComponents();

        });

        lstTables.setCellRenderer(new DefaultListCellRenderer(){

            @Override

            public java.awt.Component getListCellRendererComponent(JList<?> list, Object value, int index, boolean isSelected, boolean cellHasFocus) {

                JLabel label = (JLabel) super.getListCellRendererComponent(list, value, index, isSelected, cellHasFocus);

                if (value instanceof FeatureStore) {

                    label.setText(((FeatureStore) value).getName());

                }

                return label;

            }

        });

        txtName.getDocument().addDocumentListener(new DocumentListener() {

            @Override

            public void insertUpdate(DocumentEvent e) {

                doUpdateComponents();

            }

            @Override

            public void removeUpdate(DocumentEvent e) {

                doUpdateComponents();

            }

            @Override

            public void changedUpdate(DocumentEvent e) {

                doUpdateComponents();

            }

        });

        this.fieldForLabelPicker = DALSwingLocator.getDataSwingManager().createAttributeDescriptorPickerController(

                cboFieldForLabel, 

                (FeatureAttributeDescriptor t) -> {

            if (t.getType() == DataTypes.GEOMETRY) {

                return false;

            }

            return !t.isComputed();

        });

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

            doUpdateComponents();

        });

        if(this.cboView.getModel().getSize() > 0){

            this.cboView.setEnabled(true);

            this.cboView.setSelectedIndex(0);

            this.chkAddLayerToView.setSelected(true);

        } else {

            this.cboView.setEnabled(false);

            this.chkAddLayerToView.setSelected(false);

        }

        this.chkAddTableToProject.setSelected(true);

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    @Override

    public void setDialog(Dialog dialog) {

        this.dialog = dialog;

        this.doUpdateComponents();

    }

    private void updateTextName() {

        FeatureStore value = getFeatureStore();

        if (value != null) {

            txtName.setText(value.getName());

        } else {

            txtName.setText("");

        }

    }

    private void updateFieldsForLabel() {

        FeatureStore store = getFeatureStore();

        if (store != null){

            this.fieldForLabelPicker.setFeatureType(store.getDefaultFeatureTypeQuietly());

        }

    }

    private void doUpdateComponents() {

        try {

            VCSGisWorkspace workspace = getWorkspace();

            LabeledValue selectedView = (LabeledValue)cboView.getSelectedItem();

            boolean enableComponents = !processing && workspace != null ;

            boolean enableInit = enableComponents && 

                    getFeatureStore() != null &&

                    StringUtils.isNotBlank(txtName.getText()) &&

                    workspace.getEntity(txtName.getText())==null ;

            if (tabAddToWorkspace.getSelectedIndex() == TAB_LAYERS_INDEX) {

                enableInit = enableInit && (!chkAddLayerToView.isSelected()

                        || (chkAddLayerToView.isSelected() && selectedView != null && selectedView.getValue() != null));

            }

            this.workspacePicker.setEnabled( !processing );

            this.tabAddToWorkspace.setEnabled(enableComponents);

            this.txtName.setEnabled(enableComponents);

            this.cboFieldForLabel.setEnabled(enableComponents);

            this.txtLabel.setEnabled(enableComponents);

            this.cboCategory.setEnabled(enableComponents);

            this.cboView.setEnabled(enableComponents);

            this.chkAddLayerToView.setEnabled(enableComponents);

            this.chkAddTableToProject.setEnabled(enableComponents);

            if (dialog != null) {

                this.dialog.setButtonEnabled(WindowManager_v2.BUTTON_OK,

                        enableInit

                );

            }

            this.pbStatus.setVisible(processing);

            this.lblStatusTitle.setVisible(processing);

        } catch(VCSGisRuntimeException e1) {

            LOGGER.warn("Can't updating components.",e1);

            if( showAuthenticationErrors("_VCS_Add_to_workspace",e1) ) {

                this.workspacePicker.set(null);

                doChangeWorkspace();

            }

        } catch(Exception e2) {

            LOGGER.warn("Can't updating components.",e2);

        }

    }

    private void translate() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        toolsSwingManager.translate(this.lblWorkspace);

        toolsSwingManager.translate(this.lblName);

        toolsSwingManager.translate(this.tabAddToWorkspace);

        toolsSwingManager.translate(this.lblFieldForLabel);

        toolsSwingManager.translate(this.chkAddLayerToView);

        toolsSwingManager.translate(this.chkAddTableToProject);

        toolsSwingManager.translate(this.lblCategory);

        toolsSwingManager.translate(this.lblLabel);

    }

    @Override

    public VCSGisWorkspace getWorkspace() {

        VCSGisWorkspace workspace = this.workspacePicker.get();

        return workspace;

    }

    @Override

    public FeatureStore getFeatureStore() {

        switch (tabAddToWorkspace.getSelectedIndex()) {

            case TAB_LAYERS_INDEX:

                TreePath selection = treeLayers.getSelectionModel().getSelectionPath();

                if(selection != null) {

                    Object[] path = selection.getPath();

                    boolean isLeaf = treeLayers.getModel().isLeaf(path[path.length - 1]);

                    Object obj = path[path.length - 1];

                    if (isLeaf && obj instanceof FeatureStore) {

                        return (FeatureStore) obj;

                    }

                }

            case TAB_TABLES_INDEX:

                Object selected = lstTables.getSelectedValue();

                if(selected instanceof FeatureStore) {

                    return (FeatureStore) selected;

                }

        }

        return null;

    }

    @Override

    public String getTabletName() {

        return this.txtName.getText();

    }

    @Override

    public int add() {

        VCSGisWorkspace workspace = getWorkspace();        

        if (workspace == null) {

            return VCSGisManager.ERR_CANT_ADD_TABLE;

        }

        FeatureStore featureStore = getFeatureStore();

        if (featureStore == null) {

            return VCSGisManager.ERR_CANT_ADD_TABLE;

        }

        try {

            processing = true;

            this.doUpdateComponents();

            String label = StringUtils.defaultIfBlank(this.txtLabel.getText(), null);

            String category = StringUtils.defaultIfBlank((String)(this.cboCategory.getSelectedItem()), null);

            FeatureAttributeDescriptor descriptor = this.fieldForLabelPicker.get();

            String attrNameForLabel = descriptor == null ? null : descriptor.getName();

            //TODO: Adicionamos pkName si hace falta

            int result = workspace.add(txtName.getText(), featureStore, attrNameForLabel, category, label);

            this.postAdd(result, workspace);

            return result;

        } finally {

            processing = false;

            this.doUpdateComponents();

        }

    }

    private void postAdd(int addStatus, VCSGisWorkspace workspace) {

        if( notInSwingThreadInvokeLater(() -> {postAdd(addStatus, workspace);}) ) {

            return;

        }

        VCSGisSwingManager swingManager = VCSGisSwingLocator.getVCSGisSwingManager();

        VCSGisSwingServices services = swingManager.getDefaultServices();

        if(addStatus == ERR_OK){

            switch (tabAddToWorkspace.getSelectedIndex()) {

                case TAB_LAYERS_INDEX:

                    if(chkAddLayerToView.isSelected()){

                        FeatureStore store = workspace.getFeatureStore(txtName.getText());

                        String layerName = store.getName();

                        services.addLayerToView(store, (LabeledValue)cboView.getSelectedItem(), layerName);

                    }

                    break;

                case TAB_TABLES_INDEX:

                    if(chkAddTableToProject.isSelected()){

                        FeatureStore store = workspace.getFeatureStore(txtName.getText());

                        services.addTableToProject(workspace, store);

                    }

                    break;

            }

        }

    }

    public void setVisibleStatus(boolean visible) {

        if( notInSwingThreadInvokeLater(() -> {setVisibleStatus(visible);}) ) {

            return;

        }

        this.lblStatusTitle.setVisible(visible);

        this.pbStatus.setVisible(visible);

        this.lblStatusMessages.setVisible(true);

    }

    @Override

    public boolean isProcessing() {

        return processing;

    }

    private void message(String message) {

        if( notInSwingThreadInvokeLater(() -> {message(message);}) ) {

            return;

        }

        this.lblStatusMessages.setText(message);

    }

    private void doChangeWorkspace() {

        VCSGisWorkspace ws = this.workspacePicker.get();

        DefaultComboBoxModel<String> model = new DefaultComboBoxModel<>();

        if( ws!=null ) {

            for (String category : ws.getEntityCategories()) {

                model.addElement(category);

            }

        }

        this.cboCategory.setModel(model);

        doUpdateComponents();

    }

}