/*
    * Copyright (c) 2001 Mats Nyberg, Teknik i Media  <mn@tim.se>
    *
    * $Id: HTTPProtocolHandler.java,v 1.1.1.1 2004/05/19 12:26:42 pra Exp $
    *
    * This library is free software; you can redistribute it and/or
    * modify it under the terms of the GNU Lesser General Public
    * License as published by the Free Software Foundation; either
    * version 2 of the License, or (at your option) any later version
   * 
   * This library 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.  See the GNU
   * Lesser General Public License for more details.
   * 
   * You should have received a copy of the GNU Lesser General Public
   * License along with this library; if not, write to the Free Software
   * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
   */
  package org.backsource.amsterdam.plugins.http;
  
  import java.util.*;
  import java.io.*;
  import java.net.URL;
  import HTTPClient.*;
  
  import org.w3c.dom.Element;
  
  import org.backsource.amsterdam.service.protocol.ProtocolHandler;
  import org.backsource.amsterdam.service.protocol.GETProtocolHandler;
  import org.backsource.amsterdam.service.protocol.GETProtocolHandlerSupport;
  import org.backsource.amsterdam.service.protocol.ListEntry;
  import org.backsource.amsterdam.service.protocol.ProtocolConfigure;
  import org.backsource.amsterdam.service.ServiceMessage;
  import org.backsource.amsterdam.service.ServiceInvoker;
  import org.backsource.amsterdam.service.ServiceException;
  
  import HTTPClient.*;
  
  /*** HTTPProtocolHandler - retrieve GET/HTTP resources.
   *
   * @author Mats Nyberg <mn@tim.se>
   * @version $Revision: 1.1.1.1 $
   */
  
  public class HTTPProtocolHandler extends GETProtocolHandlerSupport {
  
      public static final int DEFAULT_TIMEOUT = 3000;
      
      private String host = null;
      private int port = 80;
      private String baseURL = "/";
      private int retryTimes = 0;
      private int retryDelay = 0;
      private int timeOut = 0;
      private String user = null;
      private String pass = null;
      private String scheeme = null;
      private String realm = null;
  
  
  
      public HTTPProtocolHandler(ProtocolConfigure conf)  throws ServiceException{
  	setConfiguration(conf);
      }
  
          
      private int getIntConfigParam(String entry, int defaultValue){
  	try{
  	    
  	    return new Integer(entry).intValue();
  	    
  	}catch(Exception e){}
  	
  	return defaultValue;
      }
      
      
      private void setConfiguration(ProtocolConfigure conf) throws ServiceException {
  		String host = conf.getEntry("Host");
  		String port = conf.getEntry("Port");
  		String baseURL = conf.getEntry("BaseURL");
  		String timeOut = conf.getEntry("TimeOut");
  		String retryTimes = conf.getEntry("RetryTimes");
  		String retryDelay = conf.getEntry("RetryDelay");
  
  		user = conf.getEntry("User");
  		pass = conf.getEntry("Pass");
  		scheeme = conf.getEntry("Scheeme");
  		realm = conf.getEntry("Realm");
  
  		
  		if(host != null)
  			this.host = host;
  		else
  			throw new ServiceException("No host specified");
  		
  		if(port != null)
  			this.port = getIntConfigParam(port, 8080);
 		
 		if(baseURL != null)
 			this.baseURL = baseURL;
 		else
 			throw new ServiceException("No URL specified");
 		
 		if(timeOut != null)
 			this.timeOut = getIntConfigParam(timeOut, DEFAULT_TIMEOUT);
 		
 		if(retryTimes != null)
 			this.retryTimes = getIntConfigParam(retryTimes, 0);
 		
 		if(retryDelay != null)
 			this.retryDelay = getIntConfigParam(retryDelay, 0);
 
     }
     
     
     public void close() {}
 
     public Iterator list(String url) throws ServiceException{
 	invoker.getService().getLogger().debug("Doing list");
 	if(url == null) {
 	    invoker.getService().getLogger().warn("No url, returning empty iterator");
 	    return new Vector().iterator();
 	}else if ( url.endsWith("/") || url.equals(".") ){
 	    return new Vector().iterator();
 	}
 	Vector vector = new Vector();
 	try{
 	    ListEntry entry = new ListEntry();
 	    HTTPConnection connection = new HTTPConnection(host, port);
 	    connection.setTimeout(timeOut);
 	    URI uri = new URI( HTTPProtocolHandlerFactory.PROTOCOL, host, port, baseURL);
 	    uri = new URI(uri, url);
 	    
 	    //mind the time out
 	    HTTPResponse response = connection.Head(uri.getPath());
 	    Date lastModified = response.getHeaderAsDate("Last-Modified");
 	    //entry.setUrl(uri.toString());
 	    entry.setUrl(url);
 	    entry.setTimestamp(lastModified);
 	    vector.add(entry);
 	}catch(Exception ex){
 	    throw new ServiceException("Could not list url: " + url + " " + ex);
 	}
 	return vector.iterator();
     }
 
 
 
 
     ServiceMessage get(URL url)throws ServiceException {
 
 	HTTPConnection connection = null;
 	
 	try{
 	    //System.out.println(url.toString());
 	    connection = new HTTPConnection(url);
 	    connection.setTimeout(timeOut);
 	    
 	} catch (ProtocolNotSuppException notSuported){
 	    System.out.println(notSuported);
 	    throw new ServiceException("Could not fetch url: " + url.toString(),notSuported);
 	} catch (Exception something){
 	    //something.printStackTrace();
 	    throw new ServiceException("Could not fetch url: " + url.toString(),something);
 	}
 	
 	// number of times to try this connection
 	Exception lastEx = null;
 	for(int i=0; i <= retryTimes; i++){
 	    //System.out.println("connecting..");
 	    try{
 		HTTPResponse response = connection.Get(url.getFile());
 		if (response.getStatusCode() == 200) {
 		    //System.out.println("connected");
 		    String contentType = response.getHeader("Content-Type");
 		    String data = new String( response.getData() );
 		    return new ServiceMessage(url.toString(), contentType, response.getInputStream());
 		} else {
 		    throw new ServiceException("Error in getting " + url.toString() + response.getReasonLine() + " code: " + response.getStatusCode());
 		}
 	    } catch(Exception exception){
 		lastEx = exception;
 		invoker.getService().getLogger().warn(exception.toString());
 		//System.out.println(exception);
 		connection.stop();
 		try{
 		    if(retryDelay > 0){
 			synchronized(this){ 
 			    wait(retryDelay);
 			}
 		    }
 		} catch(InterruptedException interrupted){ 
 		    // someone Notify'ed on this, strangely enough
 		}
 	    }
 	}
 	
 	// or else, if we got here, we where not able to get message
 	throw new ServiceException(lastEx);
     }
     
 
     
     
     public ServiceMessage get(ListEntry entry) throws ServiceException { return get(entry.getUrl()); }
 
     public ServiceMessage get(String end) throws ServiceException{
 	invoker.getService().getLogger().debug("Getting " + baseURL + "/" +end);
 	try{
 	    java.net.URL url = new java.net.URL("http", host, port, baseURL + "/" + end);
 	    return get(url);
 	}catch(java.net.MalformedURLException e){
 	    //System.out.println(e);
 	    throw new ServiceException(e);
 	}
     }
     
     //--- Servable ---
     /*
       public void init() throws Exception {}
       public void start() throws Exception {}
       public void stop() throws Exception {}
       public void destroy() throws Exception {}
     */
     
     //---- XmlConfigurable ---
     /***
     public void importXml(Element element) throws DeploymentException {
 	globalConf.importXml(element);
 	getConfiguration();
     }
     */
 }