/*
    * $Id: WorkflowUploadServlet.java 1537 2011-07-27 15:34:04Z hoheisel $
    *
    * Copyright (c) 2007
    * Fraunhofer Institute for Computer Architecture and Software Technology
    * See http://www.first.fraunhofer.de for more details.
    */
   
   package net.kwfgrid.gwes.servlet;
  
  import org.antlr.stringtemplate.StringTemplate;
  import org.apache.commons.fileupload.FileItem;
  import org.apache.commons.fileupload.FileItemFactory;
  import org.apache.commons.fileupload.FileUploadException;
  import org.apache.commons.fileupload.disk.DiskFileItemFactory;
  import org.apache.commons.fileupload.servlet.ServletFileUpload;
  import org.apache.log4j.Logger;
  
  import javax.servlet.http.HttpServlet;
  import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpServletResponse;
  import javax.servlet.http.HttpSession;
  import java.io.IOException;
  import java.io.PrintWriter;
  import java.util.List;
  
  import net.kwfgrid.gwes.client.RemoteGWES;
  import net.kwfgrid.gwes.exception.GWESException;
  
  /**
   * @author Andreas Hoheisel
   *         (<a href="http://www.andreas-hoheisel.de">www.andreas-hoheisel.de</a>)
   * @version $Id: WorkflowUploadServlet.java 1537 2011-07-27 15:34:04Z hoheisel $
   */
  public class WorkflowUploadServlet extends HttpServlet {
  
      final static Logger logger = Logger.getLogger(WorkflowUploadServlet.class);
      private static GWESProperties pr;
  
      protected void doGet(HttpServletRequest request, HttpServletResponse response) throws IOException {
          ServletLogger.log(request);
          
          response.setContentType("text/html");
          PrintWriter out = response.getWriter();
  
          //read gwes.properties
          if (pr == null) pr = GWESProperties.getInstance("/gwes.properties");
  
          //get request attributes
          String error = (String) request.getAttribute("error");
          String success = (String) request.getAttribute("newworkflowinfo");
          String workflowID = (String) request.getAttribute("workflowid");
  
          // print html
          StringTemplate t = Template.getStringTemplateGroup("html_gwes-servlets").getInstanceOf("workflowUpload");
          t.setAttribute("menu", new Menu(pr.gwesBaseUrlExternal, "New"));
          if (error != null) t.setAttribute("error", error);
          if (success != null) t.setAttribute("success", success);
          if (workflowID != null) t.setAttribute("workflowID", workflowID);
          t.setAttribute("gwesBaseUrlExternal", pr.gwesBaseUrlExternal);
          out.print(t.toString());
      }
  
      public void doPost(HttpServletRequest request, HttpServletResponse response) throws IOException {
          //read gwes.properties
          if (pr == null) pr = GWESProperties.getInstance("/gwes.properties");
  
          // test for file upload request
          boolean isMultipart = ServletFileUpload.isMultipartContent(request);
          if (isMultipart) logger.debug("=============> This is an file upload");
  
          // Create a factory for disk-based file items
          FileItemFactory factory = new DiskFileItemFactory();
  
          // Create a new file upload handler
          ServletFileUpload upload = new ServletFileUpload(factory);
  
          // Parse the request
          try {
              String workflowXml = null;
              String fileName = null;
              long sizeInBytes = 0;
              List items = upload.parseRequest(request);
              // Process the uploaded items
              for (Object item1 : items) {
                  FileItem item = (FileItem) item1;
  
                  if (item.isFormField()) {
                      String name = item.getFieldName();
                      String value = item.getString();
                      logger.debug("=========> " + name + "=" + value);
                  } else {
                      // String fieldName = item.getFieldName();
                      fileName = item.getName();
                      String contentType = item.getContentType();
                      // boolean isInMemory = item.isInMemory();
                      sizeInBytes = item.getSize();
                      logger.info("=========> file upload: " + fileName + " (" + sizeInBytes + "Bytes)");
  //                    if (!contentType.equals("text/xml"))
 //                        request.setAttribute("error", "Wrong file type \"" + contentType + "\". Must be of type \"text/xml\".");
 //                    else {
                         workflowXml = new String(item.get());
                         // logger.info("=========> GWorkflowDL XML: "+workflowXml);
 //                    }
                 }
             }
             if (workflowXml != null) {
                 // initiate new workflow
                 String userID = request.getUserPrincipal().getName();
                 String workflowid = initiateWorkflow(userID, workflowXml);
                 HttpSession session = request.getSession();
                 // parameter set for session (visible to other Servlets of same web application)
                 logger.debug("session workflow = "+workflowid);
                 session.setAttribute("workflowid", workflowid);
                 request.setAttribute("workflowid", workflowid);
                 StringBuffer newworkflowinfo = new StringBuffer();
                 newworkflowinfo.append("A new workflow with id \"")
                         .append(workflowid)
                         .append("\" has been initiated based on the uploaded workflow description \"")
                         .append(fileName)
                         .append("\" (").append(sizeInBytes).append("Bytes). ");
                 request.setAttribute("newworkflowinfo", newworkflowinfo.toString());
             }
         } catch (FileUploadException e) {
             request.setAttribute("error", "Could not upload file: <p>" + e);
             logger.error("Could not upload file: " + e, e);
         } catch (Exception e) {
             request.setAttribute("error", "Exception when uploading a workflow: <p>" + e);
             logger.error("Exception when uploading a workflow: "+e,e);
         } catch (GWESException e) {
             request.setAttribute("error", "Exception when uploading a workflow: <p>" + e);
             logger.error("Exception when uploading a workflow: "+e,e);
         }
 
         doGet(request, response);
     }
 
     /**
      * Initiate a new workflow.
      */
     private String initiateWorkflow(String userID, String xml) throws Exception, GWESException {
         return RemoteGWES.getInstance().initiate(xml, userID);
     }
 
 }