content: /**
   * Copyright (c) 2013, Andre Steingress
   * All rights reserved.
   *
   * Redistribution and use in source and binary forms, with or without modification, are permitted provided that the
   * following conditions are met:
   *
   * 1.) Redistributions of source code must retain the above copyright notice, this list of conditions and the following
   * disclaimer.
   * 2.) Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following
   * disclaimer in the documentation and/or other materials provided with the distribution.
   * 3.) Neither the name of Andre Steingress nor the names of its contributors may be used to endorse or
   * promote products derived from this software without specific prior written permission.
   *
   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
   * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
   * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
   * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
   * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
   * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
   * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
   */
  package org.gcontracts.generation;
  
  import org.codehaus.groovy.GroovyBugError;
  import org.codehaus.groovy.ast.ClassHelper;
  import org.codehaus.groovy.ast.ClassNode;
  import org.codehaus.groovy.ast.Parameter;
  import org.codehaus.groovy.ast.expr.*;
  import org.codehaus.groovy.ast.stmt.*;
  import org.codehaus.groovy.syntax.Token;
  import org.codehaus.groovy.syntax.Types;
  
  /**
   * Creates a try-catch block around a given {@link org.codehaus.groovy.ast.stmt.AssertStatement} and catches
   * a PowerAssertionError to reuse the generated visual output.
   *
   * @author ast
   */
  public class TryCatchBlockGenerator {
  
      public static BlockStatement generateTryCatchBlockForInlineMode(final ClassNode assertionErrorClass, final String message, final Statement assertStatement)  {
  
          final Class powerAssertionErrorClass = loadPowerAssertionErrorClass();
  
          if (powerAssertionErrorClass == null) throw new GroovyBugError("GContracts >= 1.1.2 needs Groovy 1.7 or above!");
  
          VariableExpression newErrorVariableExpression = new VariableExpression("newError", assertionErrorClass);
          newErrorVariableExpression.setAccessedVariable(newErrorVariableExpression);
  
          ExpressionStatement expr = new ExpressionStatement(new DeclarationExpression(newErrorVariableExpression, Token.newSymbol(Types.ASSIGN, -1, -1),
                  new ConstructorCallExpression(assertionErrorClass,
                          new ArgumentListExpression(new BinaryExpression(new ConstantExpression(message), Token.newSymbol(Types.PLUS, -1, -1), new MethodCallExpression(new VariableExpression(new Parameter(ClassHelper.makeWithoutCaching(powerAssertionErrorClass), "error")), "getMessage", ArgumentListExpression.EMPTY_ARGUMENTS))))));
  
          ExpressionStatement exp2 = new ExpressionStatement(new MethodCallExpression(newErrorVariableExpression, "setStackTrace", new ArgumentListExpression(
                  new MethodCallExpression(new VariableExpression(new Parameter(ClassHelper.makeWithoutCaching(powerAssertionErrorClass), "error")), "getStackTrace", ArgumentListExpression.EMPTY_ARGUMENTS)
          )));
  
  
          final BlockStatement catchBlock = new BlockStatement();
          catchBlock.addStatement(expr);
          catchBlock.addStatement(exp2);
          catchBlock.addStatement(new ThrowStatement(newErrorVariableExpression));
  
          final TryCatchStatement tryCatchStatement = new TryCatchStatement(assertStatement, EmptyStatement.INSTANCE);
          tryCatchStatement.addCatch(new CatchStatement(new Parameter(ClassHelper.makeWithoutCaching(powerAssertionErrorClass), "error"), catchBlock));
  
          final BlockStatement assertBlockStatement = new BlockStatement();
          assertBlockStatement.addStatement(tryCatchStatement);
  
          return assertBlockStatement;
      }
  
      public static BlockStatement generateTryCatchBlock(final ClassNode assertionErrorClass, final String message, final Statement assertStatement)  {
  
          final String $_gc_closure_result = "$_gc_closure_result";
  
          final VariableExpression variableExpression = new VariableExpression($_gc_closure_result, ClassHelper.Boolean_TYPE);
          variableExpression.setAccessedVariable(variableExpression);
  
          // if the assert statement is successful the return variable will be true else false
          final BlockStatement overallBlock = new BlockStatement();
          overallBlock.addStatement(new ExpressionStatement(new DeclarationExpression(variableExpression, Token.newSymbol(Types.ASSIGN, -1, -1), ConstantExpression.FALSE)));
  
          final BlockStatement assertBlockStatement = new BlockStatement();
          assertBlockStatement.addStatement(assertStatement);
          assertBlockStatement.addStatement(new ExpressionStatement(new BinaryExpression(variableExpression, Token.newSymbol(Types.ASSIGN, -1, -1), ConstantExpression.TRUE)));
  
          final Class powerAssertionErrorClass = loadPowerAssertionErrorClass();
  
          if (powerAssertionErrorClass == null) throw new GroovyBugError("GContracts >= 1.1.2 needs Groovy 1.7 or above!");
  
          VariableExpression newErrorVariableExpression = new VariableExpression("newError", assertionErrorClass);
          newErrorVariableExpression.setAccessedVariable(newErrorVariableExpression);
  
          ExpressionStatement expr = new ExpressionStatement(new DeclarationExpression(newErrorVariableExpression, Token.newSymbol(Types.ASSIGN, -1, -1),
                  new ConstructorCallExpression(assertionErrorClass,
                          new ArgumentListExpression(new BinaryExpression(new ConstantExpression(message), Token.newSymbol(Types.PLUS, -1, -1), new MethodCallExpression(new VariableExpression(new Parameter(ClassHelper.makeWithoutCaching(powerAssertionErrorClass), "error")), "getMessage", ArgumentListExpression.EMPTY_ARGUMENTS))))));
  
          ExpressionStatement exp2 = new ExpressionStatement(new MethodCallExpression(newErrorVariableExpression, "setStackTrace", new ArgumentListExpression(
                  new MethodCallExpression(new VariableExpression(new Parameter(ClassHelper.makeWithoutCaching(powerAssertionErrorClass), "error")), "getStackTrace", ArgumentListExpression.EMPTY_ARGUMENTS)
          )));
  
  
          final BlockStatement catchBlock = new BlockStatement();
          catchBlock.addStatement(expr);
          catchBlock.addStatement(exp2);
  
          final TryCatchStatement tryCatchStatement = new TryCatchStatement(assertBlockStatement, EmptyStatement.INSTANCE);
          tryCatchStatement.addCatch(new CatchStatement(new Parameter(ClassHelper.makeWithoutCaching(powerAssertionErrorClass), "error"), catchBlock));
  
          overallBlock.addStatement(tryCatchStatement);
          overallBlock.addStatement(new ReturnStatement(variableExpression));
  
          return overallBlock;
      }
  
      private static Class loadPowerAssertionErrorClass() {
  
          Class result = null;
  
          try {
              result = TryCatchBlockGenerator.class.getClassLoader().loadClass("org.codehaus.groovy.transform.powerassert.PowerAssertionError");
          } catch (ClassNotFoundException e) {
              try {
                  result = TryCatchBlockGenerator.class.getClassLoader().loadClass("org.codehaus.groovy.runtime.powerassert.PowerAssertionError");
              } catch (ClassNotFoundException e1) {
              }
          }
  
          return result;
      }
  }
  