Java Examples for org.springframework.boot.context.embedded.FilterRegistrationBean

@Bean
public FilterRegistrationBean authenticationFilter() {
    FilterRegistrationBean casFilter = new FilterRegistrationBean();
    Map<String, String> filterInitParam = Maps.newHashMap();
    filterInitParam.put("redisClusterName", "casClientPrincipal");
    filterInitParam.put("serverName", portalConfig.portalServerName());
    filterInitParam.put("casServerLoginUrl", portalConfig.casServerLoginUrl());
    //we don't want to use session to store login information, since we will be deployed to a cluster, not a single instance
    filterInitParam.put("useSession", "false");
    filterInitParam.put("/openapi.*", "exclude");
    casFilter.setInitParameters(filterInitParam);
    casFilter.setFilter(filter("com.ctrip.framework.apollo.sso.filter.ApolloAuthenticationFilter"));
    casFilter.addUrlPatterns("/*");
    casFilter.setOrder(2);
    return casFilter;
}

@Bean
public FilterRegistrationBean multipartFilterRegistrationBean() {
    log.debug("New instance of " + FilterRegistrationBean.class);
    final MultipartFilter multipartFilter = new MultipartFilter();
    final FilterRegistrationBean filterRegistrationBean = new FilterRegistrationBean(multipartFilter);
    filterRegistrationBean.addInitParameter("multipartResolverBeanName", "multipartResolver");
    return filterRegistrationBean;
}

@Bean
public FilterRegistrationBean jerseyFilter() {
    FilterRegistrationBean filter = new FilterRegistrationBean(new ServletContainer());
    filter.addInitParameter("jersey.config.server.provider.packages", "flapjack.controller");
    filter.addInitParameter("jersey.config.servlet.filter.forwardOn404", "true");
    filter.addInitParameter("jersey.config.server.provider.classnames", "flapjack.conf.CambridgeMvcFeature");
    filter.addInitParameter("javax.ws.rs.Application", "flapjack.conf.JerseyConfig");
    filter.setOrder(2);
    return filter;
}

@Bean
public FilterRegistrationBean aFilter(Reporter reporter) {
    FilterRegistrationBean filter = new FilterRegistrationBean();
    filter.setName("beforeAfterForwardLogging");
    filter.setOrder(FilterRegistrationBean.LOWEST_PRECEDENCE);
    filter.setFilter(new OncePerRequestFilter() {

        @Override
        protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
            try {
                reporter.filterEnter();
                filterChain.doFilter(request, response);
            } finally {
                reporter.filterExit();
            }
        }
    });
    return filter;
}

/**
   * Register PerformanceLogFilter to log running time of requests.
   *
   * @return filter
   */
@Bean
public FilterRegistrationBean performanceLogFilter() {
    FilterRegistrationBean registration = new FilterRegistrationBean();
    Filter performanceLogFilter = new PerformanceLogFilter();
    this.beanFactory.autowireBean(performanceLogFilter);
    registration.setFilter(performanceLogFilter);
    registration.addUrlPatterns("/*");
    return registration;
}

/**
   * Register PerformanceLogFilter to log running time of requests.
   *
   * @return filter
   */
@Bean
public FilterRegistrationBean performanceLogFilter() {
    FilterRegistrationBean registration = new FilterRegistrationBean();
    Filter performanceLogFilter = new PerformanceLogFilter();
    this.beanFactory.autowireBean(performanceLogFilter);
    registration.setFilter(performanceLogFilter);
    registration.addUrlPatterns("/*");
    return registration;
}

@Bean
public FilterRegistrationBean securityHeadersFilter(@Value("${pds.piwik.url}") String urlPiwik) {
    return filter(2, new OncePerRequestFilter() {

        @Override
        protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
            filterChain.doFilter(request, response);
            response.setHeader("X-XSS-Protection", "0");
            response.setHeader("X-Content-Type-Options", "nosniff");
            response.setHeader("Content-Security-Policy", "script-src: 'self' 'unsafe-inline' '" + urlPiwik + "' 'barra.brasil.gov.br'; default-src: 'self'");
        }
    });
}

@Bean
public FilterRegistrationBean filterRegistrationBean() {
    FilterRegistrationBean filterRegistrationBean = new FilterRegistrationBean();
    filterRegistrationBean.setFilter(new WebStatFilter());
    filterRegistrationBean.addUrlPatterns("/*");
    filterRegistrationBean.addInitParameter("exclusions", "*.js,*.gif,*.jpg,*.png,*.css,*.ico,/druid/*");
    return filterRegistrationBean;
}

