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 3

 * 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.lib.impl;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.io.InputStream;

import java.io.Reader;

import java.nio.charset.Charset;

import java.util.Locale;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.io.IOUtils;

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

import org.gvsig.fmap.dal.DatabaseWorkspaceManager;

import org.gvsig.fmap.dal.PersonalDatabaseServerExplorerFactory;

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

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

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

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

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.task.SimpleTaskStatus;

import org.gvsig.tools.util.HasAFile;

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

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

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

@SuppressWarnings("UseSpecificCatch")

public class DefaultXml2dbManager implements Xml2dbManager {

    public DefaultXml2dbManager() {

    }

    @Override

    public XMLInfo extractStructure(File xml, Charset encoding, IProjection projection, Locale locale, String tablePrefix, SimpleTaskStatus status) throws IOException, FileNotFoundException {

        StructureExtractorImpl extractor = new StructureExtractorImpl();

        extractor.setTablePrefix(tablePrefix);

        XMLInfo xmlinfo = extractor.extractStructure(xml, encoding, projection, locale, SimpleTaskStatus.get(status));

        return xmlinfo;

    }

    @Override

    public XMLInfo extractStructure(InputStream xml, Charset encoding, IProjection projection, Locale locale, String tablePrefix, SimpleTaskStatus status) throws IOException  {

        StructureExtractorImpl extractor = new StructureExtractorImpl();

        extractor.setTablePrefix(tablePrefix);

        XMLInfo xmlinfo = extractor.extractStructure(xml, encoding, projection, locale, SimpleTaskStatus.get(status));

        return xmlinfo;

    }

    @Override

    public XMLInfo extractStructure(Reader xml, IProjection projection, Locale locale, String tablePrefix, SimpleTaskStatus status)  throws IOException {

        StructureExtractorImpl extractor = new StructureExtractorImpl();

        extractor.setTablePrefix(tablePrefix);

        XMLInfo xmlinfo = extractor.extractStructure(xml, projection, locale, SimpleTaskStatus.get(status));

        return xmlinfo;

    }

    @Override

    public void copyXml2Db(File xml, Charset encoding, JDBCServerExplorerParameters dbparams, String tablePrefix, SimpleTaskStatus taskStatus) {

        CopyXML2dbImpl helper = new CopyXML2dbImpl();

        taskStatus = SimpleTaskStatus.get(taskStatus);

        helper.copyData(xml, encoding, dbparams, tablePrefix, taskStatus);

    }

    @Override

    public void copyXml2Db(InputStream xml, Charset encoding, JDBCServerExplorerParameters dbparams, String tablePrefix, SimpleTaskStatus taskStatus) {

        CopyXML2dbImpl helper = new CopyXML2dbImpl();

        taskStatus = SimpleTaskStatus.get(taskStatus);

        helper.copyData(xml, encoding, dbparams, tablePrefix, taskStatus);

    }

    @Override

    public JDBCServerExplorerParameters createDatabase(String dbprovider, File dbfile, XMLInfo xmlinfo, SimpleTaskStatus taskStatus) {

        return addToDatabase(dbprovider, dbfile, xmlinfo, taskStatus, true);

    }

    @Override

    public JDBCServerExplorerParameters addToDatabase(String dbprovider, File dbfile, XMLInfo xmlinfo, SimpleTaskStatus taskStatus) {

        return addToDatabase(dbprovider, dbfile, xmlinfo, taskStatus, false);

    }

    private JDBCServerExplorerParameters addToDatabase(String dbprovider, File dbfile, XMLInfo xmlinfo, SimpleTaskStatus taskStatus, boolean createDB) {

        try {

            taskStatus = SimpleTaskStatus.get(taskStatus);

            DataManager dataManager = DALLocator.getDataManager();

            String repoID = xmlinfo.getRepositoryID();

            if( StringUtils.isBlank(repoID) ) {

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

            }

            I18nManager i18n = ToolsLocator.getI18nManager();

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

            taskStatus.setIndeterminate();

            if( StringUtils.isBlank(dbprovider) ) {

                dbprovider = FeatureStore.H2SPATIAL_PROVIDER_NAME;

            }

            PersonalDatabaseServerExplorerFactory factory = (PersonalDatabaseServerExplorerFactory) dataManager.getServerExplorerRegister().getFactory(dbprovider);

            JDBCServerExplorerParameters serverParams = (JDBCServerExplorerParameters) dataManager.createServerExplorerParameters(dbprovider);

            ((HasAFile)serverParams).setFile(dbfile);

            DatabaseWorkspaceManager dbworkspace = dataManager.createDatabaseWorkspaceManager(serverParams);

            JDBCServerExplorer server = (JDBCServerExplorer) dbworkspace.getServerExplorer();

            if( !factory.existsdb(dbfile) ) {

                server.dropCaches();

            }

            if(createDB) {

                dbworkspace.create(repoID, null);

            }

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

            taskStatus.setRangeOfValues(0, xmlinfo.size());

            for (XMLTableInfo tableInfo : xmlinfo) {

                String tablename = xmlinfo.getNameWithPrefix(tableInfo.getName());

                FeatureType ftype = tableInfo.getFeatureType();

                JDBCNewStoreParameters tableParams = server.getAddParameters();

                tableParams.setTable(tablename);

                tableParams.setDefaultFeatureType(ftype);

                server.add(tableParams.getProviderName(), tableParams, false);

                taskStatus.incrementCurrentValue();

            }

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

            taskStatus.setRangeOfValues(0, xmlinfo.size());

            for (DataStoreParameters storeParams : server.list()) {

                JDBCStoreParameters featureStoreParams = (JDBCStoreParameters)storeParams;

                String tableName = featureStoreParams.getTable();

                if( xmlinfo.get(tableName)==null ) { 

                    continue;

                }

                dbworkspace.writeStoresRepositoryEntry(tableName, featureStoreParams);

                taskStatus.incrementCurrentValue();

            }

            return serverParams;

        } catch (Exception ex) {

            throw new RuntimeException("Can't create databse '"+dbfile.getAbsolutePath()+"'.", ex);

        }

    }

    @Override

    public Charset detectCharset(InputStream is) {

        return Xml2dbCommons.detectCharset(is);

    }

    @Override

    public Charset detectCharset(File file) {

        FileInputStream is = null;

        try {

            is = new FileInputStream(file);

            return detectCharset(is);

        } catch (Exception ex) {

            return null;

        } finally {

            IOUtils.closeQuietly(is);

        }

    }

    public IProjection detectProjection(File file, SimpleTaskStatus status) {

        return new ProjectionExtractorImpl().extractProjection(file, null, status);

    }

}