OLEBatchProcessJobDetailsController

package org.kuali.ole.batch.controller;

import org.apache.commons.io.IOUtils;
import org.kuali.ole.OLEConstants;
import org.kuali.ole.batch.bo.OLEBatchProcessJobDetailsBo;
import org.kuali.ole.batch.bo.OLEBatchProcessScheduleBo;
import org.kuali.ole.batch.document.OLEBatchProcessDefinitionDocument;
import org.kuali.ole.batch.form.OLEBatchProcessJobDetailsForm;
import org.kuali.ole.batch.helper.OLEBatchProcessDataHelper;
import org.kuali.ole.batch.service.OLEBatchSchedulerService;
import org.kuali.ole.sys.batch.BatchFile;
import org.kuali.ole.sys.batch.BatchFileUtils;
import org.kuali.rice.core.api.resourceloader.GlobalResourceLoader;
import org.kuali.rice.krad.service.BusinessObjectService;
import org.kuali.rice.krad.service.KRADServiceLocator;
import org.kuali.rice.krad.util.GlobalVariables;
import org.kuali.rice.krad.web.controller.TransactionalDocumentControllerBase;
import org.kuali.rice.krad.web.form.UifFormBase;
import org.quartz.CronExpression;
import org.springframework.stereotype.Controller;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.sql.Timestamp;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.*;

/**
 * Created with IntelliJ IDEA.
 * User: krishnamohanv
 * Date: 7/12/13
 * Time: 5:06 PM
 * To change this template use File | Settings | File Templates.
 */


@Controller
@RequestMapping(value = "/oleBatchProcessJobController")
public class OLEBatchProcessJobDetailsController extends TransactionalDocumentControllerBase {

    private OLEBatchProcessDataHelper oleBatchProcessDataHelper;
    private BusinessObjectService businessObjectService;
    private OLEBatchSchedulerService schedulerService;

    public OLEBatchSchedulerService getSchedulerService() {
        if(schedulerService == null ) {
            schedulerService = GlobalResourceLoader.getService("oleSchedulerService");
        }
        return schedulerService;
    }


    public BusinessObjectService getBusinessObjectService() {
        if(businessObjectService==null)
            businessObjectService = KRADServiceLocator.getBusinessObjectService();
        return businessObjectService;
    }

    private OLEBatchProcessDataHelper getOLEBatchProcessDataHelper(){

        if(oleBatchProcessDataHelper==null){
            oleBatchProcessDataHelper= OLEBatchProcessDataHelper.getInstance();
        }
        return oleBatchProcessDataHelper;
    }

    @Override
    protected OLEBatchProcessJobDetailsForm createInitialForm(HttpServletRequest request) {
        OLEBatchProcessJobDetailsForm oleBatchProcessJobDetailsForm = new OLEBatchProcessJobDetailsForm();
        GlobalVariables.getUserSession().addObject("formId", oleBatchProcessJobDetailsForm.getFormKey());
        List<OLEBatchProcessJobDetailsBo> oLEBatchProcessJobDetailsBoList = (List<OLEBatchProcessJobDetailsBo>) getBusinessObjectService().findAll(OLEBatchProcessJobDetailsBo.class);
        oleBatchProcessJobDetailsForm.setoLEBatchProcessJobDetailsBoList(oLEBatchProcessJobDetailsBoList);
        return oleBatchProcessJobDetailsForm;
    }

