/**
 * 
 */
package at.knowcenter.wag.egov.egiz.web.servlets;

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;

import javax.servlet.RequestDispatcher;
import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import at.gv.egiz.pdfas.exceptions.framework.SignatorException;
import at.gv.egiz.pdfas.framework.SignatorFactory;
import at.gv.egiz.pdfas.framework.signator.Signator;
import at.gv.egiz.pdfas.web.SignSessionInformation;
import at.gv.egiz.pdfas.web.VerifySessionInformation;
import at.gv.egiz.pdfas.web.helper.SessionHelper;
import at.gv.egiz.pdfas.web.helper.SignServletHelper;
import at.gv.egiz.pdfas.web.helper.TempDirHelper;
import at.knowcenter.wag.egov.egiz.PdfASID;
import at.knowcenter.wag.egov.egiz.exceptions.ConnectorException;
import at.knowcenter.wag.egov.egiz.exceptions.InvalidIDException;
import at.knowcenter.wag.egov.egiz.exceptions.PresentableException;
import at.knowcenter.wag.egov.egiz.exceptions.SignatorFactoryException;
import at.knowcenter.wag.egov.egiz.pdf.SignatureHolder;
import at.knowcenter.wag.egov.egiz.sig.SignatureResponse;
import at.knowcenter.wag.egov.egiz.sig.connectors.ConnectorChooser;
import at.knowcenter.wag.egov.egiz.sig.connectors.LocalConnector;
import at.knowcenter.wag.egov.egiz.sig.connectors.bku.BKUPostConnection;
import at.knowcenter.wag.egov.egiz.web.FormFields;
import at.knowcenter.wag.egov.egiz.web.LocalRequest;

/**
 * @author wprinz
 * 
 */
public class DataURLServlet extends HttpServlet
{

  /**
   * SVUID.
   */
  private static final long serialVersionUID = -5846618335843762752L;

  /**
   * The log.
   */
  private static Log log = LogFactory.getLog(DataURLServlet.class);

  protected void dispatch(HttpServletRequest request,
      HttpServletResponse response, String resource) throws ServletException, IOException
  {
    dispatch(request, response, resource, getServletContext());
  }

  protected static void dispatch(HttpServletRequest request,
      HttpServletResponse response, String resource, ServletContext context) throws ServletException, IOException
  {
    response.setContentType("text/html");
    response.setCharacterEncoding("UTF-8");

    RequestDispatcher disp = context.getRequestDispatcher(resource);
    disp.forward(request, response);
  }

  protected void dispatchToResults(List results, HttpServletRequest request,
      HttpServletResponse response, String backToListURL) throws ServletException, IOException
  {
    request.setAttribute("results", results);
    request.setAttribute("btlurl", backToListURL);
    dispatch(request, response, "/jsp/results.jsp");
  }

  /**
   * @see javax.servlet.http.HttpServlet#doPost(javax.servlet.http.HttpServletRequest,
   *      javax.servlet.http.HttpServletResponse)
   */
  protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException
  {
    log.debug("Data URL is accessed."); //$NON-NLS-1$

    try
    {
      Object sessionObject = SessionHelper.getSession(request);

      checkRequestCharacterEncoding(request);
      
      if (sessionObject instanceof SignSessionInformation)
      {
        SignSessionInformation si = (SignSessionInformation)sessionObject;
        processSign(request, response, si);
      }
      else
      {
        VerifySessionInformation si = (VerifySessionInformation) sessionObject;
        processVerify(request, response, si);
      }

    }
    catch (PresentableException e)
    {
      log.error(e);
      SignServlet.prepareDispatchToErrorPage(e, request);
      dispatch(request, response, "/jsp/error.jsp");
    }

    log.debug("DataURL access finished."); //$NON-NLS-1$
  }
  
  protected void checkRequestCharacterEncoding(HttpServletRequest request) throws UnsupportedEncodingException
  {
    log.debug("Request character encoding = " + request.getCharacterEncoding()); //$NON-NLS-1$
//  if (request.getCharacterEncoding() == null || request.getCharacterEncoding().equals("UTF-8")) //$NON-NLS-1$
  if (request.getCharacterEncoding() == null || request.getCharacterEncoding().length() <= 0) //$NON-NLS-1$
  {
    log.error("The BKU didn't set a character encoding for the request."); //$NON-NLS-1$

    log.warn("Manually setting character encoding to UTF-8"); //$NON-NLS-1$
    request.setCharacterEncoding("UTF-8"); //$NON-NLS-1$
  }
  }

  protected boolean isNullResponse(String xml_response)
  {
    return xml_response.indexOf("NullOperationResponse") >= 0;
  }
  
