/*
    * $Id: DefaultErrorHandler.java,v 1.11 2007/07/04 19:42:48 edankert Exp $
    *
    * The contents of this file are subject to the Mozilla Public License 
    * Version 1.1 (the "License"); you may not use this file except in 
    * compliance with the License. You may obtain a copy of the License at 
    * http://www.mozilla.org/MPL/ 
    *
    * Software distributed under the License is distributed on an "AS IS" basis, 
   * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License 
   * for the specific language governing rights and limitations under the License.
   *
   * The Original Code is XML Hammer code. (org.xmlhammer.*)
   *
   * The Initial Developer of the Original Code is Edwin Dankert. Portions created 
   * by the Initial Developer are Copyright (C) 2005 - 2006 the Initial Developer. 
   * All Rights Reserved.
   *
   * Contributor(s): Edwin Dankert <edankert@gmail.com>
   */
  package org.xmlhammer;
  
  import java.io.IOException;
  import java.net.URI;
  import java.util.ArrayList;
  
  import javax.xml.transform.ErrorListener;
  import javax.xml.transform.SourceLocator;
  import javax.xml.transform.TransformerException;
  
  import org.bounce.util.URIUtils;
  import org.xml.sax.ErrorHandler;
  import org.xml.sax.SAXParseException;
  
  public class DefaultErrorHandler implements ErrorHandler, ErrorListener {
      private ArrayList<Problem> problems = null;
      private ArrayList<Exception> exceptions = null;
      private URI systemId = null;
      
      public DefaultErrorHandler(URI systemId) {
          problems = new ArrayList<Problem>();
          exceptions = new ArrayList<Exception>();
          
          this.systemId = systemId;
      }
  
      public void message(String type, String message) {
          problems.add(new Message(type, message));
      }
  
      public void warning(SAXParseException e) {
          if (!exceptions.contains(e) && !exceptions.contains(e.getException())) {
              problems.add(new Warning(e));
              exceptions.add(e);
          }
      }
  
      public void warning(TransformerException e) {
          if (!exceptions.contains(e) && !exceptions.contains(e.getException())) {
              addLocator(e);
              
              problems.add(new Warning(e));
              exceptions.add(e);
          }
      }
  
      public void error(SAXParseException e) {
          if (!exceptions.contains(e) && !exceptions.contains(e.getException())) {
              problems.add(new Error(e));
              exceptions.add(e);
          }
      }
  
      public void error(TransformerException e) {
          if (!exceptions.contains(e) && !exceptions.contains(e.getException())) {
              addLocator(e);
  
              problems.add(new Error(e));
              exceptions.add(e);
          }
      }
  
      public void fatalError(SAXParseException e) {
          if (!exceptions.contains(e) && !exceptions.contains(e.getException())) {
              problems.add(new Fatal(e));
              exceptions.add(e);
          }
      }
      
      public void fatalError(TransformerException e) {
          if (!exceptions.contains(e) && !exceptions.contains(e.getException())) {
              addLocator(e);
  
              problems.add(new Fatal(e));
              exceptions.add(e);
          }
      }
  
      public void fatalError(IOException e) {
         if (!exceptions.contains(e)) {
             problems.add(new Fatal(e));
             exceptions.add(e);
         }
     }
 
     public ArrayList<Problem> getProblems() {
         return problems;
     }
     
     public Exception getFatal() {
         for (Problem problem : problems) {
             if (problem instanceof Fatal) {
                 return problem.getException();
             }
         }
         
         return null;
     }
     
     private void addLocator(TransformerException e){
         if (e.getLocator() == null && systemId != null) {
             e.setLocator(new DummySourceLocator(systemId));
         }
     }
 
 
     public class Warning extends Problem {
         public Warning(SAXParseException exception) {
             super( exception);
         }
 
         public Warning(TransformerException exception) {
             super( exception);
         }
     }
 
     public class Error extends Problem {
         public Error(SAXParseException exception) {
             super( exception);
         }
 
         public Error(TransformerException exception) {
             super( exception);
         }
     }
 
     public class Fatal extends Problem {
         public Fatal(SAXParseException exception) {
             super( exception);
         }
 
         public Fatal(TransformerException exception) {
             super( exception);
         }
 
         public Fatal(IOException exception) {
             super( exception);
         }
     }
 
     public class Problem {
         Exception exception = null;
         
         public Problem(Exception exception) {
             this.exception = exception;
         }
         
         public Exception getException() {
             return exception;
         }
     }
 
     public class Message extends Problem {
         String message = null;
         String type = null;
         
         public Message(String type, String message) {
             super(null);
             this.type = type;
             this.message = message;
         }
         
         public String getType() {
             return type;
         }
 
         public String getMessage() {
             return message;
         }
     }
     
     private class DummySourceLocator implements SourceLocator {
         private URI uri = null;
         
         public DummySourceLocator(URI uri) {
             this.uri = uri;
         }
 
         public String getPublicId() {
             return null;
         }
 
         public String getSystemId() {
             return URIUtils.toString(uri);
         }
 
         public int getLineNumber() {
             return -1;
         }
 
         public int getColumnNumber() {
             return -1;
         }
         
     }
 }