Application: gvSIG desktop

        this.provider.dispose();

                ((FeatureSelection) this.getSelection()).deselectAll();

                if(featureManager.isSelectionCompromised()) {

                    ((FeatureSelection) this.getSelection()).deselectAll();

                };

        DefaultEditableFeatureType ftype = new DefaultEditableFeatureType();

import java.util.HashSet;

import java.util.Set;

		this("default");

		this("default");

		this("default");

/**

 * 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.

 */

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

*

* 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.

*

*/

/*

* AUTHORS (In addition to CIT):

* 2009 IVER T.I   {{Task}}

*/

/**

 *

 */

package org.gvsig.fmap.dal.store.h2;

import java.io.File;

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

/**

 * @author jmvivo

 *

 */

public interface H2ConnectionParameters extends

		JDBCConnectionParameters {

	public static final String PARAMETER_NAME_USESSL = "UseSSL";

        public static final String PARAMETER_NAME_BASEFOLDER = "BaseFolder";

        public File getBaseFolder();

	public Boolean getUseSSL();

}

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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.

 *

 */

package org.gvsig.fmap.dal.store.h2;

import java.io.File;

import java.io.IOException;

import org.apache.commons.io.FileUtils;

import org.apache.commons.lang3.StringUtils;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class H2ParametersHelper {

    private static Logger logger = LoggerFactory.getLogger(H2ParametersHelper.class);

    public static String getConnectionString(H2ConnectionParameters params) {

        String connectionString = null;

        File baseFolder = params.getBaseFolder();

        String baseFolderPath = null;

        if ( baseFolder == null ) {

            baseFolder = getDefaultBaseFolder();

        }

        if ( baseFolder == null ) {

            baseFolderPath = baseFolder.getAbsolutePath();

            baseFolderPath = baseFolderPath.replace("\\", "/");

        }

        String dbname = params.getDBName();

        if ( dbname == null ) {

            logger.warn("dbname is null.");

            throw new IllegalArgumentException("dbname is null");

        }

        String host = params.getHost();

        Integer port = params.getPort();

        if ( port == null ) {

            port = new Integer(9123);

        }

        if ( StringUtils.isBlank(host) ) {

            if( baseFolderPath == null ) {

                throw new IllegalArgumentException("baseFolder required");

            }

            connectionString = "jdbc:h2:" + baseFolderPath + "/" + dbname + ";MODE=PostgreSQL;AUTO_SERVER=TRUE";

        } else {

            if ( host.trim().equalsIgnoreCase("localhost") ) {

                if( baseFolderPath == null ) {

                    connectionString = "jdbc:h2:tcp://localhost:" + port.intValue() + "/" + dbname + ";MODE=PostgreSQL";

                } else {

                    connectionString = "jdbc:h2:" + baseFolderPath + "/" + dbname + ";MODE=PostgreSQL;AUTO_SERVER=TRUE;AUTO_SERVER_PORT=" + port;

                }

            } else {

                if( baseFolderPath == null ) {

                    connectionString = "jdbc:h2:tcp://" + host + ":" + port.intValue() + "/" + dbname + ";MODE=PostgreSQL";

                } else {

                    connectionString = "jdbc:h2:tcp://" + host + ":" + port.intValue() + "/" + dbname + ":" + baseFolder + ";MODE=PostgreSQL";

                }

            }

        }

        return connectionString;

    }

    public static File getDefaultBaseFolder() {

        File baseFolder = null;

        String s = System.getProperty("user.home");

        File baseFolderInHome = new File(s + File.separator + "gvSIG" + File.separator + "dbs.h2");

        if ( !baseFolderInHome.exists() ) {

            try {

                FileUtils.forceMkdir(baseFolderInHome);

                baseFolder = baseFolderInHome;

            } catch (IOException ex) {

            }

        }

        if ( baseFolder == null ) {

            s = System.getProperty("java.io.tmpdir");

            File baseFolderInTemp = new File(s + File.separator + "gvSIG" + File.separator + "dbs.h2");

            if ( !baseFolderInTemp.exists() ) {

                try {

                    FileUtils.forceMkdir(baseFolderInTemp);

                    baseFolder = baseFolderInTemp;

                } catch (IOException ex) {

                }

            }

        }

        return baseFolder;

    }

}

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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.

 *

 */

package org.gvsig.fmap.dal.store.h2;