  protected void processSign(HttpServletRequest request, HttpServletResponse response, SignSessionInformation si) throws ServletException, IOException, ConnectorException, SignatorException, SignatorFactoryException
  {
    log.trace("processSign");
    
    String xml_response = request.getParameter("XMLResponse"); //$NON-NLS-1$
    log.debug("xml_response = " + xml_response); //$NON-NLS-1$

    if (isNullResponse(xml_response))
    {
      log.debug("Received a NullOperationResponse -> answering with the first request."); //$NON-NLS-1$

      assert si.outputAvailable == false;
      assert si.response_properties == null;
      
      log.debug("There are still requests to be performed -> answering with request."); //$NON-NLS-1$

      LocalRequest local_request = si.localRequest;

      String request_string = local_request.getRequestString();

      response.setContentType("text/xml");
      response.setCharacterEncoding("UTF-8");
      response.getWriter().println(request_string);
    }
    else
    {
      log.debug("Recieved a normal response -> storing the response."); //$NON-NLS-1$

      Properties response_properties = new Properties();
      response_properties.setProperty(BKUPostConnection.RESPONSE_STRING_KEY, xml_response);
      si.response_properties = response_properties;
      
      log.debug("All requests have been processed -> processing the responses."); //$NON-NLS-1$

      // Sign
      
      if (!si.outputAvailable)
      {
        LocalConnector c = ConnectorChooser.chooseLocalConnectorForSign(si.connector, si.type, "loc ref content not needed here"); //$NON-NLS-1$

        si.si.setSignSignatureObject(c.analyzeSignResponse(si.response_properties));

        PdfASID algorithm = FormFields.translateSignatureModeToPdfASID(si.mode);
        Signator signator = SignatorFactory.createSignator(algorithm);

        si.output = TempDirHelper.createTempDataSink(si.filename + "_signed.pdf"); 

        signator.finishSign(si.si, si.output);
        
        si.outputAvailable = true;
      }

      SignServletHelper.returnSignResponse(si, response);

    }
  }
  
  protected void processVerify(HttpServletRequest request, HttpServletResponse response, VerifySessionInformation si) throws ServletException, IOException, ConnectorException, InvalidIDException
  {
    log.trace("processVerify");
    
    String xml_response = request.getParameter("XMLResponse"); //$NON-NLS-1$
    log.debug("xml_response = " + xml_response); //$NON-NLS-1$

    if (isNullResponse(xml_response))
    {
      log.debug("Received a NullOperationResponse -> answering with the first request."); //$NON-NLS-1$

      assert si.currentLocalOperation.current_operation == 0;
    }
    else
    {
      log.debug("Recieved a normal response -> storing the response."); //$NON-NLS-1$

      Properties response_properties = new Properties();
      response_properties.setProperty(BKUPostConnection.RESPONSE_STRING_KEY, xml_response);

      si.currentLocalOperation.finishCurrentOperation(response_properties);
    }

    if (!si.currentLocalOperation.isFinished())
    {
      log.debug("There are still requests to be performed -> answering with request #" + si.currentLocalOperation.current_operation); //$NON-NLS-1$

      LocalRequest local_request = si.currentLocalOperation.getCurrentLocalRequest();

      String request_string = local_request.getRequestString();

      response.setContentType("text/xml");
      response.setCharacterEncoding("UTF-8");
      response.getWriter().println(request_string);
    }
    else
    {
      log.debug("All requests have been processed -> processing the responses."); //$NON-NLS-1$

        
        List results = new ArrayList();

        for (int i = 0; i < si.currentLocalOperation.response_properties.length; i++)
        {
          SignatureHolder sh = (SignatureHolder) si.currentLocalOperation.holders_to_be_verified.get(i);
          
          PdfASID sig_kz = sh.getSignatureObject().getKZ();
          String sig_id = sh.getSignatureObject().getSignationIds();
          
          LocalConnector c = ConnectorChooser.chooseLocalConnectorForVerify(si.connector, sig_kz, sig_id, si.type, "loc ref content not needed here"); //$NON-NLS-1$

          SignatureResponse sig_resp = c.analyzeVerifyResponse(si.currentLocalOperation.response_properties[i]);
          results.add(sig_resp);
        }
        
        si.currentLocalOperation = null;

        URL btlURL = new URL(request.getScheme(), request.getServerName(), request.getServerPort(), request.getContextPath() + "/jsp/verifylist.jsp");
        String backToListURL = response.encodeURL(btlURL.toString());

        dispatchToResults(results, request, response, backToListURL);
    }

  }
}