View Javadoc

1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  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,
13   * software distributed under the License is distributed on an
14   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   * KIND, either express or implied.  See the License for the
16   * specific language governing permissions and limitations
17   * under the License.
18   */
19  package org.apache.shiro.guice.web;
20  
21  import com.google.common.base.Function;
22  import com.google.common.collect.Iterators;
23  import com.google.inject.Injector;
24  import com.google.inject.Key;
25  import org.apache.shiro.util.PatternMatcher;
26  import org.apache.shiro.web.filter.mgt.FilterChainResolver;
27  import org.apache.shiro.web.util.WebUtils;
28  
29  import javax.servlet.Filter;
30  import javax.servlet.FilterChain;
31  import javax.servlet.ServletRequest;
32  import javax.servlet.ServletResponse;
33  import java.util.Map;
34  
35  class SimpleFilterChainResolver implements FilterChainResolver {
36      private final Map<String, Key<? extends Filter>[]> chains;
37      private final Injector injector;
38      private final PatternMatcher patternMatcher;
39  
40      SimpleFilterChainResolver(Map<String, Key<? extends Filter>[]> chains, Injector injector, PatternMatcher patternMatcher) {
41          this.chains = chains;
42          this.injector = injector;
43          this.patternMatcher = patternMatcher;
44      }
45  
46      public FilterChain getChain(ServletRequest request, ServletResponse response, final FilterChain originalChain) {
47          String path = WebUtils.getPathWithinApplication(WebUtils.toHttp(request));
48          for (final String pathPattern : chains.keySet()) {
49              if (patternMatcher.matches(pathPattern, path)) {
50                  return new SimpleFilterChain(originalChain, Iterators.transform(Iterators.forArray(chains.get(pathPattern)),
51                          new Function<Key<? extends Filter>, Filter>() {
52                              public Filter apply(Key<? extends Filter> input) {
53                                  return injector.getInstance(input);
54                              }
55                          }));
56              }
57          }
58          return null;
59      }
60  
61  }