PatronBillMaintenanceDocumentController

package org.kuali.ole.patron.bill;

import org.kuali.ole.OLEConstants;
import org.kuali.ole.service.impl.OlePatronBillMaintenanceDocumentServiceImpl;
import org.kuali.rice.core.api.resourceloader.GlobalResourceLoader;
import org.kuali.rice.krad.bo.PersistableAttachment;
import org.kuali.rice.krad.bo.PersistableBusinessObject;
import org.kuali.rice.krad.maintenance.MaintenanceDocument;
import org.kuali.rice.krad.maintenance.MaintenanceUtils;
import org.kuali.rice.krad.service.BusinessObjectService;
import org.kuali.rice.krad.service.KRADServiceLocator;
import org.kuali.rice.krad.service.MaintenanceDocumentService;
import org.kuali.rice.krad.util.GlobalVariables;
import org.kuali.rice.krad.util.KRADConstants;
import org.kuali.rice.krad.web.controller.MaintenanceDocumentController;
import org.kuali.rice.krad.web.form.DocumentFormBase;
import org.kuali.rice.krad.web.form.MaintenanceForm;
import org.kuali.rice.krad.web.form.UifFormBase;
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.math.BigDecimal;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Created with IntelliJ IDEA.
 * User: ?
 * Date: 10/5/12
 * Time: 2:21 PM
 * To change this template use File | Settings | File Templates.
 */
@Controller
@RequestMapping(value = "/patronBillMaintenance")
public class PatronBillMaintenanceDocumentController extends MaintenanceDocumentController {

    private static final org.apache.log4j.Logger LOG = org.apache.log4j.Logger.getLogger(PatronBillMaintenanceDocumentController.class);
    private BusinessObjectService businessObjectService;

    /**
     *  Gets the businessObjectService attribute.
     * @return  Returns the businessObjectService
     */
    public BusinessObjectService getBusinessObjectService() {
        if (null == businessObjectService) {
            businessObjectService = KRADServiceLocator.getBusinessObjectService();
        }
        return businessObjectService;
    }

    @RequestMapping(params = "methodToCall=" + KRADConstants.Maintenance.METHOD_TO_CALL_NEW)
    @Override
    public ModelAndView start(@ModelAttribute("KualiForm") UifFormBase form, BindingResult result,
                              HttpServletRequest request, HttpServletResponse response) {
        Date billDate = new Date();
        MaintenanceForm maintenanceForm = (MaintenanceForm) form;
        setupMaintenance(maintenanceForm, request, KRADConstants.MAINTENANCE_NEW_ACTION);
        MaintenanceDocument document = (MaintenanceDocument) maintenanceForm.getDocument();
        PatronBillPayment patronBillPayment=(PatronBillPayment)   document.getDocumentDataObject();
        patronBillPayment.setPatronId(request.getParameter("patronId"));
        patronBillPayment.setFirstName(request.getParameter("firstName"));
        patronBillPayment.setLastName(request.getParameter("lastName"));

        patronBillPayment.setBillDate(billDate);
        return getUIFModelAndView(maintenanceForm);
    }


    /**
     * This method returns the instance of olePatronBillMaintenanceDocumentService
     * @return olePatronMaintenanceDocumentService(MaintenanceDocumentService)
     */
    @Override
    protected MaintenanceDocumentService getMaintenanceDocumentService() {
        return GlobalResourceLoader.getService("olePatronBillMaintenanceDocumentService");
    }


    /**
     * This method will add the patron bill  document to the existing list and also stores the attachment
     *   to the specified path.
     *
     * @param uifForm - MaintenanceForm
     * @return ModelAndView
     */
    @RequestMapping(params = "methodToCall=addToTotalAmount")
    public ModelAndView addToTotalAmount(@ModelAttribute("KualiForm") UifFormBase uifForm, BindingResult result,
                                                HttpServletRequest request, HttpServletResponse response) throws Exception{
        ModelAndView modelAndView=super.addLine(uifForm, result, request, response);
        MaintenanceForm form = (MaintenanceForm) uifForm;
        MaintenanceDocument document = (MaintenanceDocument) form.getDocument();
        PatronBillPayment patronBillPayment = (PatronBillPayment)document.getNewMaintainableObject().getDataObject();
        BigDecimal amount = new BigDecimal(0.0);
        if(patronBillPayment.getFeeType() !=null ) {
            List<FeeType> feeTypes = patronBillPayment.getFeeType();
            for (int i=0;i<feeTypes.size();i++) {
                amount = amount.add(feeTypes.get(i).getFeeAmount());
            }
            patronBillPayment.setTotalAmount(amount);

        }
        return modelAndView;
    }


