package org.doxla.spring.automock.xml.namespace;

 import org.doxla.spring.automock.AutoMockExposingPostProcessor;

 import org.doxla.spring.automock.resolver.MockClassResolver;

 import org.springframework.beans.factory.config.BeanDefinition;

 import org.springframework.beans.factory.support.BeanDefinitionBuilder;

 import org.springframework.beans.factory.xml.AbstractBeanDefinitionParser;

 import org.springframework.beans.factory.xml.ParserContext;

 import org.springframework.context.ApplicationContext;

 import org.w3c.dom.Element;

 /**

  * Does all the work involved in registering a {@link AutoMockExposingPostProcessor} and the {@link MockClassResolver} into 

  * the Spring {@link ApplicationContext} it is defined in.

  * 

  * @see AutoMockExposingPostProcessor

  * @see MockClassResolver

  * @see NamespaceElement

  * @author danoxlade

  */

 public class BasicAutoMockBeanDefinitionParser

         extends AbstractBeanDefinitionParser{

         /**

          * The Spring bean name of the {@link AutoMockExposingPostProcessor} exposed when using the "automock" namespace.

          */

         public static final String AUTOMOCK_POST_PROCESSOR_BEAN_NAME = "org.doxla.spring.automock.AutoMockPostProcessorBean";

         private static final String MOCK_CLASS_RESOLVER = "mockClassResolver";

         private final NamespaceElement autoMockSchemaElement;

         /**

          * @param element The {@link NamespaceElement} which defines which {@link MockClassResolver} the exposed 

          * {@link AutoMockExposingPostProcessor} bean should use as a strategy.

          * 

          * @see AutoMockExposingPostProcessor

          * @see NamespaceElement

          */

         public BasicAutoMockBeanDefinitionParser(NamespaceElement element) {

                 this.autoMockSchemaElement = element;

         }

         /**

          * Handle the 'automock:elementNameHere' XML Element

          */

         @Override

         protected BeanDefinition parseInternal(Element element, ParserContext parserContext) {

                 element.setAttribute(ID_ATTRIBUTE,AUTOMOCK_POST_PROCESSOR_BEAN_NAME);

                 registerMockClassResolverBean(element,parserContext);

                 BeanDefinitionBuilder postProcessorDefinitionBuilder = BeanDefinitionBuilder.rootBeanDefinition(getPostProcessorBeanClass());

                 postProcessorDefinitionBuilder.addPropertyReference(MOCK_CLASS_RESOLVER, MOCK_CLASS_RESOLVER);

                 return postProcessorDefinitionBuilder.getBeanDefinition();

         }

         /**

          * Work out which {@link MockClassResolver} instance to use and register it with Spring

          * 

          * @param element

          * @param parserContext

          */

         private void registerMockClassResolverBean(Element element, ParserContext parserContext) {

                 BeanDefinitionBuilder strategyDefinitionBuilder = BeanDefinitionBuilder.rootBeanDefinition(getStrategyBeanClass());

                 for(String[] propertyName : this.autoMockSchemaElement.propertyToAttributeMappings()){

                         strategyDefinitionBuilder.addPropertyValue(propertyName[0], element.getAttribute( propertyName[1] ));

                 }

                 parserContext.getRegistry().registerBeanDefinition(MOCK_CLASS_RESOLVER,strategyDefinitionBuilder.getBeanDefinition());

         }

         private Class getPostProcessorBeanClass() {

                 return AutoMockExposingPostProcessor.class;

         }

         private Class<? extends MockClassResolver> getStrategyBeanClass() {

                 return this.autoMockSchemaElement.getStrategyClass();

         }

 }