    @RequestMapping(params = "methodToCall=stopButton")
    public ModelAndView stopButton(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
                                   HttpServletRequest request, HttpServletResponse response) throws Exception {
        OLEBatchProcessJobDetailsForm oLEBatchProcessJobDetailsForm = (OLEBatchProcessJobDetailsForm) form;
        OLEBatchProcessJobDetailsBo jobDetailsBo = getJobBo(oLEBatchProcessJobDetailsForm);
        if(jobDetailsBo != null ) {
            jobDetailsBo.setStatus(OLEConstants.OLEBatchProcess.JOB_STATUS_STOPPED);
            getBusinessObjectService().save(jobDetailsBo);
        }
        List<OLEBatchProcessJobDetailsBo> oleBatchProcessJobDetailsBoList = (List<OLEBatchProcessJobDetailsBo> )getBusinessObjectService().findAll(OLEBatchProcessJobDetailsBo.class);
        oLEBatchProcessJobDetailsForm.setoLEBatchProcessJobDetailsBoList(oleBatchProcessJobDetailsBoList);
        getSchedulerService().stopJob(jobDetailsBo.getJobId());
        return getUIFModelAndView(oLEBatchProcessJobDetailsForm);
    }

    @RequestMapping(params = "methodToCall=startButton")
    public ModelAndView restartButton(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
                                      HttpServletRequest request, HttpServletResponse response) throws Exception{
        OLEBatchProcessJobDetailsForm oLEBatchProcessJobDetailsForm = (OLEBatchProcessJobDetailsForm) form;
        OLEBatchProcessJobDetailsBo jobDetailsBo = getJobBo(oLEBatchProcessJobDetailsForm);
        jobDetailsBo.setStatus(OLEConstants.OLEBatchProcess.JOB_STATUS_RUNNING);
        getBusinessObjectService().save(jobDetailsBo);
        List<OLEBatchProcessJobDetailsBo> oleBatchProcessJobDetailsBoList = (List<OLEBatchProcessJobDetailsBo> )getBusinessObjectService().findAll(OLEBatchProcessJobDetailsBo.class);
        oLEBatchProcessJobDetailsForm.setoLEBatchProcessJobDetailsBoList(oleBatchProcessJobDetailsBoList);
        getSchedulerService().startJob(jobDetailsBo.getJobId());
        return getUIFModelAndView(oLEBatchProcessJobDetailsForm);
    }

    @RequestMapping(params = "methodToCall=resumeButton")
    public ModelAndView resumeButton(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
                                     HttpServletRequest request, HttpServletResponse response) throws Exception {
        OLEBatchProcessJobDetailsForm oLEBatchProcessJobDetailsForm = (OLEBatchProcessJobDetailsForm) form;
        OLEBatchProcessJobDetailsBo jobDetailsBo = getJobBo(oLEBatchProcessJobDetailsForm);
        jobDetailsBo.setStatus(OLEConstants.OLEBatchProcess.JOB_STATUS_RUNNING);
        getBusinessObjectService().save(jobDetailsBo);
        List<OLEBatchProcessJobDetailsBo> oleBatchProcessJobDetailsBoList = (List<OLEBatchProcessJobDetailsBo> )getBusinessObjectService().findAll(OLEBatchProcessJobDetailsBo.class);
        oLEBatchProcessJobDetailsForm.setoLEBatchProcessJobDetailsBoList(oleBatchProcessJobDetailsBoList);
        getSchedulerService().resumeJob(jobDetailsBo.getJobId());
        return getUIFModelAndView(oLEBatchProcessJobDetailsForm);
    }

    @RequestMapping(params = "methodToCall=removeButton")
    public ModelAndView removeButton(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
                                     HttpServletRequest request, HttpServletResponse response) throws Exception {
        OLEBatchProcessJobDetailsForm oLEBatchProcessJobDetailsForm = (OLEBatchProcessJobDetailsForm) form;
        OLEBatchProcessJobDetailsBo jobDetailsBo = getJobBo(oLEBatchProcessJobDetailsForm);
        getOLEBatchProcessDataHelper().deleteBatchFailureFile(jobDetailsBo.getBatchProcessType(),jobDetailsBo.getJobId() + "_FailureRecord" + "_" + jobDetailsBo.getUploadFileName());
        jobDetailsBo.setStatus(OLEConstants.OLEBatchProcess.JOB_STATUS_CANCELLED);
        getBusinessObjectService().save(jobDetailsBo);
        List<OLEBatchProcessJobDetailsBo> oleBatchProcessJobDetailsBoList = (List<OLEBatchProcessJobDetailsBo> )getBusinessObjectService().findAll(OLEBatchProcessJobDetailsBo.class);
        oLEBatchProcessJobDetailsForm.setoLEBatchProcessJobDetailsBoList(oleBatchProcessJobDetailsBoList);
        getSchedulerService().deleteJob(jobDetailsBo.getJobId());
        return getUIFModelAndView(oLEBatchProcessJobDetailsForm);
    }