import org.gvsig.fmap.dal.DALLibrary;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.resource.spi.ResourceManagerProviderServices;

import org.gvsig.fmap.dal.spi.DataManagerProviderServices;

import org.gvsig.fmap.dal.store.db.DBHelper;

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

import org.gvsig.metadata.exceptions.MetadataException;

import org.gvsig.tools.library.AbstractLibrary;

import org.gvsig.tools.library.LibraryException;

public class H2Library extends AbstractLibrary {

    static String DEFAULT_H2_DRIVER_NAME = "org.h2.Driver";

    @Override

    public void doRegistration() {

        registerAsServiceOf(DALLibrary.class);

        require(JDBCLibrary.class);

    }

    @Override

    protected void doInitialize() throws LibraryException {

    }

    @Override

    protected void doPostInitialize() throws LibraryException {

        LibraryException ex = null;

        //new JDBCLibrary().postInitialize();

        DBHelper.registerParametersDefinition(

                H2StoreParameters.PARAMETERS_DEFINITION_NAME,

                H2StoreParameters.class,

                "H2SQLParameters.xml"

        );

        DBHelper.registerParametersDefinition(

                H2NewStoreParameters.PARAMETERS_DEFINITION_NAME,

                H2NewStoreParameters.class,

                "H2SQLParameters.xml"

        );

        DBHelper.registerParametersDefinition(

                H2ServerExplorerParameters.PARAMETERS_DEFINITION_NAME,

                H2ServerExplorerParameters.class,

                "H2SQLParameters.xml"

        );

        DBHelper.registerParametersDefinition(

                H2ResourceParameters.PARAMETERS_DEFINITION_NAME,

                H2ResourceParameters.class,

                "H2SQLParameters.xml"

        );

        try {

            DBHelper.registerMetadataDefinition(

                    H2StoreProvider.METADATA_DEFINITION_NAME,

                    H2StoreProvider.class,

                    "H2SQLMetadata.xml"

            );

        } catch (MetadataException e) {

            ex = new LibraryException(this.getClass(), e);

        }

        ResourceManagerProviderServices resman = (ResourceManagerProviderServices) DALLocator

                .getResourceManager();

        if ( !resman.getResourceProviders().contains(H2Resource.NAME) ) {

            resman.register(H2Resource.NAME,

                    H2Resource.DESCRIPTION, H2Resource.class,

                    H2ResourceParameters.class);

        }

        DataManagerProviderServices dataman = (DataManagerProviderServices) DALLocator

                .getDataManager();

        if ( !dataman.getStoreProviders().contains(H2StoreProvider.NAME) ) {

            dataman.registerStoreProvider(H2StoreProvider.NAME,

                    H2StoreProvider.class,

                    H2StoreParameters.class);

        }

        if ( !dataman.getExplorerProviders().contains(

                H2StoreProvider.NAME) ) {

            dataman.registerExplorerProvider(H2ServerExplorer.NAME,

                    H2ServerExplorer.class,

                    H2ServerExplorerParameters.class);

        }

        if ( ex != null ) {

            throw ex;

        }

    }

}

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

*

* 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.

*

*/

/*

* AUTHORS (In addition to CIT):

* 2009 IVER T.I   {{Task}}

*/

/**

 *

 */

package org.gvsig.fmap.dal.store.h2;

import java.util.ArrayList;

import java.util.List;

import org.cresques.cts.IProjection;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

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

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

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

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

import org.gvsig.fmap.dal.feature.exception.CreateGeometryException;

import org.gvsig.fmap.dal.store.db.DBStoreParameters;

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

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

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.tools.evaluator.Evaluator;

import org.gvsig.tools.evaluator.EvaluatorFieldValue;

import org.gvsig.tools.evaluator.EvaluatorFieldsInfo;

/**

 * @author jmvivo

 *

 */

public class H2SetProvider extends JDBCSetProvider {

    private static Logger logger = LoggerFactory.getLogger(H2SetProvider.class);

