package org.wamblee.cdi;

import javax.enterprise.inject.spi.BeanManager;
import javax.naming.InitialContext;
import javax.naming.NamingException;

public class BeanManagerLookup {

    private static final String BEAN_MANAGER_JNDI = "java:comp/BeanManager";
    private static BeanManager mgr = null;
    
    /**
     * Sets the bean manager (mainly for testability). 
     * @param aMgr Bean manager. 
     */
    public static void setBeanManager(BeanManager aMgr) { 
        mgr = aMgr; 
    }

    public static BeanManager lookup() {
        if (mgr == null) {
            try {
                InitialContext ctx = new InitialContext();
                mgr = (BeanManager) ctx.lookup(BEAN_MANAGER_JNDI);
            } catch (NamingException e) {
                mgr = null;
            }
        }
        return mgr;
    }
}