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

import java.awt.Color;

import java.awt.Cursor;

import java.awt.GridBagConstraints;

import java.awt.event.ActionEvent;

import java.awt.event.ItemEvent;

import java.awt.event.ItemListener;

import java.io.File;

import java.nio.charset.Charset;

import java.util.ArrayList;

import java.util.Collections;

import java.util.List;

import java.util.Locale;

import javax.swing.AbstractListModel;

import javax.swing.DefaultComboBoxModel;

import javax.swing.JComponent;

import javax.swing.JOptionPane;

import javax.swing.SwingUtilities;

import javax.swing.event.ChangeEvent;

import javax.swing.event.DocumentEvent;

import javax.swing.event.DocumentListener;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang3.StringUtils;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

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

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.fmap.dal.swing.featuretype.FeatureTypePanel;

import static org.gvsig.fmap.dal.swing.featuretype.FeatureTypePanel.MODE_EDIT_ALL;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

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

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

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class CreatedbFromXmlPanel 

        extends CreatedbFromXmlPanelView 

        implements Xml2dbPanel

    {

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

    private static final int MODE_NEW = 0;

    private static final int MODE_OVERWRITE = 1;

    private static final int MODE_ADD = 2;

    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<>(xmlinfo.getNameWithPrefix(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 CreatedbFromXmlPanel() {

        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.setTablesModel(null);

        this.pickerXMLFile = toolsSwingManager.createFilePickerController(

                this.txtXMLFile, 

                this.btnXMLFile,

                null,

                "Xml2db_xmlFile",

                null,

                true

        );

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

            doChangeXMLfile();

        });

        this.pickerCharset = toolsSwingManager.createCharsetPickerController(

                null,

                this.cboCharset

        );

        this.pickerProjection = dataSwingManager.createProjectionPickerController(

                this.txtProjection, 

                this.btnProjection

        );

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

            setTablesModel(null);

        });

        this.txtPrefixesForTables.getDocument().addDocumentListener(new DocumentListener() {

            @Override

            public void insertUpdate(DocumentEvent e) {

                setTablesModel(null);

            }

            @Override

            public void removeUpdate(DocumentEvent e) {

                setTablesModel(null);

            }

            @Override

            public void changedUpdate(DocumentEvent e) {

                setTablesModel(null);

            }

        });

        this.pickerDbfile = toolsSwingManager.createFilePickerController(

                this.txtDatabaseFile, 

                this.btnDataBaseFile

        );

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

            doChangeDBfile();

        });

        this.btnAnalizeXML.addActionListener((ActionEvent e) -> {

            doAnalizeXML();

        });

        this.btnAnalizeXML.setCursor(new Cursor(Cursor.HAND_CURSOR));

        this.btnViewTableStructure.addActionListener((ActionEvent e) -> {

            doViewTableStructure();

        });

        this.btnViewTableStructure.setCursor(new Cursor(Cursor.HAND_CURSOR));

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

        this.chkRegisterConnection.setSelected(true);

        this.cboMode.addItemListener((ItemEvent e) -> {

            if(getMode() == MODE_ADD){

                chkRegisterConnection.setSelected(false);

            }

            updateStateComponents();

        });

        I18nManager i18n = ToolsLocator.getI18nManager();

        DefaultComboBoxModel<LabeledValue<Integer>> model = new DefaultComboBoxModel<>();

        model.addElement(new LabeledValueImpl<>(i18n.getTranslation("_New_database"), MODE_NEW));

        model.addElement(new LabeledValueImpl<>(i18n.getTranslation("_Overwrite_database"), MODE_OVERWRITE));

        model.addElement(new LabeledValueImpl<>(i18n.getTranslation("_Add_to_existing_database"), MODE_ADD));

        this.cboMode.setModel(model);

        this.cboMode.setSelectedIndex(0);

        this.cboCharset.setSelectedIndex(-1);

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

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

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

            setTablesModel(null);

        });

        toolsSwingManager.addClearButton(this.txtPrefixesForTables);

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

        ToolsSwingUtils.ensureRowsCols(this, 5, 80, 6, 100);

    }

    private void translate() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        toolsSwingManager.translate(this.lblCharset);

        toolsSwingManager.translate(this.lblRepositoryName);

        toolsSwingManager.translate(this.lblCreatedb);

        toolsSwingManager.translate(this.lblFileXML);

        toolsSwingManager.translate(this.lblDatabaseFile);

        toolsSwingManager.translate(this.lblProjection);

        toolsSwingManager.translate(this.lblTables);

        toolsSwingManager.translate(this.btnAnalizeXML);

        toolsSwingManager.translate(this.btnViewTableStructure);

        toolsSwingManager.translate(this.lblRegisterConnection);

        toolsSwingManager.translate(this.lblMode);

        toolsSwingManager.translate(this.lblLocale);

        toolsSwingManager.translate(this.lblPrefixesForTables);

    }

    private boolean isProcessing() {

        return this.lastTask.isProcessing();

    }

    private void updateStateComponents() {

        if( !SwingUtilities.isEventDispatchThread() ) {

            SwingUtilities.invokeLater(this::updateStateComponents);

            return;

        }

        message(null);

        if( this.isProcessing() ) {

            this.pickerXMLFile.setEnabled(false);

            this.pickerCharset.setEnabled(false);

            this.pickerProjection.setEnabled(false);

            this.btnAnalizeXML.setEnabled(false);

            this.lstTables.setEnabled(false);

            this.btnViewTableStructure.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);

        I18nManager i18n = ToolsLocator.getI18nManager();

        File xmlfile = this.pickerXMLFile.get();

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

            if( xmlfile==null ) {

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

            } else {

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

            }

            this.btnAnalizeXML.setEnabled(false);

            this.lstTables.setEnabled(false);

            this.btnViewTableStructure.setEnabled(false);

            if( this.dialog!=null ) {

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

            }

            return;

        }

        this.btnAnalizeXML.setEnabled(true);

        if( this.lstTables.getModel().getSize()<1 ) {

            if( this.xmlinfo == null ) {

                this.message(i18n.getTranslation("_Analize_XML_required"), JOptionPane.WARNING_MESSAGE);

            } else {

                this.message(i18n.getTranslation("_No_tables_recognized_in_the_XML"), JOptionPane.WARNING_MESSAGE);

            }

            this.btnViewTableStructure.setEnabled(false);

            if( this.dialog!=null ) {

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

            }

            return;

        }

        if( this.xmlinfo!=null && this.xmlinfo.hasGeometries() && this.pickerProjection.get()==null ) {

            this.message(i18n.getTranslation("_Projection_is_required"), JOptionPane.WARNING_MESSAGE);

            if( this.dialog!=null ) {

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

            }

            return;

        }

        this.lstTables.setEnabled(true);

        this.btnViewTableStructure.setEnabled(true);

        File dbfile = this.pickerDbfile.get();

        if( dbfile == null ) {

            this.message(i18n.getTranslation("_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.getMode() == MODE_NEW ) {

            this.message(i18n.getTranslation("_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(i18n.getTranslation("_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);        

        }

        this.taskStatusController.setVisible(this.lastTask.needToShowTheStatus());        

    }

    @Override

    public void setDialog(Dialog dialog) {

        this.dialog = dialog;

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    @Override

    public void performAction() {

        doCreatedb();

    }

    private void doCreatedb() {

        if( this.xmlinfo==null ) {

            return;

        }

        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();

        Xml2dbManager manager = Xml2dbLocator.getXml2dbManager();

        I18nManager i18n = ToolsLocator.getI18nManager();

        switch(getMode()){

            default:

            case MODE_NEW:

                if( H2SpatialUtils.existsH2db(dbfile) ) {

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

                    return;

                }

                break;

            case MODE_OVERWRITE:

                break;

            case MODE_ADD:

                if( !H2SpatialUtils.existsH2db(dbfile) ) {

                    message(i18n.getTranslation("Database not exists"), JOptionPane.WARNING_MESSAGE);

                    return;

                }

                break;

        }

//        if( H2SpatialUtils.existsH2db(dbfile) ) {

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

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

//                return;

//            }

//        }        

        this.lastTask = new Task(                

                "Xml2db_Creating_database", 

                i18n.getTranslation("_Create_database"), 

                this::updateStateComponents, 

                taskStatusController

            ) {

            @Override

            protected void task(SimpleTaskStatus taskStatus) throws Throwable{

                try {

                    JDBCServerExplorerParameters params;

                    switch(getMode()){

                        default:

                        case MODE_NEW:

                            if (H2SpatialUtils.existsH2db(dbfile)) {

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

                                return;

                            }

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

                            break;

                        case MODE_OVERWRITE:

                            if( H2SpatialUtils.existsH2db(dbfile) ) {

                                taskStatus.message(i18n.getTranslation("_Removing_existing_database"));

                                H2SpatialUtils.server_stop();

                                H2SpatialUtils.removeH2db(dbfile);

                                H2SpatialUtils.server_start();

                            }        

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

                            break;

                        case MODE_ADD:

                            params = manager.addToDatabase(dbfile, xmlinfo, taskStatus);

                            break;

                    }

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

                    if( params == null ) {

                        throw new RuntimeException("Can't create database");

                    }

                    if( registerConnection ) {

                        addToConnectionPool(params,repoName);   

                    }

                } catch(Exception ex) {

                    LOGGER.warn("Can't create 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 doAnalizeXML() {

        Xml2dbManager manager = Xml2dbLocator.getXml2dbManager();

        File xmlfile = this.pickerXMLFile.get();

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

            return;

        }

        IProjection proj = this.pickerProjection.get();

        Charset charset = this.pickerCharset.get();

        Locale locale = this.pickerLocale.get();

        String tablesPrefix = StringUtils.defaultIfBlank(this.txtPrefixesForTables.getText(), null);

        I18nManager i18n = ToolsLocator.getI18nManager();

        this.lastTask = new Task(                

                "Xml2db_AnalizeXML", 

                i18n.getTranslation("_Analize_xml"), 

                this::updateStateComponents, 

                taskStatusController

            ) {

            @Override

            protected void task(SimpleTaskStatus taskStatus) throws Throwable{

                try {

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

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

                        pickerProjection.set(xmlinfo.getSrid());

                    }

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

                        pickerCharset.set(xmlinfo.getCharset());

                    }

                    setTablesModel(xmlinfo);

                } catch(Throwable ex) {

                    LOGGER.warn("Can't analize xml",ex);

                    setTablesModel(null);

                    I18nManager i18n = ToolsLocator.getI18nManager();

                    ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

                    dialogs.messageDialog(

                            i18n.getTranslation("_Problems_processing_XML_file")

                                    +"\n\n"

                                    + ex.getLocalizedMessage(), 

                            i18n.getTranslation("_Xml2db_Create_database_from_xml"), 

                            JOptionPane.WARNING_MESSAGE

                    );

                    throw ex;

                }

            }

        };

        this.lastTask.start();

    }

    private void setTablesModel(XMLInfo xmlinfo) {

        if( !SwingUtilities.isEventDispatchThread() ) {

            SwingUtilities.invokeLater(() -> {this.setTablesModel(xmlinfo);});

            return;

        }

        this.xmlinfo = xmlinfo;

        if( xmlinfo == null ) {

            lstTables.setModel(new XMLInfoListModel()); 

            return;

        }

        lstTables.setModel(new XMLInfoListModel(xmlinfo)); 

    }

    private void doViewTableStructure() {

        LabeledValue<XMLTableInfo> x = (LabeledValue<XMLTableInfo>)this.lstTables.getSelectedValue();

        if( x == null ) {

            return;

        }

//        I18nManager i18n = ToolsLocator.getI18nManager();

        DataSwingManager dataSwingManager = DALSwingLocator.getDataSwingManager();

        WindowManager_v2 windowManager = (WindowManager_v2) ToolsSwingLocator.getWindowManager();

        XMLTableInfo tableInfo = x.getValue();

        EditableFeatureType featureType = tableInfo.getFeatureType();

        FeatureTypePanel panel = dataSwingManager.createFeatureTypePanel();

        panel.setMode(MODE_EDIT_ALL);

//        panel.setRepository(xmlinfo.getRepository());

        panel.put(featureType);

        Dialog theDialog = windowManager.createDialog(

                panel.asJComponent(),

                ToolsSwingUtils.makeTitle("_Xml2db_Column_manager", tableInfo.getName(), featureType.getLabel()),

                null, 

                WindowManager_v2.BUTTONS_OK_CANCEL

        );

        theDialog.addActionListener((ActionEvent e) -> {

            if( theDialog.getAction()==WindowManager_v2.BUTTON_OK) {

                panel.fetch(x.getValue().getFeatureType());

            }

        });

        theDialog.show(

                WindowManager.MODE.WINDOW,

                Collections.singletonMap("align", GridBagConstraints.CENTER)

        );

    }

    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(String msg, int type) {

        if( StringUtils.isBlank(msg) ) {

            this.lblStatusMsg.setText("");

            return;

        }

        if( type == JOptionPane.WARNING_MESSAGE ) {

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

        }        

        this.lblStatusMsg.setText(msg);

        this.lblStatusMsg.setVisible(true);

    }

    private int getMode() {

        LabeledValue<Integer> item = (LabeledValue) this.cboMode.getSelectedItem();

        if (item == null){

            return MODE_NEW;

        }

        return item.getValue();

    }

}