/* Copyright 2006-2008 Sun Microsystems, Inc.  All rights reserved.  You may not modify, use, reproduce, or distribute this software except in compliance with the terms of the License at:
 http://developer.sun.com/berkeley_license.html
 $Id: HttpClient.java 2631 2008-10-27 17:25:05Z heiko.braun@jboss.com $
*/package jmaki.xhp;

import java.io.*;
import java.util.Map;
import java.util.Iterator;
import java.net.*;
import java.util.logging.*;
import java.security.Security;

/**
 * @author Yutaka Yoshida, Greg Murray
 *
 * Minimum set of HTTPclient supporting both http and https.
 * It's aslo capable of POST, but it doesn't provide doGet because
 * the caller can just read the inputstream.
 */
public class HttpClient {

   private static Logger logger;
   private String proxyHost = null;
   private int proxyPort = -1;
   private boolean isHttps = false;
   private boolean isProxy = false;
   private HttpURLConnection urlConnection = null;
   private Map headers;

   /**
    * @param phost PROXY host name
    * @param pport PROXY port string
    * @param url URL string
    * @param headers Map
    */
   public HttpClient(
         String phost,
         int pport,
         String url,
         Map headers,
         String method)
         throws MalformedURLException
   {
      if (phost != null && pport != -1)
      {
         this.isProxy = true;
      }

      this.proxyHost = phost;
      this.proxyPort = pport;

      if (url.trim().startsWith("https:")) {
         isHttps = true;
      }

      this.urlConnection = getURLConnection(url);
      try {
         this.urlConnection.setRequestMethod(method);
      } catch (java.net.ProtocolException pe) {
         HttpClient.getLogger().severe("Unable protocol method to " + method + " : " + pe);
      }
      this.headers = headers;
      // seat headers
      if (headers != null) {
         Iterator it = headers.keySet().iterator();
         if (it != null) {
            while (it.hasNext()) {
               String key = (String)it.next();
               String value = (String)headers.get(key);
               this.urlConnection.setRequestProperty (key, value);
            }
         }
      }
   }

   /**
    * @param phost PROXY host name
    * @param pport PROXY port string
    * @param url URL string
    * @param headers Map
    * @param userName string
    * @param password string
    */
   public HttpClient(String phost,
                     int pport,
                     String url,
                     Map headers,
                     String method,
                     String userName,
                     String password)
         throws MalformedURLException {
      try
      {
         if (phost != null && pport != -1) {
            this.isProxy = true;
         }

         this.proxyHost = phost;
         this.proxyPort = pport;
         if (url.trim().startsWith("https:")) {
            isHttps = true;
         }
         this.urlConnection = getURLConnection(url);
         try {
            this.urlConnection.setRequestMethod(method);
         } catch (java.net.ProtocolException pe) {
            HttpClient.getLogger().severe("Unable protocol method to " + method + " : " + pe);
         }
         // set basic authentication information
         String auth = userName + ":" +  password;
         String encoded = new sun.misc.BASE64Encoder().encode (auth.getBytes());
         // set basic authorization
         this.urlConnection.setRequestProperty ("Authorization", "Basic " + encoded);
         this.headers = headers;
         // seat headers
         if (headers != null) {
            Iterator it = headers.entrySet().iterator();
            if (it != null) {
               while (it.hasNext()) {
                  String key = (String)it.next();
                  String value = (String)headers.get(key);
                  this.urlConnection.setRequestProperty (key, value);
               }
            }
         }
      } catch (Exception ex) {
         HttpClient.getLogger().severe("Unable to set basic authorization for " + userName  + " : " +ex);
      }
   }

   /**
    * private method to get the URLConnection
    * @param str URL string
    */
   private HttpURLConnection getURLConnection(String str)
         throws MalformedURLException {
      try {

         if (isHttps) {
            /* when communicating with the server which has unsigned or invalid
            * certificate (https), SSLException or IOException is thrown.
            * the following line is a hack to avoid that
            */
            Security.addProvider(new com.sun.net.ssl.internal.ssl.Provider());
            System.setProperty("java.protocol.handler.pkgs", "com.sun.net.ssl.internal.www.protocol");
            if (isProxy) {
               System.setProperty("https.proxyHost", proxyHost);
               System.setProperty("https.proxyPort", proxyPort + "");
            }
         }
         else
         {
            if (isProxy)
            {
               System.setProperty("http.proxyHost", proxyHost);
               System.setProperty("http.proxyPort", proxyPort  + "");
            }
         }

         URL url = new URL(str);
         HttpURLConnection uc = (HttpURLConnection)url.openConnection();
         // if this header has not been set by a request set the user agent.
         if (headers == null ||
               (headers != null &&  headers.get("user-agent") == null)) {
            // set user agent to mimic a common browser
            String ua="Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; .NET CLR 1.1.4322)";
            uc.setRequestProperty("user-agent", ua);
         }
         return uc;
      }
      catch (MalformedURLException me)
      {
         throw new MalformedURLException(str + " is not a valid URL");
      }
      catch (Exception e)
      {
         throw new RuntimeException("Unknown error creating UrlConnection: " + e);
      }
   }

   /**
    * returns the inputstream from URLConnection
    * @return InputStream
    */
   public InputStream getInputStream() {
      try
      {
         // logger doesnt work, because it writes to stderr,
         // which causes GwtTest to interpret it as failure
         System.out.println(
               this.urlConnection.getRequestMethod()+ " " +
                     this.urlConnection.getURL() +": "+
                     this.urlConnection.getResponseCode()
         );
         return (this.urlConnection.getInputStream());
      } catch (Exception e) {
         e.printStackTrace();
         return null;
      }
   }

   /**
    * return the OutputStream from URLConnection
    * @return OutputStream
    */
   public OutputStream getOutputStream() {

      try {
         return (this.urlConnection.getOutputStream());
      } catch (Exception e) {
         e.printStackTrace();
         return null;
      }
   }

   /**
    * posts data to the inputstream and returns the InputStream.
    * @param postData data to be posted. must be url-encoded already.
    * @param contentType allows you to set the contentType of the request.
    * @return InputStream input stream from URLConnection
    */
   public InputStream doPost(String postData, String contentType) {
      this.urlConnection.setDoOutput(true);
      if (contentType != null) this.urlConnection.setRequestProperty( "Content-type", contentType );

      OutputStream os = this.getOutputStream();
      PrintStream ps = new PrintStream(os);
      ps.print(postData);
      ps.close();
      return (this.getInputStream());
   }

   public String getContentEncoding() {
      if (this.urlConnection == null) return null;
      return (this.urlConnection.getContentEncoding());
   }
   public int getContentLength() {
      if (this.urlConnection == null) return -1;
      return (this.urlConnection.getContentLength());
   }
   public String getContentType() {
      if (this.urlConnection == null) return null;
      return (this.urlConnection.getContentType());
   }
   public long getDate() {
      if (this.urlConnection == null) return -1;
      return (this.urlConnection.getDate());
   }
   public String getHeader(String name) {
      if (this.urlConnection == null) return null;
      return (this.urlConnection.getHeaderField(name));
   }
   public long getIfModifiedSince() {
      if (this.urlConnection == null) return -1;
      return (this.urlConnection.getIfModifiedSince());
   }

   public static Logger getLogger() {
      if (logger == null) {
         logger = Logger.getLogger("jmaki.xhp.Log");
      }
      return logger;
   }
}