io.github.oliviercailloux.javagrade.bytecode.SourceScanner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jgradlib Show documentation
The newest version!
package io.github.oliviercailloux.javagrade.bytecode;

import static com.google.common.base.Preconditions.checkArgument;
import static com.google.common.base.Preconditions.checkNotNull;
import static com.google.common.base.Verify.verify;
import static io.github.oliviercailloux.jaris.exceptions.Unchecker.IO_UNCHECKER;

import com.google.common.collect.ImmutableSet;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Stream;

public class SourceScanner {
  private static final Pattern PACKAGE_PATTERN =
      Pattern.compile("^\\h*package\\h+(?[^\\h]*)\\h*;");

  public static ImmutableSet scan(Path start) {
    checkNotNull(start);
    final ImmutableSet javaPaths;
    try (Stream found = IO_UNCHECKER.getUsing(() -> Files.find(start, Integer.MAX_VALUE, (p,
        a) -> p.getFileName() == null ? false : p.getFileName().toString().endsWith(".java")))) {
      verify(found != null);
      assert found != null;
      javaPaths = found.collect(ImmutableSet.toImmutableSet());
    }
    return javaPaths.stream().map(SourceScanner::asSourceClass)
        .collect(ImmutableSet.toImmutableSet());
  }

  public static SourceClass asSourceClass(Path sourcePath) {
    final String shortClassNameFromFileName =
        sourcePath.getFileName().toString().replace(".java", "");
    // String content = JavaGradeUtils.read(sourcePath);
    String content;
    try {
      content = Files.readString(sourcePath);
    } catch (IOException e) {
      throw new IllegalStateException(e);
    }
    final Matcher matcher = PACKAGE_PATTERN.matcher(content);
    final boolean found = matcher.find();
    final String packageName = found ? matcher.group("packageName") : "";
    checkArgument(!matcher.find());
    return new SourceClass(sourcePath, packageName, shortClassNameFromFileName);
  }

  public static class SourceClass {
    private Path path;
    private String packageName;
    private String shortClassName;

    SourceClass(Path path, String packageName, String shortClassName) {
      this.path = checkNotNull(path);
      this.packageName = checkNotNull(packageName);
      this.shortClassName = checkNotNull(shortClassName);
    }

    public Path getPath() {
      return path;
    }

    public String getPackageName() {
      return packageName;
    }

    public String getShortClassName() {
      return shortClassName;
    }

    public String getFullClassName() {
      final String separator = packageName.equals("") ? "" : ".";
      return packageName + separator + shortClassName;
    }
  }
}