package org.codehaus.plexus.compiler.ajc;
   
   import javax.inject.Named;
   
   import java.io.File;
   import java.io.IOException;
   import java.net.MalformedURLException;
   import java.net.URL;
   import java.net.URLClassLoader;
  import java.util.ArrayList;
  import java.util.Arrays;
  import java.util.HashSet;
  import java.util.LinkedList;
  import java.util.List;
  import java.util.Map;
  import java.util.Set;
  
  import org.aspectj.ajdt.ajc.BuildArgParser;
  import org.aspectj.ajdt.internal.core.builder.AjBuildConfig;
  import org.aspectj.ajdt.internal.core.builder.AjBuildManager;
  import org.aspectj.bridge.AbortException;
  import org.aspectj.bridge.IMessage;
  import org.aspectj.bridge.ISourceLocation;
  import org.aspectj.bridge.MessageHandler;
  import org.aspectj.org.eclipse.jdt.internal.compiler.classfmt.ClassFileConstants;
  import org.aspectj.tools.ajc.Main;
  import org.codehaus.plexus.compiler.AbstractCompiler;
  import org.codehaus.plexus.compiler.CompilerConfiguration;
  import org.codehaus.plexus.compiler.CompilerException;
  import org.codehaus.plexus.compiler.CompilerMessage;
  import org.codehaus.plexus.compiler.CompilerOutputStyle;
  import org.codehaus.plexus.compiler.CompilerResult;
  import org.codehaus.plexus.util.DirectoryScanner;
  
  /**
   * <p>
   * Options
   * </p>
   * <p>
   * -injars JarList
   * </p>
   * <p>
   * Accept as source bytecode any .class files inside the specified .jar files. The output will include these
   * classes, possibly as woven with any applicable aspects. JarList, like classpath, is a single argument
   * containing a list of paths to jar files, delimited by the platform- specific classpath delimiter.
   * </p>
   * <p>
   * -aspectpath JarList
   * </p>
   * <p>
   * Weave binary aspects from JarList zip files into all sources. The aspects should have been output by
   * the same version of the compiler. To run the output classes requires putting all the aspectpath entries on
   * the run classpath. JarList, like classpath, is a single argument containing a list of paths to jar files,
   * delimited by the platform- specific classpath delimiter.
   * </p>
   * <p>
   * -argfile File
   * </p>
   * <p>
   * The file is a line-delimited list of arguments. These arguments are inserted into the argument list.
   * </p>
   * <p>
   * -outjar output.jar
   * </p>
   * <p>
   * Put output classes in zip file output.jar.
   * </p>
   * <p>
   * -incremental
   * </p>
   * <p>
   * Run the compiler continuously. After the initial compilation, the compiler will wait to recompile until it
   * reads a newline from the standard input, and will quit when it reads a 'q'. It will only recompile necessary
   * components, so a recompile should be much faster than doing a second compile. This requires -sourceroots.
   * </p>
   * <p>
   * -sourceroots DirPaths
   * </p>
   * <p>
   * Find and build all .java or .aj source files under any directory listed in DirPaths. DirPaths, like
   * classpath, is a single argument containing a list of paths to directories, delimited by the platform-
   * specific classpath delimiter. Required by -incremental.
   * </p>
   * <p>
   * -emacssym
   * </p>
   * <p>
   * Generate .ajesym symbol files for emacs support
   * </p>
   * <p>
   * -Xlint
   * </p>
   * <p>
   * Same as -Xlint:warning (enabled by default)
   * </p>
   * <p>
   * -Xlint:{level}
   * </p>
   * <p>
  * Set default level for messages about potential programming mistakes in crosscutting code. {level} may be
  * ignore, warning, or error. This overrides entries in org/aspectj/weaver/XlintDefault.properties from
  * aspectjtools.jar, but does not override levels set using the -Xlintfile option.
  * </p>
  * <p>
  * -Xlintfile PropertyFile
  * </p>
  * <p>
  * Specify properties file to set levels for specific crosscutting messages. PropertyFile is a path to a
  * Java .properties file that takes the same property names and values as
  * org/aspectj/weaver/XlintDefault.properties from aspectjtools.jar, which it also overrides.
  * -help
  * </p>
  * <p>
  * Emit information on compiler options and usage
  * </p>
  * <p>
  * -version
  * </p>
  * <p>
  * Emit the version of the AspectJ compiler
  * </p>
  * <p>
  * -classpath Path
  * </p>
  * <p>
  * Specify where to find user class files. Path is a single argument containing a list of paths to zip files
  * or directories, delimited by the platform-specific path delimiter.
  * </p>
  * <p>
  * -bootclasspath Path
  * </p>
  * <p>
  * Override location of VM's bootclasspath for purposes of evaluating types when compiling. Path is a single
  * argument containing a list of paths to zip files or directories, delimited by the platform-specific path
  * delimiter.
  * </p>
  * <p>
  * -extdirs Path
  * </p>
  * <p>
  * Override location of VM's extension directories for purposes of evaluating types when compiling. Path is
  * a single argument containing a list of paths to directories, delimited by the platform-specific path
  * delimiter.
  * </p>
  * <p>
  * -d Directory
  * </p>
  * <p>
  * Specify where to place generated .class files. If not specified, Directory defaults to the current
  * working dir.
  * </p>
  * <p>
  * -target [1.1|1.2]
  * </p>
  * <p>
  * Specify classfile target setting (1.1 or 1.2, default is 1.1)
  * </p>
  * <p>
  * -1.3
  * </p>
  * <p>
  * Set compliance level to 1.3 (default)
  * -1.4
  * </p>
  * <p>
  * Set compliance level to 1.4
  * -source [1.3|1.4]
  * </p>
  * <p>
  * Toggle assertions (1.3 or 1.4, default is 1.3 in -1.3 mode and 1.4 in -1.4 mode). When using -source 1.3,
  * an assert() statement valid under Java 1.4 will result in a compiler error. When using -source 1.4, treat
  * assert as a keyword and implement assertions according to the 1.4 language spec.
  * </p>
  * <p>
  * -nowarn
  * </p>
  * <p>
  * Emit no warnings (equivalent to '-warn:none') This does not suppress messages generated by declare warning
  * or Xlint.
  * </p>
  * <p>
  * -warn: items
  * </p>
  * <p>
  * Emit warnings for any instances of the comma-delimited list of questionable code
  * (eg '-warn:unusedLocals,deprecation'):
  * </p>
  * <p>
  * constructorName        method with constructor name
  * packageDefaultMethod   attempt to override package-default method
  * deprecation            usage of deprecated type or member
  * maskedCatchBlocks      hidden catch block
  * unusedLocals           local variable never read
  * unusedArguments        method argument never read
  * unusedImports          import statement not used by code in file
  * none                   suppress all compiler warnings
  * </p>
  * <p>
  * -warn:none does not suppress messages generated by declare warning or Xlint.
  * </p>
  * <p>
  * -deprecation
  * </p>
  * <p>
  * Same as -warn:deprecation
  * </p>
  * <p>
  * -noImportError
  * </p>
  * <p>
  * Emit no errors for unresolved imports
  * </p>
  * <p>
  * -proceedOnError
  * </p>
  * <p>
  * Keep compiling after error, dumping class files with problem methods
  * </p>
  * <p>
  * -g:[lines,vars,source]
  * </p>
  * <p>
  * debug attributes level, that may take three forms:
  * </p>
  * <p>
  * -g         all debug info ('-g:lines,vars,source')
  * -g:none    no debug info
  * -g:{items} debug info for any/all of [lines, vars, source], e.g.,
  * -g:lines,source
  * </p>
  * <p>
  * -preserveAllLocals
  * </p>
  * <p>
  * Preserve all local variables during code generation (to facilitate debugging).
  * </p>
  * <p>
  * -referenceInfo
  * </p>
  * <p>
  * Compute reference information.
  * </p>
  * <p>
  * -encoding format
  * </p>
  * <p>
  * Specify default source encoding format. Specify custom encoding on a per file basis by suffixing each
  * input source file/folder name with '[encoding]'.
  * </p>
  * <p>
  * -verbose
  * </p>
  * <p>
  * Emit messages about accessed/processed compilation units
  * </p>
  * <p>
  * -log file Specify a log file for compiler messages.
  * </p>
  * <p>
  * -progress Show progress (requires -log mode).
  * </p>
  * <p>
  * -time Display speed information.
  * </p>
  * <p>
  * -noExit Do not call System.exit(n) at end of compilation (n=0 if no error)
  * </p>
  * <p>
  * -repeat N Repeat compilation process N times (typically to do performance analysis).
  * </p>
  * <p>
  * -Xnoweave (Experimental) produce unwoven class files for input using -injars.
  * </p>
  * <p>
  * -Xnoinline (Experimental) do not inline around advice
  * </p>
  * <p>
  * -XincrementalFile file
  * </p>
  * <p>
  * (Experimental) This works like incremental mode, but using a file rather than standard input to control
  * the compiler. It will recompile each time file is changed and and halt when file is deleted.
  * </p>
  * <p>
  * -XserializableAspects (Experimental) Normally it is an error to declare aspects Serializable. This option removes that restriction.
  * </p>
  *
  * @author <a href="mailto:jason@maven.org">Jason van Zyl</a>
  */
 @Named("aspectj")
 public class AspectJCompiler extends AbstractCompiler {
 
     // ----------------------------------------------------------------------
     //
     // ----------------------------------------------------------------------
 
     public AspectJCompiler() {
         // Input file ending "" means: Give me all files, I am going to filter them myself later. We are doing this,
         // because in method 'getSourceFiles' we need to search for both ".java" and ".aj" files.
         super(CompilerOutputStyle.ONE_OUTPUT_FILE_PER_INPUT_FILE, "", ".class", null);
     }
 
     @Override
     public String getCompilerId() {
         return "aspectj";
     }
 
     public CompilerResult performCompile(CompilerConfiguration config) throws CompilerException {
         File destinationDir = new File(config.getOutputLocation());
 
         if (!destinationDir.exists()) {
             destinationDir.mkdirs();
         }
 
         String[] sourceFiles = getSourceFiles(config);
 
         if (sourceFiles.length == 0) {
             return new CompilerResult();
         }
 
         logCompiling(sourceFiles, config);
 
         //        String[] args = buildCompilerArguments( config, sourceFiles );
         AjBuildConfig buildConfig = buildCompilerConfig(config);
         return new CompilerResult().compilerMessages(compileInProcess(buildConfig));
     }
 
     private static class AspectJMessagePrinter extends Main.MessagePrinter {
         public AspectJMessagePrinter(boolean verbose) {
             super(verbose);
         }
     }
 
     private AjBuildConfig buildCompilerConfig(CompilerConfiguration config) throws CompilerException {
         BuildArgParser buildArgParser = new BuildArgParser(new AspectJMessagePrinter(config.isVerbose()));
         AjBuildConfig buildConfig = new AjBuildConfig(buildArgParser);
         // Avoid NPE when AjBuildConfig.getCheckedClasspaths() is called later during compilation
         buildArgParser.populateBuildConfig(buildConfig, new String[0], true, null);
         buildConfig.setIncrementalMode(false);
 
         String[] files = getSourceFiles(config);
         if (files != null) {
             buildConfig.setFiles(buildFileList(Arrays.asList(files)));
         }
 
         String releaseVersion = config.getReleaseVersion();
         setSourceVersion(buildConfig, releaseVersion == null ? config.getSourceVersion() : releaseVersion);
         setTargetVersion(buildConfig, releaseVersion == null ? config.getTargetVersion() : releaseVersion);
 
         if (config.isDebug()) {
             buildConfig.getOptions().produceDebugAttributes =
                     ClassFileConstants.ATTR_SOURCE + ClassFileConstants.ATTR_LINES + ClassFileConstants.ATTR_VARS;
         }
 
         Map<String, String> javaOpts = config.getCustomCompilerArgumentsAsMap();
         if (javaOpts != null && !javaOpts.isEmpty()) {
             // TODO support customCompilerArguments
             // buildConfig.setJavaOptions( javaOpts );
         }
 
         List<String> cp = new LinkedList<>(config.getClasspathEntries());
 
         File javaHomeDir = new File(System.getProperty("java.home"));
         File[] jars = new File(javaHomeDir, "lib").listFiles();
         if (jars != null) {
             for (File jar : jars) {
                 if (jar.getName().endsWith(".jar") || jar.getName().endsWith(".zip")) {
                     cp.add(0, jar.getAbsolutePath());
                 }
             }
         }
         jars = new File(javaHomeDir, "../Classes").listFiles();
         if (jars != null) {
             for (File jar : jars) {
                 if (jar.getName().endsWith(".jar") || jar.getName().endsWith(".zip")) {
                     cp.add(0, jar.getAbsolutePath());
                 }
             }
         }
 
         checkForAspectJRT(cp);
         if (cp != null && !cp.isEmpty()) {
             List<String> elements = new ArrayList<>(cp.size());
             for (String path : cp) {
                 elements.add((new File(path)).getAbsolutePath());
             }
 
             buildConfig.setClasspath(elements);
         }
 
         String outputLocation = config.getOutputLocation();
         if (outputLocation != null) {
             File outDir = new File(outputLocation);
             if (!outDir.exists()) {
                 outDir.mkdirs();
             }
 
             buildConfig.setOutputDir(outDir);
         }
 
         if (config instanceof AspectJCompilerConfiguration) {
             AspectJCompilerConfiguration ajCfg = (AspectJCompilerConfiguration) config;
 
             Map<String, File> sourcePathResources = ajCfg.getSourcePathResources();
             if (sourcePathResources != null && !sourcePathResources.isEmpty()) {
                 buildConfig.setSourcePathResources(sourcePathResources);
             }
 
             Map<String, String> ajOptions = ajCfg.getAJOptions();
             if (ajOptions != null && !ajOptions.isEmpty()) {
                 // TODO not supported
                 // buildConfig.setAjOptions( ajCfg.getAJOptions() );
             }
 
             List<File> aspectPath = buildFileList(ajCfg.getAspectPath());
             if (aspectPath != null && !aspectPath.isEmpty()) {
                 buildConfig.setAspectpath(buildFileList(ajCfg.getAspectPath()));
             }
 
             List<File> inJars = buildFileList(ajCfg.getInJars());
             if (inJars != null && !inJars.isEmpty()) {
                 buildConfig.setInJars(buildFileList(ajCfg.getInJars()));
             }
 
             List<File> inPaths = buildFileList(ajCfg.getInPath());
             if (inPaths != null && !inPaths.isEmpty()) {
                 buildConfig.setInPath(buildFileList(ajCfg.getInPath()));
             }
 
             String outJar = ajCfg.getOutputJar();
             if (outJar != null) {
                 buildConfig.setOutputJar(new File(ajCfg.getOutputJar()));
             }
         }
 
         return buildConfig;
     }
 
     private List<CompilerMessage> compileInProcess(AjBuildConfig buildConfig) throws CompilerException {
 
         MessageHandler messageHandler = new MessageHandler();
 
         AjBuildManager manager = new AjBuildManager(messageHandler);
 
         try {
             manager.batchBuild(buildConfig, messageHandler);
         } catch (AbortException | IOException e) {
             throw new CompilerException("Unknown error while compiling", e);
         }
 
         // We need the location of the maven so we have a couple of options
         // here.
         //
         // The aspectjrt jar is something this component needs to function so we
         // can either
         // bake it into the plugin and retrieve it somehow or use a system
         // property or we
         // could pass in a set of parameters in a Map.
 
         boolean errors = messageHandler.hasAnyMessage(IMessage.ERROR, true);
 
         List<CompilerMessage> messages = new ArrayList<>();
         if (errors) {
             IMessage[] errorMessages = messageHandler.getMessages(IMessage.ERROR, true);
 
             for (IMessage m : errorMessages) {
                 ISourceLocation sourceLocation = m.getSourceLocation();
                 CompilerMessage error;
 
                 if (sourceLocation == null) {
                     error = new CompilerMessage(m.getMessage(), true);
                 } else {
                     error = new CompilerMessage(
                             sourceLocation.getSourceFile().getPath(),
                             true,
                             sourceLocation.getLine(),
                             sourceLocation.getColumn(),
                             sourceLocation.getEndLine(),
                             sourceLocation.getColumn(),
                             m.getMessage());
                 }
                 messages.add(error);
             }
         }
 
         return messages;
     }
 
     private void checkForAspectJRT(List<String> cp) {
         if (cp == null || cp.isEmpty()) {
             throw new IllegalStateException("AspectJ Runtime not found in supplied classpath");
         } else {
             try {
                 URL[] urls = new URL[cp.size()];
                 for (int i = 0; i < urls.length; i++) {
                     urls[i] = (new File(cp.get(i))).toURL();
                 }
 
                 URLClassLoader cloader = new URLClassLoader(urls);
 
                 cloader.loadClass("org.aspectj.lang.JoinPoint");
             } catch (MalformedURLException e) {
                 throw new IllegalArgumentException("Invalid classpath entry");
             } catch (ClassNotFoundException e) {
                 throw new IllegalStateException("AspectJ Runtime not found in supplied classpath");
             }
         }
     }
 
     private List<File> buildFileList(List<String> locations) {
         List<File> fileList = new LinkedList<>();
         for (String location : locations) {
             fileList.add(new File(location));
         }
 
         return fileList;
     }
 
     /**
      * Set the source version in AspectJ compiler
      *
      * @param buildConfig
      * @param sourceVersion
      */
     private void setSourceVersion(AjBuildConfig buildConfig, String sourceVersion) throws CompilerException {
         buildConfig.getOptions().sourceLevel = versionStringToMajorMinor(sourceVersion);
     }
 
     /**
      * Set the target version in AspectJ compiler
      *
      * @param buildConfig
      * @param targetVersion
      */
     private void setTargetVersion(AjBuildConfig buildConfig, String targetVersion) throws CompilerException {
         buildConfig.getOptions().targetJDK = versionStringToMajorMinor(targetVersion);
     }
 
     private static long versionStringToMajorMinor(String version) throws CompilerException {
         if (version == null) {
             version = "";
         }
         // Note: We avoid using org.codehaus.plexus:plexus-java here on purpose, because Maven Compiler might depend on
         // a different (older) versionm, e.g. not having the 'asMajor' method yet. This can cause problems for users
         // trying to compile their AspectJ code using Plexus.
 
         version = version.trim()
                 // Cut off leading "1.", focusing on the Java major
                 .replaceFirst("^1[.]", "")
                 // Accept, but cut off trailing ".0", as ECJ/ACJ explicitly support versions like 5.0, 8.0, 11.0
                 .replaceFirst("[.]0$", "");
 
         switch (version) {
                 // Java 1.6 as a default source/target seems to make sense. Maven Compiler should set its own default
                 // anyway, so this probably never needs to be used. But not having a default feels bad, too.
             case "":
                 return ClassFileConstants.JDK1_6;
             case "1":
                 return ClassFileConstants.JDK1_1;
             case "2":
                 return ClassFileConstants.JDK1_2;
             case "3":
                 return ClassFileConstants.JDK1_3;
             case "4":
                 return ClassFileConstants.JDK1_4;
             case "5":
                 return ClassFileConstants.JDK1_5;
             case "6":
                 return ClassFileConstants.JDK1_6;
             case "7":
                 return ClassFileConstants.JDK1_7;
             case "8":
                 return ClassFileConstants.JDK1_8;
             case "9":
                 return ClassFileConstants.JDK9;
             case "10":
                 return ClassFileConstants.JDK10;
             case "11":
                 return ClassFileConstants.JDK11;
             case "12":
                 return ClassFileConstants.JDK12;
             case "13":
                 return ClassFileConstants.JDK13;
             case "14":
                 return ClassFileConstants.JDK14;
             case "15":
                 return ClassFileConstants.JDK15;
             case "16":
                 return ClassFileConstants.JDK16;
         }
         throw new CompilerException("Unknown Java source/target version number: " + version);
     }
 
     /**
      * @return null
      */
     public String[] createCommandLine(CompilerConfiguration config) throws CompilerException {
         return null;
     }
 
     protected static String[] getSourceFiles(CompilerConfiguration config) {
         Set<String> sources = new HashSet<>();
 
         Set<File> sourceFiles = config.getSourceFiles();
 
         if (sourceFiles != null && !sourceFiles.isEmpty()) {
             for (File sourceFile : sourceFiles) {
                 if (sourceFile.getName().endsWith(".java")
                         || sourceFile.getName().endsWith(".aj")) {
                     sources.add(sourceFile.getAbsolutePath());
                 }
             }
         } else {
             for (String sourceLocation : config.getSourceLocations()) {
                 sources.addAll(getSourceFilesForSourceRoot(config, sourceLocation));
             }
         }
 
         String[] result;
 
         if (sources.isEmpty()) {
             result = new String[0];
         } else {
             result = sources.toArray(new String[sources.size()]);
         }
 
         return result;
     }
 
     protected static Set<String> getSourceFilesForSourceRoot(CompilerConfiguration config, String sourceLocation) {
         DirectoryScanner scanner = new DirectoryScanner();
 
         scanner.setBasedir(sourceLocation);
 
         Set<String> includes = config.getIncludes();
 
         if (includes != null && !includes.isEmpty()) {
             String[] inclStrs = includes.toArray(new String[includes.size()]);
             scanner.setIncludes(inclStrs);
         } else {
             scanner.setIncludes(new String[] {"**/*.java", "**/*.aj"});
         }
 
         Set<String> excludes = config.getExcludes();
 
         if (excludes != null && !excludes.isEmpty()) {
             String[] exclStrs = excludes.toArray(new String[excludes.size()]);
             scanner.setExcludes(exclStrs);
         }
 
         scanner.scan();
 
         String[] sourceDirectorySources = scanner.getIncludedFiles();
 
         Set<String> sources = new HashSet<>();
 
         for (String sourceDirectorySource : sourceDirectorySources) {
             File f = new File(sourceLocation, sourceDirectorySource);
 
             sources.add(f.getPath());
         }
 
         return sources;
     }
 }