HttpMimeTypeListener.java

/*
 * This program is free software; you can redistribute it and/or modify it under the 
 * terms of the GNU Lesser General Public License, version 2.1 as published by the Free Software 
 * Foundation.
 *
 * You should have received a copy of the GNU Lesser General Public License along with this 
 * program; if not, you can obtain a copy at http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html 
 * or from the Free Software Foundation, Inc., 
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * This program 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.
 *
 * Copyright 2005 - 2009 Pentaho Corporation.  All rights reserved.
 *
*/
package org.pentaho.platform.web.servlet;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.pentaho.platform.api.engine.IMimeTypeListener;
import org.pentaho.platform.util.messages.LocaleHelper;
import org.pentaho.platform.util.web.MimeHelper;

public class HttpMimeTypeListener implements IMimeTypeListener {

  private HttpServletResponse response;

  private HttpServletRequest request;

  private String title;

  public HttpMimeTypeListener(final HttpServletRequest request, final HttpServletResponse response) {
    this(request, response, null);
  }

  public HttpMimeTypeListener(final HttpServletRequest request, final HttpServletResponse response, final String title) {
    this.request = request;
    this.response = response;
    this.title = title;
  }

  public void setName( String name ) {
	  title = name;
  }

  public void setMimeType(final String mimeType) {
    response.setContentType(mimeType);
    response.setCharacterEncoding(LocaleHelper.getSystemEncoding());
    //
    // Since this is an HTTP thing, allow users to override default behavior by allowing them
    // the mime object as an attachment. Forcing everything non-html to be an attachment
    // means that users can't have their excel reports appear in their IE browser, it makes
    // SVG not work unless you have an external SVG viewer, it makes PDF files no longer
    // be viewable in the browser. Please see PLATFORM-1018.
    // also see: http://support.microsoft.com/kb/260519 for info on content-disposition header
    //
    String forceAttachment = request.getParameter("forceAttachment"); //$NON-NLS-1$
    String fileName = title;
    if (fileName == null) {
      fileName = "default"; //$NON-NLS-1$
    }
    fileName = fileName.replace(' ', '_');
    fileName += MimeHelper.getExtension(mimeType);
    if ((forceAttachment != null) && ("true".equalsIgnoreCase(forceAttachment))) { //$NON-NLS-1$
      response.setHeader("content-disposition", "attachment;filename=" + fileName); //$NON-NLS-1$ //$NON-NLS-2$
    } else {
      response.setHeader("content-disposition", "inline;filename=" + fileName); //$NON-NLS-1$ //$NON-NLS-2$
    }
  }

}