	public H2SetProvider(JDBCStoreProvider store, FeatureQuery query,

			FeatureType featureType) throws DataException {

		super(store, query, featureType);

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see org.gvsig.fmap.dal.feature.spi.FeatureSetProvider#canFilter()

	 */

	public boolean canFilter() {

		// TODO more checks

		if (!super.canFilter()) {

			return false;

		}

		return true;

	}

    protected String getSqlForEvaluator(Evaluator filter) {

        String resp = null;

        if (filter != null && filter.getSQL() != null) {

            // =================================================

            EvaluatorFieldsInfo info = filter.getFieldsInfo();

            String filterString = filter.getSQL();

            String[] filterNames = null;

            String[] finalNames = null;

            if (info == null) {

                filterNames = getFieldNames(getFeatureType());

            } else {

                filterNames = info.getFieldNames();

            }

            finalNames = new String[filterNames.length];

            FeatureAttributeDescriptor attr;

            for (int i = 0; i < filterNames.length; i++) {

                attr = getFeatureType().getAttributeDescriptor(filterNames[i]);

                if (attr == null) {

                    finalNames[i] = filterNames[i];

                    continue;

                }

                finalNames[i] = getEscapedFieldName(attr.getName());

            }

            for (int i = 0; i < filterNames.length; i++) {

                if (!filterNames[i].equals(finalNames[i])) {

                    filterString = filterString.replaceAll(

                            "\\b" + filterNames[i] + "\\b",

                            finalNames[i]);

                }

            }

            resp = filterString;        

        }

        // ================================

        // In any case, append working area filter

        try {

            resp = appendWorkingAreaCondition(resp);

        } catch (Exception e) {

            logger.error("While appending working area condition.", e);

        }

        return resp;

    }	

    private String[] getFieldNames(FeatureType ft) {

        if (ft == null) {

            return new String[0];

        }

        FeatureAttributeDescriptor[] atts = ft.getAttributeDescriptors();

        String[] resp = new String[atts.length];

        for (int i=0; i<atts.length; i++) {

            resp[i] = atts[i].getName();

        }

        return resp;

    }

    private String getFunctionName(String newFunctionName) {

        H2StoreProvider pg_sto_prov = (H2StoreProvider) this.getStore();

        H2Helper hpr = pg_sto_prov.getPgHelper();

        if (hpr == null) {

            logger.info("Unable to get PG helper.", new Exception("Helper is null"));

            return newFunctionName;

        } else {

            return hpr.getFunctionName(newFunctionName);

        }

    }

    private String appendWorkingAreaCondition(String sql) throws Exception {

        DBStoreParameters dbParams = 

        (DBStoreParameters) getStore().getParameters();

        Envelope wa = dbParams.getWorkingArea(); 

        if (wa == null) {

            return sql;

        } else {

            FeatureStore fstore = this.getStore().getFeatureStore();

            String geoname =

                fstore.getDefaultFeatureType().getDefaultGeometryAttributeName();

            StringBuffer strbuf = new StringBuffer();

            if (sql == null)  {

                strbuf.append(

                    getFunctionName("ST_Intersects") + "("

                    + getFunctionName("ST_GeomFromText") + "('");

            } else {

                strbuf.append("(");

                strbuf.append(sql);

                strbuf.append(") AND "

                    + getFunctionName("ST_Intersects") + "("

                    + getFunctionName("ST_GeomFromText") + "('");

            }

            String workAreaWkt = null;

            workAreaWkt = wa.getGeometry().convertToWKT();

            strbuf.append(workAreaWkt);

            strbuf.append("', ");

            H2StoreProvider sprov = (H2StoreProvider) getStore();

            H2Helper helper = sprov.getPgHelper();

            IProjection proj = dbParams.getCRS();

            int sridInt = helper.getProviderSRID(proj); 

            if (sridInt == -1) {

                throw new CreateGeometryException(

                        new Exception("CRS is null or unknown."));

            } else {

                strbuf.append(Integer.toString(sridInt));

            }

            strbuf.append("), " + getFunctionName("ST_Envelope") + "(");

            strbuf.append(helper.escapeFieldName(geoname));

            strbuf.append("))");

            return strbuf.toString();

        }

    }

}

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

*

* 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.

*

*/

/*

* AUTHORS (In addition to CIT):

* 2009 IVER T.I   {{Task}}

*/

package org.gvsig.fmap.dal.store.h2;

import java.sql.SQLException;

import java.text.MessageFormat;

import javax.sql.DataSource;

import org.apache.commons.dbcp.BasicDataSource;

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

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

import org.gvsig.fmap.dal.resource.exception.AccessResourceException;

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

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

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCDriverClassNotFoundException;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCException;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCSQLException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class H2Resource extends JDBCResource {

	final static private Logger logger = LoggerFactory

			.getLogger(H2Resource.class);

	public final static String NAME = "H2SQLResource";

	public static final String DESCRIPTION = "H2SQL Connection";

	public H2Resource(H2ResourceParameters parameters)

			throws InitializeException {

		super(parameters);

	}

	public String getName() throws AccessResourceException {

		H2ResourceParameters params = (H2ResourceParameters) this.getParameters();

		return MessageFormat.format("H2SQLResource({0},{1})",

				new Object[] { params.getUrl(),params.getUser() });

	}

        public String toString() {

            try {

                return this.getName();

            } catch(Exception ex) {

                return super.toString();

            }

        }

	protected void connectToDB() throws DataException {

		if (this.dataSource != null) {

			return;

		}

		JDBCResourceParameters jdbcParams = (JDBCResourceParameters) this

				.getParameters();

		BasicDataSource dataSource = new BasicDataSource();

		dataSource.setDriverClassName(jdbcParams.getJDBCDriverClassName());

		dataSource.setUsername(jdbcParams.getUser());

		dataSource.setPassword(jdbcParams.getPassword());

		dataSource.setUrl(jdbcParams.getUrl());

		dataSource.setMaxWait(60L * 1000); // FIXME

		// FIXME Set Pool parameters:

		/*

		dataSource.setMaxActive(maxActive);

		dataSource.setMaxIdle(maxActive);

		dataSource.setMaxOpenPreparedStatements(maxActive);

		dataSource.setMaxWait(maxActive);

		dataSource.setInitialSize(initialSize);

		dataSource.setDefaultReadOnly(defaultReadOnly);

		dataSource.setDefaultTransactionIsolation(defaultTransactionIsolation);

		dataSource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);

		dataSource.setMinIdle(minIdle);

		dataSource.setTestOnBorrow(testOnBorrow);

		dataSource.setTestOnReturn(testOnReturn);

		dataSource.setTestWhileIdle(testOnReturn);

		dataSource

				.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);

		dataSource.setAccessToUnderlyingConnectionAllowed(allow);

		dataSource.setLoginTimeout(seconds);

		dataSource.setLogWriter(out);

		*/

		this.dataSource = dataSource;

	}	

	protected void registerJDBCDriver() throws InitializeException {

		String className = ((JDBCResourceParameters) getParameters())

				.getJDBCDriverClassName();

		if (className == null) {

			return;

		}

		Class theClass = null;

		try {

			theClass = Class.forName(className);

		} catch (Exception e){

			throw new InitializeException(e);

		}

		if (theClass == null) {

			try {

				throw new JDBCDriverClassNotFoundException(this.getName(),

						className);

			} catch (AccessResourceException e) {

				throw new InitializeException(e);

			}

		}

	}

	protected DataSource createDataSource() {

		H2ResourceParameters jdbcParams = (H2ResourceParameters) this

				.getParameters();

		BasicDataSource dataSource = new BasicDataSource();

		dataSource.setDriverClassName(jdbcParams.getJDBCDriverClassName());

		dataSource.setUsername(jdbcParams.getUser());

		dataSource.setPassword(jdbcParams.getPassword());

		dataSource.setUrl(jdbcParams.getUrl());

		dataSource.setMaxWait(60L * 1000); // FIXME

		// FIXME Set Pool parameters:

		/*

		dataSource.setMaxActive(maxActive);

		dataSource.setMaxIdle(maxActive);

		dataSource.setMaxOpenPreparedStatements(maxActive);

		dataSource.setMaxWait(maxActive);

		dataSource.setInitialSize(initialSize);

		dataSource.setDefaultReadOnly(defaultReadOnly);

		dataSource.setDefaultTransactionIsolation(defaultTransactionIsolation);

		dataSource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);

		dataSource.setMinIdle(minIdle);

		dataSource.setTestOnBorrow(testOnBorrow);

		dataSource.setTestOnReturn(testOnReturn);

		dataSource.setTestWhileIdle(testOnReturn);

		dataSource

			.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);

		dataSource.setAccessToUnderlyingConnectionAllowed(allow);

		dataSource.setLoginTimeout(seconds);

		dataSource.setLogWriter(out);

		 */

		return dataSource;

	}

	public boolean isConnected() {

		if (dataSource == null) {

			return false;

		}

		if (dataSource instanceof BasicDataSource) {

			return ((BasicDataSource) dataSource).getNumActive() > 0

					|| ((BasicDataSource) dataSource).getNumIdle() > 0;

		}

		return true;

	}	

	private void logPoolStatus(String src) {

		if (logger.isDebugEnabled() && dataSource instanceof BasicDataSource) {

			BasicDataSource ds = (BasicDataSource) dataSource;

			logger.debug(src + "  actives:" + ds.getNumActive() + "("

					+ ds.getMaxActive() + ") idle:" + ds.getNumIdle() + "("

					+ ds.getMaxIdle() + ")");

		}

	}

        private static class CanGetConnectionException extends JDBCSQLException {

            public CanGetConnectionException(String datasource, SQLException cause) {

                super("Can't get a connection to the data source (%(datasource))", cause, "_CanGetConnectionException", 0);

                setValue("datasource", datasource);

            }

        }

	protected synchronized Object getTheConnection() throws DataException {

		try {

			Object conn = this.dataSource.getConnection();

			logPoolStatus("getTheConnection");

			return conn;

		} catch (SQLException e) {

			throw new CanGetConnectionException(this.toString(),e);

		}

	}	

}

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * 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.

