//
   //  ========================================================================
   //  Copyright (c) 1995-2013 Mort Bay Consulting Pty. Ltd.
   //  ------------------------------------------------------------------------
   //  All rights reserved. This program and the accompanying materials
   //  are made available under the terms of the Eclipse Public License v1.0
   //  and Apache License v2.0 which accompanies this distribution.
   //
   //      The Eclipse Public License is available at
  //      http://www.eclipse.org/legal/epl-v10.html
  //
  //      The Apache License v2.0 is available at
  //      http://www.opensource.org/licenses/apache2.0.php
  //
  //  You may elect to redistribute this code under either of these licenses.
  //  ========================================================================
  //
  
  package org.eclipse.jetty.servlet;
  
  import java.util.ArrayList;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  import javax.servlet.RequestDispatcher;
  import javax.servlet.ServletContext;
  import javax.servlet.ServletException;
  import javax.servlet.http.HttpServletRequest;
  
  import org.eclipse.jetty.server.handler.ContextHandler;
  import org.eclipse.jetty.server.handler.ErrorHandler;
  
  /* ------------------------------------------------------------ */
  /** Error Page Error Handler
   *
   * An ErrorHandler that maps exceptions and status codes to URIs for dispatch using
   * the internal ERROR style of dispatch.
   *
   */
  public class ErrorPageErrorHandler extends ErrorHandler implements ErrorHandler.ErrorPageMapper
  {
      public final static String GLOBAL_ERROR_PAGE = "org.eclipse.jetty.server.error_page.global";
  
      protected ServletContext _servletContext;
      private final Map<String,String> _errorPages= new HashMap<String,String>(); // code or exception to URL
      private final List<ErrorCodeRange> _errorPageList=new ArrayList<ErrorCodeRange>(); // list of ErrorCode by range
  
      /* ------------------------------------------------------------ */
      public ErrorPageErrorHandler()
      {}
  
      @Override
      public String getErrorPage(HttpServletRequest request)
      {
          String error_page= null;
          Class<?> exClass= (Class<?>)request.getAttribute(RequestDispatcher.ERROR_EXCEPTION_TYPE);
  
          if (ServletException.class.equals(exClass))
          {
              error_page= (String)_errorPages.get(exClass.getName());
              if (error_page == null)
              {
                  Throwable th= (Throwable)request.getAttribute(RequestDispatcher.ERROR_EXCEPTION);
                  while (th instanceof ServletException)
                      th= ((ServletException)th).getRootCause();
                  if (th != null)
                      exClass= th.getClass();
              }
          }
  
          while (error_page == null && exClass != null )
          {
              error_page= (String)_errorPages.get(exClass.getName());
              exClass= exClass.getSuperclass();
          }
  
          if (error_page == null)
          {
              // look for an exact code match
              Integer code=(Integer)request.getAttribute(RequestDispatcher.ERROR_STATUS_CODE);
              if (code!=null)
              {
                  error_page= (String)_errorPages.get(Integer.toString(code));
  
                  // if still not found
                  if ((error_page == null) && (_errorPageList != null))
                  {
                      // look for an error code range match.
                      for (int i = 0; i < _errorPageList.size(); i++)
                      {
                          ErrorCodeRange errCode = (ErrorCodeRange) _errorPageList.get(i);
                          if (errCode.isInRange(code))
                          {
                              error_page = errCode.getUri();
                              break;
                          }
                      }
                  }
             }
         }
 
         //try new servlet 3.0 global error page
         if (error_page == null)
         {
             error_page = _errorPages.get(GLOBAL_ERROR_PAGE);
         }
 
         return error_page;
     }
 
     /* ------------------------------------------------------------ */
     /**
      * @return Returns the errorPages.
      */
     public Map<String,String> getErrorPages()
     {
         return _errorPages;
     }
 
     /* ------------------------------------------------------------ */
     /**
      * @param errorPages The errorPages to set. A map of Exception class name  or error code as a string to URI string
      */
     public void setErrorPages(Map<String,String> errorPages)
     {
         _errorPages.clear();
         if (errorPages!=null)
             _errorPages.putAll(errorPages);
     }
 
     /* ------------------------------------------------------------ */
     /** Add Error Page mapping for an exception class
      * This method is called as a result of an exception-type element in a web.xml file
      * or may be called directly
      * @param exception The exception
      * @param uri The URI of the error page.
      */
     public void addErrorPage(Class<? extends Throwable> exception,String uri)
     {
         _errorPages.put(exception.getName(),uri);
     }
 
     /* ------------------------------------------------------------ */
     /** Add Error Page mapping for an exception class
      * This method is called as a result of an exception-type element in a web.xml file
      * or may be called directly
      * @param exceptionClassName The exception
      * @param uri The URI of the error page.
      */
     public void addErrorPage(String exceptionClassName,String uri)
     {
         _errorPages.put(exceptionClassName,uri);
     }
 
     /* ------------------------------------------------------------ */
     /** Add Error Page mapping for a status code.
      * This method is called as a result of an error-code element in a web.xml file
      * or may be called directly
      * @param code The HTTP status code to match
      * @param uri The URI of the error page.
      */
     public void addErrorPage(int code,String uri)
     {
         _errorPages.put(Integer.toString(code),uri);
     }
 
     /* ------------------------------------------------------------ */
     /** Add Error Page mapping for a status code range.
      * This method is not available from web.xml and must be called
      * directly.
      * @param from The lowest matching status code
      * @param to The highest matching status code
      * @param uri The URI of the error page.
      */
     public void addErrorPage(int from, int to, String uri)
     {
         _errorPageList.add(new ErrorCodeRange(from, to, uri));
     }
 
     /* ------------------------------------------------------------ */
     @Override
     protected void doStart() throws Exception
     {
         super.doStart();
         _servletContext=ContextHandler.getCurrentContext();
     }
 
     /* ------------------------------------------------------------ */
     /* ------------------------------------------------------------ */
     private class ErrorCodeRange
     {
         private int _from;
         private int _to;
         private String _uri;
 
         ErrorCodeRange(int from, int to, String uri)
             throws IllegalArgumentException
         {
             if (from > to)
                 throw new IllegalArgumentException("from>to");
 
             _from = from;
             _to = to;
             _uri = uri;
         }
 
         boolean isInRange(int value)
         {
             if ((value >= _from) && (value <= _to))
             {
                 return true;
             }
 
             return false;
         }
 
         String getUri()
         {
             return _uri;
         }
 
         @Override
         public String toString()
         {
             return "from: " + _from + ",to: " + _to + ",uri: " + _uri;
         }
     }
 }