/*
    * 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.library.web.spring;
  
  import java.io.ByteArrayInputStream;
  import java.text.SimpleDateFormat;
  import java.util.Date;
  import java.util.Iterator;
  
  import javax.servlet.ServletException;
  import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpServletResponse;
  import javax.servlet.http.HttpSession;
  
  import org.jdiagnose.RemoteResult;
  import org.jdiagnose.remote.RemoteResultStore;
  import org.jdiagnose.remote.file.ResultImporter;
  import org.jdiagnose.remote.provider.DefaultSystemHolder;
  import org.jdiagnose.remote.provider.MutableSystemProvider;
  import org.jdiagnose.remote.system.InMemoryRemoteSystem;
  import org.springframework.validation.BindException;
  import org.springframework.web.bind.ServletRequestDataBinder;
  import org.springframework.web.multipart.support.ByteArrayMultipartFileEditor;
  import org.springframework.web.servlet.ModelAndView;
  import org.springframework.web.servlet.mvc.SimpleFormController;
  import org.springframework.web.servlet.view.RedirectView;
  
  /***
   * @author jmccrindle
   */
  public class ImportController extends SimpleFormController {
      
      private static final org.apache.log4j.Category log = org.apache.log4j.Category
              .getInstance(ImportController.class);
      
      private SimpleDateFormat systemDateFormat = new SimpleDateFormat("HH:mm:ss");
      private ResultImporter importer = null;
      private MutableSystemProvider provider = null;
      private RemoteResultStore resultStore = null;
      private String sessionSystemsKey = "sessionSystemsKey";
      private SessionSystemsFactory sessionSystemsFactory = null;
  
  
      /* (non-Javadoc)
       * @see org.springframework.web.servlet.mvc.AbstractFormController#formBackingObject(javax.servlet.http.HttpServletRequest)
       */
      protected Object formBackingObject(HttpServletRequest request)
              throws Exception {
          ImportCommand importCommand = new ImportCommand();
          importCommand.setSystem("Import at " + systemDateFormat.format(new Date()));
          importCommand.setRemoteSystem(true);
          return importCommand;
      }
  
      protected ModelAndView onSubmit(HttpServletRequest request,
              HttpServletResponse response, Object command, BindException errors)
              throws Exception {
  
          // cast the bean
          ImportCommand importCommand = (ImportCommand) command;
  
          // let's see if there's content there
          byte[] file = importCommand.getFile();
          if (file == null) {
              // hmm, that's strange, the user did not upload anything
          } else {
              Iterator results = importer.getResults(new ByteArrayInputStream(file));
              InMemoryRemoteSystem remoteSystem = null; 
              String systemName = importCommand.getSystem();
              HttpSession session = request.getSession(true);
              if(importCommand.isPersist() && importCommand.isRemoteSystem()) {
                  remoteSystem = new InMemoryRemoteSystem();
                  while (results.hasNext()) {
                      RemoteResult result = (RemoteResult) results.next();
                      remoteSystem.addRemoteResult(result);
                      resultStore.addRemoteResult(result);
                  }
                  String systemHolderId = addRemoteSystem(remoteSystem, systemName);
                  addToSession(session, systemHolderId);
                  return new ModelAndView(new RedirectView(request.getContextPath() + "/hosts.htm?system=" + systemHolderId));
              } else if(importCommand.isRemoteSystem()) {
                  remoteSystem = new InMemoryRemoteSystem();
                  remoteSystem.addRemoteResults(results);
                  String systemHolderId = addRemoteSystem(remoteSystem, systemName);
                  addToSession(session, systemHolderId);
                  return new ModelAndView(new RedirectView(request.getContextPath() + "/hosts.htm?system=" + systemHolderId));
             } else if(importCommand.isPersist()) {
                 resultStore.addRemoteResults(results);
                 return new ModelAndView(new RedirectView(request.getContextPath() + "/hosts.htm"));
             }
         }
         return new ModelAndView(getFormView());
     }
     
     private void addToSession(HttpSession session, String id) {
         SessionSystems sessionSystems = (SessionSystems) session.getAttribute(sessionSystemsKey);
         if(sessionSystems == null) {
             sessionSystems = sessionSystemsFactory.create();
             session.setAttribute(sessionSystemsKey, sessionSystems);
         }
         sessionSystems.getIds().add(id);
     }
 
     /***
      * @param remoteSystem
      * @param systemName
      */
     private String addRemoteSystem(InMemoryRemoteSystem remoteSystem, String systemName) {
         String systemHolderId = provider.nextId();
         DefaultSystemHolder holder = new DefaultSystemHolder();
         holder.setId(systemHolderId);
         holder.setDescription(systemName);
         holder.setRemoteSystem(remoteSystem);
         provider.addHolder(holder);
         return systemHolderId;
     }
 
     protected void initBinder(HttpServletRequest request,
             ServletRequestDataBinder binder) throws ServletException {
         // to actually be able to convert Multipart instance to byte[]
         // we have to register a custom editor (in this case the
         // ByteArrayMultipartEditor
         binder.registerCustomEditor(byte[].class,
                 new ByteArrayMultipartFileEditor());
         // now Spring knows how to handle multipart object and convert them
     }
 
     public void setImporter(ResultImporter importer) {
         this.importer = importer;
     }
     public void setSystemDateFormat(SimpleDateFormat systemDateFormat) {
         this.systemDateFormat = systemDateFormat;
     }
     public void setProvider(MutableSystemProvider provider) {
         this.provider = provider;
     }
     public void setResultStore(RemoteResultStore resultStore) {
         this.resultStore = resultStore;
     }
     public void setSessionSystemsFactory(
             SessionSystemsFactory sessionSystemsFactory) {
         this.sessionSystemsFactory = sessionSystemsFactory;
     }
     public void setSessionSystemsKey(String sessionSystemsKey) {
         this.sessionSystemsKey = sessionSystemsKey;
     }
 }