 *

 */

/*

 * AUTHORS (In addition to CIT):

 * 2009 IVER T.I   {{Task}}

 */

/**

 *

 */

package org.gvsig.fmap.dal.store.h2;

import java.sql.Connection;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import java.util.logging.Level;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.NewDataStoreParameters;

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

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

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

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

import org.gvsig.fmap.dal.resource.exception.AccessResourceException;

import org.gvsig.fmap.dal.spi.DataServerExplorerProviderServices;

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

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

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

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

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCExecuteSQLException;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCSQLException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * @author jmvivo

 *

 */

public class H2ServerExplorer extends JDBCServerExplorer {

    final static private Logger logger = LoggerFactory

            .getLogger(H2ServerExplorer.class);

    public static final String NAME = "H2SQLExplorer";

    public H2ServerExplorer(

            H2ServerExplorerParameters parameters,

            DataServerExplorerProviderServices services)

            throws InitializeException {

        super(parameters, services);

    }

    private H2ServerExplorerParameters getH2SQLParameters() {

        return (H2ServerExplorerParameters) getParameters();

    }

    protected JDBCHelper createHelper() throws InitializeException {

        return new H2Helper(this, getH2SQLParameters());

    }

    public String getStoreName() {

        return H2StoreProvider.NAME;

    }

