Application: gvSIG desktop

package org.gvsig.app.extension.copytable;

import java.awt.Component;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.Objects;

import javax.swing.AbstractCellEditor;

import javax.swing.JComboBox;

import javax.swing.JTable;

import javax.swing.ListSelectionModel;

import javax.swing.SwingUtilities;

import javax.swing.event.TreeSelectionEvent;

import javax.swing.table.AbstractTableModel;

import javax.swing.table.DefaultTableColumnModel;

import javax.swing.table.TableCellEditor;

import javax.swing.table.TableColumn;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.Expression;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingLocator;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingManager;

import org.gvsig.expressionevaluator.swing.ExpressionPickerController;

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

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

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

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

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

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

import org.gvsig.tools.ToolsLocator;

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

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

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

import org.gvsig.tools.task.SimpleTaskStatus;

/**

 *

 * @author jjdelcerro

 */

public class CopyTablePanel extends CopyTablePanelView {

    private AttributeDescriptorPickerController pickerSourceFieldJoin;

    private FeatureStore source;

    private FeatureStore target;

    private AttributeDescriptorPickerController pickerTargetFieldJoin;

    private ExpressionPickerController pickerSourceFilter;

    private ExpressionPickerController pickerTargetUpdateExpression;

    private DefaultTableColumnModel fieldsColumnModel;

    private SimpleTaskStatus taskStatus;

    private TaskStatusController status;

    private StoresRepositoryController treeSourceTableController;

    private StoresRepositoryController treeTargetTableController;

    private class FeatureAttributeCellEditor extends AbstractCellEditor implements TableCellEditor {

        private final AttributeDescriptorPickerController picker;

        private final JComboBox combo;

        public FeatureAttributeCellEditor() {

            DataSwingManager dataSwingManager = DALSwingLocator.getSwingManager();

            this.combo = new JComboBox();

            this.picker = dataSwingManager.createAttributeDescriptorPickerController(this.combo);

        }

        @Override

        public Component getTableCellEditorComponent(JTable table, Object value, boolean isSelected, int row, int column) {

            if( value == null ) {

                this.picker.clean();

                return this.combo;

            }

            this.picker.setFeatureType(source.getDefaultFeatureTypeQuietly());

            this.picker.setAllowNull(true);

            this.picker.set(Objects.toString(value,"")); 

            return this.combo;

        }

        @Override

        public Object getCellEditorValue() {

            return this.picker.getName();

        }        

    }

    private class MyTableModel extends AbstractTableModel implements Iterable<MyTableModel.Row>{

        private final List<Row> rows;

        private final String[] columnNames;

        private final Class[] columnClass;

        private class Row {

            String source;

            String target;

            boolean update;

            public Object get(int index) {

                switch(index) {

                    case 0:

                        return target;

                    case 1:

                        return source;

                    case 2:

                        return update;

                    default:

                        return null;

                }

            }

            public void set(int index, Object value) {

                switch(index) {

                    case 0:

                        this.target = (String) value;

                        break;

                    case 1:

                        this.source = (String) value;

                        break;

                    case 2:

                        this.update = (boolean) value;

                        break;

                    default:

                        break;

                }

            }

        }

        public MyTableModel() {

            this.rows = new ArrayList<>();

            FeatureType targetType = target.getDefaultFeatureTypeQuietly();

            FeatureType sourceType = source.getDefaultFeatureTypeQuietly();

            for (FeatureAttributeDescriptor targetAttr : targetType) {

                Row row = new Row();

                row.target = targetAttr.getName();

                row.source = null;

                if( sourceType.getAttributeDescriptor(targetAttr.getName())!=null ) {

                    row.source = row.target;

                }

                if( targetAttr.isAutomatic() || targetAttr.isReadOnly() || targetAttr.isPrimaryKey() ) {

                    row.update = false;

                } else {

                    row.update = true;

                }

                this.rows.add(row);

            }

            this.columnNames = new String[] {

                "Campo destino",

                "Campo origen",

                "Actualizar"

            };

            this.columnClass = new Class[] {

                String.class,

                String.class,

                Boolean.class

            };

        }

