package org.codehaus.gmavenplus.mojo;
   
   import org.apache.maven.shared.model.fileset.FileSet;
   import org.apache.maven.shared.model.fileset.util.FileSetManager;
   
   import java.io.File;
   import java.util.*;
   
   import static java.util.Collections.singletonList;
  
  
  /**
   * This mojo provides access to the Groovy sources.
   *
   * @author Keegan Witt
   */
  public abstract class AbstractGroovySourcesMojo extends AbstractGroovyMojo {
  
      /**
       * Main source directory name.
       */
      protected static final String MAIN = "main";
  
      /**
       * Test source directory name.
       */
      protected static final String TEST = "test";
  
      /**
       * Gets the set of included files from the specified source files or source directory (if sources are null).
       *
       * @param fromSources        The sources to get the included files from
       * @param includeJavaSources Whether to include Java sources
       * @return The included files from the specified sources
       */
      protected SortedSet<File> getFiles(final FileSet[] fromSources, final boolean includeJavaSources) {
          SortedSet<File> files = new TreeSet<>();
          FileSetManager fileSetManager = new FileSetManager();
  
          for (FileSet fileSet : getFilesets(fromSources, includeJavaSources)) {
              for (String include : fileSetManager.getIncludedFiles(fileSet)) {
                  files.add(new File(fileSet.getDirectory(), include));
              }
          }
  
          return files;
      }
  
      /**
       * Gets the set of included files from the specified source files or source directory (if sources are null).
       *
       * @param fromSources        The sources to get the included files from
       * @param includeJavaSources Whether to include Java sources
       * @return The included files from the specified sources
       */
      protected SortedSet<File> getTestFiles(final FileSet[] fromSources, final boolean includeJavaSources) {
          SortedSet<File> files = new TreeSet<>();
          FileSetManager fileSetManager = new FileSetManager();
  
          for (FileSet fileSet : getTestFilesets(fromSources, includeJavaSources)) {
              for (String include : fileSetManager.getIncludedFiles(fileSet)) {
                  files.add(new File(fileSet.getDirectory(), include));
              }
          }
  
          return files;
      }
  
      /**
       * Gets the set of included filesets from the specified source files or source directory (if sources are null).
       *
       * @param fromSources        The sources to get the included files from
       * @param includeJavaSources Whether to include Java sources
       * @return The included filesets from the specified sources
       */
      protected FileSet[] getFilesets(final FileSet[] fromSources, final boolean includeJavaSources) {
          FileSet[] result;
          FileSet[] groovyFileSets;
  
          if (fromSources != null) {
              groovyFileSets = fromSources;
          } else {
              FileSet groovyFileSet = new FileSet();
              String groovyDirectory = "src" + File.separator + MAIN + File.separator + "groovy";
              groovyFileSet.setDirectory(project.getBasedir() + File.separator + groovyDirectory);
              groovyFileSet.setIncludes(singletonList(GROOVY_SOURCES_PATTERN));
              groovyFileSets = new FileSet[]{groovyFileSet};
          }
  
          if (includeJavaSources) {
              List<FileSet> javaFileSets = new ArrayList<>();
              for (String sourceRoot : project.getCompileSourceRoots()) {
                  FileSet javaFileSet = new FileSet();
                  javaFileSet.setDirectory(sourceRoot);
                  javaFileSet.setIncludes(singletonList(JAVA_SOURCES_PATTERN));
                  javaFileSets.add(javaFileSet);
              }
              FileSet[] javaFileSetsArr = javaFileSets.toArray(new FileSet[0]);
              result = Arrays.copyOf(groovyFileSets, groovyFileSets.length + javaFileSetsArr.length);
             System.arraycopy(javaFileSetsArr, 0, result, groovyFileSets.length, javaFileSetsArr.length);
         } else {
             result = groovyFileSets;
         }
 
         return result;
     }
 
     /**
      * Gets the set of included filesets from the specified source files or source directory (if sources are null).
      *
      * @param fromSources        The sources to get the included files from
      * @param includeJavaSources Whether to include Java sources
      * @return The included filesets from the specified sources
      */
     protected FileSet[] getTestFilesets(final FileSet[] fromSources, final boolean includeJavaSources) {
         FileSet[] result;
         FileSet[] groovyFileSets;
 
         if (fromSources != null) {
             groovyFileSets = fromSources;
         } else {
             FileSet groovyFileSet = new FileSet();
             String groovyDirectory = "src" + File.separator + TEST + File.separator + "groovy";
             groovyFileSet.setDirectory(project.getBasedir() + File.separator + groovyDirectory);
             groovyFileSet.setIncludes(singletonList(GROOVY_SOURCES_PATTERN));
             groovyFileSets = new FileSet[]{groovyFileSet};
         }
 
         if (includeJavaSources) {
             List<FileSet> javaFileSets = new ArrayList<>();
             for (String sourceRoot : project.getTestCompileSourceRoots()) {
                 FileSet javaFileSet = new FileSet();
                 javaFileSet.setDirectory(sourceRoot);
                 javaFileSet.setIncludes(singletonList(JAVA_SOURCES_PATTERN));
                 javaFileSets.add(javaFileSet);
             }
             FileSet[] javaFileSetsArr = javaFileSets.toArray(new FileSet[0]);
             result = Arrays.copyOf(groovyFileSets, groovyFileSets.length + javaFileSetsArr.length);
             System.arraycopy(javaFileSetsArr, 0, result, groovyFileSets.length, javaFileSetsArr.length);
         } else {
             result = groovyFileSets;
         }
 
         return result;
     }
 }