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;

import org.gvsig.expressionevaluator.PrettyFormatter;

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);

        PrettyFormatter builder = PrettyFormatter.get(formatter);

        try {

            builder.push();

            if( declareCode != null ) {

                builder.append("DECLARE ");

                builder.indent();

                if( Code.isFunction(exceptionCode,"BLOCK") ) {

                    Code.Callable block = (Code.Callable)declareCode;

                    for (Code parameter : block.parameters()) {

                        builder.nl();                

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

                        builder.append("; ");

                    }

                } else {

                    builder.nl();

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

                }

                builder.unindent();

                builder.nl();

            }

            builder.append("BEGIN ");

            builder.indent();

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

                Code statement = args.get(argn);

                builder.nl();

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

                builder.append("; ");

            }

            if( exceptionCode != null ) {

                builder.unindent();

                builder.nl();

                builder.append("EXCEPT ");

                builder.indent();

                if( Code.isFunction(exceptionCode,"BLOCK") ) {

                    Code.Callable block = (Code.Callable)exceptionCode;

                    for (Code parameter : block.parameters()) {

                        builder.nl();

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

                        builder.append("; ");

                    }

                } else {

                    builder.nl();

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

                    builder.append(" ");

                }

            }

            builder.unindent();

            builder.nl();

            builder.append("END ");

            builder.nl();

            return builder.build();

        } finally {

            builder.pop();

        }

    }

}