001    /**
002     * Copyright 2004-2012 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.docsearch;
017    
018    import java.util.ArrayList;
019    import java.util.List;
020    
021    import org.apache.commons.lang.StringUtils;
022    import org.kuali.hr.time.assignment.Assignment;
023    import org.kuali.hr.time.service.base.TkServiceLocator;
024    import org.kuali.hr.time.timesheet.TimesheetDocument;
025    import org.kuali.rice.kew.api.document.DocumentWithContent;
026    import org.kuali.rice.kew.api.extension.ExtensionDefinition;
027    import org.kuali.rice.kew.docsearch.SearchableAttributeLongValue;
028    import org.kuali.rice.kew.docsearch.SearchableAttributeValue;
029    import org.kuali.rice.kew.docsearch.xml.StandardGenericXMLSearchableAttribute;
030    
031    public class WorkAreaSearchableAttribute extends StandardGenericXMLSearchableAttribute{
032    
033            private static String WORK_AREA = "workArea";
034            /**
035             * 
036             */
037            private static final long serialVersionUID = 1L;
038    
039            @Override
040        public List extractDocumentAttributes(ExtensionDefinition extensionDefinition, DocumentWithContent documentWithContent) {
041                    List<SearchableAttributeValue> searchableAttributeValues = new ArrayList<SearchableAttributeValue>();
042                    
043                    String documentId = StringUtils.substringBetween(documentWithContent.getDocumentContent().toString(), "<string>", "</string>");
044                    TimesheetDocument timesheetDocument = TkServiceLocator.getTimesheetService().getTimesheetDocument(documentId);
045                    List<Long> workAreasIncluded = new ArrayList<Long>();
046                    for(Assignment assign : timesheetDocument.getAssignments()){
047                            if(!workAreasIncluded.contains(assign.getWorkArea())){
048                                    SearchableAttributeValue attValue = new SearchableAttributeLongValue();
049                                    attValue.setSearchableAttributeKey(WORK_AREA);
050                                    attValue.setupAttributeValue(StringUtils.upperCase(assign.getWorkArea().toString()));
051                                    searchableAttributeValues.add(attValue);
052                                    workAreasIncluded.add(assign.getWorkArea());
053                            }
054                    }
055                    return searchableAttributeValues;
056            }
057    }