/*
 * Copyright (c) 2003, KNOPFLERFISH project
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following
 * conditions are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 *
 * - Redistributions in binary form must reproduce the above
 *   copyright notice, this list of conditions and the following
 *   disclaimer in the documentation and/or other materials
 *   provided with the distribution.
 *
 * - Neither the name of the KNOPFLERFISH project nor the names of its
 *   contributors may be used to endorse or promote products derived
 *   from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 * OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package org.knopflerfish.bundle.http;

import java.util.Hashtable;
import java.util.Enumeration;
import java.io.IOException;
import java.io.InputStream;
import java.io.BufferedInputStream;

import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpUtils;


public class RequestBase extends HeaderBase {

  // protected constants

  protected static final String POST_METHOD = "POST";
  protected static final String HTTP_1_0_PROTOCOL = "HTTP/1.0";
  protected static final String HTTP_1_1_PROTOCOL = "HTTP/1.1";

  protected static final String FORM_MIME_TYPE =
      "application/x-www-form-urlencoded";


  // private fields

  private String method = null;
  private String uri = null;
  private String protocol = null;
  private String queryString = null;

  private Hashtable queryParameters = null;
  private Hashtable parameters = null;

  private ServletInputStreamImpl body = null;


  // constructors

  RequestBase() { }


  // public methods

  public void init(InputStream is) throws HttpException, IOException {

    ServletInputStreamImpl in =
        new ServletInputStreamImpl(new BufferedInputStream(is));

    parseRequestLine(in);
    super.init(in);

    body = in;
  }

  public void destroy() {

    method = null;
    uri = null;
    protocol = null;
    queryString = null;

    queryParameters = null;
    parameters = null;

    body = null;

    super.destroy();
  }

  public String getMethod() {
    return method;
  }

  public String getURI() {
    return uri;
  }

  public String getProtocol() {
    return protocol;
  }

  public String getQueryString() {
    return queryString;
  }

  public Hashtable getQueryParameters() {

    if (queryParameters == null)
      queryParameters = parseQueryString();

    return queryParameters;
  }

  public ServletInputStreamImpl getBody() {
    return body;
  }

  public Hashtable getParameters() {

    if (parameters == null)
      parameters = parseParameters();

    return parameters;
  }


  // private methods

  private void parseRequestLine(ServletInputStreamImpl in)
      throws HttpException, IOException {

    String line;
    int index;

    // parse method
    line = in.readLine();
    index = line.indexOf(' ');
    if (index == -1)
      throw new HttpException(HttpServletResponse.SC_BAD_REQUEST);
    method = line.substring(0, index);

    // parse uri
    line = line.substring(index + 1);
    index = line.indexOf(' ');
    if (index == -1)
      throw new HttpException(HttpServletResponse.SC_BAD_REQUEST);
    uri = line.substring(0, index);

    // parse protocol
    protocol = line.substring(index + 1).trim();

    // parse query string
    index = uri.indexOf('?');
    if (index != -1) {
      queryString = uri.substring(index + 1);
      uri = uri.substring(0, index);
    }
  }

  private Hashtable parseQueryString() {

    if (queryString != null) {
      try {
        return HttpUtils.parseQueryString(queryString);
      } catch (IllegalArgumentException ignore) { }
    }

    return new Hashtable();
  }

  private Hashtable parseParameters() {

    Hashtable parameters = getQueryParameters();

    if (POST_METHOD.equals(method) &&
        FORM_MIME_TYPE.equals(getContentType())) {
      // Check that the input stream has not been touched
      try {
        Hashtable p = HttpUtils.parsePostData(getContentLength(),
                                              getBody());
        Enumeration e = p.keys();
        while (e.hasMoreElements()) {
          String key = (String) e.nextElement();
          String[] val = (String[]) p.get(key);
          String[] valArray;
          String oldVals[] = (String []) parameters.get(key);
          if (oldVals == null) {
            valArray = val;
          } else {
            valArray = new String[oldVals.length + val.length];
            for (int i = 0; i < val.length; i++)
              valArray[i] = val[i];
            for (int i = 0; i < oldVals.length; i++)
              valArray[val.length + i] = oldVals[i];
          }
          parameters.put(key, valArray);
        }
      } catch (IllegalArgumentException ignore) { }
    }

    return parameters;
  }

} // RequestBase