org.pitest.classpath.ClassPath Maven / Gradle / Ivy

/*
 * 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.classpath;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Optional;
import java.util.Set;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.logging.Logger;
import java.util.zip.ZipException;

import org.pitest.functional.FCollection;
import org.pitest.util.Log;
import org.pitest.util.ManifestUtils;
import org.pitest.util.PitError;
import org.pitest.util.StreamUtil;

public class ClassPath {

  private static final Logger         LOG = Log.getLogger();

  private final CompoundClassPathRoot root;

  public ClassPath() {
    this(ClassPath.getClassPathElementsAsFiles());
  }

  public ClassPath(final ClassPathRoot... roots) {
    this(Arrays.asList(roots));
  }

  public ClassPath(final Collection files) {
    this(createRoots(FCollection.filter(files, exists())));
  }

  ClassPath(List roots) {
    this.root = new CompoundClassPathRoot(roots);
  }

  public Collection classNames() {
    return this.root.classNames();
  }

  // fixme should not be determining type here
  private static List createRoots(final Collection files) {
    File lastFile = null;
    try {
      final List rs = new ArrayList<>();

      for (final File f : files) {
        lastFile = f;
        if (f.isDirectory()) {
          rs.add(new DirectoryClassPathRoot(f));
        } else {
          handleArchive(rs, f);
        }
      }
      return rs;
    } catch (final IOException ex) {
      throw new PitError("Error handling file " + lastFile, ex);
    }
  }

  private static void handleArchive(final List rs, final File f)
      throws IOException {
    try {
      if (!f.canRead()) {
        throw new IOException("Can't read the file " + f);
      }
      rs.add(new ArchiveClassPathRoot(f));
    } catch (final ZipException ex) {
      LOG.warning("Can't open the archive " + f);
    }
  }

  public byte[] getClassData(final String classname) throws IOException {
    try (InputStream is = this.root.getData(classname)) {
      if (is != null) {
        return StreamUtil.streamToByteArray(is);
      } else {
        return null;
      }
    }
  }

  public URL findResource(final String name) {
    try {
      return this.root.getResource(name);
    } catch (final IOException exception) {
      return null;
    }
  }

  public static Collection getClassPathElementsAsPaths() {
    final Set filesAsString = new LinkedHashSet<>();
    FCollection.mapTo(getClassPathElementsAsFiles(), file -> file.getPath(),
        filesAsString);
    return filesAsString;
  }


  public static Collection getClassPathElementsAsFiles() {
    final Set us = new LinkedHashSet<>();
    FCollection.mapTo(getClassPathElementsAsAre(), stringToCanonicalFile(), us);
    
    addEntriesFromClasspathManifest(us);
    return us;
  }

  /**
   * Because classpaths can become longer than the OS supports pitest creates temporary jar files and places the classpath
   * in the manifest where there is no size limit.
   * 
   * We must therefore parse them out again here. 
   * 
   * @param elements existing elements
   */
  private static void addEntriesFromClasspathManifest(final Set elements) {
    Optional maybeJar = elements.stream().filter( f -> f.getName().startsWith("classpath") && f.getName().endsWith(".jar"))
    .findFirst();
    maybeJar.ifPresent(file -> elements.addAll(ManifestUtils.readClasspathManifest(file)));
  }

  public Collection findClasses(final Predicate nameFilter) {
    return FCollection.filter(classNames(), nameFilter);
  }

  public String getLocalClassPath() {
    return this.root.cacheLocation().get();
  }

  public ClassPath getComponent(final Predicate predicate) {
    return new ClassPath(FCollection.filter(this.root, predicate).toArray(
        new ClassPathRoot[0]));
  }

  private static Predicate exists() {
    return a -> a.exists() && a.canRead();
  }

  private static Function stringToCanonicalFile() {
    return fileAsString -> {
        try {
          return new File(fileAsString).getCanonicalFile();
        } catch (final IOException ex) {
          throw new PitError("Error transforming classpath element "
              + fileAsString, ex);
        }
      };
  }

  /** FIXME move somewhere common */
  private static List getClassPathElementsAsAre() {
    final String classPath = System.getProperty("java.class.path");
    final String separator = File.pathSeparator;
    if (classPath != null) {
      return Arrays.asList(classPath.split(separator));
    } else {
      return new ArrayList<>();
    }

  }

}