* 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.

*/

* @version $Rev: 823835 $ $Date: 2009-10-10 02:47:59 -0700 (Sat, 10 Oct 2009) $

*/

    public static final Logger logger = Logger.getInstance(LogCategory.OPENEJB_STARTUP, AnnotationDeployer.class.getPackage().getName());

    public static final Logger startupLogger = Logger.getInstance(LogCategory.OPENEJB_STARTUP_CONFIG, "org.apache.openejb.util.resources");

    private static final ThreadLocal<DeploymentModule> currentModule = new ThreadLocal<DeploymentModule>();

    private final DiscoverAnnotatedBeans discoverAnnotatedBeans;

    private final ProcessAnnotatedBeans processAnnotatedBeans;

    private final EnvEntriesPropertiesDeployer envEntriesPropertiesDeployer;

    public AnnotationDeployer() {

        discoverAnnotatedBeans = new DiscoverAnnotatedBeans();

        processAnnotatedBeans = new ProcessAnnotatedBeans();

        envEntriesPropertiesDeployer = new EnvEntriesPropertiesDeployer();

    }

    public AppModule deploy(AppModule appModule) throws OpenEJBException {

        setModule(appModule);

        try {

            appModule = discoverAnnotatedBeans.deploy(appModule);

            appModule = envEntriesPropertiesDeployer.deploy(appModule);

            appModule = processAnnotatedBeans.deploy(appModule);

            return appModule;

        } finally {

            removeModule();

        }

    }

    public WebModule deploy(WebModule webModule) throws OpenEJBException {

        setModule(webModule);

        try {

            webModule = discoverAnnotatedBeans.deploy(webModule);

            webModule = envEntriesPropertiesDeployer.deploy(webModule);

            webModule = processAnnotatedBeans.deploy(webModule);

            return webModule;

        } finally {

            removeModule();

        }

    }

    public static DeploymentModule getModule() {

        return currentModule.get();

    }

    private static void setModule(DeploymentModule module) {

        currentModule.set(module);

    }

    private static void removeModule() {

        currentModule.remove();

    }

    private static ValidationContext getValidationContext() {

        return getModule().getValidation();

    }

    public static class DiscoverAnnotatedBeans implements DynamicDeployer {

        public static final Set<String> knownResourceEnvTypes = new TreeSet<String>(asList(

                "javax.ejb.SessionContext",

                "javax.ejb.EntityContext",

                "javax.ejb.MessageDrivenContext",

                "javax.transaction.UserTransaction",

                "javax.jms.Queue",

                "javax.jms.Topic",

                "javax.xml.ws.WebServiceContext",

                "javax.ejb.TimerService"

        ));

        public static final Set<String> knownEnvironmentEntries = new TreeSet<String>(asList(

                "boolean", "java.lang.Boolean",

                "char", "java.lang.Character",

                "byte", "java.lang.Byte",

                "short", "java.lang.Short",

                "int", "java.lang.Integer",

                "long", "java.lang.Long",

                "float", "java.lang.Float",

                "double", "java.lang.Double",

                "java.lang.String"

        ));

        public AppModule deploy(AppModule appModule) throws OpenEJBException {

            for (EjbModule ejbModule : appModule.getEjbModules()) {

                setModule(ejbModule);

                try {

                    deploy(ejbModule);

                } finally {

                    removeModule();

                }

            }

            for (ClientModule clientModule : appModule.getClientModules()) {

                setModule(clientModule);

                try {

                    deploy(clientModule);

                } finally {

                    removeModule();

                }

            }

            for (ConnectorModule connectorModule : appModule.getResourceModules()) {

                setModule(connectorModule);

                try {

                    deploy(connectorModule);

                } finally {

                    removeModule();

                }

            }

            for (WebModule webModule : appModule.getWebModules()) {

                setModule(webModule);

                try {

                    deploy(webModule);

                } finally {

                    removeModule();

                }

            }

            return appModule;

        }

        public ClientModule deploy(ClientModule clientModule) throws OpenEJBException {

            if (clientModule.getApplicationClient() == null){

                clientModule.setApplicationClient(new ApplicationClient());

            }

           

            // Lots of jars have main classes so this might not even be an app client.

            // We're not going to scrape it for @LocalClient or @RemoteClient annotations

            // unless they flag us specifically by adding a META-INF/application-client.xml

            //

            // ClientModules that already have a ClassFinder have been generated automatically

            // from an EjbModule, so we don't skip those ever.

            if (clientModule.getFinder() == null && clientModule.getAltDDs().containsKey("application-client.xml"))

            if (clientModule.getApplicationClient() != null && clientModule.getApplicationClient().isMetadataComplete()) return clientModule;

            ClassFinder finder = clientModule.getFinder();

            if (finder == null) {

                try {

                    if (clientModule.getJarLocation() != null) {

                        String location = clientModule.getJarLocation();

                        File file = new File(location);

                        URL url;

                        if (file.exists()) {

                            url = file.toURL();

                        } else {

                            url = new URL(location);

                        }

                        finder = new ClassFinder(clientModule.getClassLoader(), url);

                    } else {

                        finder = new ClassFinder(clientModule.getClassLoader());

                    }

                } catch (MalformedURLException e) {

                    startupLogger.warning("startup.scrapeFailedForClientModule.url", clientModule.getJarLocation());

                    return clientModule;

                } catch (Exception e) {

                    startupLogger.warning("startup.scrapeFailedForClientModule", e, clientModule.getJarLocation());

                    return clientModule;

                }

            }

            // This method is also called by the deploy(EjbModule) method to see if those

            // modules have any @LocalClient or @RemoteClient classes

            for (Class<?> clazz : finder.findAnnotatedClasses(LocalClient.class)) {

                clientModule.getLocalClients().add(clazz.getName());

            }

            for (Class<?> clazz : finder.findAnnotatedClasses(RemoteClient.class)) {

                clientModule.getRemoteClients().add(clazz.getName());

            }

            if (clientModule.getApplicationClient() == null){

                if (clientModule.getRemoteClients().size() > 0 || clientModule.getLocalClients().size() > 0) {

                    clientModule.setApplicationClient(new ApplicationClient());

                }

            }

            return clientModule;

        }

        public ConnectorModule deploy(ConnectorModule connectorModule) throws OpenEJBException {

            return connectorModule;

        }

        public WebModule deploy(WebModule webModule) throws OpenEJBException {

            WebApp webApp = webModule.getWebApp();

            if (webApp != null && (webApp.isMetadataComplete())) return webModule;

            ClassFinder finder;

            try {

                File file = new File(webModule.getJarLocation());

                URL[] urls = DeploymentLoader.getWebappUrls(file);

                final ClassLoader webClassLoader = webModule.getClassLoader();

                finder = new ClassFinder(webClassLoader, asList(urls));

                ClassFinder finder2 = new ClassFinder(webClassLoader);

                webModule.setFinder(finder);

            } catch (Exception e) {

                startupLogger.warning("Unable to scrape for @WebService or @WebServiceProvider annotations. ClassFinder failed.", e);

                return webModule;

            }

            if (webApp == null) {