/**

  * 

  */

 package org.kuali.student.lum.workflow.qualifierresolver;

 import java.util.ArrayList;

 import java.util.HashSet;

 import java.util.List;

 import java.util.Set;

 import javax.xml.namespace.QName;

 import javax.xml.xpath.XPath;

 import javax.xml.xpath.XPathConstants;

 import javax.xml.xpath.XPathExpressionException;

 import org.apache.commons.lang.StringUtils;

 import org.kuali.rice.core.resourceloader.GlobalResourceLoader;

 import org.kuali.rice.kew.engine.RouteContext;

 import org.kuali.rice.kew.engine.node.RouteNodeUtils;

 import org.kuali.rice.kew.role.QualifierResolver;

 import org.kuali.rice.kew.rule.xmlrouting.XPathHelper;

 import org.kuali.rice.kew.util.KEWConstants;

 import org.kuali.rice.kew.util.XmlHelper;

 import org.kuali.rice.kim.bo.types.dto.AttributeSet;

 import org.kuali.rice.student.bo.KualiStudentKimAttributes;

 import org.kuali.student.core.organization.service.OrganizationService;

 import org.kuali.student.core.search.dto.SearchParam;

 import org.kuali.student.core.search.dto.SearchRequest;

 import org.kuali.student.core.search.dto.SearchResult;

 import org.kuali.student.core.search.dto.SearchResultCell;

 import org.kuali.student.core.search.dto.SearchResultRow;

 import org.w3c.dom.Document;

 import org.w3c.dom.Element;

 import org.w3c.dom.Node;

 import org.w3c.dom.NodeList;

 /**

  * An abstract base class that consolidates convenience methods for using the {@link OrganizationService} class.

  * 

  */

 public abstract class AbstractOrganizationServiceQualifierResolver implements QualifierResolver {

     private static final org.apache.log4j.Logger LOG = org.apache.log4j.Logger.getLogger(AbstractOrganizationServiceQualifierResolver.class);

     protected static final String DOCUMENT_CONTENT_XML_DEFAULT_ORG_ID_KEY = "orgId";

     protected static final String DOCUMENT_CONTENT_XML_ORG_ID_KEY = "organizationIdDocumentContentKey";

     // below string MUST match

     // org.kuali.student.core.assembly.transform.WorkflowFilter.DOCUMENT_CONTENT_XML_ROOT_ELEMENT_NAME constant

     public static final String DOCUMENT_CONTENT_XML_ROOT_ELEMENT_NAME = "info";

     private OrganizationService organizationService;

     protected OrganizationService getOrganizationService() {

         if (null == organizationService) {

             organizationService = (OrganizationService) GlobalResourceLoader.getService(new QName("http://student.kuali.org/wsdl/organization", "OrganizationService"));

         }

         return organizationService;

     }

     protected void setOrganizationService(OrganizationService orgSvc) {

         organizationService = orgSvc;

     }

     /**

      * Method to fetch the organization ids from the KEW document content XML

      * 

      * @param context

      *            - RouteContext class that holds data about the current document's routing and data

      * @return A list of organization ids that are listed in the XML (may have duplicates if duplicates are allowed by

      *         KS code)

      */

     protected Set<String> getOrganizationIdsFromDocumentContent(RouteContext context) {

         String baseXpathExpression = "/" + KEWConstants.DOCUMENT_CONTENT_ELEMENT + "/" + KEWConstants.APPLICATION_CONTENT_ELEMENT + "/" + DOCUMENT_CONTENT_XML_ROOT_ELEMENT_NAME;

         String orgXpathExpression = "./" + getOrganizationIdDocumentContentFieldKey(context);

         Document xmlContent = context.getDocumentContent().getDocument();

         XPath xPath = XPathHelper.newXPath();

         try {

             NodeList baseElements = (NodeList) xPath.evaluate(baseXpathExpression, xmlContent, XPathConstants.NODESET);

             if (LOG.isDebugEnabled()) {

                 LOG.debug("Found " + baseElements.getLength() + " baseElements to parse for AttributeSets using document XML:");

                 XmlHelper.printDocumentStructure(xmlContent);

             }

             Set<String> distinctiveOrganizationIds = new HashSet<String>();

             for (int i = 0; i < baseElements.getLength(); i++) {

                 Node baseNode = baseElements.item(i);

                 NodeList attributes = (NodeList) xPath.evaluate(orgXpathExpression, baseNode, XPathConstants.NODESET);

                 for (int j = 0; j < attributes.getLength(); j++) {

                     Element attributeElement = (Element) attributes.item(j);

                     distinctiveOrganizationIds.add(attributeElement.getTextContent());

                 }

             }

             return distinctiveOrganizationIds;

         } catch (XPathExpressionException e) {

             throw new RuntimeException("Encountered an issue executing XPath.", e);

         }

     }

     protected String getOrganizationIdDocumentContentFieldKey(RouteContext context) {

         String organizationIdFieldKey = RouteNodeUtils.getValueOfCustomProperty(context.getNodeInstance().getRouteNode(), DOCUMENT_CONTENT_XML_ORG_ID_KEY);

         if (StringUtils.isBlank(organizationIdFieldKey)) {

             LOG.info("Cannot find element '" + DOCUMENT_CONTENT_XML_ORG_ID_KEY + "' on Route Node XML configuration. Will use default value of '" + DOCUMENT_CONTENT_XML_DEFAULT_ORG_ID_KEY + "'.");

             organizationIdFieldKey = DOCUMENT_CONTENT_XML_DEFAULT_ORG_ID_KEY;

         }

         return organizationIdFieldKey;

     }

     protected List<SearchResultRow> relatedOrgsFromOrgId(String orgId, String relationType, String relatedOrgType) {

         List<SearchResultRow> results = null;

         if (null != orgId) {

             List<SearchParam> queryParamValues = new ArrayList<SearchParam>(2);

             SearchParam qpRelType = new SearchParam();

             qpRelType.setKey("org.queryParam.relationType");

             qpRelType.setValue(relationType);

             queryParamValues.add(qpRelType);

             SearchParam qpOrgId = new SearchParam();

             qpOrgId.setKey("org.queryParam.orgId");

             qpOrgId.setValue(orgId);

             queryParamValues.add(qpOrgId);

             SearchParam qpRelOrgType = new SearchParam();

             qpRelOrgType.setKey("org.queryParam.relatedOrgType");

             qpRelOrgType.setValue(relatedOrgType);

             queryParamValues.add(qpRelOrgType);

             SearchRequest searchRequest = new SearchRequest();

             searchRequest.setSearchKey("org.search.orgQuickViewByRelationTypeRelatedOrgTypeOrgId");

             searchRequest.setParams(queryParamValues);

             try {

                 SearchResult result = getOrganizationService().search(searchRequest);

                 results = result.getRows();

             } catch (Exception e) {

                 LOG.error("Error calling org service");

                 throw new RuntimeException(e);

             }

         }

         return results;

     }

     protected List<AttributeSet> attributeSetFromSearchResult(List<SearchResultRow> results, String orgShortNameKey, String orgIdKey) {

         List<AttributeSet> returnAttrSetList = new ArrayList<AttributeSet>();

         if (results != null) {

             for (SearchResultRow result : results) {

                 AttributeSet attributeSet = new AttributeSet();

                 String resolvedOrgId = "";

                 String resolvedOrgShortName = "";

                 for (SearchResultCell resultCell : result.getCells()) {

                     if ("org.resultColumn.orgId".equals(resultCell.getKey())) {

                         resolvedOrgId = resultCell.getValue();

                     } else if ("org.resultColumn.orgShortName".equals(resultCell.getKey())) {

                         resolvedOrgShortName = resultCell.getValue();

                     }

                 }

                 if (orgShortNameKey != null) {

                     attributeSet.put(orgShortNameKey, resolvedOrgShortName);

                 }

                 if (orgIdKey != null) {

                     attributeSet.put(orgIdKey, resolvedOrgId);

                 }

                 attributeSet.put(KualiStudentKimAttributes.QUALIFICATION_ORG, resolvedOrgShortName);

                 attributeSet.put(KualiStudentKimAttributes.QUALIFICATION_ORG_ID, resolvedOrgId);

                 returnAttrSetList.add(attributeSet);

             }

         }

         return returnAttrSetList;

     }

 }