/*

  * Copyright 1999-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.apache.commons.latka.servlet;

 import java.io.InputStream;

 import java.io.IOException;

 import java.io.OutputStream;

 import java.util.Collections;

 import java.util.Comparator;

 import java.util.List;

 import javax.servlet.http.HttpServlet;

 import javax.servlet.http.HttpServletRequest;

 import javax.servlet.http.HttpServletResponse;

 import javax.servlet.http.HttpSession;

 import org.apache.commons.latka.http.Response;

 import org.apache.log4j.Category;

 /**

  * A servlet to view the result of running a Latka suite

  *

  * @author Morgan Delagrange

  * @version $Id: ViewResponseServlet.java 155424 2005-02-26 13:09:29Z dirkv $

  */

 public class ViewResponseServlet extends HttpServlet {

     /** log4 category for output */

     public static final Category _log = Category.getInstance(

         ViewResponseServlet.class);

     /**

      * Perform a post request

      * 

      * @param req the http request

      * @param res the http response

      * @throws IOException when an error occurs writing the response

      */

     public void doPost(HttpServletRequest req, HttpServletResponse res) throws

         IOException {

         doGet(req, res);

     }

     /**

      * Perform a get request

      * 

      * @param req the http request

      * @param res the http response

      * @throws IOException when an error occurs writing the response

      */

     public void doGet(HttpServletRequest req, HttpServletResponse res) 

         throws IOException {

         HttpSession session = req.getSession();

         List list = (List) session.getAttribute("latka.failedResponses");

         String responseHash = req.getParameter("responseId");

         Collections.sort(list, new ToStringComparator());

         int i = Collections.binarySearch(list, responseHash, 

             new ToStringComparator());

         Response response = (Response) list.get(i);

         // set the content type, if specified

         String contentType = response.getHeader("Content-type");

         _log.debug("Content type = " + contentType);

         if (contentType != null) {

             res.setHeader("Content-type", contentType);

         }

         InputStream responseStream = response.getStream();

         OutputStream servletStream = res.getOutputStream();

         _log.debug("got input and output streams");

         int someByte = responseStream.read();

         _log.debug("read first byte: " + someByte);

         while (someByte > -1) {

           servletStream.write(someByte);

           someByte = responseStream.read();

         }

         _log.debug("end doGet(HttpServletRequest,HttpServletResponse)");

     }

     /**

      * A comparator that converts the objects to strings before comparing

      */

     private class ToStringComparator implements Comparator {

         /**

          * compare the two objects

          *

          * @param o1 the object being compared with

          * @param o2 the object being compared to

          * @return 0 if equal, -1 if o1 < o2, 1 if o1 > o2

          */

         public int compare(Object o1, Object o2) {

             return o1.toString().compareTo(o2.toString());

         }

     }

 }