gvSIG projects pool

/*

 * gvSIG. Desktop Geographic Information System.

 * 

 * Copyright (C) 2007-2020 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, see <https://www.gnu.org/licenses/>. 

 * 

 * 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.vcsgis.repository.localdb;

import java.util.HashMap;

import java.util.Map;

import java.util.UUID;

import javax.json.JsonObject;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

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

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

import org.gvsig.json.Json;

import org.gvsig.json.JsonObjectBuilder;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.task.SimpleTaskStatus;

import org.gvsig.vcsgis.lib.client.VCSGisWorkspaceEntity;

import org.gvsig.vcsgis.lib.repository.VCSGisRepository;

import org.gvsig.vcsgis.lib.repository.requests.VCSGisCheckoutRequest;

import org.gvsig.vcsgis.lib.repository.requests.VCSGisCommitRequest;

import org.gvsig.vcsgis.lib.repository.requests.VCSGisEntitiesRequest;

import org.gvsig.vcsgis.lib.repository.requests.VCSGisLogRequest;

import org.gvsig.vcsgis.lib.repository.requests.VCSGisUpdateRequest;

import org.gvsig.vcsgis.repository.localdb.requests.CheckoutRequestLocaldb;

import org.gvsig.vcsgis.repository.localdb.requests.CommitRequestLocaldb;

import org.gvsig.vcsgis.repository.localdb.requests.EntitiesRequestLocaldb;

import org.gvsig.vcsgis.repository.localdb.requests.LogRequestLocaldb;

import org.gvsig.vcsgis.repository.localdb.requests.UpdateRequestLocaldb;

import org.gvsig.vcsgis.repository.localdb.tables.DataRepoTable;

import org.gvsig.vcsgis.repository.localdb.tables.EntitiesRepoTable;

import org.gvsig.vcsgis.repository.localdb.tables.HooksRepoTable;

import org.gvsig.vcsgis.repository.localdb.tables.RevisionsRepoTable;

import org.gvsig.vcsgis.repository.localdb.tables.ServerConfigRepoTable;

import org.gvsig.vcsgis.repository.localdb.tables.TopologyplanRepoTable;

import org.gvsig.vcsgis.repository.localdb.tables.UsersRepoTable;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class VCSGisRepositoryLocaldb implements VCSGisRepository {

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

    public static int create(JDBCServerExplorer server, SimpleTaskStatus status) {

        if( status == null ) {

            status = ToolsLocator.getTaskStatusManager().createDefaultSimpleTaskStatus("vcsgis init");

            status.add();

        }

        int r = 3;

        status.push();

        try {

            status.setTitle("vcsgis init");

            FeatureType[] tables = new FeatureType[] {

                ServerConfigRepoTable.featureType(),

                UsersRepoTable.featureType(),

                TopologyplanRepoTable.featureType(),

                EntitiesRepoTable.featureType(),

                RevisionsRepoTable.featureType(),

                DataRepoTable.featureType(),

                HooksRepoTable.featureType(),

            };

            r = 4;

            status.setRangeOfValues(0, tables.length);

            int step = 1;

            for (FeatureType table : tables) {

                status.message(table.getLabel());

                status.setCurValue(step++);

                r = 4+step;

                JDBCNewStoreParameters table_params = server.getAddParameters();

                table_params.setDefaultFeatureType(table.getEditable());

                table_params.setTable(table.getTags().getString("ID"));

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

            }

            status.terminate();

        } catch(Exception ex) {

            status.abort();

            LOGGER.warn("Can't init repository.",ex);

            status.message(ex.getMessage());

            return r;

        } finally {

            status.pop();

        }

        return 0;

    }

    private JDBCServerExplorer server;

    private Map<String,FeatureStore> stores;

    public VCSGisRepositoryLocaldb() {

        // For persistence

        this.stores = null;

        this.server = null;

    }

    public VCSGisRepositoryLocaldb(JDBCServerExplorer server) {

        this.stores = null;

        this.server = server;

    }

    @Override

    public String getLabel() {

        return this.server.getParameters().getUrl();

    }

    public JDBCServerExplorer getServerExplorer() {

        return this.server;

    }

    public static void selfRegister() {

        Json.registerSerializer(VCSGisRepositoryLocaldb.class);

    }

    @Override

    public VCSGisCommitRequest createCommitRequest() {

        return new CommitRequestLocaldb(this);

    }

    @Override

    public VCSGisUpdateRequest createUpdateRequest(VCSGisWorkspaceEntity localEntity) {

        return new UpdateRequestLocaldb(this, localEntity.getEntityName(), localEntity.getLocalRevisionCode());

    }

    @Override

    public VCSGisCheckoutRequest createCheckoutRequest(String entityName) {

        return new CheckoutRequestLocaldb(this, entityName);

    }

    @Override

    public void fromJson(JsonObject json) {

        try {

            DataManager dataManager = DALLocator.getDataManager();

            DataServerExplorerParameters explorerParams = (DataServerExplorerParameters) Json.toObject(json, "explorerParams");

            this.server = (JDBCServerExplorer) dataManager.openServerExplorer(

                    explorerParams.getProviderName(), 

                    explorerParams

            );

            this.stores = new HashMap<>();

        } catch (Exception ex) {

            throw new RuntimeException("Can't restore repository from json object.", ex);

        }

    }

    @Override

    public JsonObject toJson() {

        return this.toJsonBuilder().build();

    }

    @Override

    public JsonObjectBuilder toJsonBuilder() {

        JsonObjectBuilder builder = Json.createObjectBuilder();

        builder.add_class(this);

        builder.add("explorerParams", this.server.getParameters());

        return builder;

    }

    public String createUniqueID() {

        UUID x = UUID.randomUUID();

        String s = x.toString().replace("-", "");

        return s;

    }

    public FeatureStore getFeatureStore(String tableName) {

        if( this.stores==null ) {

            this.stores = new HashMap<>();

        }

        FeatureStore store = this.stores.get(tableName);

        if( store==null ) {

            store = this.openFeatureStore(tableName, null);

            this.stores.put(tableName, store);

        }

        DisposeUtils.bind(store);

        return store;

    }

    public FeatureStore openFeatureStore(String tableName, String sql) {

        FeatureStore store;

        try {

            JDBCStoreParameters params = this.server.get(tableName);

            params.setSQL(sql);

            store = (FeatureStore) DALLocator.getDataManager().openStore(params.getProviderName(), params);

            return store;

        } catch (Exception ex) {

            throw new RuntimeException("Can't open store '"+tableName+"'.", ex);

        }

    }

    @Override

    public VCSGisEntitiesRequest createEntitiesRequest() {

        return new EntitiesRequestLocaldb(this);

    }

    @Override

    public VCSGisLogRequest createLogRequest(String entityName) {

        return new LogRequestLocaldb(this, entityName);

    }

}