Application: gvSIG desktop

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright (c) 2007-2023 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.xml2db.swing.impl.copyxml2db;

import org.gvsig.xml2db.swing.impl.Task;

import java.awt.Color;

import java.io.File;

import java.util.ArrayList;

import java.util.List;

import javax.swing.AbstractListModel;

import javax.swing.JComponent;

import org.cresques.cts.IProjection;

import java.nio.charset.Charset;

import java.util.Locale;

import javax.swing.JOptionPane;

import javax.swing.SwingUtilities;

import javax.swing.event.ChangeEvent;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.store.h2.H2SpatialUtils;

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

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

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

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

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

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

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

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

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.util.LabeledValue;

import org.gvsig.tools.util.LabeledValueImpl;

import org.gvsig.xml2db.lib.api.Xml2dbLocator;

import org.gvsig.xml2db.lib.api.Xml2dbManager;

import org.gvsig.xml2db.lib.api.xmlinfo.XMLInfo;

import org.gvsig.xml2db.lib.api.xmlinfo.XMLTableInfo;

import org.gvsig.xml2db.swing.Xml2dbPanel;

import org.gvsig.xml2db.swing.impl.Xml2dbSwingCommons;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class CopyXml2dbPanel 

        extends CopyXml2dbPanelView 

        implements Xml2dbPanel

    {

    protected static final Logger LOGGER = LoggerFactory.getLogger(CopyXml2dbPanel.class);

    private PickerController<Locale> pickerLocale;

    private static class XMLInfoListModel extends AbstractListModel<LabeledValue<XMLTableInfo>> {

        private final XMLInfo xmlinfo;

        private final List<LabeledValue<XMLTableInfo>> tables;

        public XMLInfoListModel() {

            this(null);

        }

        public XMLInfoListModel(XMLInfo xmlinfo) {

            this.xmlinfo = xmlinfo;

            if( xmlinfo != null ) {

                this.tables = new ArrayList<>(xmlinfo.size());

                for (XMLTableInfo tableInfo : xmlinfo) {

                    this.tables.add(new LabeledValueImpl<>(tableInfo.getName(),tableInfo));

                }

            } else {

                this.tables = null;

            }

        }

        @Override

        public LabeledValue<XMLTableInfo> getElementAt(int index) {

            if( this.tables == null ) {

                return null;

            }

            return tables.get(index);

        }

        @Override

        public int getSize() {

            if( this.tables == null ) {

                return 0;

            }

            return this.tables.size();

        }

    }

    private Dialog dialog;

    private FilePickerController pickerXMLFile;

    private FilePickerController pickerDbfile;

    private ProjectionPickerController pickerProjection;

    private CharsetPickerController pickerCharset;

    private TaskStatusController taskStatusController;

    private Task lastTask;

    private XMLInfo xmlinfo;

    public CopyXml2dbPanel() {

        this.createComponents();

    }

    private void createComponents() {    

        this.lastTask = Task.FAKE_TASK;

        this.xmlinfo = null;

        this.initComponents();

    }

    private void initComponents() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        DataSwingManager dataSwingManager = DALSwingLocator.getDataSwingManager();

        this.taskStatusController = ToolsSwingLocator.getTaskStatusSwingManager().createTaskStatusController(

                lblStatusCaption, 

                lblStatusMsg,

                pbStatus

        );

        this.taskStatusController.setVisible(false);

        this.translate();

        this.pickerXMLFile = toolsSwingManager.createFilePickerController(

                this.txtXMLFile, 

                this.btnXMLFile

        );

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

            doChangeXMLfile();

        });

        this.pickerCharset = toolsSwingManager.createCharsetPickerController(

                null,

                this.cboCharset

        );

        this.pickerProjection = dataSwingManager.createProjectionPickerController(

                this.txtProjection, 

                this.btnProjection

        );

        this.pickerDbfile = toolsSwingManager.createFilePickerController(

                this.txtDatabaseFile, 

                this.btnDataBaseFile

        );

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

            doChangeDBfile();

        });

        this.pickerCharset.set(Charset.defaultCharset());

        this.chkRegisterConnection.setSelected(true);

        this.chkConnectToRepository.setSelected(true);

        this.chkOverwriteDatabase.setSelected(false);

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

            updateStateComponents();

        });

        this.cboCharset.setSelectedIndex(-1);

        this.pickerLocale = toolsSwingManager.createLocalePickerController(this.cboLocale);

        this.pickerLocale.set(Locale.getDefault());

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

        ToolsSwingUtils.ensureRowsCols(this, 14, 80, 16, 100);

    }

    private void translate() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        toolsSwingManager.translate(this.lblCharset);

        toolsSwingManager.translate(this.lblRepositoryName);

        toolsSwingManager.translate(this.lblFileXML);

        toolsSwingManager.translate(this.lblDatabaseFile);

        toolsSwingManager.translate(this.lblProjection);

        toolsSwingManager.translate(this.lblRegisterConnection);

        toolsSwingManager.translate(this.lblConnectToRepository);

        toolsSwingManager.translate(this.lblOverwriteDatabase);

        toolsSwingManager.translate(this.lblLocale);

    }

    private boolean isProcessing() {

        return this.lastTask.isProcessing();

    }

    private void updateStateComponents() {

        if( !SwingUtilities.isEventDispatchThread() ) {

            SwingUtilities.invokeLater(this::updateStateComponents);

            return;

        }

        if( this.isProcessing() ) {

            this.pickerXMLFile.setEnabled(false);

            this.pickerCharset.setEnabled(false);

            this.pickerProjection.setEnabled(false);

            if( this.dialog!=null ) {

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

            }

            return;

        }

        this.pickerXMLFile.setEnabled(true);

        this.pickerCharset.setEnabled(true);

        this.pickerProjection.setEnabled(true);

        File xmlfile = this.pickerXMLFile.get();

        if( xmlfile == null || !xmlfile.exists() ) {

            if( xmlfile==null ) {

                this.message("_XML_file_is_required", JOptionPane.WARNING_MESSAGE);

            } else {

                this.message("_XML_file_not_exists", JOptionPane.WARNING_MESSAGE);

            }

            if( this.dialog!=null ) {

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

            }

            return;

        }

        File dbfile = this.pickerDbfile.get();

        if( dbfile == null ) {

            this.message("_Database_file_is_required", JOptionPane.WARNING_MESSAGE);

            if( this.dialog!=null ) {

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

            }

            return;

        }

        if( H2SpatialUtils.existsH2db(dbfile) && !this.chkOverwriteDatabase.isSelected() ) {

            this.message("_Database_file_already_exists", JOptionPane.WARNING_MESSAGE);

            if( this.dialog!=null ) {

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

            }

            return;

        }

        if( StringUtils.isBlank(this.txtRepositoryName.getText()) ) {

            this.message("_Repository_name_is_required", JOptionPane.WARNING_MESSAGE);

            if( this.dialog!=null ) {

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

            }

            return;

        }

        if( this.dialog!=null ) {

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

        }

        if( this.lastTask.needToShowTheStatus()) {

            this.taskStatusController.setVisible(true);        

        } else {

            message(null);

            this.taskStatusController.setVisible(false);        

        }

    }

    @Override

    public void setDialog(Dialog dialog) {

        this.dialog = dialog;

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    @Override

    public void performAction() {

        doCopy();

    }

    private void doCopy() {

        File xmlfile = this.pickerXMLFile.get();

        if( xmlfile == null || !xmlfile.exists() ) {

            return;

        }

        IProjection proj = this.pickerProjection.get();

        Charset charset = this.pickerCharset.get();

        File dbfile = this.pickerDbfile.get();

        if( dbfile==null || dbfile.exists()) {

            return;

        }

        String s = this.txtRepositoryName.getText();

        if( StringUtils.isBlank(s) ) {

            s = FilenameUtils.getBaseName(dbfile.getName());

        } 

        String repoName = s;

        boolean registerConnection = this.chkRegisterConnection.isSelected();

        boolean connectToRepository = this.chkConnectToRepository.isSelected();

        Xml2dbManager manager = Xml2dbLocator.getXml2dbManager();

        if( H2SpatialUtils.existsH2db(dbfile) ) {

            if( !this.chkOverwriteDatabase.isSelected() ) {

                message("Database already exists", JOptionPane.WARNING_MESSAGE);

                return;

            }

        }  

        Locale locale = this.pickerLocale.get();

        this.lastTask = new Task(                

                "Xml2db_copy", 

                "Copy xml to database", 

                this::updateStateComponents, 

                taskStatusController

            ) {

            @Override

            protected void task(SimpleTaskStatus taskStatus) throws Throwable{

                try {

                    if( H2SpatialUtils.existsH2db(dbfile) ) {

                        taskStatus.message("Removing existing database...");

                        H2SpatialUtils.server_stop();

                        H2SpatialUtils.removeH2db(dbfile);

                        H2SpatialUtils.server_start();

                    }        

                    XMLInfo xmlinfo = manager.extractStructure(xmlfile, charset, proj, locale, taskStatus);

                    if( xmlinfo==null ) {

                        throw new RuntimeException("Can't process xml file");

                    }

                    if( proj==null && xmlinfo.getSrid()!=null ) {

                        pickerProjection.set(xmlinfo.getSrid());

                    }

                    if( xmlinfo.getCharset()!=null && charset==null ) {

                        pickerCharset.set(xmlinfo.getCharset());

                    }

                    JDBCServerExplorerParameters params = manager.createDatabase(dbfile, xmlinfo, taskStatus);

                    if( params == null ) {

                        throw new RuntimeException("Can't copy xml to database");

                    }

                    if( registerConnection ) {

                        addToConnectionPool(params,repoName);   

                    }

                    manager.copyXml2Db(xmlfile, xmlinfo.getCharset(), params, taskStatus);

                    if( connectToRepository) {

                        DataSwingManager dataSwingManager = DALSwingLocator.getDataSwingManager();

                        dataSwingManager.connectToDatabaseWorkspace(params);                        

                    }

                } catch(Exception ex) {

                    LOGGER.warn("Can't copy xml to database",ex);

                    throw ex;

                }

            }

            @Override

            protected void postTask() {

//                message(null);

            }

        };

        this.lastTask.start();

    }

    public void addToConnectionPool(JDBCServerExplorerParameters params, String id) {

        DataManager dataManager = DALLocator.getDataManager();

        dataManager.getDataServerExplorerPool().add(id, params);

    }

    private void doChangeXMLfile() {

        File xmlfile = this.pickerXMLFile.get();

        if( xmlfile != null ) {

            File dbfile = this.pickerDbfile.get();

            if( dbfile == null ) {

                dbfile = new File(FilenameUtils.removeExtension(xmlfile.getAbsolutePath()));

                this.pickerDbfile.set(dbfile);

                this.txtRepositoryName.setText(FilenameUtils.getBaseName(xmlfile.getName()));

            }

        }

        updateStateComponents();

    }

    private void doChangeDBfile() {

        File dbfile = this.pickerDbfile.get();

        if( dbfile != null ) {

            String repoName = this.txtRepositoryName.getText();

            if( StringUtils.isBlank(repoName) ) {

                this.txtRepositoryName.setText(FilenameUtils.getBaseName(dbfile.getName()));

            }

        }

        updateStateComponents();

    }

    private void message(String msg) {

        message(msg,JOptionPane.INFORMATION_MESSAGE);

    }

    private void message(final String msg, final int type) {

        if( !SwingUtilities.isEventDispatchThread() ) {

            SwingUtilities.invokeLater(() -> { this.message(msg, type);});

            return;

        }

        if( StringUtils.isBlank(msg) ) {

            this.lblStatusMsg.setText("");

            return;

        }

        String s = msg;

        if( type == JOptionPane.WARNING_MESSAGE ) {

            s = Xml2dbSwingCommons.getHTMLColorTag(Color.RED.darker(), msg);

        }        

        this.lblStatusMsg.setText(s);

        this.lblStatusMsg.setVisible(true);

    }

}