package org.apache.maven.surefire.booter;
   
   /*
    * 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.
   */
  
  import org.apache.maven.surefire.cli.CommandLineOption;
  import org.apache.maven.surefire.providerapi.ProviderParameters;
  import org.apache.maven.surefire.report.ConsoleLogger;
  import org.apache.maven.surefire.report.DefaultDirectConsoleReporter;
  import org.apache.maven.surefire.report.ReporterConfiguration;
  import org.apache.maven.surefire.report.ReporterFactory;
  import org.apache.maven.surefire.testset.DirectoryScannerParameters;
  import org.apache.maven.surefire.testset.RunOrderParameters;
  import org.apache.maven.surefire.testset.TestArtifactInfo;
  import org.apache.maven.surefire.testset.TestRequest;
  import org.apache.maven.surefire.util.DefaultDirectoryScanner;
  import org.apache.maven.surefire.util.DefaultRunOrderCalculator;
  import org.apache.maven.surefire.util.DefaultScanResult;
  import org.apache.maven.surefire.util.DirectoryScanner;
  import org.apache.maven.surefire.util.RunOrderCalculator;
  import org.apache.maven.surefire.util.ScanResult;
  
  import java.util.Collections;
  import java.util.List;
  import java.util.Map;
  
  /**
   * @author Kristian Rosenvold
   */
  public class BaseProviderFactory
      implements DirectoryScannerParametersAware, ReporterConfigurationAware, SurefireClassLoadersAware, TestRequestAware,
      ProviderPropertiesAware, ProviderParameters, TestArtifactInfoAware, RunOrderParametersAware, MainCliOptionsAware,
      FailFastAware, ShutdownAware
  {
      private static final int ROOT_CHANNEL = 0;
  
      private final ReporterFactory reporterFactory;
  
      private final boolean insideFork;
  
      private List<CommandLineOption> mainCliOptions = Collections.emptyList();
  
      private Map<String, String> providerProperties;
  
      private DirectoryScannerParameters directoryScannerParameters;
  
      private ReporterConfiguration reporterConfiguration;
  
      private RunOrderParameters runOrderParameters;
  
      private ClassLoader testClassLoader;
  
      private TestRequest testRequest;
  
      private TestArtifactInfo testArtifactInfo;
  
      private int skipAfterFailureCount;
  
      private Shutdown shutdown;
  
      public BaseProviderFactory( ReporterFactory reporterFactory, boolean insideFork )
      {
          this.reporterFactory = reporterFactory;
          this.insideFork = insideFork;
      }
  
      public DirectoryScanner getDirectoryScanner()
      {
          return directoryScannerParameters == null
                  ? null : new DefaultDirectoryScanner( directoryScannerParameters.getTestClassesDirectory(),
                                              directoryScannerParameters.getIncludes(),
                                              directoryScannerParameters.getExcludes(),
                                              directoryScannerParameters.getSpecificTests() );
      }
  
      public ScanResult getScanResult()
      {
          return DefaultScanResult.from( providerProperties );
      }
  
      private int getThreadCount()
      {
          final String threadcount = providerProperties.get( ProviderParameterNames.THREADCOUNT_PROP );
         return threadcount == null ? 1 : Math.max( Integer.parseInt( threadcount ), 1 );
     }
 
     public RunOrderCalculator getRunOrderCalculator()
     {
         return directoryScannerParameters == null
                 ? null : new DefaultRunOrderCalculator( runOrderParameters, getThreadCount() );
     }
 
     public ReporterFactory getReporterFactory()
     {
         return reporterFactory;
     }
 
     public void setDirectoryScannerParameters( DirectoryScannerParameters directoryScannerParameters )
     {
         this.directoryScannerParameters = directoryScannerParameters;
     }
 
     public void setReporterConfiguration( ReporterConfiguration reporterConfiguration )
     {
         this.reporterConfiguration = reporterConfiguration;
     }
 
     public void setClassLoaders( ClassLoader testClassLoader )
     {
         this.testClassLoader = testClassLoader;
     }
 
     public ConsoleLogger getConsoleLogger()
     {
         return insideFork
                 ? new ForkingRunListener( reporterConfiguration.getOriginalSystemOut(), ROOT_CHANNEL,
                                            reporterConfiguration.isTrimStackTrace() )
                 : new DefaultDirectConsoleReporter( reporterConfiguration.getOriginalSystemOut() );
     }
 
     public void setTestRequest( TestRequest testRequest )
     {
         this.testRequest = testRequest;
     }
 
     public DirectoryScannerParameters getDirectoryScannerParameters()
     {
         return directoryScannerParameters;
     }
 
     public ReporterConfiguration getReporterConfiguration()
     {
         return reporterConfiguration;
     }
 
     public TestRequest getTestRequest()
     {
         return testRequest;
     }
 
     public ClassLoader getTestClassLoader()
     {
         return testClassLoader;
     }
 
     public void setProviderProperties( Map<String, String> providerProperties )
     {
         this.providerProperties = providerProperties;
     }
 
     public Map<String, String> getProviderProperties()
     {
         return providerProperties;
     }
 
     public TestArtifactInfo getTestArtifactInfo()
     {
         return testArtifactInfo;
     }
 
     public void setTestArtifactInfo( TestArtifactInfo testArtifactInfo )
     {
         this.testArtifactInfo = testArtifactInfo;
     }
 
     public void setRunOrderParameters( RunOrderParameters runOrderParameters )
     {
         this.runOrderParameters = runOrderParameters;
     }
 
     public List<CommandLineOption> getMainCliOptions()
     {
         return mainCliOptions;
     }
 
     public void setMainCliOptions( List<CommandLineOption> mainCliOptions )
     {
         this.mainCliOptions = mainCliOptions == null ? Collections.<CommandLineOption>emptyList() : mainCliOptions;
     }
 
     public int getSkipAfterFailureCount()
     {
         return skipAfterFailureCount;
     }
 
     public void setSkipAfterFailureCount( int skipAfterFailureCount )
     {
         this.skipAfterFailureCount = skipAfterFailureCount;
     }
 
     public boolean isInsideFork()
     {
         return insideFork;
     }
 
     public Shutdown getShutdown()
     {
         return shutdown;
     }
 
     public void setShutdown( Shutdown shutdown )
     {
         this.shutdown = shutdown;
     }
 }