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.online.swing.impl.download;

import java.util.ArrayList;

import java.util.Collections;

import java.util.List;

import java.util.stream.Collectors;

import javax.swing.ImageIcon;

import javax.swing.JComponent;

import javax.swing.SwingUtilities;

import javax.swing.event.ChangeEvent;

import javax.swing.event.TableModelEvent;

import org.apache.commons.collections4.CollectionUtils;

import org.apache.commons.lang3.StringUtils;

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

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

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.online.lib.api.OnlineLayer;

import org.gvsig.online.lib.api.OnlineLocator;

import org.gvsig.online.lib.api.OnlineManager;

import static org.gvsig.online.lib.api.OnlineManager.ERR_NO_ERROR;

import org.gvsig.online.lib.api.OnlineProject;

import org.gvsig.online.lib.api.workingcopy.OnlineWorkingcopy;

import org.gvsig.online.lib.api.workingcopy.WorkingArea;

import org.gvsig.online.swing.api.OnlineJDownload;

import org.gvsig.online.swing.api.OnlineSwingLocator;

import org.gvsig.online.swing.api.OnlineSwingManager;

import org.gvsig.online.swing.api.OnlineSwingServices;

import org.gvsig.online.swing.impl.OnlineSwingCommons;

import static org.gvsig.online.swing.impl.OnlineSwingCommons.notInSwingThreadInvokeLater;

import org.gvsig.online.swing.impl.initworkspace.LayersTableModel;

import org.gvsig.online.swing.impl.workingarea.WorkingAreaPickerControllerImpl;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

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.ToolsSwingUtils;

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.task.SimpleTaskStatus;

import org.gvsig.tools.task.UserCancelTaskException;

import org.gvsig.tools.util.LabeledValue;