        @Override

        public String getColumnName(int column) {

            return this.columnNames[column];

        }

        @Override

        public Class<?> getColumnClass(int columnIndex) {

            return this.columnClass[columnIndex];

        }

        @Override

        public int getRowCount() {

            return this.rows.size();

        }

        @Override

        public int getColumnCount() {

            return 3;

        }

        @Override

        public Object getValueAt(int rowIndex, int columnIndex) {

            return this.rows.get(rowIndex).get(columnIndex);

        }

        @Override

        public void setValueAt(Object aValue, int rowIndex, int columnIndex) {

            this.rows.get(rowIndex).set(columnIndex, aValue);

        }

        @Override

        public boolean isCellEditable(int rowIndex, int columnIndex) {

            switch(columnIndex) {

                case 0:

                default:

                    return false;

                case 1:

                case 2:

                    return true;

            }

        }

        @Override

        public Iterator<Row> iterator() {

            return this.rows.iterator();

        }

    }

    public CopyTablePanel() {

        this.initComponents();

    }

    private void initComponents() {

//        DataManager dataManager = DALLocator.getDataManager();

        DataSwingManager dataSwingManager = DALSwingLocator.getSwingManager();

        ExpressionEvaluatorSwingManager expressionSwingManager = ExpressionEvaluatorSwingLocator.getManager();

        this.status = ToolsSwingLocator.getTaskStatusSwingManager().createTaskStatusController(lblStatusLabel, lblStatusMessage, pbStatus);

        this.treeSourceTableController = new StoresRepositoryController(this.treeSourceTable);

        this.treeSourceTableController.addTreeSelectionListener((TreeSelectionEvent e) -> {

            doChangeSource();

        });

        this.pickerSourceFieldJoin = dataSwingManager.createAttributeDescriptorPickerController(cboSourceJoinField);

        this.treeTargetTableController = new StoresRepositoryController(this.treeTargetTable);

        this.treeTargetTableController.addTreeSelectionListener((TreeSelectionEvent e) -> {

            doChangeTarget();

        });

        this.pickerTargetFieldJoin = dataSwingManager.createAttributeDescriptorPickerController(cboTargetJoinField);

        this.pickerSourceFilter = expressionSwingManager.createExpressionPickerController(

                txtSourceFilter, 

                btnSourceFilter, 

                btnSourceFilterBookmarks, 

                btnSourceFilterHistory

        );

        this.pickerTargetUpdateExpression = expressionSwingManager.createExpressionPickerController(

                txtTargetUpdateWhenExpression, 

                btnTargetUpdateWhenExpression, 

                btnTargetUpdateWhenExpressionBookmarks, 

                btnTargetUpdateWhenExpressionHistory

        );

        this.chkTargetInsertIfNotExistsInTarget.setSelected(false);

        this.tableFields.setAutoCreateColumnsFromModel(false);

        this.tableFields.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

        this.fieldsColumnModel = new DefaultTableColumnModel();

        TableColumn column = new TableColumn();

        column.setHeaderValue("Campo destino");

        column.setCellRenderer(null);

        column.setCellEditor(new FeatureAttributeCellEditor());

        column.setResizable(true);

        column.setModelIndex(0);

        this.fieldsColumnModel.addColumn(column);

        column = new TableColumn();

        column.setHeaderValue("Campo origen");

        column.setCellRenderer(null);

        column.setCellEditor(new FeatureAttributeCellEditor());

        column.setResizable(true);

        column.setModelIndex(1);

        this.fieldsColumnModel.addColumn(column);

        column = new TableColumn();

        column.setHeaderValue("Actualizar");

        column.setCellRenderer(null);

        column.setResizable(true);

        column.setModelIndex(2);

        this.fieldsColumnModel.addColumn(column);

        this.tableFields.setColumnModel(this.fieldsColumnModel);

        SwingUtilities.invokeLater(() -> {

            lblStatusLabel.setText("");

            lblStatusMessage.setText("");

            pbStatus.setVisible(false);

        });

        ToolsSwingUtils.ensureRowsCols(this, 20, 80, 24, 100);

    }

