/*
    * Copyright 2006-2011 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.rice.krms.test;
  
  import org.kuali.rice.krad.service.BusinessObjectService;
  import org.kuali.rice.krad.service.KRADServiceLocator;
  import org.kuali.rice.krms.api.repository.context.ContextDefinition;
  import org.kuali.rice.krms.api.repository.type.KrmsAttributeDefinition;
  import org.kuali.rice.krms.api.repository.type.KrmsTypeAttribute;
  import org.kuali.rice.krms.api.repository.type.KrmsTypeDefinition;
  import org.kuali.rice.krms.api.repository.type.KrmsTypeRepositoryService;
  import org.kuali.rice.krms.impl.repository.ContextAttributeBo;
  import org.kuali.rice.krms.impl.repository.ContextBoService;
  import org.kuali.rice.krms.impl.repository.KrmsAttributeDefinitionService;
  import org.kuali.rice.test.BaselineTestCase.BaselineMode;
  import org.kuali.rice.test.BaselineTestCase.Mode;
  
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.List;
  import java.util.Map;
  
  @BaselineMode(Mode.ROLLBACK)
  public abstract class AbstractBoTest extends KRMSTestCase {
  	
      protected ContextBoService contextRepository;
      protected KrmsTypeRepositoryService krmsTypeRepository;
      protected KrmsAttributeDefinitionService krmsAttributeDefinitionService;
  
      protected BusinessObjectService getBoService() {
  		return KRADServiceLocator.getBusinessObjectService();
  	}
  
      protected ContextDefinition createContextDefinition(String nameSpace, String name,
              Map<String, String> contextAttributes) {
          // Attributes for context;
          List<KrmsTypeAttribute.Builder> contextAttributeBuilders = new ArrayList<KrmsTypeAttribute.Builder>();
          int contextAttrSequenceIndex = 0;
  
          List<KrmsAttributeDefinition> attributeDefintions = new ArrayList<KrmsAttributeDefinition>();
  
          if (contextAttributes != null) for (Map.Entry<String,String> entry : contextAttributes.entrySet()) {
              KrmsAttributeDefinition.Builder contextTypeAttributeDefnBuilder = KrmsAttributeDefinition.Builder.create(null, entry.getKey(), nameSpace);
              contextTypeAttributeDefnBuilder.setLabel(entry.getKey() + " attribute label");
              KrmsAttributeDefinition contextTypeAttributeDefinition = krmsAttributeDefinitionService.createAttributeDefinition(contextTypeAttributeDefnBuilder.build());
              attributeDefintions.add(contextTypeAttributeDefinition);
  
              // Attr for context;
              contextAttributeBuilders.add(KrmsTypeAttribute.Builder.create(null, contextTypeAttributeDefinition.getId(),
                      contextAttrSequenceIndex));
              contextAttrSequenceIndex += 1;
          }
  
          // KrmsType for context
          KrmsTypeDefinition krmsContextTypeDefinition = krmsTypeRepository.getTypeByName(nameSpace, "KrmsTestContextType");
  
          if (krmsContextTypeDefinition == null) {
              KrmsTypeDefinition.Builder krmsContextTypeDefnBuilder = KrmsTypeDefinition.Builder.create("KrmsTestContextType", nameSpace);
              krmsContextTypeDefnBuilder.setAttributes(contextAttributeBuilders);
              krmsContextTypeDefinition = krmsContextTypeDefnBuilder.build();
              krmsContextTypeDefinition = krmsTypeRepository.createKrmsType(krmsContextTypeDefinition);
          }
  
          // Context
          ContextDefinition.Builder contextBuilder = ContextDefinition.Builder.create(nameSpace, name);
          contextBuilder.setTypeId(krmsContextTypeDefinition.getId());
          ContextDefinition contextDefinition = contextBuilder.build();
          contextDefinition = contextRepository.createContext(contextDefinition);
  
          // Context Attribute
          // TODO: do this fur eel
          for (KrmsAttributeDefinition contextTypeAttributeDefinition : attributeDefintions) {
              ContextAttributeBo contextAttribute = new ContextAttributeBo();
              contextAttribute.setAttributeDefinitionId(contextTypeAttributeDefinition.getId());
              contextAttribute.setContextId(contextDefinition.getId());
              contextAttribute.setValue(contextAttributes.get(contextTypeAttributeDefinition.getName()));
              getBoService().save(contextAttribute);
          }
  
          return contextDefinition;
      }
  
      protected KrmsTypeDefinition createKrmsGenericTypeDefinition(String nameSpace, String serviceName,
              String attributeDefinitionLabel, String attributeDefinitionName) {
          KrmsTypeDefinition krmsGenericTypeDefinition = krmsTypeRepository.getTypeByName(nameSpace, "KrmsTestGenericType");
  
         if (null == krmsGenericTypeDefinition) {
 
             // Attribute Defn for generic type;
             KrmsAttributeDefinition.Builder genericTypeAttributeDefnBuilder = KrmsAttributeDefinition.Builder.create(null,
                     attributeDefinitionName, nameSpace);
             genericTypeAttributeDefnBuilder.setLabel(attributeDefinitionLabel);
             KrmsAttributeDefinition genericTypeAttributeDefinition1 = krmsAttributeDefinitionService.createAttributeDefinition(genericTypeAttributeDefnBuilder.build());
 
             // Attr for generic type;
             KrmsTypeAttribute.Builder genericTypeAttrBuilder = KrmsTypeAttribute.Builder.create(null, genericTypeAttributeDefinition1.getId(), 1);
 
             // Can use this generic type for KRMS bits that don't actually rely on services on the bus at this point in time
             KrmsTypeDefinition.Builder krmsGenericTypeDefnBuilder = KrmsTypeDefinition.Builder.create("KrmsTestGenericType", nameSpace);
             krmsGenericTypeDefnBuilder.setServiceName(serviceName);
             krmsGenericTypeDefnBuilder.setAttributes(Collections.singletonList(genericTypeAttrBuilder));
             krmsGenericTypeDefinition = krmsTypeRepository.createKrmsType(krmsGenericTypeDefnBuilder.build());
 
         }
 
         return krmsGenericTypeDefinition;
     }
 	
 }