    public String getProviderName() {

        return NAME;

    }

    protected JDBCStoreParameters createStoreParams()

            throws InitializeException, ProviderNotRegisteredException {

        H2StoreParameters orgParams = (H2StoreParameters) super

                .createStoreParams();

        orgParams.setUseSSL(getH2SQLParameters().getUseSSL());

        return orgParams;

    }

	// ****************************

    public boolean canAdd() {

        return true;

    }

    public DataStoreParameters getOpenParameters() throws DataException {

        H2ServerExplorerParameters parameters = getH2SQLParameters();

        H2StoreParameters params = new H2StoreParameters();

        params.setHost(parameters.getHost());

        params.setPort(parameters.getPort());

        params.setDBName(parameters.getDBName());

        params.setUser(parameters.getUser());

        params.setPassword(parameters.getPassword());

        params.setCatalog(parameters.getCatalog());

        params.setSchema(parameters.getSchema());

        params.setJDBCDriverClassName(parameters.getJDBCDriverClassName());

        params.setUrl(parameters.getUrl());

        return params;

    }

    protected void checkIsMine(DataStoreParameters dsp) {

        if (!(dsp instanceof H2ConnectionParameters)) {

            // FIXME Excpetion ???

            throw new IllegalArgumentException(

                    "not instance of H2SQLConnectionParameters");

        }

        super.checkIsMine(dsp);

        H2ConnectionParameters pgp = (H2ConnectionParameters) dsp;

        if (pgp.getUseSSL().booleanValue() != getH2SQLParameters()

                .getUseSSL()) {

            throw new IllegalArgumentException("worng explorer: Host");

        }

    }

