package org.kuali.ole.batch.impl;
   
   import org.apache.commons.lang.StringUtils;
   import org.apache.log4j.Logger;
   import org.apache.velocity.Template;
   import org.apache.velocity.VelocityContext;
   import org.apache.velocity.app.VelocityEngine;
   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.helper.OLEBatchProcessDataHelper;
  import org.kuali.ole.deliver.batch.OleMailer;
  import org.kuali.ole.sys.batch.AbstractStep;
  import org.kuali.ole.sys.batch.service.SchedulerService;
  import org.kuali.rice.core.api.criteria.QueryByCriteria;
  import org.kuali.rice.core.api.mail.EmailBody;
  import org.kuali.rice.core.api.mail.EmailFrom;
  import org.kuali.rice.core.api.mail.EmailSubject;
  import org.kuali.rice.core.api.mail.EmailToList;
  import org.kuali.rice.core.api.resourceloader.GlobalResourceLoader;
  import org.kuali.rice.kns.service.KNSServiceLocator;
  import org.kuali.rice.krad.service.KRADServiceLocator;
  import org.kuali.rice.krad.service.KRADServiceLocatorWeb;
  
  import java.io.StringWriter;
  import java.util.*;
  
  /**
   * Created with IntelliJ IDEA.
   * User: meenrajd
   * Date: 7/25/13
   * Time: 4:02 PM
   * To change this template use File | Settings | File Templates.
   */
  public class OLEBatchProcessEmailStep extends AbstractStep {
      private static final Logger LOG = Logger.getLogger(OLEBatchProcessEmailStep.class);
      private static final String FROMEMAIL = "BATCHJOBADMIN@KUALI.ORG";
  
      private OleMailer oleMailer;
  
      private OleMailer getOleMailer() {
          if (oleMailer == null) {
              oleMailer = GlobalResourceLoader.getService("oleMailer");
          }
          return oleMailer;
      }
  
      private OLEBatchProcessDataHelper oleBatchProcessDataHelper;
  
      private OLEBatchProcessDataHelper getOLEBatchProcessDataHelper() {
  
          if (oleBatchProcessDataHelper == null) {
              oleBatchProcessDataHelper = OLEBatchProcessDataHelper.getInstance();
          }
          return oleBatchProcessDataHelper;
      }
  
      @Override
      public boolean execute(String jobName, Date jobRunDate) throws InterruptedException {
          buildJobReport(jobName);
          return true;
      }
  
      private void buildJobReport(String jobName) {
          try {
              String jobId = StringUtils.substringAfter(jobName, SchedulerService.SCHEDULED_GROUP + "." + OLEConstants.OLEBatchProcess.ADHOC_BATCH_JOB);
              String scheduleId = StringUtils.substringAfter(jobName, SchedulerService.SCHEDULED_GROUP + "." + OLEConstants.OLEBatchProcess.BATCH_JOB);
              if (StringUtils.isNotEmpty(jobId)) {
                  OLEBatchProcessJobDetailsBo jobDetailsBo = readJobRecord(jobId);
                  OLEBatchProcessDefinitionDocument processDefDoc = readOLEBatchProcessDefinitionDocument(jobDetailsBo.getBatchProcessId());
                  sendJobReportMail(processDefDoc, jobDetailsBo);
              } else if (StringUtils.isNotEmpty(scheduleId)) {
                  OLEBatchProcessScheduleBo scheduleBo = readScheduleRecord(scheduleId);
                  OLEBatchProcessJobDetailsBo jobDetailsBo = readJobRecordForSch(scheduleBo.getBatchProcessId());
                  if(jobDetailsBo==null){
                      LOG.error("Error while Running step OLEBatchProcessEmailStep:: Unable to get job details for process id:: "+scheduleBo.getBatchProcessId());
                  }else{
                      OLEBatchProcessDefinitionDocument processDefDoc = readOLEBatchProcessDefinitionDocument(jobDetailsBo.getBatchProcessId());
                      sendJobReportMail(processDefDoc, jobDetailsBo);
                  }
              }
          } catch (Exception ex) {
              LOG.error("Error while Running step OLEBatchProcessEmailStep:: ", ex);
          }
      }
  
      /**
       * reads the job record which is being run
       *
       * @param processId
       * @return
       */
      private OLEBatchProcessJobDetailsBo readJobRecordForSch(String processId) {
          Map<String, String> jobMap = new HashMap<String, String>();
          jobMap.put("bat_prcs_id", processId);
          List<OLEBatchProcessJobDetailsBo> jobDetailsBoList = (List<OLEBatchProcessJobDetailsBo>) KRADServiceLocatorWeb.getLegacyDataAdapter().findMatchingOrderBy(OLEBatchProcessJobDetailsBo.class, jobMap, "JOB_ID", false);
          if (jobDetailsBoList.isEmpty()) {
              return null;
         } else {
             return jobDetailsBoList.get(0);
         }
     }
 
     /**
      * reads the job record which is being run
      *
      * @param jobId
      * @return
      */
     private OLEBatchProcessJobDetailsBo readJobRecord(String jobId) {
         Map<String, String> jobMap = new HashMap<String, String>();
         jobMap.put("job_id", jobId);
         OLEBatchProcessJobDetailsBo jobDetailsBo = KRADServiceLocatorWeb.getLegacyDataAdapter().findByPrimaryKey(OLEBatchProcessJobDetailsBo.class, jobMap);
         return jobDetailsBo;
     }
 
     /**
      * reads the schedule record which is being run
      *
      * @param scheduleId
      * @return
      */
     private OLEBatchProcessScheduleBo readScheduleRecord(String scheduleId) {
         Map<String, String> schMap = new HashMap<String, String>();
         schMap.put("ole_bat_prcs_schdule_id", scheduleId);
         OLEBatchProcessScheduleBo scheduleBo = KRADServiceLocator.getDataObjectService().findUnique(OLEBatchProcessScheduleBo.class, QueryByCriteria.Builder.andAttributes(schMap).build());
         return scheduleBo;
     }
 
     /**
      * Read the OLEBatchProcessDefinitionDocument record by using batchProcessId.
      *
      * @param batchProcessId
      * @return
      */
     private OLEBatchProcessDefinitionDocument readOLEBatchProcessDefinitionDocument(String batchProcessId) {
         Map<String, String> jobMap = new HashMap<String, String>();
         jobMap.put("bat_prcs_id", batchProcessId);
         OLEBatchProcessDefinitionDocument processDefDoc = KRADServiceLocatorWeb.getLegacyDataAdapter().findByPrimaryKey(OLEBatchProcessDefinitionDocument.class, jobMap);
         return processDefDoc;
     }
 
     /**
      * Sends mail to the given recipients
      *
      * @param processDefDoc, job
      * @return
      */
     private void sendJobReportMail(OLEBatchProcessDefinitionDocument processDefDoc, OLEBatchProcessJobDetailsBo job) throws Exception {
         if (StringUtils.isNotBlank(processDefDoc.getEmailIds())) {
             String[] emailsIdsArray = processDefDoc.getEmailIds().split(",");
             if (Integer.parseInt(job.getNoOfSuccessRecords()) <= 0) {
                 if (processDefDoc.getBatchProcessType().equals(OLEConstants.OLEBatchProcess.BATCH_DELETE)){
                     getOleMailer().sendEmail(new EmailFrom(FROMEMAIL), new EmailToList(Arrays.asList(emailsIdsArray)),
                             new EmailSubject(OLEConstants.OLEBatchProcess.BATCH_DELETION_JOB + (StringUtils.isEmpty(processDefDoc.getBatchProcessName()) ? "" : processDefDoc.getBatchProcessName()) + OLEConstants.OLEBatchProcess.FAILED),
                             new EmailBody(getReportByVM(job,processDefDoc)), null, null, true);
                 }
                 else {
                     getOleMailer().sendEmail(new EmailFrom(FROMEMAIL), new EmailToList(Arrays.asList(emailsIdsArray)),
                             new EmailSubject((StringUtils.isEmpty(processDefDoc.getBatchProcessName()) ? "" : processDefDoc.getBatchProcessName() + "-") + processDefDoc.getBatchProcessType() + OLEConstants.OLEBatchProcess.FAILED),
                             new EmailBody(getReportByVM(job,processDefDoc)), null, null, true);
                 }
             } else {
                 if (processDefDoc.getBatchProcessType().equals(OLEConstants.OLEBatchProcess.BATCH_DELETE)){
                     getOleMailer().sendEmail(new EmailFrom(FROMEMAIL), new EmailToList(Arrays.asList(emailsIdsArray)),
                             new EmailSubject(OLEConstants.OLEBatchProcess.BATCH_DELETION_JOB + (StringUtils.isEmpty(processDefDoc.getBatchProcessName()) ? "" : processDefDoc.getBatchProcessName()) + OLEConstants.OLEBatchProcess.COMPLETE),
                             new EmailBody(getReportByVM(job,processDefDoc)), null, null, true);
                 }
                 else {
                     getOleMailer().sendEmail(new EmailFrom(FROMEMAIL), new EmailToList(Arrays.asList(emailsIdsArray)),
                             new EmailSubject((StringUtils.isEmpty(processDefDoc.getBatchProcessName()) ? "" : processDefDoc.getBatchProcessName() + "-") + processDefDoc.getBatchProcessType() + OLEConstants.OLEBatchProcess.COMPLETE),
                             new EmailBody(getReportByVM(job,processDefDoc)), null, null, true);
                 }
             }
         }
     }
     /**
      *
      * @param job
      * @return
      * @throws Exception
      */
     private String getReportByVM(OLEBatchProcessJobDetailsBo job,OLEBatchProcessDefinitionDocument processDefDoc) throws Exception {
         VelocityEngine ve = GlobalResourceLoader.getService("velocityEngine");
         ve.init();
         VelocityContext context = new VelocityContext();
         context.put("job", job);
         context.put("processDefDoc", processDefDoc);
         context.put("dataHelper", OLEBatchProcessDataHelper.getInstance());
         context.put("Integer", Integer.class);
         context.put("StringUtils", StringUtils.class);
         Template t = ve.getTemplate("jobreport.vm");
         StringWriter writer = new StringWriter();
         t.merge(context, writer);
         return writer.toString();
     }
 
 
 }