001    /**
002     *   GRANITE DATA SERVICES
003     *   Copyright (C) 2006-2013 GRANITE DATA SERVICES S.A.S.
004     *
005     *   This file is part of the Granite Data Services Platform.
006     *
007     *   Granite Data Services is free software; you can redistribute it and/or
008     *   modify it under the terms of the GNU Lesser General Public
009     *   License as published by the Free Software Foundation; either
010     *   version 2.1 of the License, or (at your option) any later version.
011     *
012     *   Granite Data Services is distributed in the hope that it will be useful,
013     *   but WITHOUT ANY WARRANTY; without even the implied warranty of
014     *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
015     *   General Public License for more details.
016     *
017     *   You should have received a copy of the GNU Lesser General Public
018     *   License along with this library; if not, write to the Free Software
019     *   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
020     *   USA, or see <http://www.gnu.org/licenses/>.
021     */
022    package org.granite.cdi;
023    
024    import javax.enterprise.inject.spi.BeanManager;
025    import javax.naming.InitialContext;
026    import javax.naming.NameNotFoundException;
027    import javax.servlet.ServletContext;
028    
029    
030    public class CDIUtils {
031            
032        public static BeanManager lookupBeanManager(ServletContext servletContext) {
033                    BeanManager manager = (BeanManager)servletContext.getAttribute(BeanManager.class.getName());
034                    if (manager != null)
035                            return manager;         
036                    manager = (BeanManager)servletContext.getAttribute("org.jboss.weld.environment.servlet.javax.enterprise.inject.spi.BeanManager");
037                    if (manager != null)
038                            return manager;
039                    
040                    InitialContext ic = null;
041                try {
042                            ic = new InitialContext();
043                    // Standard JNDI binding
044                    return (BeanManager)ic.lookup("java:comp/BeanManager");
045                }
046                catch (NameNotFoundException e) {
047                    if (ic == null)
048                            throw new RuntimeException("No InitialContext");
049                    
050                    // Weld/Tomcat
051                    try {
052                            return (BeanManager)ic.lookup("java:comp/env/BeanManager"); 
053                    }
054                    catch (Exception e1) {          
055                            // JBoss 5/6 (maybe obsolete in Weld 1.0+)
056                            try {
057                                    return (BeanManager)ic.lookup("java:app/BeanManager");
058                            }
059                        catch (Exception e2) {
060                            throw new RuntimeException("Could not find Bean Manager", e2);
061                        }
062                    }
063                }
064                catch (Exception e) {
065                    throw new RuntimeException("Could not find Bean Manager", e);
066                }
067        }
068    }