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.lib.server.handlers;

import java.io.IOException;

import java.io.Reader;

import java.io.Writer;

import java.util.Collections;

import java.util.Map;

import javax.json.JsonException;

import javax.json.JsonObject;

import javax.json.JsonReader;

import javax.json.stream.JsonGenerator;

import javax.json.stream.JsonGeneratorFactory;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.BooleanUtils;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.lang3.mutable.MutableObject;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.crs.CRSFactory;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryException;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.json.Json;

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

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

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

import org.gvsig.vcsgis.lib.repository.localdb.VCSGisRepositoryLocaldbImpl;

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

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

import static org.gvsig.vcsgis.lib.server.handlers.AbstractVCSGisServertHandler.LOGGER;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class CheckoutHandler extends AbstractVCSGisServertHandler {

    public CheckoutHandler(VCSGisRepository repository) {

        super(repository, "CheckoutHandler");

    }

    @Override

    protected void requestProducer(MutableObject<VCSGisRequest>req, Map<String,String> params, Reader request_contents) throws IOException {

        JsonObject jsonParameters = null;

        try {

            JsonReader reader = Json.createReader(request_contents);

            JsonObject jsonRequest = reader.readObject();

            jsonParameters = jsonRequest.getJsonObject("Parameters");

        } catch (JsonException e) {

            //Do nothing

        }

        String entityName = params.get("EntityName");

        if( jsonParameters != null ) {

            entityName = jsonParameters.getString("EntityName", entityName);

        }

        VCSGisCheckoutRequest request = this.getRepository().createCheckoutRequest(entityName);  

        request.setAuthenticationTokenAndUser(params);

        if( jsonParameters == null ) {

            request.setEfectiveDate(params.get("EfectiveDate"));

            request.setRevisionCode(params.get("RevisionCode"));

        } else {

            request.setEfectiveDate(jsonParameters.getString("EfectiveDate", params.get("EfectiveDate")));

            request.setRevisionCode(jsonParameters.getString("RevisionCode", params.get("RevisionCode")));

        }        

        String roi_hexwkb = params.get("ROI");

        if( jsonParameters != null ) {

            roi_hexwkb = jsonParameters.getString("ROI", null);

        }

        String roi_proj = params.get("ROIProjection");

        if( jsonParameters != null ) {

            roi_proj = jsonParameters.getString("ROIProjection", null);

        }

        if( StringUtils.isNotBlank(roi_hexwkb) ) {

            try {

                Geometry roi = GeometryLocator.getGeometryManager().createFrom(roi_hexwkb);

                if( StringUtils.isNotBlank(roi_proj)) {

                    IProjection proj = CRSFactory.getCRS(roi_proj);

                    roi.setProjection(proj);

                }

                request.setROI(roi.getEnvelope());

            } catch (GeometryException ex) {

                throw new RuntimeException("Can't parse roi ("+roi_hexwkb+").",ex);

            }

        }

        req.setValue(request);

        notifyRequestConsumers();

    }

    @Override

    protected void responseProducer(VCSGisRequest req, Writer response_contents) throws IOException {

        final VCSGisCheckoutRequest request = (VCSGisCheckoutRequest) req;

        JsonGenerator gen = null;

        try {

            boolean use_revision_number = false;

            try {

                if( this.getRepository() instanceof VCSGisRepositoryLocaldb ) {

                    VCSGisRepositoryLocaldb repo = (VCSGisRepositoryLocaldb) this.getRepository();

                    String use_revision_number_s = repo.getConfigValue(VCSGisRepositoryLocaldbImpl.CONFIG_CHECKOUT_WITH_REVNUMBER);

                    use_revision_number = BooleanUtils.toBoolean(use_revision_number_s);

                }

            } catch(Exception ex) {

                LOGGER.debug("Can't retrieve use_revision_number value",ex);

            }

            JsonGeneratorFactory genFactory = Json.createGeneratorFactory(

                Collections.singletonMap(JsonGenerator.PRETTY_PRINTING, true)

            );

            gen = genFactory.createGenerator(response_contents);

            gen.writeStartObject();

            gen.writeStartObject("Parameters");

            gen.write("StatusCode", request.getLastErrorCode());

            notifyResponseConsumers();

            gen.flush();

            if( StringUtils.isBlank(request.getLastErrorMessage()) ) {

                gen.writeNull("StatusMessage");

            } else {

                gen.write("StatusMessage", request.getLastErrorMessage());

            }

            gen.write("Entity", request.getEntity().toJson());

            gen.write("UsershashCode", request.getUsersHashCode());

            gen.write("TopologyPlansHashCode", request.getTopologyPlansHashCode());

            gen.writeEnd(); // Paramaters

            gen.writeStartArray("Data");

            for (VCSGisRepositoryData data : request.getData()) {

                gen.writeStartObject();

                gen.write("DataCode", data.getDataCode());

                gen.write("Data", data.getData());

                if( use_revision_number ) {

                    gen.write("RevisionNumber", data.getRevisionNumber());

                }

                gen.writeEnd(); 

                gen.flush();

            }

            gen.writeEnd(); // Data

            gen.writeEnd(); 

            gen.flush();

            response_contents.flush();

        } catch (Exception ex) {

            LOGGER.warn("Can't produce Json data for "+this.getName()+" response.",ex);

        } finally {

            IOUtils.closeQuietly(gen);

            gen.close();

//            IOUtils.closeQuietly(response_contents);

        }

    }

}