/* $Id: MultipartWrapper.java 988245 2010-08-23 18:39:35Z kwright $ */

/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.apache.manifoldcf.ui.multipart;

import org.apache.manifoldcf.core.interfaces.*;
import org.apache.manifoldcf.ui.beans.AdminProfile;
import org.apache.commons.fileupload2.core.FileUploadException;
import org.apache.commons.fileupload2.core.DiskFileItem;
import org.apache.commons.fileupload2.jakarta.servlet6.JakartaServletDiskFileUpload;
import org.apache.commons.fileupload2.core.DiskFileItemFactory;
import jakarta.servlet.*;
import jakarta.servlet.http.*;
import java.util.*;
import java.io.*;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;

/** This class provides abstract parameter service, including support for uploaded files and
* multipart forms.  It is styled much like HttpServletRequest, but wraps this interface so
* that code can access either standard post data or multipart data transparently.
*/
public class MultipartWrapper implements IPostParameters
{
  public static final String _rcsid = "@(#)$Id: MultipartWrapper.java 988245 2010-08-23 18:39:35Z kwright $";

  /** The Admin Profile bean, for password mapping. */
  protected final AdminProfile adminProfile;
  /** This is the HttpServletRequest object, which will be used for parameters only if
  * the form is not multipart. */
  protected HttpServletRequest request = null;
  protected Map variableMap = new HashMap();
  protected String characterEncoding = null;

  /** Constructor.
  */
  public MultipartWrapper(HttpServletRequest request, AdminProfile adminProfile)
    throws ManifoldCFException
  {
    // First of all make sure to have a correct charset
    if(request.getCharacterEncoding() == null) {
      try {
        request.setCharacterEncoding("UTF-8");
      } catch (UnsupportedEncodingException e) {
        throw new ManifoldCFException(e.getMessage(), e);
      }
    }
    
    this.adminProfile = adminProfile;

    // Check that we have a file upload request
    boolean isMultipart = JakartaServletDiskFileUpload.isMultipartContent(request);
    if (!isMultipart)
    {
      this.request = request;
      return;
    }

    // It is multipart!

    // Create a factory for disk-based file items
    DiskFileItemFactory factory = DiskFileItemFactory.builder().get();

    // Create a new file upload handler
    JakartaServletDiskFileUpload upload = new JakartaServletDiskFileUpload(factory);

    // Parse the request
    try
    {
      characterEncoding = request.getCharacterEncoding();
      // Handle broken tomcat; characterEncoding always comes back null for tomcat4
      if (characterEncoding == null)
        characterEncoding = "utf8";
      List<DiskFileItem> items = upload.parseRequest(request);
      for (DiskFileItem item : items)
      {
        String name = item.getFieldName();
        ArrayList list = (ArrayList)variableMap.get(name);
        if (list == null)
        {
          list = new ArrayList();
          variableMap.put(name,list);
        }
        else
        {
          if (((DiskFileItem)list.get(0)).isFormField() != item.isFormField())
            throw new ManifoldCFException("Illegal form data; posted form has the same name for different data types ('"+name+"')!");
        }
        list.add(item);
      }
    }
    catch (IOException e)
    {
      throw new ManifoldCFException("Problem reading multipart request: "+e.getMessage(),e);
    }
  }

  /** Get multiple parameter values.
  */
  @Override
  public String[] getParameterValues(String name)
  {
    // Expect multiple items, all strings
    ArrayList list = (ArrayList)variableMap.get(name);
    if (list == null)
    {
      if (request != null)
        return request.getParameterValues(name);
      return null;
    }

    Object x = list.get(0);

    if ((x instanceof DiskFileItem) && !((DiskFileItem)x).isFormField())
      return null;

    String[] rval = new String[list.size()];
    int i = 0;
    while (i < rval.length)
    {
      x = list.get(i);
      if (x instanceof String)
        rval[i] = (String)x;
      else
      {
        try
        {
          rval[i] = ((DiskFileItem)x).getString(Charset.forName(characterEncoding));
        }
        catch (IOException e)
        {
          throw new RuntimeException("Error reading multipart parameter: "+e.getMessage(),e);
        }
      }
      i++;
    }
    return rval;
  }

  /** Get single parameter value.
  */
  @Override
  public String getParameter(String name)
  {
    // Get it as a parameter.
    ArrayList list = (ArrayList)variableMap.get(name);
    if (list == null)
    {
      if (request != null)
        return request.getParameter(name);
      return null;
    }

    Object x = list.get(0);
    if (x instanceof String)
      return (String)x;

    DiskFileItem item = (DiskFileItem)x;
    if (!item.isFormField())
      return null;

    try
    {
      return item.getString(Charset.forName(characterEncoding));
    }
    catch (IOException e)
    {
      throw new RuntimeException("Error reading multipart parameter: "+e.getMessage(),e);
    }
  }

  /** Get a file parameter, as a binary input.
  */
  @Override
  public BinaryInput getBinaryStream(String name)
    throws ManifoldCFException
  {
    if (request != null)
      return null;

    ArrayList list = (ArrayList)variableMap.get(name);
    if (list == null)
      return null;

    Object x = list.get(0);
    if (x instanceof String)
      return null;

    DiskFileItem item = (DiskFileItem)x;
    if (item.isFormField())
      return null;

    try
    {
      InputStream uploadedStream = item.getInputStream();
      try
      {
        return new TempFileInput(uploadedStream);
      }
      finally
      {
        uploadedStream.close();
      }
    }
    catch (IOException e)
    {
      throw new ManifoldCFException("Error creating file binary stream",e);
    }
  }

  /** Get file parameter, as a byte array.
  */
  @Override
  public byte[] getBinaryBytes(String name)
  {
    if (request != null)
      return null;

    ArrayList list = (ArrayList)variableMap.get(name);
    if (list == null)
      return null;

    Object x = list.get(0);
    if (x instanceof String)
      return null;

    DiskFileItem item = (DiskFileItem)x;
    if (item.isFormField())
      return null;

    return item.get();
  }

  /** Set a parameter value
  */
  @Override
  public void setParameter(String name, String value)
  {
    ArrayList values = new ArrayList();
    if (value != null)
      values.add(value);
    variableMap.put(name,values);
  }

  /** Set an array of parameter values
  */
  @Override
  public void setParameterValues(String name, String[] values)
  {
    ArrayList valueArray = new ArrayList();
    int i = 0;
    while (i < values.length)
    {
      valueArray.add(values[i++]);
    }
    variableMap.put(name,valueArray);
  }

  // Password mapping
  
  /** Map a password to a unique key.
  * This method works within a specific given browser session to replace an existing password with
  * a key which can be used to look up the password at a later time.
  *@param password is the password.
  *@return the key.
  */
  @Override
  public String mapPasswordToKey(String password)
  {
    return adminProfile.getPasswordMapper().mapPasswordToKey(password);
  }
  
  /** Convert a key, created by mapPasswordToKey, back to the original password, within
  * the lifetime of the browser session.  If the provided key is not an actual key, instead
  * the key value is assumed to be a new password value.
  *@param key is the key.
  *@return the password.
  */
  @Override
  public String mapKeyToPassword(String key)
  {
    return adminProfile.getPasswordMapper().mapKeyToPassword(key);
  }
  
}
