AbstractGenerateStreamContentComponent.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 2006 - 2009 Pentaho Corporation.  All rights reserved.
 *
 */
package org.pentaho.platform.plugin.action.jfreereport.outputs;

import java.io.OutputStream;

import org.pentaho.platform.api.repository.IContentItem;
import org.pentaho.platform.plugin.action.jfreereport.AbstractJFreeReportComponent;
import org.pentaho.platform.plugin.action.messages.Messages;
import org.pentaho.reporting.engine.classic.core.MasterReport;

/**
 * Creation-Date: 07.07.2006, 20:50:22
 * 
 * @author Thomas Morgner
 */
public abstract class AbstractGenerateStreamContentComponent extends AbstractGenerateContentComponent {

  private static final long serialVersionUID = -4562767913928444314L;
  private IContentItem contentItem = null;

  protected AbstractGenerateStreamContentComponent() {
  }

  @Override
  protected boolean validateAction() {
    if ( !(super.validateAction()) ) {
      return false;
    }

    if (isDefinedOutput(AbstractJFreeReportComponent.REPORTGENERATESTREAM_REPORT_OUTPUT)) {
      return true;
    }

    if (getOutputNames().size() == 1) {
      return true;
    }

    if (getOutputNames().size() == 0) {
      warn(Messages.getInstance().getString("Base.WARN_NO_OUTPUT_STREAM")); //$NON-NLS-1$
      return true;
    }

    warn(Messages.getInstance().getString("AbstractGenerateStreamContentComponent.JFreeReport.ERROR_0038_NO_OUTPUT_DEFINED")); //$NON-NLS-1$
    return false;
  }

  protected abstract String getMimeType();

  protected abstract String getExtension();

  @Override
  protected final boolean performExport(final MasterReport report) {
    OutputStream outputStream = createOutputStream();
    if (outputStream == null) {
      // We could not get an output stream for the content
      error(Messages.getInstance().getErrorString("JFreeReport.ERROR_0008_INVALID_OUTPUT_STREAM")); //$NON-NLS-1$
      return false;
    }

    return performExport(report, outputStream);
  }

  protected final void close() {
    if (contentItem != null) {
      contentItem.closeOutputStream();
    }
  }

  protected abstract boolean performExport(final MasterReport report, final OutputStream outputStream);

  @SuppressWarnings("deprecation")
  protected OutputStream createOutputStream() {
    // Try to get the output from the action-sequence document.
    final String mimeType = getMimeType();

    if (isDefinedOutput(AbstractJFreeReportComponent.REPORTGENERATESTREAM_REPORT_OUTPUT)) {
      contentItem = getOutputItem(AbstractJFreeReportComponent.REPORTGENERATESTREAM_REPORT_OUTPUT, mimeType,
          getExtension());
      try {
        contentItem.setMimeType(mimeType);
        return contentItem.getOutputStream(getActionName());
      } catch (Exception e) {
        return null;
      }
    } else if (getOutputNames().size() == 1) {
      String outputName = (String) getOutputNames().iterator().next();
      contentItem = getOutputContentItem(outputName, mimeType);
      try {
        contentItem.setMimeType(mimeType);
        return contentItem.getOutputStream(getActionName());
      } catch (Exception e) {
        return null;
      }
    }
    if (getOutputNames().size() == 0) {
      // There was no output in the action-sequence document, so make a
      // default
      // outputStream.
      final OutputStream outputStream = getDefaultOutputStream(mimeType);
      return outputStream;
    }

    return null;
  }

  protected IContentItem getContentItem() {
    return contentItem;
  }

}