edu.duke.cabig.c3pr.web.admin.StudySubjectXMLFileUploadController.java Source code

Introduction

Here is the source code for edu.duke.cabig.c3pr.web.admin.StudySubjectXMLFileUploadController.java

Source

/*******************************************************************************
 * Copyright Duke Comprehensive Cancer Center and SemanticBits
 * 
 * Distributed under the OSI-approved BSD 3-Clause License.
 * See http://ncip.github.com/c3pr/LICENSE.txt for details.
 ******************************************************************************/
package edu.duke.cabig.c3pr.web.admin;

import java.io.File;
import java.util.Collection;
import java.util.Date;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
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 edu.duke.cabig.c3pr.domain.StudySubject;
import edu.duke.cabig.c3pr.exception.C3PRCodedException;
import edu.duke.cabig.c3pr.service.StudySubjectXMLImporterService;
import edu.duke.cabig.c3pr.web.ajax.StudySubjectXMLFileImportAjaxFacade;
import edu.duke.cabig.c3pr.web.beans.FileBean;

/**
 * User: kherm
 * 
 * @author kherm
 * @author kherm manav.kher@semanticbits.com
 */
public class StudySubjectXMLFileUploadController extends SimpleFormController {

    private static Log log = LogFactory.getLog(StudySubjectXMLFileUploadController.class);

    private StudySubjectXMLFileImportAjaxFacade studySubjectXMLFileAjaxFacade;

    private StudySubjectXMLImporterService studySubjectXMLImporterService;

    public StudySubjectXMLFileUploadController() {
        setBindOnNewForm(true);
    }

    @Override
    protected ModelAndView onSubmit(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse,
            Object command, BindException errors) throws Exception {
        // save it to session
        FileBean xMLFile = (FileBean) command;
        String filePath = System.getenv("CATALINA_HOME") + System.getProperty("file.separator") + "conf"
                + System.getProperty("file.separator") + "c3pr";
        File outputXMLDir = new File(filePath);
        outputXMLDir.mkdirs();
        String fileName = "importRegistration-output-" + new Date().getTime() + ".xml";
        File outputXMLFile = new File(filePath + System.getProperty("file.separator") + fileName);
        outputXMLFile.createNewFile();
        try {
            Collection<StudySubject> studySubjects = studySubjectXMLImporterService
                    .importStudySubjects(xMLFile.getInputStream(), outputXMLFile);
            Object viewData = studySubjectXMLFileAjaxFacade.getTable(null, httpServletRequest, studySubjects);
            httpServletRequest.setAttribute("registrations", viewData);
            httpServletRequest.setAttribute("filePath", fileName);
        } catch (C3PRCodedException e1) {
            e1.printStackTrace();
            log.debug("Uploaded file contains invalid registration");
            errors.reject("Could not import registrations " + e1.getCodedExceptionMesssage());
        } catch (Exception e1) {
            e1.printStackTrace();
            log.debug("Uploaded file contains invalid registration");
            errors.reject("Could not import registrations " + e1.getMessage());
        }

        return new ModelAndView(this.getSuccessView(), errors.getModel());
    }

    @Override
    protected ModelAndView showForm(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse,
            BindException e) throws Exception {
        return new ModelAndView(this.getFormView(), e.getModel());
    }

    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
        binder.registerCustomEditor(byte[].class, new ByteArrayMultipartFileEditor());
        // now Spring knows how to handle multipart object and convert them
    }

    public void setStudySubjectXMLFileAjaxFacade(
            StudySubjectXMLFileImportAjaxFacade studySubjectXMLFileAjaxFacade) {
        this.studySubjectXMLFileAjaxFacade = studySubjectXMLFileAjaxFacade;
    }

    public void setStudySubjectXMLImporterService(StudySubjectXMLImporterService studySubjectXMLImporterService) {
        this.studySubjectXMLImporterService = studySubjectXMLImporterService;
    }

}