/*
 *    Geotoolkit - An Open Source Java GIS Toolkit
 *    http://www.geotoolkit.org
 *
 *    (C) 2010, Geomatys
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation;
 *    version 2.1 of the License.
 *
 *    This library 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
 *    Lesser General Public License for more details.
 */
package org.geotoolkit.csw;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URL;
import java.net.URLConnection;
import static org.geotoolkit.csw.AbstractCSWRequest.POOL;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.datatype.DatatypeConfigurationException;
import javax.xml.datatype.DatatypeFactory;
import javax.xml.datatype.Duration;
import org.geotoolkit.csw.xml.CswXmlFactory;
import org.geotoolkit.csw.xml.Harvest;
import org.geotoolkit.security.ClientSecurity;
import org.geotoolkit.temporal.object.DefaultPeriodDuration;
import org.geotoolkit.temporal.object.TemporalUtilities;


/**
 * Abstract implementation of {@link TransactionRequest}, which defines the
 * parameters for a transaction request.
 *
 * @author Cédric Briançon (Geomatys)
 * @module
 */
public class AbstractHarvest extends AbstractCSWRequest implements HarvestRequest {

    /**
     * The version to use for this webservice request.
     */
    protected final String version;

    private String namespace = null;
    private String source = null;
    private String resourceType = null;
    private String resourceFormat = null;
    private String responseHandler = null;
    private String harvestInterval = null;

    /**
     * Defines the server url and the service version for this kind of request.
     *
     * @param serverURL The server url.
     * @param version The version of the request.
     */
    protected AbstractHarvest(final String serverURL, final String version, final ClientSecurity security){
        super(serverURL,security);
        this.version = version;
    }

    @Override
    public String getNamespace() {
        return namespace;
    }

    @Override
    public void setNamespace(final String namespace) {
        this.namespace = namespace;
    }

    @Override
    public String getSource() {
        return source;
    }

    @Override
    public void setSource(final String source) {
        this.source = source;
    }

    @Override
    public String getResourceType() {
        return resourceType;
    }

    @Override
    public void setResourceType(final String resourceType) {
        this.resourceType =  resourceType;
    }

    @Override
    public String getResourceFormat() {
        return resourceFormat;
    }

    @Override
    public void setResourceFormat(final String resourceFormat) {
        this.resourceFormat = resourceFormat;
    }

    @Override
    public String getResponseHandler() {
        return responseHandler;
    }

    @Override
    public void setResponseHandler(final String responseHandler) {
        this.responseHandler = responseHandler;
    }

    @Override
    public String getHarvestInterval() {
        return harvestInterval;
    }

    @Override
    public void setHarvestInterval(final String harvestInterval) {
        this.harvestInterval = harvestInterval;
    }

    @Override
    protected void prepareParameters() {
        super.prepareParameters();
        if (source == null) {
            throw new IllegalArgumentException("The parameter \"SOURCE\" is not defined");
        }
        if (resourceType == null) {
            throw new IllegalArgumentException("The parameter \"RESOURCETYPE\" is not defined");
        }

        requestParameters.put("SERVICE",      "CSW");
        requestParameters.put("REQUEST",      "Harvest");
        requestParameters.put("VERSION",      version);
        requestParameters.put("SOURCE",       source);
        requestParameters.put("RESOURCETYPE", resourceType);

        if (namespace != null) {
            requestParameters.put("NAMESPACE", namespace);
        }
        if (resourceFormat != null) {
            requestParameters.put("RESOURCEFORMAT", resourceFormat);
        }
        if (responseHandler != null) {
            requestParameters.put("RESPONSEHANDLER", responseHandler);
        }
        if (harvestInterval != null) {
            requestParameters.put("HARVESTINTERVAL", harvestInterval);
        }

    }

    /**
     * {@inheritDoc }
     */
    @Override
    public InputStream getResponseStream() throws IOException {
        final URL url = getURL();
        URLConnection conec = url.openConnection();
        conec = security.secure(conec);

        conec.setDoOutput(true);
        conec.setRequestProperty("Content-Type", "text/xml");

        OutputStream stream = conec.getOutputStream();
        stream = security.encrypt(stream);

        try {
            final Marshaller marsh = POOL.acquireMarshaller();
            final DefaultPeriodDuration periodDuration = (harvestInterval == null) ? null :
                    (DefaultPeriodDuration) TemporalUtilities.getDurationFromString(harvestInterval);
            final Duration duration = (periodDuration == null) ? null :
                    DatatypeFactory.newInstance().newDuration(true,
                        Integer.parseInt(periodDuration.getYears().toString()),
                        Integer.parseInt(periodDuration.getMonths().toString()),
                        Integer.parseInt(periodDuration.getDays().toString()),
                        Integer.parseInt(periodDuration.getHours().toString()),
                        Integer.parseInt(periodDuration.getMinutes().toString()),
                        Integer.parseInt(periodDuration.getSeconds().toString()));
            final Harvest harvestXml = CswXmlFactory.createHarvest(version, "CSW", source, resourceType,
                    resourceFormat, responseHandler, duration);
            marsh.marshal(harvestXml, stream);
            POOL.recycle(marsh);
        } catch (DatatypeConfigurationException ex) {
            throw new IOException(ex);
        } catch (JAXBException ex) {
            throw new IOException(ex);
        }
        stream.close();
        return security.decrypt(conec.getInputStream());
    }

}