Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.app.extension.resourcesmanager;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.util.Collection;

import java.util.Collections;

import java.util.List;

import java.util.logging.Level;

import javax.swing.DefaultListModel;

import javax.swing.DefaultListSelectionModel;

import javax.swing.ImageIcon;

import javax.swing.JComponent;

import javax.swing.JFileChooser;

import javax.swing.ListSelectionModel;

import javax.swing.event.ChangeEvent;

import javax.swing.event.ChangeListener;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.ArrayUtils;

import org.gvsig.andami.PluginsLocator;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.DatabaseWorkspaceManager;

import org.gvsig.fmap.dal.StoresRepository;

import org.gvsig.fmap.dal.exception.InitializeException;

import org.gvsig.fmap.dal.exception.ProviderNotRegisteredException;

import org.gvsig.fmap.dal.exception.ValidateDataParametersException;

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

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.folders.FoldersManager;

import org.gvsig.tools.resourcesstorage.FilesResourcesStorage;

import org.gvsig.tools.resourcesstorage.ResourcesStorage;

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

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

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.threadsafedialogs.ThreadSafeDialogsManager;

import org.gvsig.tools.util.LabeledValue;

import org.gvsig.tools.util.LabeledValueImpl;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author fdiaz

 */

public class ResourcesLoaderPanel extends ResourcesLoaderPanelView implements Component {

    private static final Logger LOGGER = LoggerFactory.getLogger(ResourcesLoaderPanel.class);

    private PickerController<JDBCServerExplorerParameters> connectionPicker;

    private JDBCServerExplorerParameters connection;

    private JDBCServerExplorer server;

