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.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.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.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.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, SimpleTaskStatus status) throws IOException, FileNotFoundException {

        StructureExtractorImpl extractor = new StructureExtractorImpl();

        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, SimpleTaskStatus status) throws IOException  {

        StructureExtractorImpl extractor = new StructureExtractorImpl();

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

        return xmlinfo;

    }

    @Override

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

        StructureExtractorImpl extractor = new StructureExtractorImpl();

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

        return xmlinfo;

    }

    @Override

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

        CopyXML2dbImpl helper = new CopyXML2dbImpl();

        taskStatus = SimpleTaskStatus.get(taskStatus);

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

    }

    @Override

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

        CopyXML2dbImpl helper = new CopyXML2dbImpl();

        taskStatus = SimpleTaskStatus.get(taskStatus);

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

    }

    @Override

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

        try {

            taskStatus = SimpleTaskStatus.get(taskStatus);

            DataManager dataManager = DALLocator.getDataManager();

            String repoID = xmlinfo.getRepositoryID();

            if( StringUtils.isBlank(repoID) ) {

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

            }

            taskStatus.message("Creating database...");

            taskStatus.setIndeterminate();

            JDBCServerExplorerParameters serverParams = (JDBCServerExplorerParameters) dataManager.createServerExplorerParameters(FeatureStore.H2SPATIAL_PROVIDER_NAME);

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

            DatabaseWorkspaceManager dbworkspace = dataManager.createDatabaseWorkspaceManager(serverParams);

            dbworkspace.create(repoID, null);

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

            taskStatus.message("Creating tables...");

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

            for (XMLTableInfo tableInfo : xmlinfo) {

                String tablename = 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("Registering tables in the repository...");

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

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

                JDBCStoreParameters featureStoreParams = (JDBCStoreParameters)storeParams;

                if( xmlinfo.get(featureStoreParams.getTable())==null ) {

                    continue;

                }

                dbworkspace.writeStoresRepositoryEntry(featureStoreParams.getTable(), featureStoreParams);

                taskStatus.incrementCurrentValue();

            }

            return serverParams;

        } catch (Exception ex) {

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

        }

    }

}