//	<filter>
//	<filter-name>wicket.filter</filter-name>
//	<filter-class>
//	org.apache.wicket.protocol.http.WicketFilter
//			</filter-class>
//	<init-param>
//	<param-name>applicationFactoryClassName</param-name>
//	<param-value>org.apache.wicket.spring.SpringWebApplicationFactory</param-value>
//	</init-param>
//	<!-- See maven profile : development or deployment -->
//	<init-param>
//	<param-name>configuration</param-name>
//	<param-value>${wicket.configuration}</param-value>
//	</init-param>
//
//	<init-param>
//	<param-name>testsMode</param-name>
//	<param-value>${tests.mode}</param-value>
//	</init-param>
//
//	<init-param>
//	<param-name>mockMode</param-name>
//	<!--<param-value>true</param-value>-->
//	<param-value>${mock.mode}</param-value>
//	</init-param>
//	</filter>
//	<!-- No "/*" because we use relative URLs for resources files (.css, .jpg, ...) inside HTML -->
//	<filter-mapping>
//	<filter-name>wicket.filter</filter-name>
//	<url-pattern>/portal/*</url-pattern>
//    </filter-mapping>
@Bean
public FilterRegistrationBean wicketFilterRegistration() {
    FilterRegistrationBean registration = new FilterRegistrationBean();
    WicketFilter wicketFilter = new WicketFilter();
    registration.setFilter(wicketFilter);
    registration.setName("wicketFilter");
    registration.addInitParameter(WicketFilter.APP_FACT_PARAM, SpringWebApplicationFactory.class.getName());
    registration.addInitParameter("configuration", configuration);
    registration.addInitParameter("testsMode", String.valueOf(testMode));
    registration.addInitParameter("mockMode", String.valueOf(mockMode));
    registration.addInitParameter(WicketFilter.FILTER_MAPPING_PARAM, "/portal/*");
    registration.addInitParameter(WicketFilter.IGNORE_PATHS_PARAM, "/favicon.ico");
    registration.addUrlPatterns("/portal/*");
    registration.setDispatcherTypes(DispatcherType.REQUEST, DispatcherType.FORWARD);
    registration.setMatchAfter(true);
    return registration;
}

@Bean
public FilterRegistrationBean serviceContextFilterRegistration() {
    FilterRegistrationBean registration = new FilterRegistrationBean();
    registration.setFilter(new ServiceContextServletFilter());
    Map<String, String> initParams = new HashMap<String, String>(1);
    initParams.put("ServiceContextFactoryImplementationClassName", "org.sculptor.framework.context.ServletContainerServiceContextFactory");
    registration.setInitParameters(initParams);
    registration.addUrlPatterns("/rest/*");
    registration.setOrder(Ordered.LOWEST_PRECEDENCE);
    return registration;
}

@Bean
public FilterRegistrationBean serviceContextFilterRegistration() {
    FilterRegistrationBean registration = new FilterRegistrationBean();
    registration.setFilter(new ServiceContextServletFilter());
    Map<String, String> initParams = new HashMap<String, String>(1);
    initParams.put("ServiceContextFactoryImplementationClassName", "org.sculptor.framework.context.ServletContainerServiceContextFactory");
    registration.setInitParameters(initParams);
    registration.addUrlPatterns("/rest/*");
    registration.setOrder(Ordered.LOWEST_PRECEDENCE);
    return registration;
}

@Bean
public FilterRegistrationBean correlationIdFilter() {
    return new FilterRegistrationBean(new Filter() {

        @Override
        public void init(FilterConfig filterConfig) throws ServletException {
        }

        @Override
        public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
            chain.doFilter(request, response);
        }

        @Override
        public void destroy() {
        }
    });
}

@Bean
public FilterRegistrationBean shiroFilterRegistrationBean() {
    FilterRegistrationBean registrationBean = new FilterRegistrationBean();
    ShiroFilter shiroFilter = new ShiroFilter();
    registrationBean.setFilter(shiroFilter);
    registrationBean.setDispatcherTypes(DispatcherType.REQUEST, DispatcherType.values());
    registrationBean.setOrder(0);
    return registrationBean;
}

private void createAndRegisterServletFilterBean(DefaultListableBeanFactory beanFactory, BeanDefinition bd, Class<?> beanClass, WebFilter webFilter) {
    BeanDefinitionBuilder builder = BeanDefinitionBuilder.rootBeanDefinition(FilterRegistrationBean.class);
    builder.addPropertyValue("filter", bd);
    builder.addPropertyValue("urlPatterns", new LinkedHashSet<>(Arrays.asList(webFilter.urlPatterns())));
    String beanName = webFilter.filterName().isEmpty() ? StringUtils.uncapitalize(beanClass.getSimpleName()) : webFilter.filterName();
    beanFactory.registerBeanDefinition(beanName, builder.getBeanDefinition());
}

