/*
    * Copyright 2007 The Kuali Foundation
    * 
    * Licensed under the Educational Community 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.opensource.org/licenses/ecl2.php
    * 
   * 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.kuali.ole.pdp.dataaccess.impl;
  
  import java.sql.Timestamp;
  import java.util.Iterator;
  import java.util.List;
  
  import org.apache.ojb.broker.query.Criteria;
  import org.apache.ojb.broker.query.QueryByCriteria;
  import org.kuali.ole.pdp.PdpConstants;
  import org.kuali.ole.pdp.PdpPropertyConstants;
  import org.kuali.ole.pdp.businessobject.PaymentGroup;
  import org.kuali.ole.pdp.businessobject.PaymentProcess;
  import org.kuali.ole.pdp.dataaccess.FormatPaymentDao;
  import org.kuali.rice.core.framework.persistence.ojb.dao.PlatformAwareDaoBaseOjb;
  import org.kuali.rice.krad.service.BusinessObjectService;
  
  public class FormatPaymentDaoOjb extends PlatformAwareDaoBaseOjb implements FormatPaymentDao {
      private static org.apache.log4j.Logger LOG = org.apache.log4j.Logger.getLogger(FormatPaymentDaoOjb.class);
  
      private BusinessObjectService businessObjectService;
  
      /**
       * @see org.kuali.ole.pdp.dataaccess.FormatPaymentDao#markPaymentsForFormat(org.kuali.ole.pdp.businessobject.PaymentProcess,
       *      java.util.List, java.util.Date, java.lang.String)
       */
      public Iterator markPaymentsForFormat(List customerIds, Timestamp paydateTs, String paymentTypes) {
          LOG.debug("markPaymentsForFormat() started");
  
          Criteria criteria = new Criteria();
  
          if (customerIds.size() > 0) {
              criteria.addIn(PdpPropertyConstants.PaymentGroup.PAYMENT_GROUP_BATCH + "." + PdpPropertyConstants.BatchConstants.CUSTOMER_ID, customerIds);
          }
  
          criteria.addEqualTo(PdpPropertyConstants.PaymentGroup.PAYMENT_GROUP_PAYMENT_STATUS_CODE, PdpConstants.PaymentStatusCodes.OPEN);
  
          if (PdpConstants.PaymentTypes.DISBURSEMENTS_WITH_SPECIAL_HANDLING.equals(paymentTypes)) {
              // special handling only
              criteria.addEqualTo(PdpPropertyConstants.PaymentGroup.PAYMENT_SPECIAL_HANDLING, Boolean.TRUE);
          }
          else if (PdpConstants.PaymentTypes.DISBURSEMENTS_NO_SPECIAL_HANDLING.equals(paymentTypes)) {
              // no special handling only
              criteria.addEqualTo(PdpPropertyConstants.PaymentGroup.PAYMENT_SPECIAL_HANDLING, Boolean.FALSE);
          }
          else if (PdpConstants.PaymentTypes.DISBURSEMENTS_WITH_ATTACHMENTS.equals(paymentTypes)) {
              // attachments only
              criteria.addEqualTo(PdpPropertyConstants.PaymentGroup.PAYMENT_ATTACHMENT, Boolean.TRUE);
          }
          else if (PdpConstants.PaymentTypes.DISBURSEMENTS_NO_ATTACHMENTS.equals(paymentTypes)) {
              // no attachments only
              criteria.addEqualTo(PdpPropertyConstants.PaymentGroup.PAYMENT_ATTACHMENT, Boolean.FALSE);
          }
  
          if (PdpConstants.PaymentTypes.PROCESS_IMMEDIATE.equals(paymentTypes)) {
              criteria.addEqualTo(PdpPropertyConstants.PaymentGroup.PROCESS_IMMEDIATE, Boolean.TRUE);
          }
          else {
              // (Payment date <= usePaydate OR immediate = TRUE)
              Criteria criteria1 = new Criteria();
              criteria1.addEqualTo(PdpPropertyConstants.PaymentGroup.PROCESS_IMMEDIATE, Boolean.TRUE);
  
              Criteria criteria2 = new Criteria();
              criteria2.addLessOrEqualThan(PdpPropertyConstants.PaymentGroup.PAYMENT_DATE, paydateTs);
              criteria1.addOrCriteria(criteria2);
  
              criteria.addAndCriteria(criteria1);
          }
  
          Iterator groupIterator = getPersistenceBrokerTemplate().getIteratorByQuery(new QueryByCriteria(PaymentGroup.class, criteria));
          return groupIterator;
      }
  
      /**
       * @see org.kuali.ole.pdp.dataaccess.FormatPaymentDao#unmarkPaymentsForFormat(org.kuali.ole.pdp.businessobject.PaymentProcess)
       */
      public Iterator unmarkPaymentsForFormat(PaymentProcess proc) {
          LOG.debug("unmarkPaymentsForFormat() started");
  
          Criteria criteria = new Criteria();
          criteria.addEqualTo(PdpPropertyConstants.PaymentGroup.PAYMENT_GROUP_PROCESS_ID, proc.getId());
          criteria.addEqualTo(PdpPropertyConstants.PaymentGroup.PAYMENT_GROUP_PAYMENT_STATUS_CODE, PdpConstants.PaymentStatusCodes.FORMAT);
  
          Iterator groupIterator = getPersistenceBrokerTemplate().getIteratorByQuery(new QueryByCriteria(PaymentGroup.class, criteria));
          return groupIterator;
     }
 
     /**
      * Gets the businessObjectService attribute.
      * 
      * @return Returns the KRADServiceLocatorWeb.getLegacyDataAdapter().
      */
     public BusinessObjectService getBusinessObjectService() {
         return businessObjectService;
     }
 
     /**
      * Sets the businessObjectService attribute value.
      * 
      * @param businessObjectService The businessObjectService to set.
      */
     public void setBusinessObjectService(BusinessObjectService businessObjectService) {
         this.businessObjectService = businessObjectService;
     }
 
 }