| /* |
| * Copyright 2014 gitblit.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.gitblit.servlet; |
| |
| import java.io.IOException; |
| import java.util.ArrayList; |
| import java.util.Iterator; |
| import java.util.List; |
| |
| import com.google.inject.Inject; |
| import com.google.inject.Singleton; |
| import javax.servlet.Filter; |
| import javax.servlet.FilterChain; |
| import javax.servlet.FilterConfig; |
| import javax.servlet.ServletException; |
| import javax.servlet.ServletRequest; |
| import javax.servlet.ServletResponse; |
| |
| import ro.fortsoft.pf4j.PluginWrapper; |
| |
| import com.gitblit.extensions.HttpRequestFilter; |
| import com.gitblit.manager.IPluginManager; |
| import com.gitblit.manager.IRuntimeManager; |
| |
| /** |
| * A request filter than allows registered extension request filters to access |
| * request data. The intended purpose is for server monitoring plugins. |
| * |
| * @author David Ostrovsky |
| * @since 1.6.0 |
| */ |
| @Singleton |
| public class ProxyFilter implements Filter { |
| private final IRuntimeManager runtimeManager; |
| |
| private final IPluginManager pluginManager; |
| |
| private final List<HttpRequestFilter> filters; |
| |
| @Inject |
| public ProxyFilter( |
| IRuntimeManager runtimeManager, |
| IPluginManager pluginManager) { |
| |
| this.runtimeManager = runtimeManager; |
| this.pluginManager = pluginManager; |
| this.filters = new ArrayList<>(); |
| |
| } |
| |
| @Override |
| public void init(FilterConfig filterConfig) throws ServletException { |
| |
| filters.addAll(pluginManager.getExtensions(HttpRequestFilter.class)); |
| for (HttpRequestFilter f : filters) { |
| // wrap the filter config for Gitblit settings retrieval |
| PluginWrapper pluginWrapper = pluginManager.whichPlugin(f.getClass()); |
| FilterConfig runtimeConfig = new FilterRuntimeConfig(runtimeManager, |
| pluginWrapper.getPluginId(), filterConfig); |
| |
| f.init(runtimeConfig); |
| } |
| } |
| |
| @Override |
| public void doFilter(ServletRequest req, ServletResponse res, final FilterChain last) |
| throws IOException, ServletException { |
| final Iterator<HttpRequestFilter> itr = filters.iterator(); |
| new FilterChain() { |
| @Override |
| public void doFilter(ServletRequest req, ServletResponse res) throws IOException, |
| ServletException { |
| if (itr.hasNext()) { |
| itr.next().doFilter(req, res, this); |
| } else { |
| last.doFilter(req, res); |
| } |
| } |
| }.doFilter(req, res); |
| } |
| |
| @Override |
| public void destroy() { |
| for (HttpRequestFilter f : filters) { |
| f.destroy(); |
| } |
| } |
| } |