package org.apache.maven.surefire.util;

 /*

  * 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.SpecificTestClassFilter;

 import java.io.File;

 import java.util.ArrayList;

 import java.util.Collections;

 import java.util.List;

 /**

  * Scans directories looking for tests.

  *

  * @author Karl M. Davis

  * @author Kristian Rosenvold

  */

 public class DefaultDirectoryScanner

     implements DirectoryScanner

 {

     private static final String FS = System.getProperty( "file.separator" );

     private static final String[] EMPTY_STRING_ARRAY = new String[0];

     private static final String JAVA_SOURCE_FILE_EXTENSION = ".java";

     private static final String JAVA_CLASS_FILE_EXTENSION = ".class";

     private final File basedir;

     private final List includes;

     private final List excludes;

     private final List specificTests;

     private final List<Class> classesSkippedByValidation = new ArrayList<Class>();

     public DefaultDirectoryScanner( File basedir, List includes, List excludes, List specificTests )

     {

         this.basedir = basedir;

         this.includes = includes;

         this.excludes = excludes;

         this.specificTests = specificTests;

     }

     public TestsToRun locateTestClasses( ClassLoader classLoader, ScannerFilter scannerFilter )

     {

         String[] testClassNames = collectTests();

         List<Class> result = new ArrayList<Class>();

         String[] specific = specificTests == null ? new String[0] : processIncludesExcludes( specificTests );

         SpecificTestClassFilter specificTestFilter = new SpecificTestClassFilter( specific );

         for ( String className : testClassNames )

         {

             Class testClass = loadClass( classLoader, className );

             if ( !specificTestFilter.accept( testClass ) )

             {

                 // FIXME: Log this somehow!

                 continue;

             }

             if ( scannerFilter == null || scannerFilter.accept( testClass ) )

             {

                 result.add( testClass );

             }

             else

             {

                 classesSkippedByValidation.add( testClass );

             }

         }

         return new TestsToRun( result );

     }

     private static Class loadClass( ClassLoader classLoader, String className )

     {

         Class testClass;

         try

         {

             testClass = classLoader.loadClass( className );

         }

         catch ( ClassNotFoundException e )

         {

             throw new NestedRuntimeException( "Unable to create test class '" + className + "'", e );

         }

         return testClass;

     }

     String[] collectTests()

     {

         String[] tests = EMPTY_STRING_ARRAY;

         if ( basedir.exists() )

         {

             org.apache.maven.shared.utils.io.DirectoryScanner scanner =

                 new org.apache.maven.shared.utils.io.DirectoryScanner();

             scanner.setBasedir( basedir );

             if ( includes != null )

             {

                 scanner.setIncludes( processIncludesExcludes( includes ) );

             }

             if ( excludes != null )

             {

                 scanner.setExcludes( processIncludesExcludes( excludes ) );

             }

             scanner.scan();

             tests = scanner.getIncludedFiles();

             for ( int i = 0; i < tests.length; i++ )

             {

                 String test = tests[i];

                 test = test.substring( 0, test.indexOf( "." ) );

                 tests[i] = test.replace( FS.charAt( 0 ), '.' );

             }

         }

         return tests;

     }

     private static String[] processIncludesExcludes( List list )

     {

         List<String> newList = new ArrayList<String>();

         for ( Object aList : list )

         {

             String include = (String) aList;

             String[] includes = include.split( "," );

             Collections.addAll( newList, includes );

         }

         String[] incs = new String[newList.size()];

         for ( int i = 0; i < incs.length; i++ )

         {

             String inc = newList.get( i );

             if ( inc.endsWith( JAVA_SOURCE_FILE_EXTENSION ) )

             {

                 inc = inc.substring( 0, inc.lastIndexOf( JAVA_SOURCE_FILE_EXTENSION ) ) + JAVA_CLASS_FILE_EXTENSION;

             }

             incs[i] = inc;

         }

         return incs;

     }

 }