[Analysis API Standalone] copy JdkClassFinder from IntelliJ sources
^KT-60884
This commit is contained in:
Ilya Kirillov
committed by
Space Team
Space Team
parent
180434951b
commit
8f824dcff9
+214
@@ -0,0 +1,214 @@
|
||||
/*
|
||||
* Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package org.jetbrains.kotlin.analysis.api.impl.base.util;
|
||||
|
||||
import com.intellij.ide.highlighter.ArchiveFileType;
|
||||
import com.intellij.openapi.fileTypes.FileTypeRegistry;
|
||||
import com.intellij.openapi.util.io.FileUtil;
|
||||
import com.intellij.openapi.util.registry.Registry;
|
||||
import com.intellij.openapi.util.text.StringUtil;
|
||||
import com.intellij.openapi.vfs.StandardFileSystems;
|
||||
import com.intellij.openapi.vfs.VirtualFile;
|
||||
import com.intellij.openapi.vfs.VirtualFileManager;
|
||||
import com.intellij.util.containers.CollectionFactory;
|
||||
import com.intellij.util.containers.ContainerUtil;
|
||||
import com.intellij.util.io.URLUtil;
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
import org.jetbrains.annotations.Nullable;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.nio.file.DirectoryStream;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import java.util.*;
|
||||
import java.util.stream.Stream;
|
||||
|
||||
/*
|
||||
Copy of corresponding classes from IntelliJ source repository.
|
||||
We cannot reuse them directly as not all those methods are parts of intellij-core which is used in Analysis API
|
||||
*/
|
||||
class JdkClassFinder {
|
||||
// copy of 'com.intellij.openapi.projectRoots.impl.JavaSdkImpl.findClasses'
|
||||
static @NotNull List<String> findClasses(@NotNull Path jdkHome, boolean isJre) {
|
||||
List<String> result = new ArrayList<>();
|
||||
|
||||
if (isExplodedModularRuntime(jdkHome)) {
|
||||
try {
|
||||
try (DirectoryStream<Path> roots = Files.newDirectoryStream(jdkHome.resolve("modules"))) {
|
||||
for (Path root : roots) {
|
||||
result.add(getUrlForLibraryRoot(root));
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (IOException ignore) {
|
||||
}
|
||||
}
|
||||
else if (isModularRuntime(jdkHome)) {
|
||||
String jrtBaseUrl = StandardFileSystems.JRT_PROTOCOL_PREFIX + vfsPath(jdkHome) + URLUtil.JAR_SEPARATOR;
|
||||
List<String> modules = readModulesFromReleaseFile(jdkHome);
|
||||
if (modules != null) {
|
||||
for (String module : modules) {
|
||||
result.add(jrtBaseUrl + module);
|
||||
}
|
||||
}
|
||||
else {
|
||||
VirtualFile jrt = VirtualFileManager.getInstance().findFileByUrl(jrtBaseUrl);
|
||||
if (jrt != null) {
|
||||
for (VirtualFile virtualFile : jrt.getChildren()) {
|
||||
result.add(virtualFile.getUrl());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
else {
|
||||
for (Path root : getJdkClassesRoots(jdkHome, isJre)) {
|
||||
result.add(getUrlForLibraryRoot(root));
|
||||
}
|
||||
}
|
||||
|
||||
Collections.sort(result);
|
||||
return result;
|
||||
}
|
||||
|
||||
// copy of 'com.intellij.openapi.projectRoots.impl.JavaSdkImpl.vfsPath'
|
||||
private static String vfsPath(@NotNull Path path) {
|
||||
return FileUtil.toSystemIndependentName(path.toAbsolutePath().toString());
|
||||
}
|
||||
|
||||
|
||||
// copy of 'com.intellij.openapi.projectRoots.JdkUtil.isExplodedModularRuntime'
|
||||
private static boolean isExplodedModularRuntime(@NotNull Path homePath) {
|
||||
return Files.isDirectory(homePath.resolve("modules/java.base"));
|
||||
}
|
||||
|
||||
// copy of 'com.intellij.openapi.projectRoots.JdkUtil.isModularRuntime'
|
||||
private static boolean isModularRuntime(@NotNull Path homePath) {
|
||||
return Files.isRegularFile(homePath.resolve("lib/jrt-fs.jar")) || isExplodedModularRuntime(homePath);
|
||||
}
|
||||
|
||||
// copy of 'com.intellij.openapi.vfs.VfsUtil.getUrlForLibraryRoot'
|
||||
private static @NotNull String getUrlForLibraryRoot(@NotNull Path libraryRoot) {
|
||||
return getUrlForLibraryRoot(libraryRoot.toAbsolutePath().toString(), libraryRoot.getFileName().toString());
|
||||
}
|
||||
|
||||
// copy of `com.intellij.openapi.vfs.VfsUtil.getUrlForLibraryRoot`
|
||||
private static @NotNull String getUrlForLibraryRoot(@NotNull String libraryRootAbsolutePath, @NotNull String libraryRootFileName) {
|
||||
String path = FileUtil.toSystemIndependentName(libraryRootAbsolutePath);
|
||||
return FileTypeRegistry.getInstance().getFileTypeByFileName(libraryRootFileName) == ArchiveFileType.INSTANCE
|
||||
? VirtualFileManager.constructUrl(StandardFileSystems.JAR_PROTOCOL, path + URLUtil.JAR_SEPARATOR)
|
||||
: VirtualFileManager.constructUrl(StandardFileSystems.FILE_PROTOCOL, path);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Tries to load the list of modules in the JDK from the 'release' file. Returns null if the 'release' file is not there
|
||||
* or doesn't contain the expected information.
|
||||
* <p>
|
||||
* Copy of `com.intellij.openapi.projectRoots.impl.JavaSdkImpl.readModulesFromReleaseFile`
|
||||
*/
|
||||
private static @Nullable List<String> readModulesFromReleaseFile(@NotNull Path jrtBaseDir) {
|
||||
try (InputStream stream = Files.newInputStream(jrtBaseDir.resolve("release"))) {
|
||||
Properties p = new Properties();
|
||||
p.load(stream);
|
||||
String modules = p.getProperty("MODULES");
|
||||
if (modules != null) {
|
||||
return StringUtil.split(StringUtil.unquoteString(modules), " ");
|
||||
}
|
||||
}
|
||||
catch (IOException | IllegalArgumentException e) {
|
||||
return null;
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
// copy of `org.jetbrains.jps.model.java.impl.JavaSdkUtil.getJdkClassesRoots`
|
||||
private static @NotNull List<Path> getJdkClassesRoots(@NotNull Path home, boolean isJre) {
|
||||
Path[] jarDirs;
|
||||
Path fileName = home.getFileName();
|
||||
if (fileName != null && "Home".equals(fileName.toString()) && Files.exists(home.resolve("../Classes/classes.jar"))) {
|
||||
Path libDir = home.resolve("lib");
|
||||
Path classesDir = home.resolveSibling("Classes");
|
||||
Path libExtDir = libDir.resolve("ext");
|
||||
Path libEndorsedDir = libDir.resolve("endorsed");
|
||||
jarDirs = new Path[] {libEndorsedDir, libDir, classesDir, libExtDir};
|
||||
}
|
||||
else if (Files.exists(home.resolve("lib/jrt-fs.jar"))) {
|
||||
jarDirs = new Path[0];
|
||||
}
|
||||
else {
|
||||
Path libDir = home.resolve(isJre ? "lib" : "jre/lib");
|
||||
Path libExtDir = libDir.resolve("ext");
|
||||
Path libEndorsedDir = libDir.resolve("endorsed");
|
||||
jarDirs = new Path[] {libEndorsedDir, libDir, libExtDir};
|
||||
}
|
||||
|
||||
List<Path> rootFiles = new ArrayList<>();
|
||||
|
||||
if (Registry.is("project.structure.add.tools.jar.to.new.jdk", false)) {
|
||||
@SuppressWarnings("IdentifierGrammar") Path toolsJar = home.resolve("lib/tools.jar");
|
||||
if (Files.isRegularFile(toolsJar)) {
|
||||
rootFiles.add(toolsJar);
|
||||
}
|
||||
}
|
||||
|
||||
Set<String> pathFilter = CollectionFactory.createFilePathSet();
|
||||
for (Path jarDir : jarDirs) {
|
||||
if (jarDir != null && Files.isDirectory(jarDir)) {
|
||||
try (DirectoryStream<Path> stream = Files.newDirectoryStream(jarDir, "*.jar")) {
|
||||
for (Path jarFile : stream) {
|
||||
String jarFileName = jarFile.getFileName().toString();
|
||||
if (jarFileName.equals("alt-rt.jar") || jarFileName.equals("alt-string.jar")) {
|
||||
continue; // filter out alternative implementations
|
||||
}
|
||||
String canonicalPath = getCanonicalPath(jarFile);
|
||||
if (canonicalPath == null || !pathFilter.add(canonicalPath)) {
|
||||
continue; // filter out duplicate (symbolically linked) .jar files commonly found in OS X JDK distributions
|
||||
}
|
||||
rootFiles.add(jarFile);
|
||||
}
|
||||
}
|
||||
catch (IOException ignored) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (ContainerUtil.exists(rootFiles, path -> path.getFileName().toString().startsWith("ibm"))) {
|
||||
// ancient IBM JDKs split JRE classes between `rt.jar` and `vm.jar`, and the latter might be anywhere
|
||||
try (Stream<Path> paths = Files.walk(isJre ? home : home.resolve("jre"))) {
|
||||
paths.filter(path -> path.getFileName().toString().equals("vm.jar"))
|
||||
.findFirst()
|
||||
.ifPresent(rootFiles::add);
|
||||
}
|
||||
catch (IOException ignored) {
|
||||
}
|
||||
}
|
||||
|
||||
Path classesZip = home.resolve("lib/classes.zip");
|
||||
if (Files.isRegularFile(classesZip)) {
|
||||
rootFiles.add(classesZip);
|
||||
}
|
||||
|
||||
if (rootFiles.isEmpty()) {
|
||||
Path classesDir = home.resolve("classes");
|
||||
if (Files.isDirectory(classesDir)) {
|
||||
rootFiles.add(classesDir);
|
||||
}
|
||||
}
|
||||
|
||||
return rootFiles;
|
||||
}
|
||||
|
||||
// copy of `org.jetbrains.jps.model.java.impl.JavaSdkUtil.getCanonicalPath`
|
||||
private static @Nullable String getCanonicalPath(Path file) {
|
||||
try {
|
||||
return file.toRealPath().toString();
|
||||
}
|
||||
catch (IOException e) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user