/*

  * Licensed to the Apache Software Foundation (ASF) under one

  * or more contributor license agreements.  See the NOTICE file

  * distributed with this work for additional information

  * regarding copyright ownership.  The ASF licenses this file

  * to you under the Apache 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.apache.org/licenses/LICENSE-2.0

  *

  * 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.apache.myfaces.flow.cdi;

 import java.util.Map;

 import java.util.concurrent.ConcurrentHashMap;

 import javax.enterprise.event.Observes;

 import javax.enterprise.inject.spi.AfterBeanDiscovery;

 import javax.enterprise.inject.spi.AfterDeploymentValidation;

 import javax.enterprise.inject.spi.AnnotatedType;

 import javax.enterprise.inject.spi.BeanManager;

 import javax.enterprise.inject.spi.BeforeBeanDiscovery;

 import javax.enterprise.inject.spi.Extension;

 import javax.enterprise.inject.spi.ProcessBean;

 import javax.faces.flow.FlowScoped;

 import org.apache.myfaces.flow.FlowReference;

 /**

  *

  * @author Leonardo Uribe

  */

 public class FlowScopeCDIExtension implements Extension

 {

     private FlowScopedContextImpl flowScopedContext;

     private Map<Class, FlowReference> flowBeanReferences;

     public FlowScopeCDIExtension()

     {

         flowBeanReferences = new ConcurrentHashMap<Class, FlowReference>();

     }

     void beforeBeanDiscovery(

         @Observes final BeforeBeanDiscovery event, BeanManager beanManager)

     {

         event.addScope(FlowScoped.class, true, true);

         // Register FlowBuilderFactoryBean as a bean with CDI annotations, so the system

         // can take it into account, and use it later when necessary.

         AnnotatedType bean = beanManager.createAnnotatedType(FlowScopeBeanHolder.class);

         event.addAnnotatedType(bean);

     }

     void onProcessBean(@Observes ProcessBean event, BeanManager manager)

     {

         // Register all beans who are annotated with FlowScoped and has a flow reference

         // restriction, to take it into account later when it is created and store it

         // in the right context so @PreDestroy is called when the referenced flow is.

         if (event.getAnnotated().isAnnotationPresent(FlowScoped.class))

         {

             FlowScoped flowScoped = event.getAnnotated().getAnnotation(FlowScoped.class);

             String flowId = flowScoped.value();

             if (flowId != null)

             {

                 flowBeanReferences.put(event.getBean().getBeanClass(), new FlowReference(

                     flowScoped.definingDocumentId(), flowId));

             }

         }

     }

     void afterBeanDiscovery(@Observes AfterBeanDiscovery event, BeanManager manager)

     {

         flowScopedContext = new FlowScopedContextImpl(manager, flowBeanReferences);

         event.addContext(flowScopedContext);

     }

     void initializeFlowContexts(@Observes AfterDeploymentValidation adv, BeanManager beanManager)

     {

         //FlowScopeBeanHolder flowScopeBeanHolder = BeanProvider.getContextualReference(

         //    beanManager, FlowScopeBeanHolder.class, false);

         //flowScopedContext.initFlowContext(flowScopeBeanHolder);

     }

 }