Application: gvSIG desktop

package org.gvsig.expressionevaluator.impl;

import java.net.URL;

import java.util.ArrayList;

import java.util.Collection;

import java.util.List;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.ExpressionEvaluatorLocator;

import org.gvsig.expressionevaluator.ExpressionEvaluatorManager;

import org.gvsig.expressionevaluator.ExpressionRuntimeException;

import org.gvsig.expressionevaluator.Function;

import org.gvsig.expressionevaluator.SymbolTable;

import org.gvsig.tools.script.Script;

/**

 *

 * @author jjdelcerro

 */

public class SimpleScript implements Script {

    private final String name;

    private String source;

    private DefaultInterpreter interpreter;

    private List<String> names;

    public SimpleScript(String name, String source) {

        this.name = name;

        this.setCode(source);

    }

    @Override

    public String getName() {

        return this.name;

    }

    @Override

    public String getCode() {

        return this.source;

    }

    @Override

    public String getTypeName() {

        return "COSA";

    }

    @Override

    public URL getURL() {

        return null;

    }

    @Override

    public void put(String name, Object value) {

        DefaultSymbolTable symbolTable = (DefaultSymbolTable) this.interpreter.getSymbolTable();

        symbolTable.setVar(name, value);

    }

    @Override

    public Object invokeFunction(String funcname, Object[] args) throws NoSuchMethodException {

        if( !this.interpreter.hasFunction(funcname) ) {

            throw new NoSuchMethodException(funcname);

        }

        try {

            return this.interpreter.call(funcname, args);

        } catch (Exception ex) {

            throw new ExpressionRuntimeException(this.interpreter.getCurrentCode(),"Error calling funcion '"+funcname+"'.", ex);

        }

    }

    @Override

    public final void setCode(String source) {

        this.source = source;

        this.names  = null;

        ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

        this.interpreter = new DefaultInterpreter();

        Code code = manager.compile(source);

        this.interpreter.run(code);

    }

    @Override

    public List<String> getNames() {

        if( this.names == null ) {

            List<String> theNames = new ArrayList<>();

            SymbolTable symbolTable = this.interpreter.getSymbolTable();

            Collection<Function> funcs = symbolTable.localfunctions();

            if( funcs!=null ) {

                for (Function function : funcs) {

                    theNames.add(function.name());

                }

            }

            Collection<String> vars = symbolTable.localvariables();

            if( vars!=null ) {

                for (String var : vars) {

                    theNames.add(var);

                }

            }

            this.names = theNames;

        }

        return this.names;

    }

}