    @RequestMapping(params = "methodToCall=pauseButton")
    public ModelAndView pauseButton(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
                                    HttpServletRequest request, HttpServletResponse response) throws Exception{
        OLEBatchProcessJobDetailsForm oLEBatchProcessJobDetailsForm = (OLEBatchProcessJobDetailsForm) form;
        OLEBatchProcessJobDetailsBo jobDetailsBo = getJobBo(oLEBatchProcessJobDetailsForm);
        jobDetailsBo.setStatus(OLEConstants.OLEBatchProcess.JOB_STATUS_PAUSED);
        getBusinessObjectService().save(jobDetailsBo);
        List<OLEBatchProcessJobDetailsBo> oleBatchProcessJobDetailsBoList = (List<OLEBatchProcessJobDetailsBo> )getBusinessObjectService().findAll(OLEBatchProcessJobDetailsBo.class);
        oLEBatchProcessJobDetailsForm.setoLEBatchProcessJobDetailsBoList(oleBatchProcessJobDetailsBoList);
        getSchedulerService().pauseJob(jobDetailsBo.getJobId());
        return getUIFModelAndView(oLEBatchProcessJobDetailsForm);
    }


    @RequestMapping(params = "methodToCall=openViewReport")
    public ModelAndView openViewReport(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
                                       HttpServletRequest request, HttpServletResponse response) {
        OLEBatchProcessJobDetailsForm oLEBatchProcessJobDetailsForm = (OLEBatchProcessJobDetailsForm) form;
        OLEBatchProcessJobDetailsBo jobDetailsBo = getJobBo(oLEBatchProcessJobDetailsForm);
        if (jobDetailsBo.getBatchProcessType().equalsIgnoreCase(OLEConstants.OLEBatchProcess.BATCH_EXPORT)) {
            jobDetailsBo.setUploadFileName(getOLEBatchProcessDataHelper().getExportPathUrl(jobDetailsBo));
        }
        else if (jobDetailsBo.getBatchProcessType().equalsIgnoreCase(OLEConstants.OLEBatchProcess.BATCH_BIB_IMPORT)) {
            jobDetailsBo.setBibErrorPath(getOLEBatchProcessDataHelper().getBibPathUrl(jobDetailsBo));
        } else if (jobDetailsBo.getBatchProcessType().equalsIgnoreCase(OLEConstants.OLEBatchProcess.BATCH_DELETE)) {
            jobDetailsBo.setBatchDeletePath(getOLEBatchProcessDataHelper().getDeletePathUrl(jobDetailsBo));

        }
        List<OLEBatchProcessJobDetailsBo> oleBatchProcessJobDetailsBoList = (List<OLEBatchProcessJobDetailsBo> )getBusinessObjectService().findAll(OLEBatchProcessJobDetailsBo.class);
        oLEBatchProcessJobDetailsForm.setoLEBatchProcessJobDetailsBoList(oleBatchProcessJobDetailsBoList);
        oLEBatchProcessJobDetailsForm.setOleBatchProcessJobDetailsBo(jobDetailsBo);

        return getUIFModelAndView(oLEBatchProcessJobDetailsForm);
    }

