/*
    * Licensed to the Apache Software Foundation (ASF) under one   *
    * or more contributor license agreements.  See the NOTICE file *
    * distributed with this work for additional information        *
    * regarding copyright ownership.  The ASF licenses this file   *
    * to you under the Apache License, Version 2.0 (the            *
    * "License"); you may not use this file except in compliance   *
    * with the License.  You may obtain a copy of the License at   *
    *                                                              *
   *   http://www.apache.org/licenses/LICENSE-2.0                 *
   *                                                              *
   * Unless required by applicable law or agreed to in writing,   *
   * software distributed under the License is distributed on an  *
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY       *
   * KIND, either express or implied.  See the License for the    *
   * specific language governing permissions and limitations      *
   * under the License.                                           *
   */
  package org.apache.rat;
  
  import java.io.File;
  import java.io.IOException;
  import java.io.PrintWriter;
  import java.io.Serial;
  import java.io.Serializable;
  import java.nio.charset.StandardCharsets;
  import java.util.Arrays;
  import java.util.Collections;
  import java.util.Comparator;
  import java.util.Map;
  import java.util.Optional;
  import java.util.TreeMap;
  import java.util.function.Consumer;
  import java.util.function.Supplier;
  import java.util.stream.Collectors;
  
  import org.apache.commons.cli.DefaultParser;
  import org.apache.commons.cli.Option;
  import org.apache.commons.cli.Options;
  import org.apache.commons.cli.ParseException;
  import org.apache.rat.api.Document;
  import org.apache.rat.commandline.Arg;
  import org.apache.rat.commandline.ArgumentContext;
  import org.apache.rat.commandline.StyleSheets;
  import org.apache.rat.config.exclusion.StandardCollection;
  import org.apache.rat.document.DocumentName;
  import org.apache.rat.document.DocumentNameMatcher;
  import org.apache.rat.document.FileDocument;
  import org.apache.rat.help.Licenses;
  import org.apache.rat.license.LicenseSetFactory;
  import org.apache.rat.report.Reportable;
  import org.apache.rat.report.claim.ClaimStatistic;
  import org.apache.rat.ui.ArgumentTracker;
  import org.apache.rat.ui.UIOption;
  import org.apache.rat.ui.UIOptionCollection;
  import org.apache.rat.utils.DefaultLog;
  import org.apache.rat.utils.Log.Level;
  import org.apache.rat.walker.ArchiveWalker;
  import org.apache.rat.walker.DirectoryWalker;
  
  import static java.lang.String.format;
  
  /**
   * The collection of standard options for the CLI as well as utility methods to manage them and methods to create the
   * ReportConfiguration from the options and an array of arguments.
   */
  public final class OptionCollection {
  
      private OptionCollection() {
          // do not instantiate
      }
  
      /**
       * The collection of UI Options.
       */
      private static final UIOptionCollection<? extends UIOption<?>> BASE_OPTION_COLLECTION = new CLIOptionCollection();
  
      /**
       * The Option comparator to sort the help.
       */
      public static final Comparator<Option> OPTION_COMPARATOR = new OptionComparator();
  
      /** The Help option */
      public static final Option HELP = new Option("?", "help", false, "Print help for the RAT command line interface and exit.");
  
      /** A mapping of {@code argName(value)} values to a description of those values. */
      @Deprecated
      private static final Map<String, Supplier<String>> ARGUMENT_TYPES;
      static {
          ARGUMENT_TYPES = new TreeMap<>();
          for (ArgumentType argType : ArgumentType.values()) {
              ARGUMENT_TYPES.put(argType.getDisplayName(), argType.description);
          }
      }
  
      /**
       * Gets the mapping of {@code argName(value)} values to a description of those values.
       * @return the mapping of {@code argName(value)} values to a description of those values.
       * @deprecated use {@link ArgumentType}
      */
     @Deprecated
     public static Map<String, Supplier<String>> getArgumentTypes() {
         return Collections.unmodifiableMap(ARGUMENT_TYPES);
     }
 
     /**
      * Join a collection of objects together as a comma separated list of their string values.
      * @param args the objects to join together.
      * @return the comma separated string.
      */
     private static String asString(final Object[] args) {
         return Arrays.stream(args).map(Object::toString).collect(Collectors.joining(", "));
     }
 
     /**
      * Parses the standard options to create a ReportConfiguration.
      *
      * @param workingDirectory The directory to resolve relative file names against.
      * @param args the arguments to parse
      * @param helpCmd the help command to run when necessary.
      * @return a ReportConfiguration or {@code null} if Help was printed.
      * @throws IOException on error.
      */
     public static ReportConfiguration parseCommands(final File workingDirectory, final String[] args, final Consumer<Options> helpCmd) throws IOException {
         return parseCommands(workingDirectory, args, helpCmd, false);
     }
 
     /**
      * Parses the standard options to create a ReportConfiguration.
      * <p>
      * This method is {@code synchronized} because it uses shared mutable state:
      * the {@link #BASE_OPTION_COLLECTION}'s {@code OptionGroup} instances (whose {@code selected}
      * field is mutated by {@link DefaultParser#parse(Options, String[])}), and
      * {@link org.apache.rat.commandline.Converters#FILE_CONVERTER} (whose
      * {@code workingDirectory} field is set during argument processing).
      * Without synchronization, parallel Maven reactor threads (e.g. {@code mvn -T4})
      * corrupt each other's parse state, causing options like {@code --input-exclude}
      * to be silently skipped.
      * </p>
      *
      * @param workingDirectory The directory to resolve relative file names against.
      * @param args the arguments to parse.
      * @param helpCmd the help command to run when necessary.
      * @param noArgs If {@code true} then the commands do not need extra arguments.
      * @return a ReportConfiguration or {@code null} if Help was printed.
      * @throws IOException on error.
      */
     public static synchronized ReportConfiguration parseCommands(final File workingDirectory, final String[] args,
                                                     final Consumer<Options> helpCmd, final boolean noArgs) throws IOException {
         Options opts = buildOptions();
         ArgumentContext argumentContext;
         try {
             argumentContext = new ArgumentContext(workingDirectory, opts, args);
         } catch (ParseException e) {
             System.exit(1);
             return null; // dummy return (won't be reached) to avoid Eclipse complaint about possible NPE
             // for "commandLine"
         }
         Arg.processLogLevel(argumentContext, BASE_OPTION_COLLECTION);
 
         if (argumentContext.hasOption(HELP)) {
             helpCmd.accept(opts);
             return null;
         }
 
         if (argumentContext.hasOption(Arg.HELP_LICENSES.option())) {
             new Licenses(createConfiguration(argumentContext), new PrintWriter(System.out, false, StandardCharsets.UTF_8)).printHelp();
             return null;
         }
 
         ReportConfiguration configuration = createConfiguration(argumentContext);
         if (!noArgs && !configuration.hasSource()) {
             String msg = "No directories or files specified for scanning. Did you forget to close a multi-argument option?";
             DefaultLog.getInstance().error(msg);
             helpCmd.accept(opts);
             return null;
         }
 
         return configuration;
     }
 
     /**
      * Create the report configuration.
      * Note: this method is visible for testing.
      * You probably want one of the {@code parseCommands(...)} methods instead.
      * @param argumentContext The context to execute in.
      * @return a ReportConfiguration
      * @see #parseCommands(File, String[], Consumer)
      * @see #parseCommands(File, String[], Consumer, boolean)
      */
     public static ReportConfiguration createConfiguration(final ArgumentContext argumentContext) {
         try {
             argumentContext.processArgs(BASE_OPTION_COLLECTION);
             final ReportConfiguration configuration = argumentContext.getConfiguration();
             Optional<Option> dirOpt = BASE_OPTION_COLLECTION.getSelected(Arg.DIR);
             dirOpt.ifPresent(opt -> {
                     File directoryName = argumentContext.getParsedOptionValue(opt);
                     configuration.addSource(getReportable(directoryName, configuration));
             });
             for (String s : argumentContext.getArgs()) {
                 Reportable reportable = getReportable(new File(s), configuration);
                 if (reportable != null) {
                     configuration.addSource(reportable);
                 }
             }
             return configuration;
         } catch (RuntimeException e) {
             try (PrintWriter pw = new PrintWriter(DefaultLog.getInstance().asWriter(Level.ERROR))) {
                 pw.println("Unable to create configuration: " + e.getMessage());
                 pw.println("=== Command line options ===");
                 for (Option opt : argumentContext.getOptions()) {
                     String[] values = opt.getValues();
                     pw.printf("   %s: %s%n", ArgumentTracker.extractKey(opt), values == null ? "" : String.join(", ", values));
                 }
             }
             throw new ConfigurationException("Unable to create configuration", e);
         }
 
     }
 
     /**
      * Create an {@code Options} object from the list of defined Options.
      * Mutually exclusive options must be listed in an OptionGroup.
      * @return the Options comprised of the Options defined in this class.
      */
     public static Options buildOptions() {
         BASE_OPTION_COLLECTION.resetSelected();
         return BASE_OPTION_COLLECTION.getOptions();
     }
 
     /**
      * Creates a Reportable object from the directory name and ReportConfiguration
      * object.
      *
      * @param base the directory that contains the files to report on.
      * @param config the ReportConfiguration.
      * @return the Reportable instance containing the files.
      */
     public static Reportable getReportable(final File base, final ReportConfiguration config) {
         File absBase = base.getAbsoluteFile();
         DocumentName documentName = DocumentName.builder(absBase).build();
         if (!absBase.exists()) {
             DefaultLog.getInstance().error("Directory '" + documentName + "' does not exist.");
             return null;
         }
         DocumentNameMatcher documentExcluder = config.getDocumentExcluder(documentName);
 
         Document doc = new FileDocument(documentName, absBase, documentExcluder);
         if (!documentExcluder.matches(doc.getName())) {
             DefaultLog.getInstance().error("Directory '" + documentName + "' is in excluded list.");
             return null;
         }
 
         if (absBase.isDirectory()) {
             return new DirectoryWalker(doc);
         }
 
         return new ArchiveWalker(doc);
     }
 
     /**
      * This class implements the {@code Comparator} interface for comparing Options.
      */
     private static final class OptionComparator implements Comparator<Option>, Serializable {
         /** The serial version UID.  */
         @Serial
         private static final long serialVersionUID = 5305467873966684014L;
 
         private String getKey(final Option opt) {
             String key = opt.getOpt();
             key = key == null ? opt.getLongOpt() : key;
             return key;
         }
 
         /**
          * Compares its two arguments for order. Returns a negative integer, zero, or a
          * positive integer as the first argument is less than, equal to, or greater
          * than the second.
          *
          * @param opt1 The first Option to be compared.
          * @param opt2 The second Option to be compared.
          * @return a negative integer, zero, or a positive integer as the first argument
          * is less than, equal to, or greater than the second.
          */
         @Override
         public int compare(final Option opt1, final Option opt2) {
             return getKey(opt1).compareToIgnoreCase(getKey(opt2));
         }
     }
 
     public enum ArgumentType {
         /**
          * A plain file.
          */
         FILE("File", () -> "A file name."),
         /**
          * An Integer.
          */
         INTEGER("Integer", () -> "An integer value."),
         /**
          * A directory or archive.
          */
         DIRORARCHIVE("DirOrArchive", () -> "A directory or archive file to scan."),
         /**
          * A matching expression.
          */
         EXPRESSION("Expression", () -> "A file matching pattern usually of the form used in Ant build files and " +
                 "'.gitignore' files (see https://ant.apache.org/manual/dirtasks.html#patterns for examples). " +
                 "Regular expression patterns may be specified by surrounding the pattern with '%regex[' and ']'. " +
                 "For example '%regex[[A-Z].*]' would match files and directories that start with uppercase Latin letters."),
         /**
          * A license filter.
          */
         LICENSEFILTER("LicenseFilter", () -> format("A defined filter for the licenses to include. Valid values: %s.",
                 asString(LicenseSetFactory.LicenseFilter.values()))),
         /**
          * A log level.
          */
         LOGLEVEL("LogLevel", () -> format("The log level to use. Valid values %s.", asString(Level.values()))),
         /**
          * A processing type.
          */
         PROCESSINGTYPE("ProcessingType", () -> format("Specifies how to process file types. Valid values are: %s%n",
                 Arrays.stream(ReportConfiguration.Processing.values())
                         .map(v -> format("\t%s: %s", v.name(), v.desc()))
                         .collect(Collectors.joining(System.lineSeparator())))),
         /**
          * A style sheet.
          */
         STYLESHEET("StyleSheet", () -> format("Either an external xsl file or one of the internal named sheets. Internal sheets are: %n%s",
                 Arrays.stream(StyleSheets.values())
                         .map(v -> format("\t%s: %s%n", v.arg(), v.desc()))
                         .collect(Collectors.joining(System.lineSeparator())))),
         /**
          * A license id.
          */
         LICENSEID("LicenseID", () -> "The ID for a license."),
         /**
          * A license family id.
          */
         FAMILYID("FamilyID", () -> "The ID for a license family."),
         /**
          * A standard collection name.
          */
         STANDARDCOLLECTION("StandardCollection", () -> format("Defines standard expression patterns (see above). Valid values are: %n%s%n",
                 Arrays.stream(StandardCollection.values())
                         .map(v -> format("\t%s: %s%n", v.name(), v.desc()))
                         .collect(Collectors.joining(System.lineSeparator())))),
         /**
          * A Counter pattern name
          */
         COUNTERPATTERN("CounterPattern", () -> format("A pattern comprising one of the following prefixes followed by " +
                         "a colon and a count (e.g. %s:5).  Prefixes are %n%s.", ClaimStatistic.Counter.UNAPPROVED,
                 Arrays.stream(ClaimStatistic.Counter.values())
                         .map(v -> format("\t%s: %s Default range [%s, %s]%n", v.name(), v.getDescription(),
                                 v.getDefaultMinValue(),
                                 v.getDefaultMaxValue() == -1 ? "unlimited" : v.getDefaultMaxValue()))
                         .collect(Collectors.joining(System.lineSeparator())))),
         /**
          * A generic argument.
          */
         ARG("Arg", () -> "A string"),
         /**
          * No argument.
          */
         NONE("", () -> "");
 
         /**
          * The display name
          */
         private final String displayName;
         /**
          * A supplier of the description
          */
         private final Supplier<String> description;
 
         ArgumentType(final String name,
                      final Supplier<String> description) {
             this.displayName = name;
             this.description = description;
         }
 
         /**
          * Gets the display name.
          * @return the display name.
          */
         public String getDisplayName() {
             return displayName;
         }
 
         /**
          * Gets the description.
          * @return the description.
          */
         public Supplier<String> description() {
             return description;
         }
 
         /**
          * Get the matching Argument type.
          * @param displayName the display name for the desired type.
          * @return An optional with the ArgumentType or an empty optional if none exists.
          */
         public static Optional<ArgumentType> forDisplayName(final String displayName) {
             return Arrays.stream(ArgumentType.values()).filter(argType -> argType.displayName.equals(displayName))
                     .findAny();
         }
     }
 }