    public ResourcesLoaderPanel() {

        translate();

        try {

            FileUtils.forceMkdir(getLocalResourcesPath());

        } catch (IOException ex) {

            LOGGER.warn("Can't create local resources path.", ex);

        }

        initComponents();

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    private void translate() {

        ToolsSwingManager swingManager = ToolsSwingLocator.getToolsSwingManager();

        swingManager.translate(this.lblConnection);

        swingManager.translate(this.lblDBResources);

        swingManager.translate(this.lblLocalResources);

        swingManager.translate(this.btnDownload);

        swingManager.translate(this.btnUpload);

        swingManager.translate(this.btnAddLocal);

        swingManager.translate(this.btnDeleteLocal);

        swingManager.translate(this.btnDeleteRemote);

        swingManager.translate(this.chkUseLocalResources);

    }

    private void initComponents() {

        this.connectionPicker = DALSwingLocator.getSwingManager().createJDBCConnectionPickerController(

                this.cboConnection,

                this.btnConnection

        );

        this.connectionPicker.addChangeListener(new ChangeListener() {

            @Override

            public void stateChanged(ChangeEvent e) {

                connection = null;

                if(!isValidSelectedConnection()){

                    connectionPicker.set(null);

                }

                chkUseLocalResources.setSelected(getUseLocalResources());

                refreshDBResources();

                doUpdateEnableComponents();

            }

        });

        DefaultListSelectionModel lstDBResourcesSelectionModel = new DefaultListSelectionModel();

        lstDBResourcesSelectionModel.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

        lstDBResources.setSelectionModel(lstDBResourcesSelectionModel);

        lstDBResourcesSelectionModel.addListSelectionListener(new ListSelectionListener() {

            @Override

            public void valueChanged(ListSelectionEvent e) {

                doUpdateEnableComponents();

            }

        });

        refreshLocalResources();

        DefaultListSelectionModel lstLocalResourcesSelectionModel = new DefaultListSelectionModel();

        lstLocalResourcesSelectionModel.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

        lstLocalResources.setSelectionModel(lstLocalResourcesSelectionModel);

        lstLocalResourcesSelectionModel.addListSelectionListener(new ListSelectionListener() {

            @Override

            public void valueChanged(ListSelectionEvent e) {

                doUpdateEnableComponents();

            }

        });

        btnDownload.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doDownload();

            }

        });

        btnUpload.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doUpload();

            }

        });

        btnDeleteLocal.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doDeleteLocal();

            }

        });

        btnDeleteRemote.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doDeleteRemote();

            }

        });

        btnAddLocal.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doAddlocal();

            }

        });

        chkUseLocalResources.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                doChangeUseLocalResources();

            }

        });

        chkUseLocalResources.setSelected(getUseLocalResources());

        doUpdateEnableComponents();

        ToolsSwingUtils.ensureRowsCols(this, 20, 60);

    }

    private void doDownload() {

        LabeledValue selected = (LabeledValue) lstDBResources.getSelectedValue();

        if (selected != null) {

            String name = selected.getLabel();

            ResourcesStorage.Resource resource = (ResourcesStorage.Resource) selected.getValue();

            FilesResourcesStorage localResourcesStorage = getLocalResourcesStorage();

            if (localResourcesStorage != null) {

                try {

                    if (localResourcesStorage.exists(name)) {

                        ResourcesStorage.Resource localResource = localResourcesStorage.getResource(name);

                        copy(resource, localResource);

                    } else {

                        localResourcesStorage.add(name, resource.asInputStream());

                    }

                    refreshLocalResources();

                    refreshDocuments();

                } catch (IOException ex) {

                    LOGGER.warn("Can't download '" + name + "' resource'", ex);

                }

            }

        }

    }

    private void doUpload() {

        LabeledValue selected = (LabeledValue) lstLocalResources.getSelectedValue();

        if (selected != null) {

            String name = selected.getLabel();

            ResourcesStorage.Resource resource = (ResourcesStorage.Resource) selected.getValue();

            ResourcesStorage remoteResourcesStorage = getDBResourcesStorage();

            if (remoteResourcesStorage != null) {

                try {

                    if (remoteResourcesStorage.exists(name)) {

                        ResourcesStorage.Resource remoteResource = remoteResourcesStorage.getResource(name);

                        copy(resource, remoteResource);

                    } else {

                        remoteResourcesStorage.add(name, resource.asInputStream());

                    }

                    refreshDBResources();

                    refreshDocuments();

                } catch (IOException ex) {

                    LOGGER.warn("Can't upload '" + name + "' resource'", ex);

                }

            }

        }

    }

    private void doDeleteLocal() {

        LabeledValue selected = (LabeledValue) lstLocalResources.getSelectedValue();

        if (selected != null) {

            String name = selected.getLabel();

            FilesResourcesStorage localResourcesStorage = getLocalResourcesStorage();

            if (localResourcesStorage != null && localResourcesStorage.exists(name)) {

                localResourcesStorage.remove(name);

                refreshLocalResources();

            }

        }

    }

    private void doDeleteRemote() {

        LabeledValue selected = (LabeledValue) lstDBResources.getSelectedValue();

        if (selected != null) {

            String name = selected.getLabel();

            ResourcesStorage remoteResourcesStorage = getDBResourcesStorage();

            if (remoteResourcesStorage != null && remoteResourcesStorage.exists(name)) {

                remoteResourcesStorage.remove(name);

                refreshDBResources();

            }

        }

    }

    private void doAddlocal() {

        ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

        FoldersManager folderManager = ToolsLocator.getFoldersManager();

        File[] f = dialogs.showChooserDialog(

                "_Resources_loader", //this.dialogTitle,

                JFileChooser.OPEN_DIALOG,

                JFileChooser.FILES_ONLY,

                false,

                null,

                null,

                true

        );

        if (ArrayUtils.isEmpty(f)) {

            return;

        }

        String name = f[0].getName();

        try {

            FileInputStream is = new FileInputStream(f[0]);

            FilesResourcesStorage localResourcesStorage = getLocalResourcesStorage();

            if (localResourcesStorage != null) {

                if (localResourcesStorage.exists(name)) {

                    ResourcesStorage.Resource localResource = localResourcesStorage.getResource(name);

                    IOUtils.copy(is, localResource.asOutputStream());

                    is.close();

                    localResource.close();

                } else {

                    localResourcesStorage.add(name, is);

                }

                refreshLocalResources();

            }

        } catch (IOException ex) {

            LOGGER.warn("Can't add '" + name + "' resource'", ex);

        }

    }

    private boolean getUseLocalResources() {

        DataManager manager = DALLocator.getDataManager();

        JDBCServerExplorerParameters con = getConnection();

        if(con == null){

            return false;

        }

        DatabaseWorkspaceManager ws = manager.getDatabaseWorkspace(con);

        if(ws == null){

            return false;

        }

        return ws.hasAlternativeResourcesStorage();

    }

    private boolean isValidSelectedConnection() {

        DataManager manager = DALLocator.getDataManager();

        JDBCServerExplorerParameters con = this.connectionPicker.get();

        if(con == null){

            return false;

        }

        DatabaseWorkspaceManager ws = manager.getDatabaseWorkspace(con);

        if(ws == null){

            return false;

        }

        return true;

    }

    private void doChangeUseLocalResources() {

        DataManager manager = DALLocator.getDataManager();

        JDBCServerExplorerParameters con = getConnection();

        if(con == null){

            return;

        }

        DatabaseWorkspaceManager ws = manager.getDatabaseWorkspace(con);

        if(chkUseLocalResources.isSelected()) {

            String s = "<%=$GVSIGHOMEFOLDER%>/plugins/org.gvsig.app.mainplugin/LocalResources";

            ws.setAlternativeResourcesStorage(s);

        } else {

            ws.setAlternativeResourcesStorage(null);

        }

        refreshDocuments();

    }

    private void refreshDocuments() {

        DataManager manager = DALLocator.getDataManager();

        JDBCServerExplorerParameters con = getConnection();

        if(con == null){

            return;

        }

        DatabaseWorkspaceManager ws = manager.getDatabaseWorkspace(con);

        StoresRepository repository = ws.getStoresRepository();

        for (String name : repository.keySet()) {

            DataStoreParameters params = repository.get(name);

            ApplicationLocator.getApplicationManager().refreshDocument(params);

        }

    }

    private void refreshDBResources() {

        ResourcesStorage resourcesStorage = getDBResourcesStorage();

        List<String> dbResourcesNames = resourcesStorage == null ? Collections.EMPTY_LIST : resourcesStorage.getResourceNames();

        DefaultListModel dBResourcesModel = new DefaultListModel();

        for (String resourceName : dbResourcesNames) {

            ResourcesStorage.Resource resource = resourcesStorage.getResource(resourceName);

            dBResourcesModel.addElement(new LabeledValueImpl<>(resourceName, resource));

        }

        lstDBResources.setModel(dBResourcesModel);

    }

    private void refreshLocalResources() {

        FilesResourcesStorage localResourcesStorage = getLocalResourcesStorage();

        List<String> localResourcesNames = localResourcesStorage.getResourceNames();

        DefaultListModel localResourcesModel = new DefaultListModel();

        for (String resourceName : localResourcesNames) {

            ResourcesStorage.Resource resource = localResourcesStorage.getResource(resourceName);

            localResourcesModel.addElement(new LabeledValueImpl<>(resourceName, resource));

        }

        lstLocalResources.setModel(localResourcesModel);

    }

    @Override

    public ImageIcon loadImage(String imageName) {

        return ToolsSwingUtils.loadImage(this, imageName);

    }  

    public JDBCServerExplorerParameters getConnection() {

        if (this.connection == null) {

            this.connection = this.connectionPicker.get();

        }

        return this.connection;

    }

    public String getConnectionLabel() {

        String label = "";

        if (this.getConnection() != null) {

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

            label = item.getLabel();

        }

        return label;

    }

    public JDBCServerExplorer getServer() {

        if (this.server == null) {

            try {

                DataManager dataManager = DALLocator.getDataManager();

                if (getConnection() != null) {

                    this.server = (JDBCServerExplorer) dataManager.openServerExplorer(

                            getConnection().getExplorerName(), getConnection()

                    );

                }

            } catch (InitializeException | ProviderNotRegisteredException | ValidateDataParametersException ex) {

                LOGGER.warn("Can't open server " + getConnection().getExplorerName());

            }

        }

        return this.server;

    }

    public ResourcesStorage getDBResourcesStorage() {

        JDBCServerExplorer theServer = getServer();

        return theServer==null?null:theServer.getResourcesStorage();

    }

    public final File getLocalResourcesPath() {

        File folder = PluginsLocator.getPluginsManager().getApplicationHomeFolder();

        File f = FileUtils.getFile(folder, "plugins","org.gvsig.app.mainplugin","LocalResources");

        return f;

    }

    public FilesResourcesStorage getLocalResourcesStorage() {

        File f = getLocalResourcesPath();

        FilesResourcesStorage resources = new FilesResourcesStorage(f.getAbsolutePath());

        return resources;

    }

    private void doUpdateEnableComponents() {

        ResourcesStorage dBResourcesStorage = getDBResourcesStorage();

        FilesResourcesStorage localResourcesStorage = getLocalResourcesStorage();

        this.chkUseLocalResources.setEnabled(dBResourcesStorage != null);

        this.btnDownload.setEnabled(

                dBResourcesStorage != null

                && localResourcesStorage != null

                && lstDBResources.getSelectedValue() != null);

        this.btnUpload.setEnabled(

                dBResourcesStorage != null

                && localResourcesStorage != null

                && lstLocalResources.getSelectedValue() != null);

        this.btnDeleteLocal.setEnabled(localResourcesStorage != null && lstLocalResources.getSelectedValue() != null);

        this.btnDeleteRemote.setEnabled(dBResourcesStorage != null && lstDBResources.getSelectedValue() != null);

    }

    private void copy(ResourcesStorage.Resource source, ResourcesStorage.Resource target) throws IOException {

        IOUtils.copy(source.asInputStream(), target.asOutputStream());

        source.close();

        target.close();

    }

}