FunctionCallWord.java

/**
 * $Id: $
 * $Date: $
 *
 */

package org.xmlsh.sh.core;

import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.io.Writer;
import java.util.ArrayList;
import java.util.List;

import org.xmlsh.core.CommandFactory;
import org.xmlsh.core.CoreException;
import org.xmlsh.core.IFunction;
import org.xmlsh.core.InvalidArgumentException;
import org.xmlsh.core.Variables;
import org.xmlsh.core.XValue;
import org.xmlsh.sh.shell.Shell;
import org.xmlsh.util.MutableInteger;
import org.xmlsh.util.Util;

/*
 * A Value that evaulates to a "cmd_word" which is either a simple string,
 * or a subprocess expression 
 * 
 */
public class FunctionCallWord extends Word {
	String		mFunction;
	WordList	mArgs ;
	
	public FunctionCallWord( String func , WordList args )
	{
		mFunction = func;
		mArgs = args ;

	}
	
	public void print( PrintWriter out )
	{
		out.print( mFunction );
		out.print("(");
		if( mArgs != null )
			mArgs.print(out);
		out.print(")");
		
		
		
	}	
	


	
	@Override
	public XValue expand(Shell shell,boolean bExpandWild , boolean bExpandWords, boolean bTongs , MutableInteger retValue, SourceLocation loc ) throws IOException, CoreException {
		
		
		// Try builtin functions first
		IFunction func = CommandFactory.getInstance().getBuiltinFunction(shell, mFunction ,  loc );

		
		if( func == null )
			func = shell.getFunction(mFunction);
		
		
		if( func == null )
			throw new InvalidArgumentException("Unknown function: " + mFunction);
		

		List<XValue> saveArgs = shell.getArgs();
		String saveArg0 = shell.getArg0();

		Variables save_vars = shell.pushLocalVars();


		ArrayList<XValue>	args = new ArrayList<XValue>();
		
		if( mArgs != null )
			for( Word arg : mArgs )
				args.addAll(arg.expand(shell,arg.isExpand(),arg.isExpand(),arg.isExpand(), !arg.isExpand(),loc));

		
		try {
			
			shell.setArg0(mFunction);
			shell.setArgs(args);
			int ret =	shell.exec(func.getBody(), loc);
			return shell.getReturnValue(true);

		} catch (Exception e) {
			if( e instanceof CoreException )
				throw (CoreException)e ;
			if( e instanceof IOException )
				throw (IOException)e;
			throw new CoreException(e);
		} finally {
			shell.popLocalVars(save_vars);
			
			shell.setArg0(saveArg0);
			shell.setArgs(saveArgs);
		}
		
		
		
	}

	
	public boolean isEmpty() {
		return Util.isEmpty(mFunction);
	}
	
	public String toString()
	{
		Writer sw = new StringWriter();
		PrintWriter pw = new PrintWriter(sw);
		print(pw);
		pw.flush();
		return sw.toString();
	}
}



//
//
//Copyright (C) 2008-2014    David A. Lee.
//
//The contents of this file are subject to the "Simplified BSD License" (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.opensource.org/licenses/bsd-license.php 
//
//Software distributed under the License is distributed on an "AS IS" basis,
//WITHOUT WARRANTY OF ANY KIND, either express or implied.
//See the License for the specific language governing rights and limitations under the License.
//
//The Original Code is: all this file.
//
//The Initial Developer of the Original Code is David A. Lee
//
//Portions created by (your name) are Copyright (C) (your legal entity). All Rights Reserved.
//
//Contributor(s): none.
//