org.baswell.httproxy.HttpMessage Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of httproxy Show documentation
A low overhead Java HTTP proxy.
The newest version!
/*
 * Copyright 2015 Corey Baswell
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.baswell.httproxy;

import gnu.trove.list.array.TByteArrayList;

import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import static org.baswell.httproxy.Constants.*;

/**
 * Base class for {@link HttpRequest} and {@link HttpResponse}.
 */
abstract public class HttpMessage
{
  /**
   * @return The status line of this message (ex. GET /path/to/file/index.html HTTP/1.0, HTTP/1.0 200 OK).
   */
  abstract public String getStatusLine();

  /**
   * When did this message start.
   */
  public final Date startedAt = new Date();

  /**
   * The HTTP headers of this message.
   */
  public final List headers = new ArrayList();

  /**
   * When did this message end.
   */
  public Date endedAt;

  /**
   * Attachments not used by HttProxy.
   */
  public final Map attachements = new HashMap();

  /**
   *
   * @param name The header name
   * @return All headers that have the given name.
   */
  public List getHeaders(String name)
  {
    List headersWithName = new ArrayList();
    for (HttpHeader header : headers)
    {
      if (header.name.equalsIgnoreCase(name))
      {
        headersWithName.add(header);
      }
    }
    return headersWithName;
  }

  /**
   *
   * @param name The HTTP header name to retrieve.
   * @return The value from the first header found that matches the given name or null if no match is found.
   */
  public String getHeaderValue(String name)
  {
    for (HttpHeader header : headers)
    {
      if (header.name.equalsIgnoreCase(name))
      {
        return header.value;
      }
    }

    return null;
  }

  /**
   * If a header with the given name exists then the value for this header is replaced. If no header matches the given
   * name then a new header is added with the given value.
   *
   * @param name The HTTP header name.
   * @param value The HTTP header value.
   */
  public void setOrAddHeader(String name, String value)
  {
    for (HttpHeader header : headers)
    {
      if (header.name.equalsIgnoreCase(name))
      {
        header.value = value;
        return;
      }
    }

    headers.add(new HttpHeader(name, value));
  }

  @Override
  public String toString()
  {
    return new String(toBytes());
  }

  byte[] toBytes()
  {
    TByteArrayList bytes = new TByteArrayList((headers.size() + 1) * AVERAGE_HEADER_LENGTH);
    bytes.add(getStatusLine().getBytes());
    bytes.add(CR);
    bytes.add(LF);

    for (HttpHeader header : headers)
    {
      header.addTo(bytes);
    }

    bytes.add(CR);
    bytes.add(LF);

    return bytes.toArray();
  }

  HttpHeader addHeader(String headerLine)
  {
    int index = headerLine.indexOf(':');
    if ((index > 0) && (index < (headerLine.length() - 1)))
    {
      String name = headerLine.substring(0, index).trim();
      String value = headerLine.substring(index + 1, headerLine.length()).trim();
      HttpHeader header = new HttpHeader(name, value);
      headers.add(header);
      return header;
    }
    else
    {
      return null;
    }
  }
}