    /**
     * This method will add the agreement document to the existing list and also stores the attachment
     *   to the specified path.
     *
     * @param uifForm - MaintenanceForm
     * @return ModelAndView
     */
    @RequestMapping(params = "methodToCall=deleteFromTotalAmount")
    public ModelAndView deleteFromTotalAmount(@ModelAttribute("KualiForm") UifFormBase uifForm, BindingResult result,
                                         HttpServletRequest request, HttpServletResponse response) {

        ModelAndView modelAndView=super.deleteLine(uifForm, result, request, response);
        MaintenanceForm form = (MaintenanceForm) uifForm;
        MaintenanceDocument document = (MaintenanceDocument) form.getDocument();
        PatronBillPayment patronBillPayment = (PatronBillPayment)document.getNewMaintainableObject().getDataObject();
        BigDecimal amount = new BigDecimal(0.0);
        String billNumber="";
        Map feeTypeId = new HashMap();
        if(patronBillPayment.getFeeType() !=null ) {
            List<FeeType> feeTypes = patronBillPayment.getFeeType();
            for (int i=0;i<feeTypes.size();i++) {
                amount = amount.add(feeTypes.get(i).getFeeAmount());
                billNumber = feeTypes.get(i).getBillNumber();
                feeTypeId.put("billNumber",billNumber);
            }
            patronBillPayment.setTotalAmount(amount);
            getBusinessObjectService().deleteMatching(FeeType.class,feeTypeId);

        }
        return modelAndView;
    }


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

        MaintenanceForm form = (MaintenanceForm) uifForm;
        MaintenanceDocument document = (MaintenanceDocument) form.getDocument();
        PatronBillPayment patronBillPayment = (PatronBillPayment)document.getNewMaintainableObject().getDataObject();
        BigDecimal amount = new BigDecimal(0.0);
        if(patronBillPayment.getFeeType() !=null ) {
            List<FeeType> feeTypes = patronBillPayment.getFeeType();
            for (int i=0;i<feeTypes.size();i++) {
                amount = amount.add(feeTypes.get(i).getFeeAmount());
            }
            patronBillPayment.setTotalAmount(amount);
        }
        return getUIFModelAndView(form);
    }


    @RequestMapping(params = "methodToCall=" +"maintenanceDelete")
    public ModelAndView maintenanceDelete(@ModelAttribute("KualiForm") MaintenanceForm form, BindingResult result,
                                          HttpServletRequest request, HttpServletResponse response) throws Exception {
        setupMaintenanceForDelete(form, request,"Delete");
        return getUIFModelAndView(form);
    }


    @RequestMapping(params = "methodToCall=" +"deleteDocument")
    public ModelAndView deleteDocument(@ModelAttribute("KualiForm") MaintenanceForm form, BindingResult result,
                                       HttpServletRequest request, HttpServletResponse response) throws Exception {
        MaintenanceDocument document = form.getDocument();
        PatronBillPayment patronBillPayment = new PatronBillPayment();
        //OleProxyPatronDocument proxyPatronDocument = new OleProxyPatronDocument();

        if(document.getDocumentDataObject() != null) {
            patronBillPayment = (PatronBillPayment)document.getDocumentDataObject();
            if(patronBillPayment != null && patronBillPayment.getPatronId() != null) {
                KRADServiceLocator.getBusinessObjectService().delete(patronBillPayment);
                GlobalVariables.getMessageMap().putInfoWithoutFullErrorPath(KRADConstants.GLOBAL_ERRORS,
                        OLEConstants.OleLicenseRequest.MSG_DELETE_DOC);
            }
            else {
                GlobalVariables.getMessageMap().putErrorWithoutFullErrorPath(KRADConstants.GLOBAL_MESSAGES, OLEConstants.OleLicenseRequest.ERROR_CHECKLIST_NOT_FOUND);
                return getUIFModelAndView(form);
            }
        }
        return close(form, result, request, response);
    }


    /**
     * This method populates confirmation to delete the document.
     * @param form
     * @param request
     * @param maintenanceAction
     */
    protected void setupMaintenanceForDelete(MaintenanceForm form, HttpServletRequest request, String maintenanceAction) {
        MaintenanceDocument document = form.getDocument();

        if (document == null) {
            document = getMaintenanceDocumentService()
                    .setupNewMaintenanceDocument(form.getDataObjectClassName(), form.getDocTypeName(),
                            maintenanceAction);

            form.setDocument(document);
            form.setDocTypeName(document.getDocumentHeader().getWorkflowDocument().getDocumentTypeName());
        }

        form.setMaintenanceAction(maintenanceAction);
        OlePatronBillMaintenanceDocumentServiceImpl olePatronBillMaintenanceDocumentService = (OlePatronBillMaintenanceDocumentServiceImpl)getMaintenanceDocumentService();
        olePatronBillMaintenanceDocumentService.setupMaintenanceObjectForDelete(document, maintenanceAction, request.getParameterMap());
        MaintenanceUtils.checkForLockingDocument(document, false);
    }

    /**
     * Override route to retrieve the maintenance object if it is an attachment
     *
     *      (DocumentFormBase, HttpServletRequest, HttpServletResponse)
     */
    @Override
    @RequestMapping(params = "methodToCall=route")
    public ModelAndView route(@ModelAttribute("KualiForm") DocumentFormBase form, BindingResult result,
                              HttpServletRequest request, HttpServletResponse response) {

        ModelAndView modelAndView;

        modelAndView = super.route(form, result, request, response);
        try{
            calculateTotal(form, result, request, response);
        }catch(Exception e){

        }

        MaintenanceDocument document = (MaintenanceDocument) form.getDocument();
        if (document.getNewMaintainableObject().getDataObject() instanceof PersistableAttachment) {
            PersistableAttachment bo = (PersistableAttachment) getBusinessObjectService()
                    .retrieve((PersistableBusinessObject) document.getNewMaintainableObject().getDataObject());
            request.setAttribute("fileName", bo.getFileName());
        }

        modelAndView = getUIFModelAndView(form);

        return modelAndView;
    }


}