Application: gvSIG desktop

package org.gvsig.expressionevaluator.impl.function.programming;

import java.util.ArrayList;

import java.util.Collection;

import java.util.Collections;

import java.util.List;

import java.util.Objects;

import org.apache.commons.lang3.Range;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.Codes;

import org.gvsig.expressionevaluator.Formatter;

import org.gvsig.expressionevaluator.Function;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.MutableSymbolTable;

import org.gvsig.expressionevaluator.SymbolTable;

import org.gvsig.expressionevaluator.impl.DefaultInterpreter;

import org.gvsig.expressionevaluator.impl.HostExpressionUtils;

import org.gvsig.expressionevaluator.impl.function.programming.BreakFunction.BreakException;

import org.gvsig.expressionevaluator.impl.function.programming.ReturnFunction.ReturnException;

import org.gvsig.expressionevaluator.spi.AbstractFunction;

public class CodeBlockWithExceptFunction extends AbstractFunction {

    public static final String NAME = "BLOCK_EXCEPT";

    public CodeBlockWithExceptFunction() {

        super(Function.GROUP_PROGRAMMING, 

                NAME, 

                Range.between(1,Integer.MAX_VALUE),

                "Evaluate each of the arguments sequentially.",

                NAME+"( {{arguments...}} )",

                null,

                "Object",

                false

        );

    }

    @Override

    public boolean isHidden() {

      return true;

    }

    @Override

    public boolean useArgumentsInsteadObjects() {

        return true;

    }

    @Override

    public boolean allowConstantFolding() {

        return false;

    }

    @Override

    public Object call(Interpreter interpreter, Object[] args) throws Exception {

        throw new UnsupportedOperationException("Not supported yet.");

    }

    @Override

    public Object call(Interpreter interpreter, Codes args) throws Exception {

        Object value = null;

        Code exceptionCode = args.get(args.size()-2);

        Code declareCode = args.get(args.size()-1);

        if( declareCode != null ) {

            try {

                value = ((DefaultInterpreter)interpreter).runCode(declareCode);

//            } catch(BreakException|ReturnException ex) {

//                throw ex;

            } catch(Exception ex) {

                if( exceptionCode==null ) {

                    LOGGER.warn("Error in DECLARE calling "+ code2string(declareCode), ex);

                    throw ex;

                } else {                    

                    value = ((DefaultInterpreter)interpreter).runCode(exceptionCode);

                }

            }

        }

        for (int argn = 0; argn < args.size()-2; argn++) {

            Code statement = args.get(argn);

            // LLamamos a runCode para que no atrape los returns.

            try {

                statement = HostExpressionUtils.resolveHostExpressions(statement, interpreter);

                value = ((DefaultInterpreter)interpreter).runCode(statement);

            } catch(BreakException|ReturnException ex) {

                throw ex;

            } catch(Exception ex) {

                if( exceptionCode==null ) {

                    LOGGER.warn("Error in BEGIN/END calling line "+ argn + ", "+ code2string(statement), ex);

                    throw ex;

                } else {     

                    SymbolTable symbolTable = interpreter.getSymbolTable();

                    if( symbolTable instanceof MutableSymbolTable ) {

                        ((MutableSymbolTable) symbolTable).setVar("@ErrorMessage", ex.getMessage());

                        ((MutableSymbolTable) symbolTable).setVar("@ErrorStatement", getErrorStatement(statement));

                        ((MutableSymbolTable) symbolTable).setVar("@ErrorContext", this.getErrorContext(symbolTable));

                    }

                    value = ((DefaultInterpreter)interpreter).runCode(exceptionCode);

                }

            }

        }

        return value;

    }

    private String getErrorContext(SymbolTable symbolTable) {

        try {

            StringBuilder builder = new StringBuilder();

            List<String> names = new ArrayList<>(symbolTable.variables());

            Collections.sort(names);

            for (String name : names) {

                if( StringUtils.equalsIgnoreCase("@ErrorContext", name) ) {

                    continue;

                }

                String value = "##ERROR##";

                try {

                    value = Objects.toString(symbolTable.value(name));

                } catch(Throwable t) {

                    LOGGER.debug("Can't retrieve value for '"+name+"'.",t);

                }

                builder.append(name);

                builder.append("=");

                builder.append(value);

                builder.append(";\n");

            }

            return builder.toString();

        } catch(Throwable t) {

            return "##ERROR##";

        }

    }

    private String getErrorStatement(Code statement) {

        try { 

            return statement.toString(); 

        } catch(Throwable t) { 

            return "Unknown";

        }

    }

    private String code2string(Code code) {

        try {

            return Objects.toString(code,"");

        } catch(Exception ex2) {

            return "unknown";

        }

    }

    @Override

    public String toString(Codes args, Formatter<Code> formatter) {

        Code exceptionCode = args.get(args.size()-2);

        Code declareCode = args.get(args.size()-1);

        StringBuilder builder = new StringBuilder();

        if( declareCode != null ) {

            builder.append("DECLARE ");

            builder.append(declareCode.toString(formatter));

        }

        builder.append("BEGIN ");

        for (int argn = 0; argn < args.size()-2; argn++) {

            Code statement = args.get(argn);

            builder.append(statement.toString(formatter));

            builder.append("; ");

        }

        if( exceptionCode != null ) {

            builder.append("EXCEPTION ");

            builder.append(exceptionCode.toString(formatter));

            builder.append(" ");

        }

        builder.append("END ");

        return builder.toString();

    }

}