    @RequestMapping(params = "methodToCall=downLoadFile")
    public ModelAndView downLoadFile(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
                                     HttpServletRequest request, HttpServletResponse response) throws Exception {

        OLEBatchProcessJobDetailsForm oleBatchProcessJobDetailsForm = (OLEBatchProcessJobDetailsForm) form;

        OLEBatchProcessJobDetailsBo jobDetailsBo = getJobBo(oleBatchProcessJobDetailsForm);
        if (jobDetailsBo != null) {
            Map map= new HashMap();
            map.put("batchProcessId",jobDetailsBo.getBatchProcessId());
            List<OLEBatchProcessDefinitionDocument> oleBatchProcessDefinitionDocumentList = (List<OLEBatchProcessDefinitionDocument>) getBusinessObjectService().findMatching(OLEBatchProcessDefinitionDocument.class, map);
            if(oleBatchProcessDefinitionDocumentList!=null && oleBatchProcessDefinitionDocumentList.size()>0){
            OLEBatchProcessDefinitionDocument oleBatchProcessDefinitionDocument=oleBatchProcessDefinitionDocumentList.get(0);
            File file = new File(getBatchProcessFilePath(oleBatchProcessDefinitionDocument.getBatchProcessType()) + jobDetailsBo.getJobId() + "_FailureRecord" + "_" + jobDetailsBo.getUploadFileName());
            if (!file.exists() || !file.isFile()) {
                throw new RuntimeException("Error: non-existent file or directory provided");
            }
            File containingDirectory = file.getParentFile();
            if (!BatchFileUtils.isDirectoryAccessible(containingDirectory.getAbsolutePath())) {
                throw new RuntimeException("Error: inaccessible directory provided");
            }

            BatchFile batchFile = new BatchFile();
            batchFile.setFile(file);
            response.setContentType("application/octet-stream");
            response.setHeader("Content-disposition", "attachment; filename=" + file.getName());
            response.setHeader("Expires", "0");
            response.setHeader("Cache-Control", "must-revalidate, post-check=0, pre-check=0");
            response.setHeader("Pragma", "public");
            response.setContentLength((int) file.length());

            InputStream fis = new FileInputStream(file);
            IOUtils.copy(fis, response.getOutputStream());
            response.getOutputStream().flush();
        }
        }
        return null;


    }

    @RequestMapping(params = "methodToCall=jobDocHandler")
    public ModelAndView jobDocHandler(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
                                   HttpServletRequest request, HttpServletResponse response) throws Exception {
        OLEBatchProcessJobDetailsForm oleBatchProcessJobDetailsForm = (OLEBatchProcessJobDetailsForm) form;
        ModelAndView modelAndView=super.docHandler(oleBatchProcessJobDetailsForm,result,request,response);
        String documentClass= request.getParameter("documentClass");
        if(documentClass!=null && OLEBatchProcessScheduleBo.class.toString().contains(documentClass)){
            List<OLEBatchProcessScheduleBo> oleBatchProcessScheduleBoList = (List<OLEBatchProcessScheduleBo>) getBusinessObjectService().findAll(OLEBatchProcessScheduleBo.class);
            for (OLEBatchProcessScheduleBo oleBatchProcessScheduleBo : oleBatchProcessScheduleBoList) {
                oleBatchProcessScheduleBo.setBatchProfileName(getBatchProcessDocument(oleBatchProcessScheduleBo.getBatchProcessId()).getBatchProcessProfileName());
            }
            oleBatchProcessJobDetailsForm.setOleBatchProcessScheduleBoList(oleBatchProcessScheduleBoList);
            oneTimeDate(oleBatchProcessScheduleBoList);
        }
        else{
            List<OLEBatchProcessJobDetailsBo> oLEBatchProcessJobDetailsBoList = (List<OLEBatchProcessJobDetailsBo>) getBusinessObjectService().findAll(OLEBatchProcessJobDetailsBo.class);
            oleBatchProcessJobDetailsForm.setoLEBatchProcessJobDetailsBoList(oLEBatchProcessJobDetailsBoList);
        }
        return modelAndView;
    }