import org.slf4j.LoggerFactory;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class OnlineJDownloadImpl extends OnlineJDownloadView

        implements Component, OnlineJDownload  {

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

    private Dialog dialog;

    private WorkingAreaPickerControllerImpl workingAreaPicker;

    private TaskStatusController taskStatusController;

    private PickerController<OnlineWorkingcopy> workingcopyPicker;

    private List<String> defaultTablesSelection;

    private boolean processing;

    public OnlineJDownloadImpl() {

        translate();

        initComponents();

    }

    public OnlineJDownloadImpl(OnlineWorkingcopy workspace, List<String> tableNames, Boolean overwrite) {

        processing = false;

        translate();

        if( tableNames==null ) {

            this.defaultTablesSelection = Collections.EMPTY_LIST;

        } else {

            this.defaultTablesSelection = tableNames;

        }

        initComponents();

        SwingUtilities.invokeLater(() -> {

            this.workingcopyPicker.set(workspace);

//            if( overwrite!=null ) {

//                this.chkOverwriteTable.setSelected(overwrite);

//            }

        });

    }

    private void initComponents() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        TaskStatusSwingManager taskStatusManager = ToolsSwingLocator.getTaskStatusSwingManager();

        DataSwingManager dataSwingManager = DALSwingLocator.getSwingManager();

        OnlineSwingManager swingManager = OnlineSwingLocator.getOnlineSwingManager();

        OnlineSwingServices onlineservices = OnlineSwingLocator.getOnlineSwingManager().getDefaultServices();

        this.taskStatusController = taskStatusManager.createTaskStatusController(

                null,

                this.lblStatusTitle,

                this.lblStatusMessages,

                this.pbStatus,

                this.btnStatusCancel,

                null

        );

        this.taskStatusController.setShowCancelButton(false);

        this.taskStatusController.setShowRemoveTaskButton(false);

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

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

        this.tblLayers.setModel(new LayersTableModel());

        this.workingcopyPicker = swingManager.createPickerWorkspaceController(cboWorkingcopy);

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

            this.clearMessage();

            doChangeWorkingcopy();

        });

        this.txtWorkspaceName.setText("");

        this.txtWorkspaceName.setEditable(false);

        this.workingAreaPicker = new WorkingAreaPickerControllerImpl(

            () -> {return this.getWorkingcopy().getSite();},

            txtWorkingAreaLabel,

            txtWorkingArea,

            btnWorkingAreaFromView,

            btnWorkingAreaDialog,

            btnWorkingAreaHistory,

            btnWorkingAreaBookmarks,

            () -> {

                LabeledValue<MapControl> n = onlineservices.getActiveMapControl();

                return (n == null) ? null : n.getValue();

            },

            () -> {

                List<OnlineLayer> selectedLayers = getLayersModel().getSelectedLayers();

                if (selectedLayers.isEmpty()) {

                    return null;

                }

                return selectedLayers;

            }

        );

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

            this.getWorkingcopy().setCurrentWorkingArea(this.workingAreaPicker.getWithoutAddHistory());

            doUpdateEnableComponents();

        });

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

        ToolsSwingUtils.ensureRowsCols(this, 15, 40, 25, 100);

    }

    @Override

    public void setDialog(Dialog dialog) {

        this.dialog = dialog;

        this.doUpdateEnableComponents();

    }

    private void doUpdateEnableComponents() {

        message("");

        boolean enableAll = true;

        boolean isThereWorkingcopy = false;

        boolean isThereSelectedsLayers = false;

        boolean isThereWorkingArea = false;

        // 0. Check is processing

        enableAll = !this.isProcessing();

        // 1. Check workingcopy

        if( enableAll ) {

            OnlineWorkingcopy w = this.getWorkingcopy();

            isThereWorkingcopy = (w != null);

            if( !isThereWorkingcopy ) {

                message2(enableAll, "_You_must_select_a_workingcopy");

            }

            enableAll = enableAll && isThereWorkingcopy;

        }

        // 2. Check layers selected

        if( enableAll ) {

            isThereSelectedsLayers = !this.getLayersModel().isSelectionEmpty();

            if( !isThereSelectedsLayers ) {

                message2(enableAll, "_You_must_select_some_layer");

            }

            enableAll = isThereWorkingcopy && isThereSelectedsLayers;

        }

        // 3. Check selected working area

        if( enableAll ) {

            WorkingArea workingArea = this.workingAreaPicker.getWithoutAddHistory();

            isThereWorkingArea = (workingArea != null && workingArea.getValue() != null);

            if( !isThereWorkingArea ) {

                message2(enableAll, "_You_must_select_a_working_area");

            }

            enableAll = isThereWorkingcopy && isThereWorkingArea;

        }

        this.tblLayers.setEnabled(isThereWorkingcopy);

        this.txtWorkingAreaLabel.setEnabled(isThereWorkingcopy);

        this.txtWorkingArea.setEnabled(isThereWorkingcopy);

        this.btnWorkingAreaFromView.setEnabled(isThereWorkingcopy);

        this.btnWorkingAreaDialog.setEnabled(isThereSelectedsLayers && isThereWorkingcopy);

        if (this.dialog != null) {

            this.dialog.setButtonEnabled(WindowManager_v2.BUTTON_OK,enableAll);

        }

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    private void translate() {

        ToolsSwingManager swingManager = ToolsSwingLocator.getToolsSwingManager();

        swingManager.translate(this.lblWorkspaceName);

        swingManager.translate(this.lblProjects);

        swingManager.translate(this.lblLayers);

        swingManager.translate(this.lblWorkingArea);

        swingManager.translate(this.btnWorkingAreaFromView);

        swingManager.translate(this.btnWorkingAreaDialog);

    }

    @Override

    public int actionPerformed() {

        I18nManager i18n = ToolsLocator.getI18nManager();

        OnlineManager manager = OnlineLocator.getOnlineManager();

        OnlineWorkingcopy workingcopy = this.getWorkingcopy();

        if( workingcopy==null ) {

            return OnlineManager.ERR_WORKINGCOPY_REQUIRED;

        }

        WorkingArea workingArea = this.workingAreaPicker.get();

        if (workingArea==null ) {

            return OnlineManager.ERR_WORKINGAREA_REQUIRED;

        }

        int err = ERR_NO_ERROR;

        SimpleTaskStatus status = this.getTaskStatus();

        try {

            status.setTitle(i18n.getTranslation("_Initializing_workspace"));

            LayersTableModel model = (LayersTableModel) this.tblLayers.getModel();

            List<OnlineLayer> selectedLayers = model.getSelectedLayers();

            if( !CollectionUtils.isEmpty(selectedLayers) ) {

                status.setRangeOfValues(0, selectedLayers.size());

                status.setCurValue(0);

                for (OnlineLayer layer : selectedLayers) {

                    status.incrementCurrentValue();

                    if( status.isCancellationRequested() ) {

                        status.cancel();

                        throw new UserCancelTaskException();

                    }

                    if( workingcopy.getEntity(layer.getName())==null ) {

                        status.setTitle(i18n.getTranslation("_Adding_layer")+" "+layer.getName());

                        String fieldForLabel = "ogc_fid"; // FIXME: Pedir en el GUI

                        String label = layer.getName(); // FIXME: Pedir en el GUI

                        int tileSize = layer.calculateTileSize(5000, status);

                        workingcopy.addEntity(

                                layer.getFeatureType(), 

                                layer.getName(), 

                                layer.getTitle(), 

                                fieldForLabel, 

                                label, 

                                layer.getPkName(),

                                tileSize

                        );

                    }

                }

                status.setRangeOfValues(0, selectedLayers.size());

                status.setCurValue(0);

                for (OnlineLayer layer : selectedLayers) {

                    status.incrementCurrentValue();

                    status.setTitle(i18n.getTranslation("_Downloading_layer")+" "+layer.getName());

                    if( status.isCancellationRequested() ) {

                        status.cancel();

                        throw new UserCancelTaskException();

                    }

                    err = workingcopy.download(layer.getName(),workingArea.getValue(), status);

                    if (err != ERR_NO_ERROR ) {

                        // TODO: Que hacer aqui si da error

                    }

                }

            }

            OnlineSwingServices services = OnlineSwingLocator.getOnlineSwingManager().getDefaultServices();

            services.refreshDocuments();

        } catch (UserCancelTaskException ex) {

            LOGGER.info("Cancelled by user.");

            err = OnlineManager.ERR_CANCELLED_BY_USER;

        } catch (Exception ex) {

            LOGGER.info("Can't initialize workspace", ex);

            err = OnlineManager.ERR_EXCEPTION;

        }

        return err;

    }

    public static void selfRegister() {

    }

    @Override

    public boolean isProcessing() {

        return false;

    }

    private void message2(boolean doMessage, String s) {

        if( doMessage ) {

            message(s);

        }

    }

    private void message(String message) {

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

            return;

        }

        String s = message;

        if( StringUtils.startsWith(s, "_") ) {

            I18nManager i18n = ToolsLocator.getI18nManager();

            s = i18n.getTranslation(s);

        }

        this.lblStatusMessages.setText(s);

//        this.lblStatusMessages.setToolTipText(toolTipText);        

    }

    @Override

    public ImageIcon loadImage(String imageName) {

        return OnlineSwingCommons.loadImage(imageName);

    }

    private void clearLayers() {

        LayersTableModel model = new LayersTableModel();

        this.tblLayers.setModel(model);

    }