@Bean
FilterRegistrationBean corsFilter(@Value("${tagit.origin:}") String origin) {
    return new FilterRegistrationBean(new Filter() {

        public void doFilter(ServletRequest req, ServletResponse res, FilterChain chain) throws IOException, ServletException {
            HttpServletRequest request = (HttpServletRequest) req;
            HttpServletResponse response = (HttpServletResponse) res;
            String method = request.getMethod();
            response.setHeader("Access-Control-Allow-Origin", "http://localhost:9000");
            response.setHeader("Access-Control-Allow-Methods", "POST,GET,OPTIONS,DELETE");
            response.setHeader("Access-Control-Max-Age", Long.toString(60 * 60));
            response.setHeader("Access-Control-Allow-Credentials", "true");
            response.setHeader("Access-Control-Allow-Headers", "Origin,Accept,X-Requested-With,Content-Type,Access-Control-Request-Method,Access-Control-Request-Headers,Authorization");
            if ("OPTIONS".equals(method)) {
                response.setStatus(HttpStatus.OK.value());
            } else {
                chain.doFilter(req, res);
            }
        }

        public void init(FilterConfig filterConfig) {
        }

        public void destroy() {
        }
    });
}

/**
   * This method obtains the method that register the openEntityManagerInView 
   * filter
   * 
   * @return MethodMetadata with the information about the new method
   */
private MethodMetadata getOpenEntityManagerInViewFilterMethod() {
    // Check if already exists
    if (openEntityManagerInViewFilterMethod != null) {
        return openEntityManagerInViewFilterMethod;
    }
    JavaType filterRegistrationBeanType = new JavaType("org.springframework.boot.context.embedded.FilterRegistrationBean");
    // Generating method body 
    InvocableMemberBodyBuilder bodyBuilder = new InvocableMemberBodyBuilder();
    // FilterRegistrationBean filterRegBean = new FilterRegistrationBean();
    bodyBuilder.appendFormalLine("%s filterRegBean = new FilterRegistrationBean();", getNameOfJavaType(filterRegistrationBeanType));
    // filterRegBean.setFilter(new OpenEntityManagerInViewFilter());
    bodyBuilder.appendFormalLine("filterRegBean.setFilter(new %s());", getNameOfJavaType(new JavaType("org.springframework.orm.jpa.support.OpenEntityManagerInViewFilter")));
    // List<String> urlPatterns = new ArrayList<String>();
    bodyBuilder.appendFormalLine("%s<String> urlPatterns = new %s<String>();", getNameOfJavaType(JavaType.LIST), getNameOfJavaType(JavaType.ARRAY_LIST));
    // urlPatterns.add(this.cxfServletPath + "/*");
    bodyBuilder.appendFormalLine("urlPatterns.add(%s() + \"/*\");", getAccessorMethod(getServletField()).getMethodName());
    // filterRegBean.setUrlPatterns(urlPatterns);
    bodyBuilder.appendFormalLine("filterRegBean.setUrlPatterns(urlPatterns);");
    // if (LOG.isDebugEnabled()) {
    bodyBuilder.appendFormalLine("if (%s().isDebugEnabled()) {", getAccessorMethod(getLoggerField()).getMethodName());
    // LOG.debug("Registering the 'OpenEntityManagerInViewFilter' filter for the '"
    bodyBuilder.indent();
    bodyBuilder.appendFormalLine("%s().debug(\"Registering the 'OpenEntityManagerInViewFilter' filter for the '\"", getAccessorMethod(getLoggerField()).getMethodName());
    // .concat(this.cxfServletPath + "/*").concat("' URL."));
    bodyBuilder.indent();
    bodyBuilder.appendFormalLine(".concat(%s() + \"/*\").concat(\"' URL.\"));", getAccessorMethod(getServletField()).getMethodName());
    bodyBuilder.indentRemove();
    bodyBuilder.indentRemove();
    // }
    bodyBuilder.appendFormalLine("}");
    // return filterRegBean;
    bodyBuilder.appendFormalLine("return filterRegBean;");
    MethodMetadataBuilder method = new MethodMetadataBuilder(getId(), Modifier.PUBLIC, new JavaSymbolName("openEntityManagerInViewFilter"), filterRegistrationBeanType, bodyBuilder);
    method.addAnnotation(new AnnotationMetadataBuilder(SpringJavaType.BEAN));
    openEntityManagerInViewFilterMethod = method.build();
    return openEntityManagerInViewFilterMethod;
}

@Bean
public FilterRegistrationBean cacheFilterRegistration(ServletRegistrationBean dispatcherRegistration) {
    FilterRegistrationBean registrationBean = new FilterRegistrationBean(new CacheFilter());
    registrationBean.addUrlPatterns("/img/*");
    return registrationBean;
}

@Bean
public FilterRegistrationBean ticketFilter(Iterator<UUID> tickets) {
    return filter(0, new TicketFilter(tickets));
}