    private OLEBatchProcessDefinitionDocument getBatchProcessDocument(String batchProcessId) {
        Map map = new HashMap();
        map.put("batchProcessId", batchProcessId);
        OLEBatchProcessDefinitionDocument batchProcessDefinitionDocument = getBusinessObjectService().findByPrimaryKey(OLEBatchProcessDefinitionDocument.class, map);
        return batchProcessDefinitionDocument;
    }

    private String getBatchProcessFilePath(String batchProceesType) {
        String batchProcessLocation =  getOLEBatchProcessDataHelper().getBatchProcessFilePath(batchProceesType);
        return batchProcessLocation;
    }

    /**
     * Runs the process for the profile selected and takes the scheduled record id as the job name
     * @param jobName
     */
    private void runJobNow(String jobName){
        OLEBatchSchedulerService schedulerService = GlobalResourceLoader.getService("oleSchedulerService");
        try {
            schedulerService.startJob(jobName);
        } catch (Exception e) {
            LOG.error("Error while starting job with job id :: "+jobName);
        }
    }

    private void oneTimeDate(List<OLEBatchProcessScheduleBo> oleBatchProcessScheduleBoList) {

        try {
            for (OLEBatchProcessScheduleBo oleBatchProcessScheduleBo : oleBatchProcessScheduleBoList) {
                CronExpression exp = new CronExpression(oleBatchProcessScheduleBo.getCronExpression());
                Date date = exp.getNextValidTimeAfter(new Date());
                if (date != null) {
                    SimpleDateFormat sdf = new SimpleDateFormat("MM/dd/yyyy HH:mm:ss");
                    Timestamp nextRunTime = new Timestamp(date.getTime());
                    oleBatchProcessScheduleBo.setNextRunTime(nextRunTime);
                }
            }
        } catch (ParseException e) {
            LOG.error("Error while validating cron exp::" + oleBatchProcessScheduleBoList.get(0).getCronExpression(), e);
        }


    }

    @RequestMapping(params = "methodToCall=removeScheduleJob")
    public ModelAndView removeScheduleJob(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
                                                HttpServletRequest request, HttpServletResponse response) throws Exception {
        OLEBatchProcessJobDetailsForm oleBatchProcessJobDetailsForm =  (OLEBatchProcessJobDetailsForm)form;
        Map map =new HashMap();
        map.put("scheduleId",request.getParameter("scheduleId"));
        OLEBatchProcessScheduleBo scheduleBo = getBusinessObjectService().findByPrimaryKey(OLEBatchProcessScheduleBo.class, map);
        getOLEBatchProcessDataHelper().deleteBatchFailureFile(scheduleBo.getBatchProcessType(),scheduleBo.getScheduleId() + "_Scheduled" + "_" + scheduleBo.getUploadFileName());
        getBusinessObjectService().delete(scheduleBo);
        removeJob(scheduleBo.getScheduleId());
        List<OLEBatchProcessScheduleBo> oleBatchProcessScheduleBoList = ( List<OLEBatchProcessScheduleBo> ) getBusinessObjectService().findAll(OLEBatchProcessScheduleBo.class);
        oleBatchProcessJobDetailsForm.setOleBatchProcessScheduleBoList(oleBatchProcessScheduleBoList);
        return  getUIFModelAndView(oleBatchProcessJobDetailsForm);

    }

    private void removeJob(String jobName){
        OLEBatchSchedulerService schedulerService = GlobalResourceLoader.getService("oleSchedulerService");
        try {
            schedulerService.deleteJob(jobName);
        } catch (Exception e) {
            LOG.error("Error while removing job with schedule id :: "+jobName);
        }
    }

    private OLEBatchProcessJobDetailsBo getJobBo(OLEBatchProcessJobDetailsForm oleBatchProcessJobDetailsForm) {
        Map map = new HashMap();
        map.put("jobId", oleBatchProcessJobDetailsForm.getJobId());
        OLEBatchProcessJobDetailsBo jobDetailsBo = getBusinessObjectService().findByPrimaryKey(OLEBatchProcessJobDetailsBo.class, map);
        return jobDetailsBo;
    }


}