/*
    * 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.maven.surefire.junitcore;
  
  import java.util.ArrayList;
  import java.util.Collection;
  import java.util.Map;
  
  import org.apache.maven.surefire.api.booter.ProviderParameterNames;
  
  /**
   * @author Kristian Rosenvold
   */
  public final class JUnitCoreParameters {
      public static final String PARALLEL_KEY = ProviderParameterNames.PARALLEL_PROP;
  
      public static final String PERCORETHREADCOUNT_KEY = "perCoreThreadCount";
  
      public static final String THREADCOUNT_KEY = ProviderParameterNames.THREADCOUNT_PROP;
  
      public static final String THREADCOUNTSUITES_KEY = ProviderParameterNames.THREADCOUNTSUITES_PROP;
  
      public static final String THREADCOUNTCLASSES_KEY = ProviderParameterNames.THREADCOUNTCLASSES_PROP;
  
      public static final String THREADCOUNTMETHODS_KEY = ProviderParameterNames.THREADCOUNTMETHODS_PROP;
  
      public static final String USEUNLIMITEDTHREADS_KEY = "useUnlimitedThreads";
  
      public static final String PARALLEL_TIMEOUT_KEY = ProviderParameterNames.PARALLEL_TIMEOUT_PROP;
  
      public static final String PARALLEL_TIMEOUTFORCED_KEY = ProviderParameterNames.PARALLEL_TIMEOUTFORCED_PROP;
  
      public static final String PARALLEL_OPTIMIZE_KEY = ProviderParameterNames.PARALLEL_OPTIMIZE_PROP;
  
      private final String parallel;
  
      private final boolean perCoreThreadCount;
  
      private final int threadCount;
  
      private final int threadCountSuites;
  
      private final int threadCountClasses;
  
      private final int threadCountMethods;
  
      private final double parallelTestsTimeoutInSeconds;
  
      private final double parallelTestsTimeoutForcedInSeconds;
  
      private final boolean useUnlimitedThreads;
  
      private final boolean parallelOptimization;
  
      public JUnitCoreParameters(Map<String, String> properties) {
          parallel = property(properties, PARALLEL_KEY, "none").toLowerCase();
          perCoreThreadCount = property(properties, PERCORETHREADCOUNT_KEY, true);
          threadCount = property(properties, THREADCOUNT_KEY, 0);
          threadCountMethods = property(properties, THREADCOUNTMETHODS_KEY, 0);
          threadCountClasses = property(properties, THREADCOUNTCLASSES_KEY, 0);
          threadCountSuites = property(properties, THREADCOUNTSUITES_KEY, 0);
          useUnlimitedThreads = property(properties, USEUNLIMITEDTHREADS_KEY, false);
          parallelTestsTimeoutInSeconds = Math.max(property(properties, PARALLEL_TIMEOUT_KEY, 0d), 0);
          parallelTestsTimeoutForcedInSeconds = Math.max(property(properties, PARALLEL_TIMEOUTFORCED_KEY, 0d), 0);
          parallelOptimization = property(properties, PARALLEL_OPTIMIZE_KEY, true);
      }
  
      private static Collection<String> lowerCase(String... elements) {
          ArrayList<String> lowerCase = new ArrayList<>();
          for (String element : elements) {
              lowerCase.add(element.toLowerCase());
          }
          return lowerCase;
      }
  
      private boolean isAllParallel() {
          return "all".equals(parallel);
      }
  
      public boolean isParallelMethods() {
          return isAllParallel()
                  || lowerCase("both", "methods", "suitesAndMethods", "classesAndMethods")
                          .contains(parallel);
     }
 
     public boolean isParallelClasses() {
         return isAllParallel()
                 || lowerCase("both", "classes", "suitesAndClasses", "classesAndMethods")
                         .contains(parallel);
     }
 
     public boolean isParallelSuites() {
         return isAllParallel()
                 || lowerCase("suites", "suitesAndClasses", "suitesAndMethods").contains(parallel);
     }
 
     /**
      * @deprecated Instead use the expression isParallelMethods() &amp;&amp; isParallelClasses().
      * @return {@code true} if classes and methods are both parallel
      */
     @Deprecated
     @SuppressWarnings("unused")
     public boolean isParallelBoth() {
         return isParallelMethods() && isParallelClasses();
     }
 
     public boolean isPerCoreThreadCount() {
         return perCoreThreadCount;
     }
 
     public int getThreadCount() {
         return threadCount;
     }
 
     public int getThreadCountMethods() {
         return threadCountMethods;
     }
 
     public int getThreadCountClasses() {
         return threadCountClasses;
     }
 
     public int getThreadCountSuites() {
         return threadCountSuites;
     }
 
     public boolean isUseUnlimitedThreads() {
         return useUnlimitedThreads;
     }
 
     public double getParallelTestsTimeoutInSeconds() {
         return parallelTestsTimeoutInSeconds;
     }
 
     public double getParallelTestsTimeoutForcedInSeconds() {
         return parallelTestsTimeoutForcedInSeconds;
     }
 
     public boolean isNoThreading() {
         return !isParallelismSelected();
     }
 
     public boolean isParallelismSelected() {
         return isParallelSuites() || isParallelClasses() || isParallelMethods();
     }
 
     public boolean isParallelOptimization() {
         return parallelOptimization;
     }
 
     @Override
     public String toString() {
         return "parallel='" + parallel + '\'' + ", perCoreThreadCount=" + perCoreThreadCount + ", threadCount="
                 + threadCount + ", useUnlimitedThreads=" + useUnlimitedThreads + ", threadCountSuites="
                 + threadCountSuites
                 + ", threadCountClasses=" + threadCountClasses + ", threadCountMethods=" + threadCountMethods
                 + ", parallelOptimization=" + parallelOptimization;
     }
 
     private static boolean property(Map<String, String> properties, String key, boolean fallback) {
         return properties.containsKey(key) ? Boolean.valueOf(properties.get(key)) : fallback;
     }
 
     private static String property(Map<String, String> properties, String key, String fallback) {
         return properties.containsKey(key) ? properties.get(key) : fallback;
     }
 
     private static int property(Map<String, String> properties, String key, int fallback) {
         return properties.containsKey(key) ? Integer.valueOf(properties.get(key)) : fallback;
     }
 
     private static double property(Map<String, String> properties, String key, double fallback) {
         return properties.containsKey(key) ? Double.valueOf(properties.get(key)) : fallback;
     }
 }