//    private WorkingArea getWorkingArea() {

//        return this.workingAreaPicker.get();

//    }

    private LayersTableModel getLayersModel() {

        return (LayersTableModel) this.tblLayers.getModel();

    }

    private SimpleTaskStatus getTaskStatus() {

        SimpleTaskStatus taskStatus = ToolsLocator.getTaskStatusManager().createDefaultSimpleTaskStatus("_Create_workingcopy");

        taskStatus.setAutoremove(true);

        taskStatus.add();

        taskStatus.setIndeterminate();

        this.taskStatusController.bind(taskStatus);

        return taskStatus;

    }

    public void setVisibleStatus(boolean visible) {

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

            return;

        }

        this.lblStatusTitle.setVisible(visible);

        this.pbStatus.setVisible(visible);

        this.lblStatusMessages.setVisible(true);

    }

    private void clearMessage() {

        message(" ");

    }

    @Override

    public OnlineWorkingcopy getWorkingcopy() {

        OnlineWorkingcopy w = this.workingcopyPicker.get();

        return w;

    }

    @Override

    public List<String> getSelectedLayers() {

        OnlineWorkingcopy w = this.getWorkingcopy();

        if( w == null ) {

            return Collections.EMPTY_LIST;

        }

        LayersTableModel model = (LayersTableModel) this.tblLayers.getModel();

        List<OnlineLayer> selectedLayers = model.getSelectedLayers();

        if( CollectionUtils.isEmpty(selectedLayers) ) {

            return Collections.EMPTY_LIST;

        }

        return selectedLayers.stream().map((OnlineLayer t) -> t.getName()).collect(Collectors.toList());

    }

    private void doChangeWorkingcopy() {

        OnlineWorkingcopy w = this.getWorkingcopy();

        if( w == null ) {

            return;

        }

        OnlineProject project = w.getProject();

        List<OnlineLayer> layers = new ArrayList<>();

        for (OnlineLayer layer : project.layers()) {

            if( layer.isHidden() ) {

                continue;

            }

            layers.add(layer);

        }

        layers.sort((OnlineLayer o1, OnlineLayer o2) -> o1.getLabel().compareTo(o2.getLabel()));

        LayersTableModel model = new LayersTableModel(layers);

        model.addTableModelListener((TableModelEvent e) -> {

            doUpdateEnableComponents();

        });

        this.txtWorkspaceName.setText(w.getLabel());

        this.tblLayers.setModel(model);

        this.workingAreaPicker.set(w.getCurrentWorkingArea());

        doUpdateEnableComponents();    

    }

}