/*
 * Copyright (c) 1998-2011 Caucho Technology -- all rights reserved
 *
 * This file is part of Resin(R) Open Source
 *
 * Each copy or derived work must preserve the copyright notice and this
 * notice unmodified.
 *
 * Resin Open Source is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * Resin Open Source is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
 * of NON-INFRINGEMENT.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Resin Open Source; if not, write to the
 *
 *   Free Software Foundation, Inc.
 *   59 Temple Place, Suite 330
 *   Boston, MA 02111-1307  USA
 *
 * @author Scott Ferguson
 */

package com.caucho.quercus.statement;

import com.caucho.quercus.Location;
import com.caucho.quercus.QuercusDieException;
import com.caucho.quercus.QuercusException;
import com.caucho.quercus.QuercusExitException;
import com.caucho.quercus.QuercusRuntimeException;
import com.caucho.quercus.env.Env;
import com.caucho.quercus.env.NullValue;
import com.caucho.quercus.env.QuercusLanguageException;
import com.caucho.quercus.env.Value;
import com.caucho.quercus.expr.AbstractVarExpr;

import java.util.ArrayList;

/**
 * Represents sequence of statements.
 */
public class TryStatement extends Statement {
  protected final Statement _block;
  protected final ArrayList<Catch> _catchList = new ArrayList<Catch>();

  public TryStatement(Location location, Statement block)
  {
    super(location);

    _block = block;

    block.setParent(this);
  }

  public void addCatch(String id, AbstractVarExpr lhs, Statement block)
  {
    _catchList.add(new Catch(id, lhs, block));
    
    block.setParent(this);
  }

  public Value execute(Env env)
  {
    try {
      return _block.execute(env);
    } catch (QuercusLanguageException e) {
      Value value = null;
      
      try {
        value = e.toValue(env);
      } catch (Throwable e1) {
        throw new QuercusRuntimeException(e1);
      }

      for (int i = 0; i < _catchList.size(); i++) {
        Catch item = _catchList.get(i);
        
        if (value != null && value.isA(item.getId())
            || item.getId().equals("Exception")) {
          if (value != null)
            item.getExpr().evalAssignValue(env, value);
          else
            item.getExpr().evalAssignValue(env, NullValue.NULL);
              
          return item.getBlock().execute(env);
        }
      }

      throw e;
      
    } catch (QuercusDieException e) {
      for (int i = 0; i < _catchList.size(); i++) {
        Catch item = _catchList.get(i);

        if (item.getId().equals("QuercusDieException")) {
          item.getExpr().evalAssignValue(env, env.createException(e));

          return item.getBlock().execute(env);
        }
      }
      
      throw e;
      
    } catch (QuercusExitException e) {
      for (int i = 0; i < _catchList.size(); i++) {
        Catch item = _catchList.get(i);

        if (item.getId().equals("QuercusExitException")) {
          item.getExpr().evalAssignValue(env, env.createException(e));

          return item.getBlock().execute(env);
        }
      }
      
      throw e;

    } catch (Exception e) {
      for (int i = 0; i < _catchList.size(); i++) {
        Catch item = _catchList.get(i);

        if (item.getId().equals("Exception")) {
          Throwable cause = e;
          
          //if (e instanceof QuercusException && e.getCause() != null)
            //cause = e.getCause();
          
          item.getExpr().evalAssignValue(env, env.createException(cause));

          return item.getBlock().execute(env);
        }
      }
      
      if (e instanceof QuercusException)
        throw (QuercusException) e;
      else
        throw new QuercusException(e);
    }
  }

  public static class Catch {
    private final String _id;
    private final AbstractVarExpr _lhs;
    private final Statement _block;

    Catch(String id, AbstractVarExpr lhs, Statement block)
    {
      _id = id;
      _lhs = lhs;
      _block = block;

      if (id == null)
        throw new NullPointerException();
    }

    public String getId()
    {
      return _id;
    }

    public AbstractVarExpr getExpr()
    {
      return _lhs;
    }

    public Statement getBlock()
    {
      return _block;
    }
  }
}