1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.kuali.hr.time.approval.web;
17
18 import java.text.SimpleDateFormat;
19 import java.util.ArrayList;
20 import java.util.Date;
21 import java.util.HashMap;
22 import java.util.LinkedList;
23 import java.util.List;
24 import java.util.Map;
25 import java.util.Set;
26
27 import javax.servlet.http.HttpServletRequest;
28 import javax.servlet.http.HttpServletResponse;
29
30 import com.google.common.collect.Lists;
31 import org.apache.commons.collections.MapUtils;
32 import org.apache.commons.lang.StringUtils;
33 import org.apache.commons.lang.time.DateUtils;
34 import org.apache.log4j.Logger;
35 import org.apache.struts.action.ActionForm;
36 import org.apache.struts.action.ActionForward;
37 import org.apache.struts.action.ActionMapping;
38 import org.hsqldb.lib.StringUtil;
39 import org.json.simple.JSONValue;
40 import org.kuali.hr.lm.leaveblock.LeaveBlock;
41 import org.kuali.hr.time.assignment.Assignment;
42 import org.kuali.hr.time.base.web.ApprovalForm;
43 import org.kuali.hr.time.base.web.TkAction;
44 import org.kuali.hr.time.detail.web.ActionFormUtils;
45 import org.kuali.hr.time.service.base.TkServiceLocator;
46 import org.kuali.hr.time.timeblock.TimeBlock;
47 import org.kuali.hr.time.timesheet.TimesheetDocument;
48 import org.kuali.hr.time.timesummary.AssignmentColumn;
49 import org.kuali.hr.time.timesummary.AssignmentRow;
50 import org.kuali.hr.time.timesummary.EarnCodeSection;
51 import org.kuali.hr.time.timesummary.EarnGroupSection;
52 import org.kuali.hr.time.timesummary.TimeSummary;
53 import org.kuali.hr.time.util.TKContext;
54 import org.kuali.hr.time.util.TKUtils;
55 import org.kuali.hr.time.workflow.TimesheetDocumentHeader;
56
57 public class TimeApprovalWSAction extends TkAction {
58
59 private static final Logger LOG = Logger.getLogger(TimeApprovalWSAction.class);
60
61
62
63
64
65
66
67 public ActionForward searchApprovalRows(ActionMapping mapping, ActionForm form, HttpServletRequest request, HttpServletResponse response) throws Exception {
68 TimeApprovalActionForm taaf = (TimeApprovalActionForm) form;
69 List<Map<String, String>> results = new LinkedList<Map<String, String>>();
70 if(StringUtils.isNotEmpty(taaf.getPayBeginDateForSearch())
71 && StringUtils.isNotEmpty(taaf.getPayEndDateForSearch()) ) {
72 Date beginDate = new SimpleDateFormat("MM/dd/yyyy").parse(taaf.getPayBeginDateForSearch());
73 Date endDate = new SimpleDateFormat("MM/dd/yyyy").parse(taaf.getPayEndDateForSearch());
74
75 endDate = DateUtils.addDays(endDate,1);
76 List<String> workAreaList = new ArrayList<String>();
77 if(StringUtil.isEmpty(taaf.getSelectedWorkArea())) {
78 Set<Long> workAreas = TkServiceLocator.getTkRoleService().getWorkAreasForApprover(TKContext.getTargetPrincipalId(), TKUtils.getCurrentDate());
79 for(Long workArea : workAreas) {
80 workAreaList.add(workArea.toString());
81 }
82 } else {
83 workAreaList.add(taaf.getSelectedWorkArea());
84 }
85 List<String> principalIds = TkServiceLocator.getTimeApproveService()
86 .getTimePrincipalIdsWithSearchCriteria(workAreaList, taaf.getSelectedPayCalendarGroup(),
87 new java.sql.Date(endDate.getTime()), new java.sql.Date(beginDate.getTime()), new java.sql.Date(endDate.getTime()));
88
89
90 if (StringUtils.equals(taaf.getSearchField(), ApprovalForm.ORDER_BY_PRINCIPAL)) {
91 for (String id : principalIds) {
92 if(StringUtils.contains(id, taaf.getSearchTerm())) {
93 Map<String, String> labelValue = new HashMap<String, String>();
94 labelValue.put("id", id);
95 labelValue.put("result", id);
96 results.add(labelValue);
97 }
98 }
99 } else if (StringUtils.equals(taaf.getSearchField(), ApprovalForm.ORDER_BY_DOCID)) {
100 Map<String, TimesheetDocumentHeader> principalDocumentHeaders =
101 TkServiceLocator.getTimeApproveService().getPrincipalDocumentHeader(principalIds, beginDate, endDate);
102
103 for (Map.Entry<String,TimesheetDocumentHeader> entry : principalDocumentHeaders.entrySet()) {
104 if (StringUtils.contains(entry.getValue().getDocumentId(), taaf.getSearchTerm())) {
105 Map<String, String> labelValue = new HashMap<String, String>();
106
107 labelValue.put("id", entry.getValue().getDocumentId());
108 labelValue.put("result", entry.getValue().getPrincipalId());
109 results.add(labelValue);
110 }
111 }
112 }
113 }
114
115 taaf.setOutputString(JSONValue.toJSONString(results));
116 return mapping.findForward("ws");
117 }
118
119 public ActionForward getTimeSummary(ActionMapping mapping, ActionForm form, HttpServletRequest request, HttpServletResponse response) throws Exception {
120 TimeApprovalActionForm taaf = (TimeApprovalActionForm) form;
121 TimesheetDocument td = TkServiceLocator.getTimesheetService().getTimesheetDocument(taaf.getDocumentId());
122 TimeSummary ts = TkServiceLocator.getTimeSummaryService().getTimeSummary(td);
123 List<Assignment> assignments = td.getAssignments();
124 List<String> assignmentKeys = new ArrayList<String>();
125 for(Assignment assignment : assignments) {
126 assignmentKeys.add(assignment.getAssignmentKey());
127 }
128 List<TimeBlock> timeBlocks = td.getTimeBlocks();
129 List<LeaveBlock> leaveBlocks = TkServiceLocator.getLeaveBlockService().getLeaveBlocksForTimeCalendar(td.getPrincipalId(), td.getAsOfDate(), td.getDocEndDate(), assignmentKeys);
130 Map<String, String> aMap = ActionFormUtils.buildAssignmentStyleClassMap(timeBlocks, leaveBlocks);
131
132 for (EarnGroupSection earnGroupSection : ts.getSections()) {
133 for (EarnCodeSection section : earnGroupSection.getEarnCodeSections()) {
134 for (AssignmentRow assignRow : section.getAssignmentsRows()) {
135 String assignmentCssStyle = MapUtils.getString(aMap, assignRow.getAssignmentKey());
136 assignRow.setCssClass(assignmentCssStyle);
137 for (AssignmentColumn assignmentColumn : assignRow.getAssignmentColumns()) {
138 assignmentColumn.setCssClass(assignmentCssStyle);
139 }
140 }
141 }
142 }
143
144
145
146 ts.setSections(Lists.reverse(ts.getSections()));
147 taaf.setOutputString(ts.toJsonString());
148 return mapping.findForward("ws");
149 }
150
151 }