    public void remove(DataStoreParameters dsp) throws RemoveException {

        final H2StoreParameters pgParams = (H2StoreParameters) dsp;

        TransactionalAction action = new TransactionalAction() {

            public boolean continueTransactionAllowed() {

                return false;

            }

            public Object action(Connection conn) throws DataException {

                Statement st;

                try {

                    st = conn.createStatement();

                } catch (SQLException e) {

                    throw new JDBCSQLException(e);

                }

                String sqlDrop = "Drop table "

                        + pgParams.tableID();

                StringBuilder strb = new StringBuilder();

                strb.append("Delete from GEOMETRY_COLUMNS where f_table_schema = ");

                if (pgParams.getSchema() == null || pgParams.getSchema().length() == 0) {

                    strb.append("current_schema() ");

                } else {

                    strb.append('\'');

                    strb.append(pgParams.getSchema());

                    strb.append("' ");

                }

                strb.append("and f_table_name = '");

                strb.append(pgParams.getTable());

                strb.append('\'');

                String sqlDeleteFromGeometry_column = strb.toString();

                try {

                    try {

                        JDBCHelper.execute(st, sqlDrop);

                    } catch (SQLException e) {

                        throw new JDBCExecuteSQLException(sqlDrop, e);

                    }

                    try {

                        JDBCHelper.execute(st, sqlDeleteFromGeometry_column);

                    } catch (SQLException e) {

                        throw new JDBCExecuteSQLException(

                                sqlDeleteFromGeometry_column, e);

                    }

                } finally {

                    try {

                        st.close();

                    } catch (SQLException e) {

                    };

                }

                return null;

            }

        };

        try {

            this.helper.doConnectionAction(action);

        } catch (Exception e) {

            throw new RemoveException(this.getProviderName(), e);

        }

    }

    public NewDataStoreParameters getAddParameters() throws DataException {

        H2ServerExplorerParameters parameters = getH2SQLParameters();

        H2NewStoreParameters params = new H2NewStoreParameters();

        params.setHost(parameters.getHost());

        params.setPort(parameters.getPort());

        params.setDBName(parameters.getDBName());

        params.setUser(parameters.getUser());

        params.setPassword(parameters.getPassword());

        params.setCatalog(parameters.getCatalog());

        params.setSchema(parameters.getSchema());

        params.setJDBCDriverClassName(parameters.getJDBCDriverClassName());

        params.setUrl(parameters.getUrl());

        params.setUseSSL(parameters.getUseSSL());

        params.setDefaultFeatureType(this.getServerExplorerProviderServices()

                .createNewFeatureType());

        return params;

    }

    public boolean hasGeometrySupport() {

        return true;

    }

    protected H2Helper getPgHelper() {

        return (H2Helper) getHelper();

    }

    @Override

    public List getDataStoreProviderNames() {

        List x = new ArrayList(1);

        x.add(H2StoreProvider.NAME);

        return x;

    }

