/*
 *    GeoTools - The Open Source Java GIS Toolkit
 *    http://geotools.org
 *
 *    (C) 2002-2008, Open Source Geospatial Foundation (OSGeo)
 *
 *    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.geotools.data.wfs.protocol.http;

import java.io.BufferedInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Map;
import java.util.zip.GZIPInputStream;

import org.apache.commons.codec.binary.Base64;

/**
 * An {@link HTTPProtocol} implementation that relies on plain {@link HttpURLConnection}
 * 
 * @author Gabriel Roldan (OpenGeo)
 * @since 2.6.x
 * @version $Id: SimpleHttpProtocol.java 35068 2010-03-20 05:46:56Z jive $
 *
 * @source $URL: http://svn.osgeo.org/geotools/branches/2.7.x/build/maven/javadoc/../../../modules/unsupported/wfs-ng/src/main/java/org/geotools/data/wfs/protocol/http/SimpleHttpProtocol.java $
 */
@SuppressWarnings("nls")
public class SimpleHttpProtocol extends AbstractHttpProtocol {

    private static class SimpleHttpResponse implements HTTPResponse {

        private HttpURLConnection conn;

        private InputStream inputStream;

        public SimpleHttpResponse(HttpURLConnection conn) {
            this.conn = conn;
        }

        public String getContentType() {
            return conn.getContentType();
        }

        public String getResponseCharset() {
            // String contentType = getContentType();
            return null;
        }

        public String getResponseHeader(String headerName) {
            String headerField = conn.getHeaderField(headerName);
            return headerField;
        }

        public InputStream getResponseStream() throws IOException {
            if (this.inputStream == null) {
                InputStream responseStream = conn.getInputStream();
                responseStream = new BufferedInputStream(responseStream);
                final String contentEncoding = conn.getContentEncoding();
                if (contentEncoding != null && contentEncoding.toLowerCase().indexOf("gzip") != -1) {
                    responseStream = new GZIPInputStream(responseStream);
                }
                this.inputStream = responseStream;
            }
            return this.inputStream;
        }

        public String getTargetUrl() {
            return conn.getURL().toExternalForm();
        }

    }

    public HTTPResponse issueGet(URL baseUrl, Map<String, String> kvp) throws IOException {
        URL targetUrl = createUrl(baseUrl, kvp);
        HttpURLConnection conn = openConnection(targetUrl, HttpMethod.GET);
        HTTPResponse response = new SimpleHttpResponse(conn);
        return response;
    }

    public HTTPResponse issuePost(URL targetUrl, POSTCallBack callback) throws IOException {
        HttpURLConnection conn = openConnection(targetUrl, HttpMethod.POST);

        long contentLength = callback.getContentLength();
        conn.setRequestProperty("Content-Length", String.valueOf(contentLength));

        String bodyContentType = callback.getContentType();
        conn.setRequestProperty("Content-Type", bodyContentType);

        OutputStream bodyOut = conn.getOutputStream();
        callback.writeBody(bodyOut);

        HTTPResponse response = new SimpleHttpResponse(conn);
        return response;
    }

    private HttpURLConnection openConnection(URL targetUrl, HttpMethod method) throws IOException {
        HttpURLConnection conn;
        try {
            conn = (HttpURLConnection) targetUrl.openConnection();
        } catch (ClassCastException wrongUrl) {
            throw new IOException("HTTP connection required for " + targetUrl);
        }
        if (0 < getTimeoutMillis()) {
            conn.setConnectTimeout(getTimeoutMillis());
            conn.setReadTimeout(getTimeoutMillis());
        }
        // remember to set the Accept-Encoding header before setting the authentication credentials
        // or an IllegalStateException is thrown
        if (isTryGzip()) {
            conn.setRequestProperty("Accept-Encoding", "gzip");
        }
        if (method == HttpMethod.POST) {
            conn.setRequestMethod("POST");
            // conn.setRequestProperty("Content-type", "text/xml, application/xml");
            conn.setDoOutput(true);
        } else {
            conn.setRequestMethod("GET");
        }
        conn.setDoInput(true);

        if (authUsername != null && authPassword != null) {
            String userPassword = authUsername + ":" + authPassword;
            byte[] encodedUserPassword = userPassword.getBytes();

            String base64UserAndPasswd = new String(Base64.encodeBase64(encodedUserPassword));
            
            conn.setRequestProperty("Authorization", "Basic " + base64UserAndPasswd);
        }

        return conn;
    }
}