content: /*
   * Copyright 2016 HuntBugs contributors
   * 
   * 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 one.util.huntbugs.repo;
  
  import java.util.function.Predicate;
  
  import com.strobel.assembler.metadata.ITypeLoader;
  
  /**
   * Repository which filters visited classes with given predicate
   * (though all classes still could be loaded).
   * 
   * @author Tagir Valeev
   */
  public class FilteredRepository implements Repository {
      private final Predicate<String> classFilter;
      private final Repository repository;
  
      /**
       * @param repository parent repository
       * @param classFilter predicate which accepts internal class name (like "a/b/c/d") and returns true if it should be visited
       */
      public FilteredRepository(Repository repository, Predicate<String> classFilter) {
          this.repository = repository;
          this.classFilter = classFilter;
      }
  
      @Override
      public ITypeLoader createTypeLoader() {
          return repository.createTypeLoader();
      }
  
      @Override
      public void visit(String rootPackage, RepositoryVisitor visitor) {
          repository.visit(rootPackage, new RepositoryVisitor() {
              
              @Override
              public boolean visitPackage(String packageName) {
                  return visitor.visitPackage(packageName);
              }
              
              @Override
              public void visitClass(String className) {
                  if(classFilter.test(className))
                      visitor.visitClass(className);
              }
          });
      }
  }
  