    private void doChangeSource() {

        this.source = this.treeSourceTableController.getSelectedFeatureStore();

        if( this.source==null ) {

            return;

        }

        DataSwingManager dataSwingManager = DALSwingLocator.getSwingManager();

        dataSwingManager.resetExpressionBuilder(this.pickerTargetUpdateExpression);

        dataSwingManager.addToExpressionBuilder(this.pickerTargetUpdateExpression, this.source, "source");

        this.pickerSourceFieldJoin.setFeatureType(this.source.getDefaultFeatureTypeQuietly());

        if( this.target!=null ) {

            this.tableFields.setModel(new MyTableModel());

            dataSwingManager.addToExpressionBuilder(this.pickerTargetUpdateExpression, this.target, "target");

        }

    }

    private void doChangeTarget() {

        this.target = this.treeTargetTableController.getSelectedFeatureStore();

        if( this.target==null ) {

            return;

        }

        DataSwingManager dataSwingManager = DALSwingLocator.getSwingManager();

        dataSwingManager.resetExpressionBuilder(this.pickerTargetUpdateExpression);

        dataSwingManager.addToExpressionBuilder(this.pickerTargetUpdateExpression, this.target, "target");

        this.pickerTargetFieldJoin.setFeatureType(this.target.getDefaultFeatureTypeQuietly());

        if( this.source!=null ) {

            this.tableFields.setModel(new MyTableModel());

            dataSwingManager.addToExpressionBuilder(this.pickerTargetUpdateExpression, this.source, "source");

        }

    }

    public FeatureStore getSource() {

        return this.source;

    }

    public String getSourceFieldJoin() {

        return this.pickerSourceFieldJoin.getName();

    }

    public Expression getSourceFilter() {

        return this.pickerSourceFilter.get();

    }

    public FeatureStore getTarget() {

        return this.target;

    }

    public String getTargetFieldJoin() {

        return this.pickerTargetFieldJoin.getName();

    }

    public Expression getUpdateIfExistsInTargetCondition() {

        return this.pickerTargetUpdateExpression.get();

    }

    public boolean getInsertIfNotExistsTarget() {

        return this.chkTargetInsertIfNotExistsInTarget.isSelected();

    }

    public boolean getUpdateIfExistsInTarget() {

        return this.chkTargetUpdateIfExistsInTarget.isSelected();

    }

    public boolean getDeleteIfNotExiststInSource() {

        return this.chkTargetDeleteIfNotExiststInSource.isSelected();

    }

    public SimpleTaskStatus getStatus() {

        if( this.taskStatus == null ) {

            this.taskStatus = ToolsLocator.getTaskStatusManager().createDefaultSimpleTaskStatus("Copy table");

            this.status.bind(this.taskStatus);

        }

        lblStatusLabel.setVisible(true);

        lblStatusMessage.setVisible(true);

        pbStatus.setVisible(true);

        return this.taskStatus;

    }

    public Map<String/*target*/, String/*source*/> getFields() {

        Map<String/*target*/, String/*source*/> fields = new HashMap<>();

        MyTableModel model = (MyTableModel) this.tableFields.getModel();

        for (MyTableModel.Row row : model) {

            if( StringUtils.isNotBlank(row.source) ) {

                fields.put(row.target, row.source);

            }

        }

        return fields;

    }

    public List<String/*target*/> getFieldsToUpdate() {

        List<String> fields = new ArrayList<>();

        MyTableModel model = (MyTableModel) this.tableFields.getModel();

        for (MyTableModel.Row row : model) {

            if( StringUtils.isNotBlank(row.source) && row.update) {

                fields.add(row.target);

            }

        }

        return fields;

    }

}