@Bean
public FilterRegistrationBean commonsRequestLoggingFilter() {
    final FilterRegistrationBean registrationBean = new FilterRegistrationBean();
    registrationBean.setFilter(new CORSFilter());
    return registrationBean;
}

@Bean
public FilterRegistrationBean resourceLogFilter() {
    log.debug("Initializing resource log filter");
    FilterRegistrationBean registration = new FilterRegistrationBean();
    registration.setFilter(new ResourceLogFilter());
    registration.addUrlPatterns("/psi/*");
    registration.setOrder(2);
    return registration;
}

@Bean
public FilterRegistrationBean xssFilterRegistrationBean() {
    FilterRegistrationBean registrationBean = new FilterRegistrationBean();
    registrationBean.setFilter(xssFilter());
    registrationBean.addUrlPatterns("/*");
    return registrationBean;
}

@Bean
public FilterRegistrationBean dandelionFilterRegistrationBean() {
    return new FilterRegistrationBean(new DandelionFilter());
}

@Bean
public FilterRegistrationBean javaMelody() {
    FilterRegistrationBean javaMelody = new FilterRegistrationBean();
    javaMelody.setFilter(new MonitoringFilter());
    javaMelody.addInitParameter(Parameter.LOG.getCode(), Boolean.toString(true));
    javaMelody.addUrlPatterns("/*");
    return javaMelody;
}

@Bean
FilterRegistrationBean brokerApiVersionFilter() {
    FilterRegistrationBean bean = new FilterRegistrationBean();
    bean.setFilter(new BrokerApiVersionFilter());
    bean.addUrlPatterns("/v2/*");
    return bean;
}

@Bean
public FilterRegistrationBean removeHeaderRegistration() {
    RemoveHeaderFilter rhf = new RemoveHeaderFilter();
    FilterRegistrationBean registrationBean = new FilterRegistrationBean(rhf);
    logger.info("Adding remove header filter!");
    registrationBean.addUrlPatterns("/*");
    return registrationBean;
}

@Bean
public FilterRegistrationBean filterRegistrationBean() {
    FilterRegistrationBean filterRegistrationBean = new FilterRegistrationBean();
    filterRegistrationBean.setFilter(basicAuthenticationProcessingFilter);
    filterRegistrationBean.setEnabled(false);
    return filterRegistrationBean;
}

@Bean
FilterRegistrationBean wro4jFilterRegistration(ConfigurableWroFilter wroFilter) {
    FilterRegistrationBean filterRegistrationBean = new FilterRegistrationBean(wroFilter);
    filterRegistrationBean.addUrlPatterns(wro4jProperties.getUrlPattern());
    return filterRegistrationBean;
}

@Bean
public FilterRegistrationBean hiddenHttpMethodFilter() {
    HiddenHttpMethodFilter hiddenHttpMethodFilter = new HiddenHttpMethodFilter();
    FilterRegistrationBean registrationBean = new FilterRegistrationBean();
    registrationBean.setFilter(hiddenHttpMethodFilter);
    return registrationBean;
}

@Bean
public FilterRegistrationBean j360FilterRegistration() {
    FilterRegistrationBean registration = new FilterRegistrationBean();
    registration.setFilter(characterEncodingFilter());
    return registration;
}

@Bean
public FilterRegistrationBean openEntityManagerInViewFilter() {
    FilterRegistrationBean reg = new FilterRegistrationBean();
    reg.setName("OpenEntityManagerInViewFilter");
    reg.setFilter(new OpenEntityManagerInViewFilter());
    return reg;
}

/*
	 * Withdrawn after issue discovered with jetty
	 */
//	@Bean
//	public ServletRegistrationBean threadStartupServlet() {
//	    ServletRegistrationBean registration =
//	    		new ServletRegistrationBean(new ThreadMonitorStartupServlet(), "/admin4j/threads");
//	    registration.setLoadOnStartup(1);
//	    return registration;
//	}
@Bean
public FilterRegistrationBean monetaPerformanceFilter() {
    FilterRegistrationBean registration = new FilterRegistrationBean(new MonetaPerformanceFilter(), monetaServlet(), monetaTopicListServlet());
    return registration;
}

@Bean
public FilterRegistrationBean resourceFilterRegistration() {
    FilterRegistrationBean bean = new FilterRegistrationBean();
    bean.setFilter(resourceFilter);
    bean.setEnabled(false);
    return bean;
}

@Bean
FilterRegistrationBean crossOriginFilter() {
    FilterRegistrationBean filterRegistrationBean = new FilterRegistrationBean();
    filterRegistrationBean.setFilter(new CrossOriginFilter());
    filterRegistrationBean.addUrlPatterns("/api/v1/*", "/management/*");
    return filterRegistrationBean;
}