/**
    * Copyright 2004-2014 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.kpme.tklm.time.workflow;
  
  import org.apache.commons.lang.StringUtils;
  import org.apache.commons.lang.math.NumberUtils;
  import org.apache.log4j.Logger;
  import org.joda.time.DateTime;
  import org.kuali.kpme.core.api.assignment.Assignment;
  import org.kuali.kpme.core.api.namespace.KPMENamespace;
  import org.kuali.kpme.core.role.KPMERole;
  import org.kuali.kpme.core.service.HrServiceLocator;
  import org.kuali.kpme.tklm.time.service.TkServiceLocator;
  import org.kuali.kpme.tklm.time.timesheet.TimesheetDocument;
  import org.kuali.rice.kew.api.identity.Id;
  import org.kuali.rice.kew.api.identity.PrincipalId;
  import org.kuali.rice.kew.api.rule.RoleName;
  import org.kuali.rice.kew.engine.RouteContext;
  import org.kuali.rice.kew.routeheader.DocumentContent;
  import org.kuali.rice.kew.rule.GenericRoleAttribute;
  import org.kuali.rice.kew.rule.QualifiedRoleName;
  import org.kuali.rice.kim.api.role.RoleMember;
  
  import java.util.*;
  
  @SuppressWarnings("unchecked")
  public class TimesheetRoleAttribute extends GenericRoleAttribute {
  
  	private static final long serialVersionUID = 6133675090144279606L;
  	
  	private static final Logger LOG = Logger.getLogger(TimesheetRoleAttribute.class);
  
  	@Override
  	public List<RoleName> getRoleNames() {
  		List<RoleName> roleNames = new ArrayList<RoleName>();
  		
  		roleNames.add(RoleName.Builder.create(getClass().getName(), KPMERole.APPROVER.getRoleName(), KPMERole.APPROVER.getRoleName()).build());
  		roleNames.add(RoleName.Builder.create(getClass().getName(), KPMERole.APPROVER_DELEGATE.getRoleName(), KPMERole.APPROVER_DELEGATE.getRoleName()).build());
  
  		return roleNames;
  	}
  	
  	@Override
      protected List<String> getRoleNameQualifiers(String roleName, DocumentContent documentContent) {
  		Set<String> roleNameQualifiers = new HashSet<String>();
  		
  		Long routeHeaderId = new Long(documentContent.getRouteContext().getDocument().getDocumentId());
  		TimesheetDocument timesheetDocument = TkServiceLocator.getTimesheetService().getTimesheetDocument(routeHeaderId.toString());
  
  		if (timesheetDocument != null) {
  			List<Assignment> assignments = timesheetDocument.getAllAssignments();
  			for (Assignment assignment : assignments) {
  				roleNameQualifiers.add(String.valueOf(assignment.getWorkArea()));
  			}
  		}
  		
  		return new ArrayList<String>(roleNameQualifiers);
      }
  	
  	@Override
      protected List<Id> resolveRecipients(RouteContext routeContext, QualifiedRoleName qualifiedRoleName) {
  		List<Id> recipients = new ArrayList<Id>();
  		
  		String roleName = qualifiedRoleName.getBaseRoleName();
  		String qualifier = qualifiedRoleName.getQualifier();
  		
  		if (StringUtils.isNotBlank(roleName) && NumberUtils.isNumber(qualifier)) {
  			Long workArea = Long.valueOf(qualifier);
  	
  			List<RoleMember> roleMembers = HrServiceLocator.getKPMERoleService().getRoleMembersInWorkArea(KPMENamespace.KPME_HR.getNamespaceCode(), roleName, workArea, new DateTime(), true);
  			
  	        for (RoleMember roleMember : roleMembers) {
  	        	recipients.add(new PrincipalId(roleMember.getMemberId()));
  		    }
  		} else {
  			LOG.error("Could not route to role " + roleName + " with work area " + qualifier);
  		}
  		
  		return recipients;
      }
  
  	@Override
  	public Map<String, String> getProperties() {
  		return null;
  	}
  
 }