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.h2spatial.restore;

import org.gvsig.h2spatial.backup.*;

import java.io.BufferedReader;

import java.io.File;

import java.io.FileInputStream;

import java.io.InputStreamReader;

import java.io.Reader;

import java.nio.charset.StandardCharsets;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.ArrayList;

import java.util.List;

import java.util.logging.Level;

import javax.swing.JOptionPane;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.io.IOUtils;

import org.apache.commons.io.output.FileWriterWithEncoding;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

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

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

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

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.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.pickercontroller.FilePickerController;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author fdiaz

 */

public class Restore extends RestoreView {

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

    private FilePickerController pickerBackupFile;

    private FilePickerController pickerDatabaseFile;

    public Restore() {

        initComponents();

    }

    private void initComponents() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        this.pickerBackupFile = toolsSwingManager.createFilePickerController(txtBackupFile, btnBackupFile);

        this.pickerDatabaseFile = toolsSwingManager.createFilePickerController(txtDatabaseFile, btnDatabaseFile);

    }

    private void translate() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        toolsSwingManager.translate(lblBackupFile);

        toolsSwingManager.translate(lblDatabaseFile);

    }

    public void restoreBackup(){

        ApplicationManager application = ApplicationLocator.getApplicationManager();

        I18nManager i18n = ToolsLocator.getI18nManager();

        Connection conn = null;

        Statement st =null;

        try {

            String h2DatabaseFile = H2SpatialUtils.removeH2FileNameExtension(pickerDatabaseFile.get().getAbsolutePath().replace("\\", "/"));

            File[] h2Files = H2SpatialUtils.getH2Files(new File(h2DatabaseFile));

            for (File h2File : h2Files) {

                if (h2File.exists()) {

                    application.messageDialog(

                            i18n.getTranslation("_The_target_file_already_exists"),

                            i18n.getTranslation("_Restore"),

                            JOptionPane.WARNING_MESSAGE

                    );

                    return;

                }

            }

            conn = DriverManager.getConnection("jdbc:h2:"+h2DatabaseFile+";MODE=PostgreSQL;SCHEMA=PUBLIC;ALLOW_LITERALS=ALL", "sa", "");

            st = conn.createStatement();

            st.execute("RUNSCRIPT FROM '"+pickerBackupFile.get().getAbsolutePath().replace("\\", "/")+"'"); //+"' CHARSET '"+StandardCharsets.UTF_8.name()+"'");

        } catch (Exception ex) {

            LOGGER.warn("Can't restore backup.", ex);

            application.messageDialog(

                    i18n.getTranslation("_Cant_restore_backup")+"\n\n"+

                    i18n.getTranslation("_see_error_log_for_more_information"), 

                    i18n.getTranslation("_Restore"), 

                    JOptionPane.WARNING_MESSAGE

            );

        } finally {

            if(st != null){

                try {

                    st.close();

                } catch (SQLException ex) {

                    LOGGER.debug("Can't close statement", ex);

                }

            }

            if(conn != null){

                try {

                    conn.close();

                } catch (SQLException ex) {

                    LOGGER.debug("Can't close connection", ex);

                }

            }

        }

    }

}