PatronBillMaintenanceDocumentController

package org.kuali.ole.patron.bill;

import org.apache.commons.lang.StringUtils;
import org.kuali.incubator.SolrRequestReponseHandler;
import org.kuali.ole.OLEConstants;
import org.kuali.ole.license.bo.OleEventLogBo;
import org.kuali.ole.patron.bo.OlePatronDocument;
import org.kuali.ole.patron.bo.OleProxyPatronDocument;
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.uif.UifParameters;
import org.kuali.rice.krad.uif.container.CollectionGroup;
import org.kuali.rice.krad.uif.util.ObjectPropertyUtils;
import org.kuali.rice.krad.uif.view.View;
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.bind.annotation.RequestMethod;
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;
    }

    /**
     * Default method for controller that setups a new
     * <code>MaintenanceView</code> with the default new action
     */
    @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(OLEConstants.PTRN_ID));
        patronBillPayment.setFirstName(request.getParameter(OLEConstants.PTRN_FN));
        patronBillPayment.setLastName(request.getParameter(OLEConstants.PTRN_LN));
        Map proxyPatronMap = new HashMap();
        String firstName = "";
        String lastName = "";
        proxyPatronMap.put("olePatronId",request.getParameter(OLEConstants.PTRN_ID));
        List<OleProxyPatronDocument> oleProxyPatronDocumentList = (List<OleProxyPatronDocument>)KRADServiceLocator.getBusinessObjectService().findMatching(OleProxyPatronDocument.class,proxyPatronMap);
        String proxyPatronId = "";
        if(oleProxyPatronDocumentList.size() > 0){
            proxyPatronId = oleProxyPatronDocumentList.get(0).getProxyPatronId();
            Map patronMap = new HashMap();
            patronMap.put("olePatronId",proxyPatronId);
            List<OlePatronDocument> olePatronDocumentList = (List<OlePatronDocument>)KRADServiceLocator.getBusinessObjectService().findMatching(OlePatronDocument.class,patronMap);
            if(olePatronDocumentList.size() > 0){
                 firstName = olePatronDocumentList.get(0).getEntity().getNames().get(0).getFirstName();
                 lastName = olePatronDocumentList.get(0).getEntity().getNames().get(0).getLastName();
                patronBillPayment.setProxyFirstName(firstName);
                patronBillPayment.setProxyLastName(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 {
        LOG.debug("Initialized addToTotalAmount method");
        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;
    }


    /**
     * Called by the add line action for a new collection line. Method
     * determines which collection the add action was selected for and invokes
     * the view helper service to add the line
     */
    @Override
    @RequestMapping(method = RequestMethod.POST, params = "methodToCall=addLine")
    public ModelAndView addLine(@ModelAttribute("KualiForm") UifFormBase uifForm, BindingResult result,
                                HttpServletRequest request, HttpServletResponse response) {
        LOG.debug("Initialized addLine method");
        MaintenanceForm form = (MaintenanceForm) uifForm;
        String selectedCollectionPath = form.getActionParamaterValue(UifParameters.SELLECTED_COLLECTION_PATH);
        CollectionGroup collectionGroup = form.getPostedView().getViewIndex().getCollectionGroupByPath(
                selectedCollectionPath);
        String addLinePath = collectionGroup.getAddLineBindingInfo().getBindingPath();
        SolrRequestReponseHandler solrRequestReponseHandler = new SolrRequestReponseHandler();
        Object eventObject = ObjectPropertyUtils.getPropertyValue(uifForm, addLinePath);
        FeeType feeType = (FeeType) eventObject;
        String itemBarcode = feeType.getItemBarcode();
        View view = uifForm.getPostedView();
        if (itemBarcode == "" || !(itemBarcode.isEmpty())) {
            List<HashMap<String, Object>> documentList = solrRequestReponseHandler.retriveResults(OLEConstants.ITM_BAR_DISP + itemBarcode);
            if (documentList != null && documentList.size() == 0) {
                GlobalVariables.getMessageMap().putError(KRADConstants.GLOBAL_ERRORS, OLEConstants.ITM_BAR_NOT_AVAL);
                return updateComponent(uifForm, result, request, response);
            }
            GlobalVariables.getMessageMap().removeFromErrorPath(KRADConstants.GLOBAL_ERRORS);
        }
        view.getViewHelperService().processCollectionAddLine(view, uifForm, selectedCollectionPath);
        return updateComponent(uifForm, result, request, response);
    }

    /**
     * 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;
    }


    /**
     * This method will add the all fee amount and calculate total
     *
     * @param uifForm - MaintenanceForm
     * @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 calculated total amount
     * <p/>
     * (DocumentFormBase, HttpServletRequest, HttpServletResponse)
     */
    @Override
   @RequestMapping(params = "methodToCall=route")
    public ModelAndView route(@ModelAttribute("KualiForm") DocumentFormBase form, BindingResult result,
                              HttpServletRequest request, HttpServletResponse response) {
        ModelAndView modelAndView;
        MaintenanceForm maintenanceForm = (MaintenanceForm) form;

       try {
            calculateTotal(form, result, request, response);
        } catch (Exception e) {
            e.printStackTrace();
        }
        MaintenanceDocument document = (MaintenanceDocument) form.getDocument();
        PatronBillPayment patronBillPayment = (PatronBillPayment) document.getNewMaintainableObject().getDataObject();
        if (patronBillPayment.getFeeType() != null) {
            List<FeeType> feeTypes = patronBillPayment.getFeeType();
            if(feeTypes.size() ==0){
                GlobalVariables.getMessageMap().putError(KRADConstants.GLOBAL_ERRORS, OLEConstants.ADD_FEE_TYPE);
                return getUIFModelAndView(maintenanceForm);
            }
        }
        if (document.getNewMaintainableObject().getDataObject() instanceof PersistableAttachment) {
            PersistableAttachment bo = (PersistableAttachment) getBusinessObjectService()
                    .retrieve((PersistableBusinessObject) document.getNewMaintainableObject().getDataObject());
            request.setAttribute("fileName", bo.getFileName());
        }
        //modelAndView = getUIFModelAndView(form);
        modelAndView = super.route(maintenanceForm, result, request, response);
        return modelAndView;
    }
}