/**
    * Copyright 2004-2012 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.hr.time.docsearch;
  
  import java.util.ArrayList;
  import java.util.Date;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  import org.apache.log4j.Logger;
  import org.kuali.hr.job.Job;
  import org.kuali.hr.time.assignment.Assignment;
  import org.kuali.hr.time.service.base.TkServiceLocator;
  import org.kuali.hr.time.timesheet.TimesheetDocument;
  import org.kuali.hr.time.util.TKContext;
  import org.kuali.hr.time.util.TKUtils;
  import org.kuali.hr.time.workarea.WorkArea;
  import org.kuali.rice.kew.api.WorkflowDocument;
  import org.kuali.rice.kew.api.WorkflowDocumentFactory;
  
  public class TkSearchableAttributeServiceImpl implements
  		TkSearchableAttributeService {
  
      private static final Logger LOG = Logger.getLogger(TkSearchableAttributeServiceImpl.class);
  
  	public void updateSearchableAttribute(TimesheetDocument document, Date asOfDate){
          WorkflowDocument workflowDocument = null;
          //
          // djunk - Need to actually look at why this call is here for every
          //         document submission. Rice does not allow save events for
          //         documents in final status. We may add more skips.
          //
          if (!document.getDocumentHeader().getDocumentStatus().equals("F")) {
              try {
                  workflowDocument = WorkflowDocumentFactory.loadDocument(document.getPrincipalId(), document.getDocumentId());
                  workflowDocument.setApplicationContent(createSearchableAttributeXml(document, asOfDate));
                  workflowDocument.saveDocument("");
                  if (!"I".equals(workflowDocument.getStatus().getCode())) {
                      //updateWorkflowTitle(document,workflowDocument);
                      if(workflowDocument.getInitiatorPrincipalId().equals(TKContext.getPrincipalId())){
                          workflowDocument.saveDocument("");
                      }else{
                      	workflowDocument.saveDocumentData();
                      }
                  }else{
                      workflowDocument.saveDocument("");
                  }
  
  
              } catch (Exception e) {
                  LOG.warn("Exception during searchable attribute update.");
                  throw new RuntimeException(e);
              }
          }
  	}
  
  	@Override
  	public String createSearchableAttributeXml(TimesheetDocument document, Date asOfDate) {
  		List<Long> workAreas = new ArrayList<Long>();
  		Map<String,List<Long>> deptToListOfWorkAreas = new HashMap<String,List<Long>>();
  		List<String> salGroups = new ArrayList<String>();
  
  		for(Assignment assign: document.getAssignments()){
  			if(!workAreas.contains(assign.getWorkArea())){
  				workAreas.add(assign.getWorkArea());
  			}
  			Job job = TkServiceLocator.getJobService().getJob(assign.getPrincipalId(), assign.getJobNumber(), document.getAsOfDate());
  
  			if(!salGroups.contains(job.getHrSalGroup())){
  				salGroups.add(job.getHrSalGroup());
  			}
  		}
  
  		for(Long workArea : workAreas){
  			WorkArea workAreaObj = TkServiceLocator.getWorkAreaService().getWorkArea(workArea, TKUtils.getTimelessDate(asOfDate));
  			if(deptToListOfWorkAreas.containsKey(workAreaObj.getDept())){
  				List<Long> deptWorkAreas = deptToListOfWorkAreas.get(workAreaObj.getDept());
  				deptWorkAreas.add(workArea);
  			} else {
  				List<Long> deptWorkAreas = new ArrayList<Long>();
  				deptWorkAreas.add(workArea);
  				deptToListOfWorkAreas.put(workAreaObj.getDept(), deptWorkAreas);
  			}
  		}
  		StringBuilder sb = new StringBuilder();
         sb.append("<documentContext><applicationContent><TimesheetDocument>");
 		sb.append("<DEPARTMENTS>");
 		for(String dept : deptToListOfWorkAreas.keySet()){
 			sb.append("<DEPARTMENT value=\""+dept+"\">");
 			List<Long> deptWorkAreas = deptToListOfWorkAreas.get(dept);
 			for(Long workArea : deptWorkAreas){
 				sb.append("<WORKAREA value=\""+workArea+"\"/>");
 			}
 			sb.append("</DEPARTMENT>");
 		}
 		sb.append("</DEPARTMENTS>");
 		for(String salGroup : salGroups){
 			sb.append("<SALGROUP value=\""+salGroup+"\"/>");
 		}
 
 		sb.append("<PAYENDDATE value=\""+asOfDate+"\"/>");
 		sb.append("</TimesheetDocument></applicationContent></documentContext>");
 
 		return sb.toString();
 	}
 
 }