Application: gvSIG desktop

package org.gvsig.geodb.workspace;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import static org.gvsig.fmap.dal.DataManager.RESOURCES_FIELD_NAME;

import static org.gvsig.fmap.dal.DataManager.RESOURCES_FIELD_RESOURCE;

import static org.gvsig.fmap.dal.DataManager.RESOURCES_TABLE_NAME;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.DataTypes;

import org.gvsig.fmap.dal.StoresRepository;

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

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

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

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

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

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

import org.gvsig.tools.resourcesstorage.ResourcesStorage;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class DefaultWorkspaceManager implements WorkspaceManager {

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

    private static final String REPOSITORY_TABLE_NAME = "gvsigd_repository";

    private static final String CONFIG_TABLE_NAME = "gvsigd_config";

    private final JDBCServerExplorerParameters connection;

    public DefaultWorkspaceManager(JDBCServerExplorerParameters connection) {

        this.connection = connection;

    }

    @Override

    public JDBCServerExplorerParameters getConnection() {

        return this.connection;

    }

    private boolean existsTable(String tableName) {

        try {

            DataManager dataManager = DALLocator.getDataManager();

            JDBCServerExplorer server = (JDBCServerExplorer) dataManager.openServerExplorer(

                    this.connection.getExplorerName(),

                    this.connection

            );

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

                String theTableName = ((JDBCStoreParameters)params).getTable();

                if( StringUtils.equals(theTableName, tableName) ) {

                    return true;

                }

            }

        } catch (Exception ex) {

            LOGGER.warn("Can't check if the table '"+tableName+"' exists.",ex);

        }

        return false;

    }

    private FeatureStore getTable(String tableName) {

        try {

            DataManager dataManager = DALLocator.getDataManager();

            JDBCServerExplorer server = (JDBCServerExplorer) dataManager.openServerExplorer(

                    this.connection.getExplorerName(),

                    this.connection

            );

            DataStoreParameters params = server.get(tableName);

            if( params!=null ) {

                FeatureStore store = (FeatureStore) dataManager.openStore(

                        params.getDataStoreName(), 

                        params

                );

                return store;

            }

        } catch (Exception ex) {

            LOGGER.warn("Can't open table '"+tableName+"'.",ex);

        }

        return null;

    }

    private void dropTable(String tableName) {

        try {

            DataManager dataManager = DALLocator.getDataManager();

            JDBCServerExplorer server = (JDBCServerExplorer) dataManager.openServerExplorer(

                    this.connection.getExplorerName(),

                    this.connection

            );

            DataStoreParameters params = server.get(tableName);

            if( params!=null ) {

                server.remove(params);

            }

        } catch (Exception ex) {

            LOGGER.warn("Can't drop table '"+tableName+"'.",ex);

        }

    }

    @Override

    public boolean existsResourcesTable() {

        return this.existsTable(RESOURCES_TABLE_NAME);

    }

    @Override

    public void createResourcesTable() {

        String tableName = RESOURCES_TABLE_NAME;

        try {

            DataManager dataManager = DALLocator.getDataManager();

            JDBCServerExplorer server = (JDBCServerExplorer) dataManager.openServerExplorer(

                    this.connection.getExplorerName(),

                    this.connection

            );

            JDBCNewStoreParameters params = server.getAddParameters();

            params.setTable(tableName);

            EditableFeatureType ft = params.getDefaultFeatureType();

            ft.add(RESOURCES_FIELD_NAME, DataTypes.STRING, 150)

                    .setAllowNull(false)

                    .setIsPrimaryKey(true);

            ft.add(RESOURCES_FIELD_RESOURCE, DataTypes.BYTEARRAY)

                    .setAllowNull(true);

            server.add(tableName, params, false);

        } catch (Exception ex) {

            LOGGER.warn("Can't create table '"+tableName+"'.",ex);

        }

    }

    @Override

    public FeatureStore getResourcesTable() {

        return this.getTable(RESOURCES_TABLE_NAME);

    }

    @Override

    public void dropResourcesTable() {

        this.dropTable(RESOURCES_TABLE_NAME);

    }

    @Override

    public String get(String name) {

        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

    }

    @Override

    public void set(String name, String value) {

        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

    }

    @Override

    public StoresRepository getStoresRepository() {

        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

    }

    @Override

    public boolean canAnonymousUserWriteInTheTables() {

        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

    }

    @Override

    public FeatureStore getRepositoryTable() {

        return this.getTable(REPOSITORY_TABLE_NAME);

    }

    @Override

    public FeatureStore getConfigurationTable() {

        return this.getTable(CONFIG_TABLE_NAME);

    }

    @Override

    public void createRepositoryTable() {

        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

    }

    @Override

    public void createConfigurationTable() {

        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.

    }

    @Override

    public boolean existsRepositoryTable() {

        return this.existsTable(REPOSITORY_TABLE_NAME);

    }

    @Override

    public boolean existsConfigurationTable() {

        return this.existsTable(CONFIG_TABLE_NAME);

    }

}