    public void updateTableStatistics(String tableName) throws JDBCExecuteSQLException {

        String sql="";

        try {

            if( tableName.startsWith("\"") ) {

                sql = "VACUUM ANALYZE " + tableName ;

            } else {

                sql = "VACUUM ANALYZE \"" + tableName + "\"";

            }

            Connection conn = this.getHelper().getConnection();

            Statement st = conn.createStatement();

            JDBCHelper.execute(st, sql);

        } catch (Exception ex) {

            throw new JDBCExecuteSQLException(sql, ex);

        }

    }

}

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

*

* 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.

*

*/

/*

* AUTHORS (In addition to CIT):

* 2009 IVER T.I   {{Task}}

*/

package org.gvsig.fmap.dal.store.h2;

import java.util.Iterator;

import java.util.List;

import java.util.Properties;

import java.util.regex.Matcher;

import java.util.regex.Pattern;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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

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

import org.gvsig.fmap.dal.feature.spi.FeatureProvider;

import org.gvsig.fmap.dal.feature.spi.FeatureSetProvider;

import org.gvsig.fmap.dal.spi.DataStoreProviderServices;

import org.gvsig.fmap.dal.store.db.DBHelper;

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

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

import org.gvsig.fmap.geom.Geometry;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class H2StoreProvider extends JDBCStoreProviderWriter {

	public final static Logger logger = LoggerFactory

			.getLogger(H2StoreProvider.class);

	public static final String NAME = "H2";

	public static final String DESCRIPTION = "H2 source";

	public static final String METADATA_DEFINITION_NAME = NAME;

	public H2StoreProvider(H2StoreParameters params,

			DataStoreProviderServices storeServices)

			throws InitializeException {

		super(params, storeServices, DBHelper.newMetadataContainer(METADATA_DEFINITION_NAME));

	}

	private H2StoreParameters getPGParameters() {

		return (H2StoreParameters) this.getParameters();

	}

	protected JDBCHelper createHelper() throws InitializeException {

	    JDBCHelper resp = new H2Helper(this, getPGParameters());

	    return resp;

	}

	protected String fixFilter(String _filter) {

		if (_filter == null) {

			return null;

		}

		String filter = fixFunctionNames(_filter);

		// Transform SRS to code

		// GeomFromText\s*\(\s*'[^']*'\s*,\s*('[^']*')\s*\)

		String geom_from_text = this.getFunctionName("ST_GeomFromText");

		Pattern pattern = Pattern

				.compile(geom_from_text + "\\s*\\(\\s*'[^']*'\\s*,\\s*'([^']*)'\\s*\\)");

		Matcher matcher = pattern.matcher(filter);

		StringBuilder strb = new StringBuilder();

		int pos = 0;

		String srsCode;

		while (matcher.find(pos)) {

			strb.append(filter.substring(pos, matcher.start(1)));

			srsCode = matcher.group(1).trim();

			if (srsCode.startsWith("'")) {

				srsCode = srsCode.substring(1);

			}

			if (srsCode.endsWith("'")) {

				srsCode = srsCode.substring(0, srsCode.length() - 1);

			}

			strb.append(helper.getProviderSRID(srsCode));

			strb.append(filter.substring(matcher.end(1), matcher.end()));

			pos = matcher.end();

		}

		strb.append(filter.substring(pos));

		return strb.toString();

	}

    public String getName() {

		return NAME;

	}

	public FeatureSetProvider createSet(FeatureQuery query,

			FeatureType featureType) throws DataException {

		return new H2SetProvider(this, query, featureType);

	}

	public DataServerExplorer getExplorer() throws ReadException {

		DataManager manager = DALLocator.getDataManager();

		H2ServerExplorerParameters exParams;

		H2StoreParameters params = getPGParameters();

		try {

			exParams = (H2ServerExplorerParameters) manager

					.createServerExplorerParameters(H2ServerExplorer.NAME);

			exParams.setUrl(params.getUrl());

			exParams.setHost(params.getHost());

			exParams.setPort(params.getPort());

			exParams.setDBName(params.getDBName());

			exParams.setUser(params.getUser());

			exParams.setPassword(params.getPassword());

			exParams.setCatalog(params.getCatalog());

			exParams.setSchema(params.getSchema());

			exParams.setJDBCDriverClassName(params.getJDBCDriverClassName());

			exParams.setUseSSL(params.getUseSSL());

			return manager.openServerExplorer(H2ServerExplorer.NAME, exParams);

		} catch (DataException e) {

			throw new ReadException(this.getName(), e);

		} catch (ValidateDataParametersException e) {

			throw new ReadException(this.getName(), e);

		}

	}

	public boolean allowAutomaticValues() {

		return true;

	}

	public boolean hasGeometrySupport() {

		return true;

	}

	protected H2Helper getPgHelper() {

		return (H2Helper) getHelper();

	}

	public boolean canWriteGeometry(int geometryType, int geometrySubtype)

			throws DataException {

		FeatureType type = getFeatureStore().getDefaultFeatureType();

		FeatureAttributeDescriptor geomAttr = type.getAttributeDescriptor(type

								.getDefaultGeometryAttributeName());

		if (geomAttr == null) {

			return false;

		}

		if (geometrySubtype != geomAttr.getGeometrySubType()) {

			return false;

		}

		switch (geomAttr.getGeometryType()) {

		case Geometry.TYPES.GEOMETRY:

			return true;

		case Geometry.TYPES.MULTISURFACE:

			return geometryType == Geometry.TYPES.MULTISURFACE

					|| geometryType == Geometry.TYPES.SURFACE;

		case Geometry.TYPES.MULTIPOINT:

			return geometryType == Geometry.TYPES.MULTIPOINT

					|| geometryType == Geometry.TYPES.POINT;

		case Geometry.TYPES.MULTICURVE:

			return geometryType == Geometry.TYPES.MULTICURVE

					|| geometryType == Geometry.TYPES.CURVE;

		case Geometry.TYPES.MULTISOLID:

			return geometryType == Geometry.TYPES.MULTISOLID

					|| geometryType == Geometry.TYPES.SOLID;

		default:

			return geometryType == geomAttr.getGeometryType();

		}

	}

	protected void addToListFeatureValues(FeatureProvider featureProvider,

			FeatureAttributeDescriptor attrOfList,

			FeatureAttributeDescriptor attr,

			List<Object> values) throws DataException {

		super.addToListFeatureValues(featureProvider, attrOfList, attr, values);

		if (attr.getType() == DataTypes.GEOMETRY) {

			values.add(helper.getProviderSRID(attr.getSRS()));

		}

	}

	protected void prepareAttributeForInsert(

			FeatureAttributeDescriptor attr, List<String> fields, List<String> values) {

		if (attr.getType() == DataTypes.GEOMETRY) {

			fields.add(helper.escapeFieldName(attr.getName()));

			values.add(getFunctionName("ST_GeomFromWKB") + "(?,?)");

		} else {

			super.prepareAttributeForInsert(attr, fields, values);

		}

	}

	protected void prepareAttributeForUpdate(FeatureAttributeDescriptor attr,

			List<String> values) {

		if (attr.getType() == DataTypes.GEOMETRY) {

			values.add(helper.escapeFieldName(attr.getName())

					+ " = " + getFunctionName("ST_GeomFromWKB") + "(?,?)");

		} else {

			super.prepareAttributeForUpdate(attr, values);

		}

	}

	protected String getSqlStatementAddField(FeatureAttributeDescriptor attr,

			List<String> additionalStatement) throws DataException {

		if (attr.getType() == DataTypes.GEOMETRY) {

			H2StoreParameters params = getPGParameters();

			additionalStatement.addAll(	((H2Helper) helper)

					.getSqlGeometyFieldAdd(attr, params.getTable(), params

							.getSchema()));

		}

		return super.getSqlStatementAddField(attr, additionalStatement);

	}

	private String getSqlGeometyFieldDrop(FeatureAttributeDescriptor attr) {

		StringBuilder strb = new StringBuilder();

		H2StoreParameters params = getPGParameters();

		strb.append("Delete from geometry_columns where f_geometry_column = '");

		strb.append(attr.getName());

		strb.append("' and f_table_nam = '");

		strb.append(params.getTable());

		strb.append("' and f_table_schema = ");

		if (params.getSchema() == null || params.getSchema().length() == 0) {

			strb.append("current_schema()");

		} else {

			strb.append("'");

			strb.append(params.getSchema());

			strb.append("'");

		}

		if (params.getCatalog() != null && params.getCatalog().length() > 0) {

			strb.append(" and f_table_catalog = '");

			strb.append(params.getCatalog());

			strb.append("'");

		}

		return strb.toString();

	}

	protected String getSqlStatementDropField(FeatureAttributeDescriptor attr,

			List<String> additionalStatement) {

		String result = super.getSqlStatementDropField(attr,

				additionalStatement);

		if (attr.getType() == DataTypes.GEOMETRY) {

			additionalStatement.add(getSqlGeometyFieldDrop(attr));

		}

		return result;

	}

	protected List<String> getSqlStatementAlterField(

			FeatureAttributeDescriptor attrOrg,

			FeatureAttributeDescriptor attrTrg, List<String> additionalStatement)

			throws DataException {

		//

		List<String> actions = super.getSqlStatementAlterField(attrOrg, attrTrg,

				additionalStatement);

		H2StoreParameters params = getPGParameters();

		if (attrOrg.getDataType() != attrTrg.getDataType()) {

			if (attrOrg.getType() == DataTypes.GEOMETRY) {

				additionalStatement.add(getSqlGeometyFieldDrop(attrOrg));

			}

			if (attrTrg.getType() == DataTypes.GEOMETRY) {

				additionalStatement.addAll(((H2Helper) helper)

						.getSqlGeometyFieldAdd(attrTrg, params.getTable(),

								params.getSchema()));

			}

		}

		if (attrOrg.getDataType() == attrTrg.getDataType()

				&& attrTrg.getType() == DataTypes.GEOMETRY) {

			// TODO Checks SRS and GeomType/Subtype

		}

		return actions;

	}

	private String getFunctionName(String newFunctionName) {