package p2;

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

/**
 * 在指定目錄中查詢包含關鍵字的檔案(或查詢字尾名為XXX的檔案)，返回包含指定關鍵字的檔案路徑.
 * @author Guozhu Zhu
 * @date 2018/6/3
 * @version 1.0
 *
 */
public class Test07 {
	
    public static List<File> searchFiles(File folder, final String keyword) {
        List<File> result = new ArrayList<File>();
        if (folder.isFile())
            result.add(folder);

        File[] subFolders = folder.listFiles(new FileFilter() {
            @Override
            public boolean accept(File file) {
                if (file.isDirectory()) {
                    return true;
                }
                if (file.getName().toLowerCase().contains(keyword)) {
                    return true;
                }
                return false;
            }
        });
        
        /*查詢字尾名
         * File[] subFolders = folder.listFiles(new FileFilter() {
            @Override
            public boolean accept(File file) {
                if (file.isDirectory()) {
                    return true;
                }
                if (file.getName().toLowerCase().endsWith(keyword)) {
                    return true;
                }
                return false;
            }
        });*/

        if (subFolders != null) {
            for (File file : subFolders) {
                if (file.isFile()) {
                    // 如果是檔案則將檔案新增到結果列表中
                    result.add(file);
                } else {
                    // 如果是資料夾，則遞迴呼叫本方法，然後把所有的檔案加到結果列表中
                    result.addAll(searchFiles(file, keyword));
                }
            }
        }
        return result;
    }

    public static void main(String[] args) {
        List<File> files = searchFiles(new File("d:/"), "01");
        System.out.println("共找到:" + files.size() + "個檔案");
        for (File file : files) {
            System.out.println(file.getAbsolutePath());
        }
    }
    
}