FileSelector.java

package org.slf4j.migrator;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

import org.slf4j.migrator.internal.ProgressListener;

public class FileSelector {

  private List<File> javaFileList = new ArrayList<File>();

  ProgressListener pl;

  FileSelector(ProgressListener pl) {
    this.pl = pl;
  }

  public List<File> selectJavaFilesInFolder(File folder) {
    if (folder.isDirectory()) {
      selectFiles(folder);
      return javaFileList;
    } else {
      throw new IllegalArgumentException("[" + folder + "] is not a directory");
    }
  }

  private void selectFiles(File file) {
    if (file.isDirectory()) {
      pl.onDirectory(file);
      File[] files = file.listFiles();
      if (files != null) {
        for (int i = 0; i < files.length; i++) {
          selectFiles(files[i]);
        }
      }
    } else {
      if (file.getName().endsWith(".java")) {
        pl.onFileAddition(file);
        javaFileList.add(file);
      }

    }
  }
}