001    /**
002     * Copyright 2004-2013 The Kuali Foundation
003     *
004     * Licensed under the Educational Community License, Version 2.0 (the "License");
005     * you may not use this file except in compliance with the License.
006     * You may obtain a copy of the License at
007     *
008     * http://www.opensource.org/licenses/ecl2.php
009     *
010     * Unless required by applicable law or agreed to in writing, software
011     * distributed under the License is distributed on an "AS IS" BASIS,
012     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013     * See the License for the specific language governing permissions and
014     * limitations under the License.
015     */
016    package org.kuali.hr.time.admin.web;
017    
018    import javax.servlet.http.HttpServletRequest;
019    import javax.servlet.http.HttpServletResponse;
020    
021    import org.apache.commons.lang.StringUtils;
022    import org.apache.struts.action.ActionForm;
023    import org.apache.struts.action.ActionForward;
024    import org.apache.struts.action.ActionMapping;
025    import org.kuali.hr.lm.workflow.LeaveCalendarDocumentHeader;
026    import org.kuali.hr.time.base.web.TkAction;
027    import org.kuali.hr.time.service.base.TkServiceLocator;
028    import org.kuali.hr.time.workflow.TimesheetDocumentHeader;
029    
030    public class DeleteDocumentAction extends TkAction {
031    
032        public ActionForward deleteDocument(ActionMapping mapping, ActionForm form, HttpServletRequest request, HttpServletResponse response) throws Exception {
033            DeleteDocumentForm deleteDocumentForm = (DeleteDocumentForm) form;
034            String documentId = deleteDocumentForm.getDeleteDocumentId();
035    
036            if (StringUtils.isNotBlank(documentId)) {
037                TimesheetDocumentHeader tdh = TkServiceLocator.getTimesheetDocumentHeaderService().getDocumentHeader(documentId);
038                LeaveCalendarDocumentHeader ldh = TkServiceLocator.getLeaveCalendarDocumentHeaderService().getDocumentHeader(documentId);
039                
040                if(tdh != null) {
041                    TkServiceLocator.getTimeBlockService().deleteTimeBlocksAssociatedWithDocumentId(documentId);
042                        TkServiceLocator.getTimesheetService().deleteTimesheet(documentId);
043                } else if (ldh != null) {
044                    TkServiceLocator.getLeaveBlockService().deleteLeaveBlocksForDocumentId(documentId);
045                    TkServiceLocator.getLeaveCalendarDocumentHeaderService().deleteLeaveCalendarHeader(documentId);
046                }
047            }
048            
049            return mapping.findForward("basic");
050        }
051    
052    }