001/*
002 * Copyright (C) 2022 - 2024, the original author or authors.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *    http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package io.github.ascopes.jct.filemanagers.config;
017
018import io.github.ascopes.jct.compilers.JctCompiler;
019import io.github.ascopes.jct.filemanagers.JctFileManager;
020import io.github.ascopes.jct.utils.SpecialLocationUtils;
021import io.github.ascopes.jct.utils.StringUtils;
022import io.github.ascopes.jct.workspaces.impl.WrappingDirectoryImpl;
023import javax.tools.StandardLocation;
024import org.slf4j.Logger;
025import org.slf4j.LoggerFactory;
026
027/**
028 * Configurer for a file manager that applies the running JVM's classpath to the file manager.
029 *
030 * <p>If classpath inheritance is disabled in the compiler, then this will not run.
031 *
032 * @author Ashley Scopes
033 * @since 0.0.1
034 */
035public final class JctFileManagerJvmClassPathConfigurer
036    implements JctFileManagerConfigurer {
037
038  private static final Logger log = LoggerFactory
039      .getLogger(JctFileManagerJvmClassPathConfigurer.class);
040
041  private final JctCompiler compiler;
042
043  /**
044   * Initialise the configurer with the desired compiler.
045   *
046   * @param compiler the compiler to wrap.
047   */
048  public JctFileManagerJvmClassPathConfigurer(JctCompiler compiler) {
049    this.compiler = compiler;
050  }
051
052  @Override
053  public JctFileManager configure(JctFileManager fileManager) {
054    log.debug("Configuring the class path");
055
056    SpecialLocationUtils
057        .currentClassPathLocations()
058        .stream()
059        .peek(loc -> log
060            .atTrace()
061            .setMessage("Adding {} ({}) to file manager class path (inherited from JVM))")
062            .addArgument(() -> StringUtils.quoted(loc.toAbsolutePath()))
063            .addArgument(() -> StringUtils.quoted(loc.toUri()))
064            .log())
065        .map(WrappingDirectoryImpl::new)
066        .forEach(dir -> fileManager.addPath(StandardLocation.CLASS_PATH, dir));
067
068    return fileManager;
069  }
070
071  @Override
072  public boolean isEnabled() {
073    return compiler.isInheritClassPath();
074  }
075}