/*
 *    Geotoolkit - An Open Source Java GIS Toolkit
 *    http://www.geotoolkit.org
 *
 *    (C) 2011, 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.wps;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URL;
import java.net.URLConnection;
import java.util.ArrayList;
import java.util.List;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import org.geotoolkit.client.AbstractRequest;
import org.geotoolkit.ows.xml.v110.CodeType;
import org.geotoolkit.security.ClientSecurity;
import org.geotoolkit.wps.xml.WPSMarshallerPool;
import org.geotoolkit.wps.xml.v100.DescribeProcess;

/**
 * Abstract get capabilities request.
 *
 * @author Quentin Boileau
 * @module
 */
public abstract class AbstractDescribeProcess extends AbstractRequest implements DescribeProcessRequest{

    protected final String version;
    protected List<String> identifiers;
    protected final boolean doGET;

    protected AbstractDescribeProcess(final String serverURL,final String version, final ClientSecurity security){
        this(serverURL, version, security, true);
    }

    protected AbstractDescribeProcess(final String serverURL,final String version, final ClientSecurity security, final boolean doGET){
        super(serverURL,security,null);
        this.version = version;
        this.doGET = doGET;
    }

    /**
     * {@inheritDoc }
     */
    @Override
    public List<String> getIdentifiers() {
        return identifiers;
    }

    /**
     * {@inheritDoc }
     */
    @Override
    public void setIdentifiers(List<String> identifiers) {
       this.identifiers = identifiers;
    }

    /**
     * {@inheritDoc }
     */
    @Override
    protected void prepareParameters() {
        super.prepareParameters();
        requestParameters.put("SERVICE",    "WPS");
        requestParameters.put("REQUEST", "DescribeProcess");
        requestParameters.put("VERSION",    "1.0.0");
    }

    /**
     * {@inheritDoc }
     */
    @Override
    public InputStream getResponseStream() throws IOException {

        final DescribeProcess request = makeRequest();

        if (doGET) {

            //GET
            final URL url = getURL(); //build GET request
            URLConnection conec = url.openConnection();
            conec = security.secure(conec);
            return conec.getInputStream();
        } else {

            //POST
            final URL url = new URL(serverURL);
            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);
            Marshaller marshaller = null;
            try {
                marshaller = WPSMarshallerPool.getInstance().acquireMarshaller();
                marshaller.marshal(request, stream);
                //marshaller.marshal(request, System.out);
                WPSMarshallerPool.getInstance().recycle(marshaller);
            } catch (JAXBException ex) {
                throw new IOException(ex);
            }
            stream.close();
            return security.decrypt(conec.getInputStream());
        }
    }

    public DescribeProcess makeRequest(){

        final DescribeProcess request = new DescribeProcess();
        request.setService("WPS");
        request.setVersion(version);
        final List<CodeType> listId = new ArrayList<CodeType>();

        for(String str : identifiers){
            listId.add(new CodeType(str));
        }
        request.getIdentifier().addAll(listId);

        return request;
    }
}