org.pitest.mutationtest.tooling.SmartSourceLocator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pitest-entry Show documentation
Pitest entry point
There is a newer version: 1.17.1
/*
 * Copyright 2010 Henry Coles
 *
 * 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 org.pitest.mutationtest.tooling;

import java.io.File;
import java.io.IOException;
import java.io.Reader;
import java.nio.charset.Charset;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.function.Function;

import org.pitest.functional.FCollection;
import java.util.Optional;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import org.pitest.mutationtest.SourceLocator;
import org.pitest.util.Unchecked;

public class SmartSourceLocator implements SourceLocator {

  private static final int                MAX_DEPTH = 4;

  private final List children;
  private final Charset inputCharset;

  public SmartSourceLocator(final Collection roots, Charset inputCharset) {
    this.inputCharset = inputCharset;
    final Collection childDirs = FCollection.flatMap(roots, collectChildren(MAX_DEPTH));
    childDirs.addAll(roots);

    this.children = FCollection.map(childDirs, f -> new DirectorySourceLocator(f, this.inputCharset));
  }

  private Function> collectChildren(final int depth) {
    return a -> collectDirectories(a, depth);
  }

  private Collection collectDirectories(Path root, int depth) {
    try {
      if (!Files.exists(root)) {
        return Collections.emptyList();
      }

      try (Stream matches = Files.find(root, depth, (unused, attributes) -> attributes.isDirectory())) {
        return matches.collect(Collectors.toList());
      }

    } catch (IOException ex) {
      throw Unchecked.translateCheckedException(ex);
    }

  }

  @Override
  public Optional locate(Collection classes, String fileName) {
    for (final SourceLocator each : this.children) {
      final Optional reader = each.locate(classes, fileName);
      if (reader.isPresent()) {
        return reader;
      }
    }
    return Optional.empty();
  }

  /** Provide hint as to where to look when dealing with multiple modules. NOT thread safe.
   */
  public void sourceRootHint(Path file) {
    children.sort(new PathComparator(file, File.separator));
  }
}