ActionSequenceParameterMgr.java

/*
 * This program is free software; you can redistribute it and/or modify it under the 
 * terms of the GNU General Public License, version 2 as published by the Free Software 
 * Foundation.
 *
 * You should have received a copy of the GNU General Public License along with this 
 * program; if not, you can obtain a copy at http://www.gnu.org/licenses/gpl-2.0.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 General Public License for more details.
 */
package org.pentaho.platform.engine.services.actionsequence;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;

import org.pentaho.actionsequence.dom.IActionInput;
import org.pentaho.actionsequence.dom.IActionOutput;
import org.pentaho.actionsequence.dom.IActionResource;
import org.pentaho.actionsequence.dom.actions.IActionParameterMgr;
import org.pentaho.commons.connection.IPentahoStreamSource;
import org.pentaho.platform.api.engine.IActionSequenceResource;
import org.pentaho.platform.api.engine.IParameterResolver;
import org.pentaho.platform.api.engine.IPentahoSession;
import org.pentaho.platform.api.engine.IRuntimeContext;
import org.pentaho.platform.api.repository.IContentItem;

public class ActionSequenceParameterMgr implements IActionParameterMgr {

  IRuntimeContext runtimeContext;

  IPentahoSession pentahoSession;

  IParameterResolver parameterResolver;

  public ActionSequenceParameterMgr(final IRuntimeContext runtimeContext, final IPentahoSession session) {
    this(runtimeContext, session, null);
  }

  public ActionSequenceParameterMgr(final IRuntimeContext runtimeContext, final IPentahoSession session,
      final IParameterResolver parameterResolver) {
    this.runtimeContext = runtimeContext;
    pentahoSession = session;
    this.parameterResolver = parameterResolver;
  }

  public Object getInputValue(final IActionInput actionInput) {
    Object value = null;
    String variableName = actionInput.getName();
    if (runtimeContext.getInputNames().contains(variableName)) {
      value = runtimeContext.getInputParameterValue(variableName);
    }
    if (value instanceof IContentItem) {
      value = ((IContentItem) value).getInputStream();
    }
    return value;
  }

  public String replaceParameterReferences(final String inputString) {
    return runtimeContext.applyInputsToFormat(inputString, parameterResolver);
  }

  public IPentahoStreamSource getDataSource(final IActionResource actionResource) throws FileNotFoundException {
    IPentahoStreamSource dataSrc = null;
    IActionSequenceResource resource = runtimeContext.getResourceDefintion(actionResource.getName());
    if (resource != null) {
      dataSrc = runtimeContext.getResourceDataSource(resource);
    }
    return dataSrc;
  }

  public String getString(final IActionResource actionResource) throws IOException {
    String resourceString = null;
    IActionSequenceResource resource = runtimeContext.getResourceDefintion(actionResource.getName());
    if (resource != null) {
      resourceString = runtimeContext.getResourceAsString(resource);
    }
    return resourceString;
  }

  public IPentahoStreamSource getDataSource(final IActionInput actionInput) {
    return runtimeContext.getDataSource(actionInput.getName());
  }

  public void setOutputValue(final IActionOutput actionOutput, final Object value) {
    runtimeContext.setOutputValue(actionOutput.getName(), value);
  }

  public InputStream getInputStream(final IActionResource actionResource) throws FileNotFoundException {
    InputStream inputStream = null;
    IActionSequenceResource resource = runtimeContext.getResourceDefintion(actionResource.getName());
    if (resource != null) {
      inputStream = runtimeContext.getResourceInputStream(resource);
    }
    return inputStream;
  }
}