Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 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.export.csv.service;

import java.util.ArrayList;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.export.ExportException;

import org.gvsig.fmap.dal.exception.DataException;

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

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

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

import org.gvsig.export.spi.ExportService;

import org.gvsig.export.spi.AbstractExportService;

import org.gvsig.export.spi.ExportServiceFactory;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.NewDataStoreParameters;

import org.gvsig.fmap.dal.OpenDataStoreParameters;

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemServerExplorer;

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemServerExplorerParameters;

import org.gvsig.tools.util.HasAFile;

/**

 * @author gvSIG Team

 * @version $Id$

 *

 */

public class ExportCSVService

        extends AbstractExportService

        implements ExportService {

    public static final int MAX_FIELD_NAME_LENGTH = 50;

    private FeatureType actualFeatureType;

    protected ExportCSVService(ExportServiceFactory factory, ExportCSVParameters parameters) {

        super(factory, parameters);

    }

    @Override

    public ExportCSVParameters getParameters() {

        return (ExportCSVParameters) super.getParameters();

    }

    @Override

    protected FilesystemServerExplorer createServerExplorer() throws ExportException {

        DataManager dataManager = DALLocator.getDataManager();

        FilesystemServerExplorerParameters explorerParams;

        try {

            explorerParams

                    = (FilesystemServerExplorerParameters) dataManager

                            .createServerExplorerParameters(FilesystemServerExplorer.NAME);

        } catch (Exception e) {

            throw new ExportException(e);

        }

        explorerParams.setRoot(this.getParameters().getFile().getParent());

        FilesystemServerExplorer explorer;

        try {

            explorer = (FilesystemServerExplorer) dataManager.openServerExplorer(

                    "FilesystemExplorer", explorerParams

            );

            return explorer;

        } catch (Exception e) {

            throw new ExportException(e);

        }

    }

    @Override

    protected NewDataStoreParameters createTargetNewStoreParameters() throws ExportException {

        try {

            FilesystemServerExplorer explorer = this.createServerExplorer();

            NewFeatureStoreParameters newStoreParameters = (NewFeatureStoreParameters) explorer.getAddParameters(

                    this.getParameters().getFile()

            );

            newStoreParameters.setDynValue("profile", this.getParameters().getProfile());

            newStoreParameters.setDynValue("quotePolicy", this.getParameters().getQuotePolicy());

            newStoreParameters.setDynValue("locale", this.getParameters().getLocale());

            newStoreParameters.setDynValue("recordSeparator", this.getParameters().getRecordSeparator());

            newStoreParameters.setDynValue("delimiter", this.getParameters().getDelimiter());

            newStoreParameters.setDynValue("quoteCharacter", this.getParameters().getQuoteCharacter());

            newStoreParameters.setDynValue("commentStartMarker", this.getParameters().getCommentStartMarker());

            newStoreParameters.setDynValue("escapeCharacter", this.getParameters().getEscapeCharacter());

            FeatureType ft = this.getParameters().getExportAttributes().getTargetFeatureType();

            if (ft == null) {

                ft = this.getParameters().getSourceFeatureType();

            }

            // FIXME: check types of fields in ft (remove Geometry?)

            newStoreParameters.setDefaultFeatureType(ft.getEditable());

            return newStoreParameters;

        } catch (DataException ex) {

            throw new ExportException(ex);

        }

    }

    private String checkEmptyStringParam(String param) {

        if (StringUtils.isEmpty(param)) {

            return null;

        } else {

            return param;

        }

    }

    @Override

    public OpenDataStoreParameters createTargetOpenStoreParameters() throws ExportException {

        try {

            DataManager dataManager = DALLocator.getDataManager();

            OpenFeatureStoreParameters openStoreParameters = (OpenFeatureStoreParameters) dataManager.createStoreParameters("CSV");

            ((HasAFile) openStoreParameters).setFile(getParameters().getFile());

            openStoreParameters.setDynValue("profile", this.getParameters().getProfile());

            openStoreParameters.setDynValue("quotePolicy", this.getParameters().getQuotePolicy());

            openStoreParameters.setDynValue("locale", this.getParameters().getLocale());

            openStoreParameters.setDynValue("recordSeparator", checkEmptyStringParam(this.getParameters().getRecordSeparator()));

            openStoreParameters.setDynValue("delimiter", checkEmptyStringParam(this.getParameters().getDelimiter()));

            openStoreParameters.setDynValue("quoteCharacter", checkEmptyStringParam(this.getParameters().getQuoteCharacter()));

            openStoreParameters.setDynValue("commentStartMarker", checkEmptyStringParam(this.getParameters().getCommentStartMarker()));

            openStoreParameters.setDynValue("escapeCharacter", checkEmptyStringParam(this.getParameters().getEscapeCharacter()));

            openStoreParameters.setDynValue("firstLineHeader", true);

            return openStoreParameters;

        } catch (DataException ex) {

            throw new ExportException(ex);

        }

    }

    @Override

    public List<OpenDataStoreParameters> getTargetOpenStoreParameters() throws ExportException {

        List<OpenDataStoreParameters> r = new ArrayList<>();

        r.add(this.createTargetOpenStoreParameters());

        return r;

    }

}