1 /*
2  * JBoss, Home of Professional Open Source.
3  * Copyright 2014 Red Hat, Inc., and individual contributors
4  * as indicated by the @author tags.
5  *
6  * Licensed under the Apache License, Version 2.0 (the "License");
7  * you may not use this file except in compliance with the License.
8  * You may obtain a copy of the License at
9  *
10  *     http://www.apache.org/licenses/LICENSE-2.0
11  *
12  *  Unless required by applicable law or agreed to in writing, software
13  *  distributed under the License is distributed on an "AS IS" BASIS,
14  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  *  See the License for the specific language governing permissions and
16  *  limitations under the License.
17  */
18
19 package io.undertow.servlet.api;
20
21 import io.undertow.servlet.UndertowServletMessages;
22 import io.undertow.servlet.util.ConstructorInstanceFactory;
23
24 import java.lang.reflect.Constructor;
25 import java.util.Set;
26
27 import javax.servlet.ServletContainerInitializer;
28
29 /**
30  * @author Stuart Douglas
31  */
32 public class ServletContainerInitializerInfo {
33
34     private final Class<? extends ServletContainerInitializer> servletContainerInitializerClass;
35     private final InstanceFactory<? extends ServletContainerInitializer> instanceFactory;
36     private final Set<Class<?>> handlesTypes;
37
38     public ServletContainerInitializerInfo(final Class<? extends ServletContainerInitializer> servletContainerInitializerClass, final InstanceFactory<? extends ServletContainerInitializer> instanceFactory, final Set<Class<?>> handlesTypes) {
39         this.servletContainerInitializerClass = servletContainerInitializerClass;
40         this.instanceFactory = instanceFactory;
41         this.handlesTypes = handlesTypes;
42     }
43
44     public ServletContainerInitializerInfo(final Class<? extends ServletContainerInitializer> servletContainerInitializerClass, final Set<Class<?>> handlesTypes) {
45         this.servletContainerInitializerClass = servletContainerInitializerClass;
46         this.handlesTypes = handlesTypes;
47
48         try {
49             final Constructor<ServletContainerInitializer> ctor = (Constructor<ServletContainerInitializer>) servletContainerInitializerClass.getDeclaredConstructor();
50             ctor.setAccessible(true);
51             this.instanceFactory = new ConstructorInstanceFactory<>(ctor);
52         } catch (NoSuchMethodException e) {
53             throw UndertowServletMessages.MESSAGES.componentMustHaveDefaultConstructor("ServletContainerInitializer", servletContainerInitializerClass);
54         }
55     }
56
57     public Class<? extends ServletContainerInitializer> getServletContainerInitializerClass() {
58         return servletContainerInitializerClass;
59     }
60
61     /**
62      * Returns the actual types present in the deployment that are handled by this ServletContainerInitializer.
63      *
64      * (i.e. not the types in the {@link javax.servlet.annotation.HandlesTypes} annotation, but rather actual types
65      * the container has discovered that meet the criteria)
66      *
67      * @return The handled types
68      */
69     public Set<Class<?>> getHandlesTypes() {
70         return handlesTypes;
71     }
72
73     public InstanceFactory<? extends ServletContainerInitializer> getInstanceFactory() {
74         return instanceFactory;
75     }
76 }
77