/*
 * Copyright (c) 1998-2008 Caucho Technology -- all rights reserved
 *
 * This file is part of Resin(R) Open Source
 *
 * Each copy or derived work must preserve the copyright notice and this
 * notice unmodified.
 *
 * Resin Open Source is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * Resin Open Source 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, or any warranty
 * of NON-INFRINGEMENT.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Resin Open Source; if not, write to the
 *
 *   Free Software Foundation, Inc.
 *   59 Temple Place, Suite 330
 *   Boston, MA 02111-1307  USA
 *
 * @author Emil Ong
 */

package com.caucho.util;

import java.io.IOException;

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

import com.caucho.util.L10N;

public class Attachment {
  private static final L10N L = new L10N(Attachment.class);

  private byte[] _contents;
  private final Map<String,List<String>> _headers = 
    new HashMap<String,List<String>>();

  Attachment()
  {
  }

  void setContents(byte[] contents)
  {
    _contents = contents;
  }

  public byte[] getContents()
  {
    return _contents;
  }

  /**
   * Decodes the attachment, if necessary, based on the 
   * Content-Transfer-Encoding header.
   **/
  public byte[] getRawContents()
    throws IOException
  {
    String encoding = getHeaderValue("Content-Transfer-Encoding");

    if (encoding == null) // XXX is this the right default?
      return _contents;

    else if ("base64".equalsIgnoreCase(encoding))
      return Base64.decodeToByteArray(new String(_contents, "US-ASCII"));
    
    else if ("binary".equalsIgnoreCase(encoding))
      return _contents;

    throw new IOException(L.l("Unknown attachment transfer encoding: {0}",
                              encoding));
  }

  public String getHeaderValue(String name)
  {
    List<String> values = _headers.get(name);

    if (values == null)
      return null;

    return values.get(0);
  }

  public Iterator getHeaderValues(String name)
  {
    List<String> values = _headers.get(name);

    if (values == null)
      return null;

    return values.iterator();
  }

  public void addHeader(String name, String value)
  {
    List<String> values = _headers.get(name);

    if (values == null) {
      values = new ArrayList<String>();
      _headers.put(name, values);
    }

    values.add(value);
  }

  public String toString()
  {
    StringBuilder sb = new StringBuilder("Attachment[");

    for (Map.Entry<String,List<String>> entry : _headers.entrySet()) {
      for (String value : entry.getValue()) {
        sb.append(entry.getKey());
        sb.append(" -> ");
        sb.append(value);
        sb.append(",");
      }
    }

    sb.append("contents=");
    sb.append(new String(_contents));
    sb.append("]");

    return sb.toString();
  }
}
