/*
* Copyright 2011 Eric F. Savage, code@efsavage.com
*
* Licensed 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 com.ajah.spring.mvc.util;
import java.util.EnumSet;
import java.util.Map;
import javax.servlet.DispatcherType;
import javax.servlet.Filter;
import javax.servlet.FilterRegistration;
import javax.servlet.FilterRegistration.Dynamic;
import javax.servlet.Registration;
import javax.servlet.ServletContext;
import org.springframework.context.ApplicationContext;
import org.springframework.web.WebApplicationInitializer;
import org.springframework.web.context.support.AnnotationConfigWebApplicationContext;
/**
* Utilities for dealing with Filters. Currently just provides easier ways to
* add filters to a context from within a {@link WebApplicationInitializer}.
*
* @author <a href="http://efsavage.com">Eric F. Savage</a>,
* <a href="mailto:code@efsavage.com">code@efsavage.com</a>.
*
*/
public class FilterUtils {
/**
* Convenience method for adding a filter that is available as a Spring
* bean/service to all requests.
*
* @see ServletContext#addFilter(String, Class)
* @param filterClass
* The class of Filter to instantiate.
* @param appContext
* The application context to find the bean in.
* @param servletContext
* @return The Dynamic Mapping created by this method.
*/
public static Dynamic add(final Class<? extends Filter> filterClass, final ApplicationContext appContext, final ServletContext servletContext) {
final Filter filter = appContext.getBean(filterClass);
final FilterRegistration.Dynamic reg = servletContext.addFilter(filterClass.getName(), filter);
reg.addMappingForUrlPatterns(EnumSet.of(DispatcherType.REQUEST), true, "/*");
return reg;
}
/**
* Convenience method for adding a filter that is available as a Spring
* bean/service to a servlet.
*
* @see ServletContext#addFilter(String, Class)
* @param filterClass
* The class of Filter to instantiate.
* @param appContext
* The application context to find the bean in.
* @param servletContext
* @param servletRegistration
* The servlet to add this filter to.
* @return The Dynamic Mapping created by this method.
*/
public static Dynamic add(final Class<? extends Filter> filterClass, final ApplicationContext appContext, final ServletContext servletContext, final Registration servletRegistration) {
final Filter filter = appContext.getBean(filterClass);
final FilterRegistration.Dynamic reg = servletContext.addFilter(filterClass.getName(), filter);
reg.addMappingForServletNames(EnumSet.of(DispatcherType.REQUEST), false, servletRegistration.getName());
return reg;
}
/**
* Convenience method for adding a filter to all requests.
*
* @see ServletContext#addFilter(String, Class)
* @param filterClass
* The class of Filter to instantiate.
* @param servletContext
*/
public static void add(final Class<? extends Filter> filterClass, final ServletContext servletContext) {
add(filterClass, servletContext, "/*");
}
/**
* Convenience method for adding a filter that is available as a Spring
* bean/service to a servlet.
*
* @see ServletContext#addFilter(String, Class)
* @param filterClass
* The class of Filter to instantiate.
* @param servletContext
* @param servletRegistration
* The servlet to add this filter to.
* @return The Dynamic Mapping created by this method.
*/
public static Dynamic add(final Class<? extends Filter> filterClass, final ServletContext servletContext, final Registration servletRegistration) {
final FilterRegistration.Dynamic reg = servletContext.addFilter(filterClass.getName(), filterClass);
reg.addMappingForServletNames(EnumSet.of(DispatcherType.REQUEST), false, servletRegistration.getName());
return reg;
}
/**
* Convenience method for adding a filter to certain requests.
*
* @see ServletContext#addFilter(String, Class)
* @param filterClass
* The class of Filter to instantiate, required.
* @param servletContext
* The servlet context of the application, required.
* @param urlPattern
* The pattern of the URL this filter should be applied to.
*/
public static void add(final Class<? extends Filter> filterClass, final ServletContext servletContext, final String urlPattern) {
final FilterRegistration.Dynamic reg = servletContext.addFilter(filterClass.getName(), filterClass);
reg.addMappingForUrlPatterns(EnumSet.of(DispatcherType.REQUEST), true, urlPattern);
}
/**
* Convenience method for adding a filter to certain requests.
*
* @see ServletContext#addFilter(String, Class)
* @param filterClass
* The class of Filter to instantiate, required.
* @param servletContext
* The servlet context of the application, required.
* @param urlPattern
* The pattern of the URL this filter should be applied to.
*/
public static void add(final Class<? extends Filter> filterClass, final ServletContext servletContext, final String urlPattern, final Map<String, String> initParameters) {
final FilterRegistration.Dynamic reg = servletContext.addFilter(filterClass.getName(), filterClass);
if (initParameters != null && initParameters.size() > 0) {
reg.setInitParameters(initParameters);
}
reg.addMappingForUrlPatterns(EnumSet.of(DispatcherType.REQUEST), true, urlPattern);
}
/**
* Convenience method for adding a filter to certain requests. This version
* will attempt to find a pre-existing bean for the specified class.
*
* @see ServletContext#addFilter(String, Class)
* @param appContest
* The application contest to look for the filter bean in.
* @param filterClass
* The class of Filter to instantiate, required.
* @param servletContext
* The servlet context of the application, required.
* @param urlPattern
* The pattern of the URL this filter should be applied to.
*/
public static void add(AnnotationConfigWebApplicationContext appContext, Class<? extends Filter> filterClass, ServletContext servletContext, String urlPattern) {
Filter filter = appContext.getBean(filterClass);
if (filter != null) {
final FilterRegistration.Dynamic reg = servletContext.addFilter(filterClass.getName(), filter);
reg.addMappingForUrlPatterns(EnumSet.of(DispatcherType.REQUEST), true, urlPattern);
} else {
add(filterClass, servletContext, urlPattern);
}
}
}