/*
 *    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.MalformedURLException;
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 org.geotoolkit.csw.xml.CswXmlFactory;
import org.geotoolkit.csw.xml.Transaction;
import org.geotoolkit.csw.xml.Delete;
import org.geotoolkit.csw.xml.Insert;
import org.geotoolkit.csw.xml.Update;
import org.geotoolkit.security.ClientSecurity;


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

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

    private Delete delete = null;
    private Insert insert = null;
    private Update update = 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 AbstractTransaction(final String serverURL, final String version, final ClientSecurity security){
        super(serverURL,security);
        this.version = version;
    }

    @Override
    public Delete getDelete() {
        return delete;
    }

    @Override
    public void setDelete(final Delete delete) {
        this.delete = delete;
    }

    @Override
    public Insert getInsert() {
        return insert;
    }

    @Override
    public void setInsert(final Insert insert) {
        this.insert = insert;
    }

    @Override
    public Update getUpdate() {
        return update;
    }

    @Override
    public void setUpdate(final Update update) {
        this.update = update;
    }

    /**
     * The transaction CSW requests do not support the KVP encondig, regarding the standards.
     * This method will always return an {@link UnsupportedOperationException}.
     * Do not call it.
     */
    @Override
    public URL getURL() throws MalformedURLException {
        throw new UnsupportedOperationException("Transaction requests do not support KVP encoding.");
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public InputStream getResponseStream() throws IOException {
        final URL url = new URL(serverURL);
        URLConnection conec = url.openConnection();
        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 Transaction transactXml;
            if (delete != null) {
                transactXml = CswXmlFactory.createTransaction(version, "CSW", delete);
            } else if (insert != null) {
                transactXml = CswXmlFactory.createTransaction(version, "CSW", insert);
            } else if (update != null) {
                transactXml = CswXmlFactory.createTransaction(version, "CSW", update);
            } else {
                throw new IllegalArgumentException("The specified requests is not valid. " +
                        "It does not contain any insert, delete or update request.");
            }
            marsh.marshal(transactXml, stream);
            POOL.recycle(marsh);
        } catch (JAXBException ex) {
            throw new IOException(ex);
        }
        stream.close();
        return security.decrypt(conec.getInputStream());
    }
}