/*
    * Copyright 2001-2004 The Apache Software Foundation.
    * 
    * Licensed under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0
    * 
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.jdiagnose.remote.comms;
  
  import java.io.ByteArrayOutputStream;
  import java.io.PrintWriter;
  import java.net.Authenticator;
  import java.net.HttpURLConnection;
  import java.net.PasswordAuthentication;
  import java.net.URL;
  import java.net.URLConnection;
  import java.net.URLEncoder;
  
  import org.jdiagnose.DiagnosticMessage;
  import org.jdiagnose.RemoteResult;
  import org.jdiagnose.library.HttpProxyConfig;
  import org.jdiagnose.remote.ExceptionListener;
  import org.jdiagnose.remote.RemoteResultException;
  import org.jdiagnose.remote.RemoteResultListener;
  import org.jdiagnose.remote.RemoteResultSender;
  import org.jdiagnose.remote.StdErrExceptionListener;
  
  /***
   * Sends Diagnostic Results to the JDiagnose Server view HTTP. 
   * 
   * @author jmccrindle
   */
  public class HttpSender implements RemoteResultSender, RemoteResultListener {
  
      private String url = null;
      private HttpProxyConfig proxyConfig = null;
      private String method = "POST";
      private ExceptionListener exceptionListener = StdErrExceptionListener.DEFAULT_STD_ERR_EXCEPTION_LISTENER;
  
      /***
       * @param url
       */
      public HttpSender(String url) {
          super();
          this.url = url;
      }
  
      /***
       */
      public HttpSender() {
          super();
      }
  
      /*
       * (non-Javadoc)
       * 
       * @see org.jdiagnose.remote.RemoteResultSender#send(org.jdiagnose.remote.RemoteResult)
       */
      public void send(RemoteResult result) throws RemoteResultException {
  
          if (url == null) {
              throw new RemoteResultException(
                      "url property must be set for HttpSender");
          }
  
          try {
              if (proxyConfig != null) {
                  System
                          .setProperty("http.proxyHost", proxyConfig
                                  .getProxyHost());
                  System
                          .setProperty("http.proxyPort", proxyConfig
                                  .getProxyPort());
                  if (proxyConfig.getProxyUsername() != null
                          && proxyConfig.getProxyPassword() != null) {
                      Authenticator.setDefault(new Authenticator() {
                          protected PasswordAuthentication getPasswordAuthentication() {
                              return new PasswordAuthentication(proxyConfig
                                      .getProxyUsername(), proxyConfig
                                      .getProxyPassword().toCharArray());
                          }
                      });
                  }
              }
              URL fullUrl = new URL(url);
              URLConnection connection = fullUrl.openConnection();
  
              HttpURLConnection urlConnection = (HttpURLConnection) connection;
  
              urlConnection.setRequestMethod(method);
              urlConnection.setUseCaches(false);
             urlConnection.setDoOutput(true);
 
             ByteArrayOutputStream byteStream = new ByteArrayOutputStream(512);
 
             // Stream that writes into buffer
             PrintWriter out = new PrintWriter(byteStream, true);
             StringBuffer postData = new StringBuffer("host="
                     + URLEncoder.encode(result.getHost())
                     + "&agent="
                     + URLEncoder.encode(result.getAgent())
                     + "&guid="
                     + URLEncoder.encode(result.getGuid())
                     + "&sequenceNumber="
                     + result.getSequenceNumber()
                     + "&resultInfo.name="
                     + URLEncoder.encode(result.getResultInfo().getName())
                     + "&resultInfo.duration="
                     + result.getResultInfo().getDuration()
                     + "&resultInfo.startTime="
                     + result.getResultInfo().getStartTime()
                     + "&resultInfo.finishTime="
                     + result.getResultInfo().getFinishTime()
                     + "&resultInfo.state="
                     + URLEncoder.encode(result.getResultInfo().getState()
                             .toString()));
 
             DiagnosticMessage message = result.getResultInfo()
                     .getMessage();
             // TODO change this, exceptionMessage and body...
             if (message != null) {
                 postData.append("&resultInfo.message.summary="
                         + URLEncoder.encode(message.getSummary()));
                 postData.append("&resultInfo.message.body="
                         + URLEncoder.encode(message.getBody()));
             }
 
             // Write POST data into local buffer
             out.print(postData);
             out.flush(); // Flush since above used print, not println
 
             // POST requests are required to have Content-Length
             String lengthString = String.valueOf(byteStream.size());
             connection.setRequestProperty("Content-Length", lengthString);
 
             connection.setRequestProperty("Content-Type",
                     "application/x-www-form-urlencoded");
 
             // Write POST data to real output stream
             byteStream.writeTo(connection.getOutputStream());
 
             int responseCode = urlConnection.getResponseCode();
             if (responseCode != 200) {
                 throw new RemoteResultException(
                         "Http sending failed with response code "
                                 + responseCode);
             }
         } catch (RemoteResultException e) {
             throw e;
         } catch (Throwable e) {
             throw new RemoteResultException(e);
         }
     }
 
     public String getUrl() {
         return url;
     }
 
     public void setUrl(String string) {
         url = string;
     }
 
     public String getMethod() {
         return method;
     }
 
     public void setMethod(String method) {
         this.method = method;
     }
 
     public HttpProxyConfig getProxyConfig() {
         return proxyConfig;
     }
 
     public void setProxyConfig(HttpProxyConfig proxyConfig) {
         this.proxyConfig = proxyConfig;
     }
 
     /* (non-Javadoc)
      * @see org.jdiagnose.remote.RemoteResultListener#onRemoteResult(org.jdiagnose.RemoteResult)
      */
     public void onRemoteResult(RemoteResult remoteResult) {
         try {
             send(remoteResult);
         } catch (Throwable e) {
             exceptionListener.onException(e);
         }
     }
 
 }