dev.jbang.net.jdkproviders.BaseFoldersJdkProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jbang-cli Show documentation
JBang Command Line Interface
The newest version!
package dev.jbang.net.jdkproviders;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.function.Predicate;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;

import dev.jbang.net.JdkProvider;
import dev.jbang.util.JavaUtil;
import dev.jbang.util.Util;

public abstract class BaseFoldersJdkProvider implements JdkProvider {
	@Nonnull
	@Override
	public List listInstalled() {
		if (Files.isDirectory(getJdksRoot())) {
			try (Stream jdkPaths = listJdkPaths()) {
				return jdkPaths
								.map(this::createJdk)
								.filter(Objects::nonNull)
								.sorted(Jdk::compareTo)
								.collect(Collectors.toList());
			} catch (IOException e) {
				Util.verboseMsg("Couldn't list installed JDKs", e);
			}
		}
		return Collections.emptyList();
	}

	@Nullable
	@Override
	public Jdk getJdkById(@Nonnull String id) {
		if (isValidId(id)) {
			try (Stream jdkPaths = listJdkPaths()) {
				return jdkPaths
								.filter(p -> jdkId(p.getFileName().toString()).equals(id))
								.map(this::createJdk)
								.filter(Objects::nonNull)
								.findFirst()
								.orElse(null);
			} catch (IOException e) {
				Util.verboseMsg("Couldn't list installed JDKs", e);
			}
		}
		return null;
	}

	@Nullable
	@Override
	public Jdk getJdkByPath(@Nonnull Path jdkPath) {
		if (jdkPath.startsWith(getJdksRoot())) {
			try (Stream jdkPaths = listJdkPaths()) {
				return jdkPaths
								.filter(jdkPath::startsWith)
								.map(this::createJdk)
								.filter(Objects::nonNull)
								.findFirst()
								.orElse(null);
			} catch (IOException e) {
				Util.verboseMsg("Couldn't list installed JDKs", e);
			}
		}
		return null;
	}

	/**
	 * Returns a path to the requested JDK. This method should never return
	 * null and should return the path where the requested JDK is
	 * either currently installed or where it would be installed if it were
	 * available. This only needs to be implemented for providers that are
	 * updatable.
	 *
	 * @param jdk The identifier of the JDK to install
	 * @return A path to the requested JDK
	 */
	@Nonnull
	protected Path getJdkPath(@Nonnull String jdk) {
		return getJdksRoot().resolve(jdk);
	}

	protected Predicate sameJdk(Path jdkRoot) {
		Path release = jdkRoot.resolve("release");
		return (Path p) -> {
			try {
				return Files.isSameFile(p.resolve("release"), release);
			} catch (IOException e) {
				return false;
			}
		};
	}

	protected Stream listJdkPaths() throws IOException {
		if (Files.isDirectory(getJdksRoot())) {
			return Files.list(getJdksRoot()).filter(this::acceptFolder);
		}
		return Stream.empty();
	}

	@Nonnull
	protected Path getJdksRoot() {
		throw new UnsupportedOperationException("Getting the JDK root folder not supported by " + getClass().getName());
	}

	@Nullable
	protected Jdk createJdk(Path home) {
		String name = home.getFileName().toString();
		Optional version = JavaUtil.resolveJavaVersionStringFromPath(home);
		if (version.isPresent() && acceptFolder(home)) {
			return createJdk(jdkId(name), home, version.get());
		}
		return null;
	}

	protected boolean acceptFolder(Path jdkFolder) {
		return Util.searchPath("javac", jdkFolder.resolve("bin").toString()) != null;
	}

	protected boolean isValidId(String id) {
		return id.endsWith("-" + name());
	}

	protected abstract String jdkId(String name);
}