/*! * 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 (c) 2002-2013 Pentaho Corporation.. All rights reserved. */ package org.pentaho.platform.plugin.condition.scriptable; import org.apache.commons.logging.Log; import org.pentaho.commons.connection.IPentahoResultSet; import org.pentaho.platform.api.engine.IActionParameter; import org.pentaho.platform.api.engine.IConditionalExecution; import org.pentaho.platform.plugin.action.messages.Messages; import javax.script.ScriptEngine; import javax.script.ScriptEngineFactory; import javax.script.ScriptEngineManager; import java.util.Iterator; import java.util.List; import java.util.Map; public class ScriptableCondition implements IConditionalExecution { private String script; private String scriptLanguage = "JavaScript"; private boolean defaultResult = true; private boolean ignoreInputNamesWithMinus = false; // Backward compatibility with old ConditionalExecution private boolean listAvailableEngines = false; public ScriptableCondition() { super(); } public String getScript() { return script; } public void setScript( final String script ) { this.script = script; } public void setScriptLanguage( final String value ) { this.scriptLanguage = value; } public void setDefaultResult( final boolean value ) { this.defaultResult = value; } public boolean getDefaultResult() { return this.defaultResult; } public String getScriptLanguage() { return this.scriptLanguage; } public void setIgnoreInputNamesWithMinus( final boolean value ) { this.ignoreInputNamesWithMinus = value; } public boolean getIgnoreInputNamesWithMinus() { return this.ignoreInputNamesWithMinus; } public void setListAvailableEngines( final boolean value ) { this.listAvailableEngines = value; if ( value ) { System.out.println( "*** DEBUG - Display Script Engine List ***" ); ScriptEngineManager manager = new ScriptEngineManager(); List<ScriptEngineFactory> factories = manager.getEngineFactories(); for ( ScriptEngineFactory factory : factories ) { System.out.println( String.format( "Engine %s, Version %s, Language %s, Registered Names: %s", factory .getEngineName(), factory.getEngineVersion(), factory.getLanguageName(), factory.getNames().toString() ) ); } } } public boolean shouldExecute( final Map currentInputs, final Log logger ) throws Exception { boolean shouldExecute = this.getDefaultResult(); ScriptEngineManager mgr = new ScriptEngineManager(); ScriptEngine engine = mgr.getEngineByName( this.getScriptLanguage() ); if ( engine == null ) { throw new IllegalArgumentException( Messages.getInstance().getErrorString( "ScriptableCondition.ERROR_0001_ENGINE_NOT_AVAILABLE", this.getScriptLanguage() ) ); //$NON-NLS-1$ } Object inputValue; IActionParameter inputParameter; String inputName = null; Iterator inputs = currentInputs.entrySet().iterator(); Map.Entry mapEntry; while ( inputs.hasNext() ) { mapEntry = (Map.Entry) inputs.next(); inputName = (String) mapEntry.getKey(); if ( this.getIgnoreInputNamesWithMinus() && inputName.indexOf( '-' ) >= 0 ) { logger.info( Messages.getInstance().getString( "ScriptableCondition.INFO_IGNORED_INPUT", inputName ) ); //$NON-NLS-1$ continue; } inputParameter = (IActionParameter) mapEntry.getValue(); inputValue = inputParameter.getValue(); engine.put( inputName, inputValue ); // What happens to resultset objects I wonder... } engine.put( "out", System.out ); engine.put( "rule", this ); Object resultObject = engine.eval( this.getScript() ); if ( resultObject instanceof Boolean ) { return ( (Boolean) resultObject ).booleanValue(); } else if ( resultObject instanceof String ) { return ( "true".equalsIgnoreCase( resultObject.toString() ) ) || ( "yes".equalsIgnoreCase( resultObject.toString() ) ); //$NON-NLS-1$ //$NON-NLS-2$ } else if ( resultObject instanceof Number ) { return ( (Number) resultObject ).intValue() > 0; } else if ( resultObject instanceof IPentahoResultSet ) { return ( (IPentahoResultSet) resultObject ).getRowCount() > 0; } logger.info( Messages.getInstance().getString( "ScriptableCondition.INFO_DEFAULT_RESULT_RETURNED" ) ); //$NON-NLS-1$ return shouldExecute; } }